Attempt to fix tokens again

UW-Macrostrat · Oct 19, 2024 · 4197270 · 4197270
1 parent 459943b
commit 4197270
Showing 1 changed file with 20 additions and 7 deletions.
diff --git a/api/routes/security.py b/api/routes/security.py
@@ -332,13 +332,26 @@ async def redirect_callback(code: str, state: Optional[str] = None):
             #         samesite="lax",
             #     )
             # else:
-            response.set_cookie(
-                access_token_key,
-                f"Bearer {access_token}",
-                domain=domain,
-                httponly=True,
-                samesite="lax",
-            )
+
+            if "localhost" in redirect_domain:
+                # Adjust host header to localhost
+                response.headers["host"] = redirect_domain
+                response.set_cookie(
+                    access_token_key,
+                    f"Bearer {access_token}",
+                    domain="localhost",
+                    httponly=True,
+                    samesite="lax",
+                )
+
+            else:
+                response.set_cookie(
+                    access_token_key,
+                    f"Bearer {access_token}",
+                    domain=domain,
+                    httponly=True,
+                    samesite="lax",
+                )
 
             return response