提交初始文件，增加授权应用路由

2026-06-10 10:47:18 +08:00
parent d8bfebdcfb
commit b24c176b88
6 changed files with 590 additions and 1 deletions
@@ -0,0 +1,124 @@
+from fastapi import APIRouter, Depends, HTTPException, Query, status
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.dependencies import get_current_user, get_db
+from app.models.user import User
+from app.schemas.user_oauth import RequestOAuthRequest, RequestOAuthResponse, UserOAuthOut
+from app.services.user_oauth_service import (
+    build_oauth_url,
+    get_account_info_by_type,
+    get_token_by_type,
+    save_oauth_token,
+)
+
+router = APIRouter(prefix="/user-oauth", tags=["user-oauth"])
+
+
+@router.post(
+    "/request_oauth",
+    summary="获取授权链接",
+    description="用户提交oauth_type，返回对应的第三方授权链接",
+    response_model=RequestOAuthResponse,
+)
+async def request_oauth(
+    req: RequestOAuthRequest,
+    current_user: User = Depends(get_current_user),
+):
+    try:
+        auth_url = await build_oauth_url(req.oauth_type, current_user.id)
+        return {"auth_url": auth_url}
+    except ValueError as e:
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail=str(e),
+        )
+
+
+@router.get(
+    "/juliang_callback",
+    summary="巨量授权回调",
+    description="巨量引擎授权回调地址，接收code和state参数，获取token并保存",
+)
+async def juliang_callback(
+    auth_code: str = Query(..., description="第三方返回的授权码"),
+    state: str = Query(..., description="请求时传递的自定义参数"),
+    db: AsyncSession = Depends(get_db),
+):
+    try:
+        parts = state.split(":")
+        if len(parts) != 4:
+            raise HTTPException(
+                status_code=status.HTTP_400_BAD_REQUEST,
+                detail="无效的state参数",
+            )
+
+        oauth_type = int(parts[0])
+        user_id = parts[1]
+        app_id = parts[2]
+        app_type = parts[3]
+
+        token = await get_token_by_type(auth_code, oauth_type, app_id, app_type)
+        account_info = await get_account_info_by_type(token, oauth_type, app_type)
+        user_oauth = await save_oauth_token(db, user_id, oauth_type, token, account_info, app_id)
+
+        return {
+            "message": "授权成功",
+            "code": 0,
+            "data": UserOAuthOut.model_validate(user_oauth),
+        }
+
+    except ValueError as e:
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail=str(e),
+        )
+    except Exception as e:
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=f"授权失败: {str(e)}",
+        )
+
+
+@router.get(
+    "/callback",
+    summary="通用授权回调",
+    description="其他平台授权回调地址，接收code和state参数",
+)
+async def oauth_callback(
+    code: str = Query(..., description="第三方返回的授权码"),
+    state: str = Query(..., description="请求时传递的自定义参数"),
+    db: AsyncSession = Depends(get_db),
+):
+    try:
+        parts = state.split(":")
+        if len(parts) != 4:
+            raise HTTPException(
+                status_code=status.HTTP_400_BAD_REQUEST,
+                detail="无效的state参数",
+            )
+
+        oauth_type = int(parts[0])
+        user_id = parts[1]
+        app_id = parts[2]
+        app_type = parts[3]
+
+        token = await get_token_by_type(code, oauth_type, app_id, app_type)
+        account_info = await get_account_info_by_type(token, oauth_type, app_type)
+        user_oauth = await save_oauth_token(db, user_id, oauth_type, token, account_info, app_id)
+
+        return {
+            "message": "授权成功",
+            "code": 0,
+            "data": UserOAuthOut.model_validate(user_oauth),
+        }
+
+    except ValueError as e:
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail=str(e),
+        )
+    except Exception as e:
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=f"授权失败: {str(e)}",
+        )