add withdraw all contract

web_app/api/user.py

@@ -1,3 +1,4 @@
+
 """
 This module handles user-related API endpoints.
 """
@@ -21,6 +22,7 @@
     TelegramUserDBConnector,
     UserDBConnector,
 )
+from web_app.contract_tools.blockchain_call import CLIENT
 
 logger = logging.getLogger(__name__)
 router = APIRouter()  # Initialize the router
@@ -298,3 +300,36 @@ async def allow_notification(
         raise HTTPException(status_code=404, detail=str(e))
     except Exception as e:
         raise HTTPException(status_code=500, detail="Internal server error")
+
+
+@router.post(
+    "/api/withdraw-all",
+    tags=["User Operations"],
+    summary="Withdraw all tokens from user's contract",
+    response_description="Status of withdrawal operations",
+)
+async def withdraw_all(wallet_id: str) -> dict:
+   """
+   Withdraws all supported tokens from the user's contract.
+
+   :param wallet_id: The wallet ID of the user.
+   :return: detail: "Successfully initiated withdrawals for all tokens"
+   """
+   # Get user's contract address
+   contract_address = await get_user_contract(wallet_id)
+   if not contract_address:
+       raise HTTPException(status_code=404, detail="Contract not found")
+
+   try:
+       # Perform withdrawals
+       results = await CLIENT.withdraw_all(contract_address)
+       return {
+           "detail": "Successfully initiated withdrawals for all tokens",
+           "results": results
+       }
+   except Exception as e:
+       logger.error(f"Contract withdrawal failed: {str(e)}")
+       raise HTTPException(
+           status_code=500, 
+           detail=f"Failed to withdraw tokens: {str(e)}"
+       )

web_app/contract_tools/blockchain_call.py

@@ -352,5 +352,37 @@ async def add_extra_deposit(self, contract_address: str, token_address: str, amo
             calldata=[self._convert_address(token_address), amount],
         )
 
+    async def withdraw_all(self, contract_address: str) -> dict[str, str]:
+        """
+        Withdraws all supported tokens from the contract by calling withdraw with amount=0.
+
+        :param contract_address: The contract address to withdraw from
+        :return: A dictionary summarizing the results for each token.
+        """
+        contract_addr_int = self._convert_address(contract_address)
+        results = {}
+
+        for token in TokenParams.tokens():  
+            try:
+                token_addr_int = self._convert_address(token.address)  
+                token_symbol = token.name
+                logger.info(f"Withdrawing {token_symbol} from contract {contract_address}")
+                await self._func_call(
+                    addr=contract_addr_int,
+                    selector="withdraw",
+                    calldata=[token_addr_int, 0] 
+                )
+                results[token_symbol] = "Success"
+
+            except ValueError as e:
+                logger.error(f"Invalid address format for {token_symbol}: {str(e)}")
+                results[token_symbol] = f"Failed: Invalid address format"
+
+            except Exception as e:
+                logger.error(f"Error withdrawing {token_symbol}: {e}")
+                results[token_symbol] = f"Failed: {e}"
+
+        return results
+
 
 CLIENT = StarknetClient()