feat(cli): replace balance check with CloudflareSolver

Updated the balance command to use the new CloudflareSolver class for retrieving account balance, improving error handling and API interaction.

Author: zetxtech

cfsolver/cli.py

@@ -403,31 +403,19 @@ def request(
 @click.pass_context
 def balance(ctx, api_key, api_base, api_proxy):
     """Check account balance."""
-    from curl_cffi.requests import Session
+    from .client import CloudflareSolver
 
     api_key = get_api_key(api_key)
     api_base = get_api_base(api_base)
 
     try:
-        with Session(verify=False, proxy=api_proxy) as session:
-            resp = session.post(
-                f"{api_base}/api/getBalance",
-                json={"clientKey": api_key},
-            )
-
-            if resp.status_code != 200:
-                click.echo(f"[x] Error: HTTP {resp.status_code}", err=True)
-                sys.exit(1)
-
-            data = resp.json()
-            if data.get("errorId"):
-                click.echo(
-                    f"[x] Error: {data.get('errorDescription', 'Unknown error')}",
-                    err=True,
-                )
-                sys.exit(1)
-
-            balance_val = data.get("balance", 0)
+        with CloudflareSolver(
+            api_key=api_key,
+            api_base=api_base,
+            api_proxy=api_proxy,
+            solve=False,
+        ) as solver:
+            balance_val = solver.get_balance()
             click.echo(f"[+] Balance: {balance_val}")
 
     except Exception as e:

cfsolver/client.py

@@ -535,6 +535,31 @@ def _wait_for_result(self, task_id: str, timeout: int = 120) -> Dict[str, Any]:
 
         raise CFSolverTimeoutError("Task timed out")
 
+    def get_balance(self) -> float:
+        """
+        Get the current account balance.
+
+        Returns:
+            The account balance as a float
+
+        Raises:
+            CFSolverAPIError: If the API request fails
+        """
+        with Session(verify=False, proxy=self.api_proxy) as api_session:
+            resp = api_session.post(
+                f"{self.api_base}/api/getBalance",
+                json={"clientKey": self.api_key},
+            )
+            _raise_for_status(resp)
+            data = resp.json()
+
+        if data.get("errorId"):
+            raise CFSolverAPIError(
+                f"Failed to get balance: {data.get('errorDescription', 'Unknown error')}"
+            )
+
+        return float(data.get("balance", 0))
+
     def solve_turnstile(self, url: str, sitekey: str) -> str:
         """
         Solve a Turnstile challenge and return the token.