Update app/google_oauth.py

app/google_oauth.py

@@ -15,7 +15,7 @@ SCOPES = [
 TOKEN_FILENAME = "google_token.json"
 DATA_DIR = Path("/data")
 FALLBACK_DATA_DIR = Path(tempfile.gettempdir()) / "tcc2_agent"
-
+OAUTH_TEMP = {}
 
 class GoogleOAuthError(RuntimeError):
     pass
@@ -76,20 +76,32 @@ def _client_config():
     }
 
 
-def build_flow(state=None):
+def build_flow(state=None, code_verifier=None):
     redirect_uri = os.environ.get("GOOGLE_REDIRECT_URI", "").strip()
-    flow = Flow.from_client_config(_client_config(), scopes=SCOPES, state=state)
+
+    flow = Flow.from_client_config(
+        _client_config(),
+        scopes=SCOPES,
+        state=state,
+        code_verifier=code_verifier,
+    )
+
     flow.redirect_uri = redirect_uri
     return flow
 
 
 def get_authorization_url():
     flow = build_flow()
+
     authorization_url, state = flow.authorization_url(
         access_type="offline",
         prompt="consent",
         include_granted_scopes="true",
     )
+
+    OAUTH_TEMP["state"] = state
+    OAUTH_TEMP["code_verifier"] = flow.code_verifier
+
     return authorization_url, state