GH-9: Add Auth to on-auth callback

Author: ArtyomVancyan

examples/demonstration/main.py

@@ -6,6 +6,7 @@
 from database import Base
 from database import engine
 from database import get_db
+from fastapi_oauth2.middleware import Auth
 from fastapi_oauth2.middleware import OAuth2Middleware
 from fastapi_oauth2.middleware import User
 from fastapi_oauth2.router import router as oauth2_router
@@ -17,7 +18,7 @@
 router = APIRouter()
 
 
-async def on_auth(user: User):
+async def on_auth(auth: Auth, user: User):
     # perform a check for user existence in
     # the database and create if not exists
     db: Session = next(get_db())

src/fastapi_oauth2/middleware.py

@@ -124,7 +124,7 @@ class OAuth2Backend(AuthenticationBackend):
     def __init__(
             self,
             config: OAuth2Config,
-            callback: Callable[[User], Union[Awaitable[None], None]] = None,
+            callback: Callable[[Auth, User], Union[Awaitable[None], None]] = None,
     ) -> None:
         Auth.set_http(config.allow_http)
         Auth.set_secret(config.jwt_secret)
@@ -153,7 +153,7 @@ async def authenticate(self, request: Request) -> Optional[Tuple[Auth, User]]:
 
         # Call the callback function on authentication
         if callable(self.callback):
-            coroutine = self.callback(user)
+            coroutine = self.callback(auth, user)
             if issubclass(type(coroutine), Awaitable):
                 await coroutine
         return auth, user
@@ -168,7 +168,7 @@ def __init__(
             self,
             app: ASGIApp,
             config: Union[OAuth2Config, dict],
-            callback: Callable[[User], Union[Awaitable[None], None]] = None,
+            callback: Callable[[Auth, User], Union[Awaitable[None], None]] = None,
             **kwargs,  # AuthenticationMiddleware kwargs
     ) -> None:
         """Initiates the middleware with the given configuration.