introduce graphql

app/auth.py

@@ -164,11 +164,11 @@ async def auth_callback(request: Request, code: str, state: str, background_task
         "access_token": jwt_token,
         "refresh_token": refresh_token,
         "token_type": "bearer",
-        "user": {
-            "email": user.email,
-            "name": user.name,
-            "avatar": get_image_url(user.avatar),
-        }
+        # "user": {
+        #     "email": user.email,
+        #     "name": user.name,
+        #     "avatar": get_image_url(user.avatar),
+        # }
     }
     
 @router.get("/me")

app/graphql/context.py

@@ -0,0 +1,26 @@
+from fastapi import Request
+from strawberry.fastapi import BaseContext
+from app.db import SessionLocal
+from app.auth.deps import get_current_user
+from app.models import User
+from typing import AsyncGenerator
+
+class CustomContext(BaseContext):
+    def __init__(self, request: Request):
+        super().__init__(request)
+        self.db = SessionLocal()
+        self.request = request
+
+    async def get_current_user(self) -> User | None:
+        # 复用现有的 auth 逻辑
+        try:
+            return await get_current_user(self.request)
+        except Exception:
+            return None
+
+async def get_context(request: Request) -> AsyncGenerator[CustomContext, None]:
+    context = CustomContext(request)
+    try:
+        yield context
+    finally:
+        context.db.close()  # 确保请求结束时关闭数据库连接

app/graphql/mutations.py

@@ -0,0 +1,18 @@
+import strawberry
+from app.auth.jwt import create_jwt_token, create_refresh_token
+
+@strawberry.type
+class AuthResponse:
+    access_token: str
+    refresh_token: str
+    token_type: str
+
+@strawberry.type
+class Mutation:
+    @strawberry.mutation
+    def logout(self) -> str:
+        return "ok"
+    
+    # 提示：像 Google OAuth 这种涉及第三方重定向的流程，
+    # 依然保留现有的 RESTful (/login/google 和 /auth/callback) 会更简单。
+    # 登录成功后重定向回前端，前端拿到 token，后续所有数据交互全部走 GraphQL。

app/graphql/queries.py

@@ -0,0 +1,18 @@
+import strawberry
+from typing import Optional
+from app.graphql.types import UserType
+from app.models import User
+
+@strawberry.type
+class Query:
+    @strawberry.field
+    async def me(self, info) -> Optional[UserType]:
+        # 从 context 获取当前登录用户
+        user = await info.context.get_current_user()
+        # Strawberry 会自动将 SQLAlchemy 实例映射为 UserType
+        return user
+
+    @strawberry.field
+    def get_user_by_handle(self, info, handle: str) -> Optional[UserType]:
+        db = info.context.db
+        return db.query(User).filter(User.handle == handle).first()

app/graphql/types.py

@@ -0,0 +1,44 @@
+import strawberry
+from typing import List, Optional
+from datetime import datetime
+from app.utils.format import get_image_url
+
+@strawberry.type
+class UserType:
+    id: strawberry.ID
+    email: str
+    name: Optional[str]
+    handle: Optional[str]
+    avatar: Optional[str]
+    following_count: int
+    follower_count: int
+    created_at: datetime
+
+    # 动态解析字段：例如自动处理头像的绝对路径
+    @strawberry.field
+    def avatar_url(self) -> Optional[str]:
+        return get_image_url(self.avatar) if self.avatar else None
+
+@strawberry.type
+class WorkspaceType:
+    id: strawberry.ID
+    name: str
+    slug: str
+    logo: Optional[str]
+    description: Optional[str]
+    storage_used: int
+    
+    # 支持前端按需加载关联的成员
+    @strawberry.field
+    def members(self, info) -> List["WorkspaceMemberType"]:
+        from app.models import WorkspaceMember
+        db = info.context.db
+        return db.query(WorkspaceMember).filter(WorkspaceMember.workspace_id == self.id).all()
+
+@strawberry.type
+class WorkspaceMemberType:
+    id: strawberry.ID
+    workspace_id: str
+    user_id: Optional[str]
+    role: str
+    status: str

app/main.py

@@ -1,10 +1,31 @@
 from fastapi import FastAPI
-from app.auth import router as auth_router
-from app.models import Base
-from app.db import engine
+from strawberry import Schema
+from strawberry.fastapi import GraphQLRouter
 
+from app.auth import router as auth_router
+from app.db import engine
+from app.models import Base
+
+# 导入 GraphQL 组件
+from app.graphql.queries import Query
+from app.graphql.mutations import Mutation
+from app.graphql.context import get_context
+
+# 自动建表
 Base.metadata.create_all(bind=engine)
 
+# 创建 FastAPI 实例
 app = FastAPI()
 
-app.include_router(auth_router)
+# 组装 GraphQL Schema
+schema = Schema(query=Query, mutation=Mutation)
+
+# 创建 GraphQL 路由并注入上下文
+graphql_router = GraphQLRouter(schema, context_getter=get_context)
+
+# 挂载路由
+# 保留现有的 auth 路由供 Google OAuth 回调使用
+app.include_router(auth_router) 
+
+# 挂载 GraphQL 终点，前端以后只需要请求 /graphql
+app.include_router(graphql_router, prefix="/graphql")