Reset hesla, doplneni mailu

2024-02-08 16:26:12 +01:00
parent 03c9f2d189
commit c26eb97288
11 changed files with 235 additions and 79 deletions
@@ -0,0 +1,49 @@
+from datetime import timedelta
+from fastapi import FastAPI, Security, Depends, BackgroundTasks, APIRouter
+from fastapi_jwt import JwtAuthorizationCredentials
+from fastapi_jwt.jwt import JwtAccessBearerCookie
+from sqlalchemy import select
+from starlette.responses import RedirectResponse
+from strawberry.fastapi import GraphQLRouter
+from config import GRAPHIQL, APP_DEBUG
+from database import async_session, models
+from graphql_schema.schema import GraphQLContext, schema
+
+
+def setup_graphql_endpoint(app: FastAPI, access_security: JwtAccessBearerCookie, api_router: APIRouter):
+    async def setup_graphql_context(credentials: JwtAuthorizationCredentials = Security(access_security)):
+        user_id = credentials['id'] if credentials else None
+        organization_ids = set()
+
+        if user_id:
+            async with async_session() as db:
+                organization_ids = set((await db.scalars(
+                    select(models.user_is_in_organization.c.organization_id)
+                    .filter(models.user_is_in_organization.c.user_id == user_id)
+                )).all())
+
+        if APP_DEBUG:
+            @api_router.get("/graphql/autologin")
+            async def autologin():
+                access_token = access_security.create_access_token(subject={"id": 6, "name": "Franta Vomacka"})
+
+                response = RedirectResponse(url="/graphql")
+                access_security.set_access_cookie(response, access_token, expires_delta=timedelta(days=14))
+
+                return response
+
+        return GraphQLContext(
+            user_id=user_id,
+            organization_ids=organization_ids,
+            jwt_auth_credentials=credentials,
+            jwt=access_security,
+            background_tasks=Depends(BackgroundTasks)
+        )
+
+    graphql_app = GraphQLRouter(
+        schema,
+        graphiql=GRAPHIQL,
+        debug=APP_DEBUG,
+        context_getter=setup_graphql_context
+    )
+    app.include_router(graphql_app, prefix="/graphql")