Add a new attribute for SSR mode

Author: ArtyomVancyan

src/fastapi_oauth2/config.py

@@ -8,6 +8,7 @@
 class OAuth2Config:
     """Configuration class of the authentication middleware."""
 
+    enable_ssr: bool
     allow_http: bool
     jwt_secret: str
     jwt_expires: int
@@ -17,6 +18,7 @@ class OAuth2Config:
     def __init__(
             self,
             *,
+            enable_ssr: bool = True,
             allow_http: bool = False,
             jwt_secret: str = "",
             jwt_expires: Union[int, str] = 900,
@@ -25,6 +27,7 @@ def __init__(
     ) -> None:
         if allow_http:
             os.environ["OAUTHLIB_INSECURE_TRANSPORT"] = "1"
+        self.enable_ssr = enable_ssr
         self.allow_http = allow_http
         self.jwt_secret = jwt_secret
         self.jwt_expires = int(jwt_expires)

src/fastapi_oauth2/middleware.py

@@ -31,6 +31,7 @@
 class Auth(AuthCredentials):
     """Extended auth credentials schema based on Starlette AuthCredentials."""
 
+    ssr: bool
     http: bool
     secret: str
     expires: int
@@ -39,6 +40,10 @@ class Auth(AuthCredentials):
     provider: OAuth2Core = None
     clients: Dict[str, OAuth2Core] = {}
 
+    @classmethod
+    def set_ssr(cls, ssr: bool) -> None:
+        cls.ssr = ssr
+
     @classmethod
     def set_http(cls, http: bool) -> None:
         cls.http = http
@@ -117,6 +122,7 @@ def __init__(
             config: OAuth2Config,
             callback: Callable[[Auth, User], Union[Awaitable[None], None]] = None,
     ) -> None:
+        Auth.set_ssr(config.enable_ssr)
         Auth.set_http(config.allow_http)
         Auth.set_secret(config.jwt_secret)
         Auth.set_expires(config.jwt_expires)

src/fastapi_oauth2/router.py

@@ -6,13 +6,17 @@
 
 
 @router.get("/{provider}/auth")
-async def login(request: Request, provider: str):
-    return await request.auth.clients[provider].login_redirect(request)
+def authorize(request: Request, provider: str):
+    if request.auth.ssr:
+        return request.auth.clients[provider].authorization_redirect(request)
+    return dict(url=request.auth.clients[provider].authorization_url(request))
 
 
 @router.get("/{provider}/token")
 async def token(request: Request, provider: str):
-    return await request.auth.clients[provider].token_redirect(request)
+    if request.auth.ssr:
+        return await request.auth.clients[provider].token_redirect(request)
+    return await request.auth.clients[provider].token_data(request)
 
 
 @router.get("/logout")