68 lines
2.1 KiB
Python
68 lines
2.1 KiB
Python
import os
|
|
from fastapi import FastAPI, Depends, HTTPException
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from strawberry import Schema
|
|
from strawberry.fastapi import GraphQLRouter
|
|
from dotenv import load_dotenv
|
|
|
|
from auth import router as auth_router, get_user_by_id, get_current_user
|
|
|
|
from app.db import engine, SessionLocal
|
|
from app.models import Base, User
|
|
from app.utils.format import get_image_url
|
|
|
|
# 导入 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()
|
|
|
|
# 获取环境变量字符串，并转成列表
|
|
load_dotenv()
|
|
origins_str = os.getenv("CORS_ORIGINS", "http://localhost:3000")
|
|
origins = [origin.strip() for origin in origins_str.split(",")]
|
|
|
|
# 配置 CORS 中间件，允许前端访问
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=origins,
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# 组装 GraphQL Schema
|
|
schema = Schema(query=Query, mutation=Mutation)
|
|
|
|
# 创建 GraphQL 路由并注入上下文
|
|
graphql_router = GraphQLRouter(schema, context_getter=get_context)
|
|
|
|
# 挂载路由
|
|
app.include_router(auth_router) # 保留现有的 auth 路由供 Google OAuth 回调使用
|
|
app.include_router(graphql_router, prefix="/graphql") # 挂载 GraphQL 终点，前端以后只需要请求 /graphql
|
|
|
|
def override_get_user_by_id(user_id: str = Depends(get_current_user)):
|
|
db = SessionLocal()
|
|
try:
|
|
user = db.query(User).filter(User.id == user_id).first()
|
|
if not user:
|
|
raise HTTPException(status_code=404, detail="User not found")
|
|
return {
|
|
"email": user.email,
|
|
"name": user.name,
|
|
"avatar": get_image_url(user.avatar),
|
|
}
|
|
finally:
|
|
db.close()
|
|
|
|
app.dependency_overrides[get_user_by_id] = override_get_user_by_id
|
|
|
|
# test code
|
|
for route in app.routes:
|
|
methods = getattr(route, "methods", "N/A")
|
|
print(f"Path: {route.path} | Name: {route.name} | Methods: {methods}") |