Fixes for return_on_mfa code

README.md

@@ -140,10 +140,10 @@ garth.login(email, password, prompt_mfa=lambda: input("Enter MFA code: "))
 For advanced use cases (like async handling), MFA can be handled separately:
 
 ```python
-result = garth.login(email, password, return_on_mfa=True)
-if isinstance(result, dict):  # MFA is required
+result1, result2 = garth.login(email, password, return_on_mfa=True)
+if result1 == "needs_mfa":  # MFA is required
     mfa_code = "123456"  # Get this from your custom MFA flow
-    oauth1, oauth2 = garth.resume_login(result['client_state'], mfa_code)
+    oauth1, oauth2 = garth.resume_login(result2, mfa_code)
 ```
 
 ### Configure

garth/http.py

@@ -164,7 +164,14 @@ def login(self, *args, **kwargs):
         self.oauth1_token, self.oauth2_token = sso.login(
             *args, **kwargs, client=self
         )
-
+        return self.oauth1_token, self.oauth2_token
+
+    def resume_login(self, *args, **kwargs):
+        self.oauth1_token, self.oauth2_token = sso.resume_login(
+            *args, **kwargs
+        )
+        return self.oauth1_token, self.oauth2_token
+
     def refresh_oauth2(self):
         assert self.oauth1_token, "OAuth1 token is required for OAuth2 refresh"
         # There is a way to perform a refresh of an OAuth2 token, but it

garth/sso.py

@@ -1,7 +1,7 @@
 import asyncio
 import re
 import time
-from typing import Callable, Dict, Tuple
+from typing import Callable, Dict, Tuple, Literal
 from urllib.parse import parse_qs
 
 import requests
@@ -74,7 +74,7 @@ def login(
     client: "http.Client | None" = None,
     prompt_mfa: Callable | None = lambda: input("MFA code: "),
     return_on_mfa: bool = False,
-) -> Tuple[OAuth1Token, OAuth2Token] | dict:
+) -> Tuple[OAuth1Token, OAuth2Token] | Tuple[Literal["needs_mfa"], dict]:
     """Login to Garmin Connect.
 
     Args:
@@ -141,13 +141,10 @@ def login(
     # Handle MFA
     if "MFA" in title:
         if return_on_mfa or prompt_mfa is None:
-            return {
-                "needs_mfa": True,
-                "client_state": {
-                    "csrf_token": csrf_token,
-                    "signin_params": SIGNIN_PARAMS,
-                    "client": client,
-                },
+            return "needs_mfa", {
+                "csrf_token": csrf_token,
+                "signin_params": SIGNIN_PARAMS,
+                "client": client,
             }
 
         handle_mfa(client, SIGNIN_PARAMS, prompt_mfa)

garth/version.py

@@ -1 +1 @@
-__version__ = "0.5.2"
+__version__ = "0.5.3"