于小丘
51b6de921b
- Added user authentication and registration endpoints with JWT support. - Created tag management routes for creating and deleting tags. - Implemented value-added service (VAS) endpoints for managing storage packs and orders. - Developed WebDAV account management routes for creating, updating, and deleting accounts. - Introduced slave router for handling file uploads, downloads, and aria2 task management. - Enhanced JWT utility functions for token creation and secret key management. - Established lifespan management for FastAPI application startup and shutdown processes. - Integrated password handling utilities with Argon2 hashing and two-factor authentication support.
298 lines
9.1 KiB
Python
298 lines
9.1 KiB
Python
from fastapi import APIRouter, Depends, Query
|
|
from fastapi.responses import PlainTextResponse, RedirectResponse
|
|
from middleware.auth import SignRequired
|
|
from models.response import ResponseBase
|
|
import service.oauth
|
|
|
|
callback_router = APIRouter(
|
|
prefix='/callback',
|
|
tags=["callback"],
|
|
)
|
|
|
|
oauth_router = APIRouter(
|
|
prefix='/callback/oauth',
|
|
tags=["callback", "oauth"],
|
|
)
|
|
|
|
pay_router = APIRouter(
|
|
prefix='/callback/pay',
|
|
tags=["callback", "pay"],
|
|
)
|
|
|
|
upload_router = APIRouter(
|
|
prefix='/callback/upload',
|
|
tags=["callback", "upload"],
|
|
)
|
|
|
|
callback_router.include_router(oauth_router)
|
|
callback_router.include_router(pay_router)
|
|
callback_router.include_router(upload_router)
|
|
|
|
@oauth_router.post(
|
|
path='/qq',
|
|
summary='QQ互联回调',
|
|
description='Handle QQ OAuth callback and return user information.',
|
|
)
|
|
def router_callback_qq() -> ResponseBase:
|
|
"""
|
|
Handle QQ OAuth callback and return user information.
|
|
|
|
Returns:
|
|
ResponseModel: A model containing the response data for the QQ OAuth callback.
|
|
"""
|
|
pass
|
|
|
|
@oauth_router.get(
|
|
path='/github',
|
|
summary='GitHub OAuth 回调',
|
|
description='Handle GitHub OAuth callback and return user information.',
|
|
)
|
|
async def router_callback_github(
|
|
code: str = Query(description="The token received from GitHub for authentication.")) -> PlainTextResponse:
|
|
"""
|
|
GitHub OAuth 回调处理
|
|
- 错误响应示例：
|
|
- {
|
|
'error': 'bad_verification_code',
|
|
'error_description': 'The code passed is incorrect or expired.',
|
|
'error_uri': 'https://docs.github.com/apps/managing-oauth-apps/troubleshooting-oauth-app-access-token-request-errors/#bad-verification-code'
|
|
}
|
|
|
|
Returns:
|
|
PlainTextResponse: A response containing the user information from GitHub.
|
|
"""
|
|
try:
|
|
access_token = await service.oauth.github.get_access_token(code)
|
|
# [TODO] 把access_token写数据库里
|
|
if not access_token:
|
|
return PlainTextResponse("Failed to retrieve access token from GitHub.", status_code=400)
|
|
|
|
user_data = await service.oauth.github.get_user_info(access_token.access_token)
|
|
# [TODO] 把user_data写数据库里
|
|
|
|
return PlainTextResponse(f"User information processed successfully, code: {code}, user_data: {user_data.json_dump()}", status_code=200)
|
|
except Exception as e:
|
|
return PlainTextResponse(f"An error occurred: {str(e)}", status_code=500)
|
|
|
|
@pay_router.post(
|
|
path='/alipay',
|
|
summary='支付宝支付回调',
|
|
description='Handle Alipay payment callback and return payment status.',
|
|
)
|
|
def router_callback_alipay() -> ResponseBase:
|
|
"""
|
|
Handle Alipay payment callback and return payment status.
|
|
|
|
Returns:
|
|
ResponseModel: A model containing the response data for the Alipay payment callback.
|
|
"""
|
|
pass
|
|
|
|
@pay_router.post(
|
|
path='/wechat',
|
|
summary='微信支付回调',
|
|
description='Handle WeChat Pay payment callback and return payment status.',
|
|
)
|
|
def router_callback_wechat() -> ResponseBase:
|
|
"""
|
|
Handle WeChat Pay payment callback and return payment status.
|
|
|
|
Returns:
|
|
ResponseModel: A model containing the response data for the WeChat Pay payment callback.
|
|
"""
|
|
pass
|
|
|
|
@pay_router.post(
|
|
path='/stripe',
|
|
summary='Stripe支付回调',
|
|
description='Handle Stripe payment callback and return payment status.',
|
|
)
|
|
def router_callback_stripe() -> ResponseBase:
|
|
"""
|
|
Handle Stripe payment callback and return payment status.
|
|
|
|
Returns:
|
|
ResponseModel: A model containing the response data for the Stripe payment callback.
|
|
"""
|
|
pass
|
|
|
|
@pay_router.get(
|
|
path='/easypay',
|
|
summary='易支付回调',
|
|
description='Handle EasyPay payment callback and return payment status.',
|
|
)
|
|
def router_callback_easypay() -> PlainTextResponse:
|
|
"""
|
|
Handle EasyPay payment callback and return payment status.
|
|
|
|
Returns:
|
|
PlainTextResponse: A response containing the payment status for the EasyPay payment callback.
|
|
"""
|
|
pass
|
|
# return PlainTextResponse("success", status_code=200)
|
|
|
|
@pay_router.get(
|
|
path='/custom/{order_no}/{id}',
|
|
summary='自定义支付回调',
|
|
description='Handle custom payment callback and return payment status.',
|
|
)
|
|
def router_callback_custom(order_no: str, id: str) -> ResponseBase:
|
|
"""
|
|
Handle custom payment callback and return payment status.
|
|
|
|
Args:
|
|
order_no (str): The order number for the payment.
|
|
id (str): The ID associated with the payment.
|
|
|
|
Returns:
|
|
ResponseModel: A model containing the response data for the custom payment callback.
|
|
"""
|
|
pass
|
|
|
|
@upload_router.post(
|
|
path='/remote/{session_id}/{key}',
|
|
summary='远程上传回调',
|
|
description='Handle remote upload callback and return upload status.',
|
|
)
|
|
def router_callback_remote(session_id: str, key: str) -> ResponseBase:
|
|
"""
|
|
Handle remote upload callback and return upload status.
|
|
|
|
Args:
|
|
session_id (str): The session ID for the upload.
|
|
key (str): The key for the uploaded file.
|
|
|
|
Returns:
|
|
ResponseModel: A model containing the response data for the remote upload callback.
|
|
"""
|
|
pass
|
|
|
|
@upload_router.post(
|
|
path='/qiniu/{session_id}',
|
|
summary='七牛云上传回调',
|
|
description='Handle Qiniu Cloud upload callback and return upload status.',
|
|
)
|
|
def router_callback_qiniu(session_id: str) -> ResponseBase:
|
|
"""
|
|
Handle Qiniu Cloud upload callback and return upload status.
|
|
|
|
Args:
|
|
session_id (str): The session ID for the upload.
|
|
|
|
Returns:
|
|
ResponseModel: A model containing the response data for the Qiniu Cloud upload callback.
|
|
"""
|
|
pass
|
|
|
|
@upload_router.post(
|
|
path='/tencent/{session_id}',
|
|
summary='腾讯云上传回调',
|
|
description='Handle Tencent Cloud upload callback and return upload status.',
|
|
)
|
|
def router_callback_tencent(session_id: str) -> ResponseBase:
|
|
"""
|
|
Handle Tencent Cloud upload callback and return upload status.
|
|
|
|
Args:
|
|
session_id (str): The session ID for the upload.
|
|
|
|
Returns:
|
|
ResponseModel: A model containing the response data for the Tencent Cloud upload callback.
|
|
"""
|
|
pass
|
|
|
|
@upload_router.post(
|
|
path='/aliyun/{session_id}',
|
|
summary='阿里云上传回调',
|
|
description='Handle Aliyun upload callback and return upload status.',
|
|
)
|
|
def router_callback_aliyun(session_id: str) -> ResponseBase:
|
|
"""
|
|
Handle Aliyun upload callback and return upload status.
|
|
|
|
Args:
|
|
session_id (str): The session ID for the upload.
|
|
|
|
Returns:
|
|
ResponseModel: A model containing the response data for the Aliyun upload callback.
|
|
"""
|
|
pass
|
|
|
|
@upload_router.post(
|
|
path='/upyun/{session_id}',
|
|
summary='又拍云上传回调',
|
|
description='Handle Upyun upload callback and return upload status.',
|
|
)
|
|
def router_callback_upyun(session_id: str) -> ResponseBase:
|
|
"""
|
|
Handle Upyun upload callback and return upload status.
|
|
|
|
Args:
|
|
session_id (str): The session ID for the upload.
|
|
|
|
Returns:
|
|
ResponseModel: A model containing the response data for the Upyun upload callback.
|
|
"""
|
|
pass
|
|
|
|
@upload_router.post(
|
|
path='/aws/{session_id}',
|
|
summary='AWS S3上传回调',
|
|
description='Handle AWS S3 upload callback and return upload status.',
|
|
)
|
|
def router_callback_aws(session_id: str) -> ResponseBase:
|
|
"""
|
|
Handle AWS S3 upload callback and return upload status.
|
|
|
|
Args:
|
|
session_id (str): The session ID for the upload.
|
|
|
|
Returns:
|
|
ResponseModel: A model containing the response data for the AWS S3 upload callback.
|
|
"""
|
|
pass
|
|
|
|
@upload_router.post(
|
|
path='/onedrive/finish/{session_id}',
|
|
summary='OneDrive上传完成回调',
|
|
description='Handle OneDrive upload completion callback and return upload status.',
|
|
)
|
|
def router_callback_onedrive_finish(session_id: str) -> ResponseBase:
|
|
"""
|
|
Handle OneDrive upload completion callback and return upload status.
|
|
|
|
Args:
|
|
session_id (str): The session ID for the upload.
|
|
|
|
Returns:
|
|
ResponseModel: A model containing the response data for the OneDrive upload completion callback.
|
|
"""
|
|
pass
|
|
|
|
@upload_router.get(
|
|
path='/ondrive/auth',
|
|
summary='OneDrive授权回调',
|
|
description='Handle OneDrive authorization callback and return authorization status.',
|
|
)
|
|
def router_callback_onedrive_auth() -> ResponseBase:
|
|
"""
|
|
Handle OneDrive authorization callback and return authorization status.
|
|
|
|
Returns:
|
|
ResponseModel: A model containing the response data for the OneDrive authorization callback.
|
|
"""
|
|
pass
|
|
|
|
@upload_router.get(
|
|
path='/google/auth',
|
|
summary='Google OAuth 完成',
|
|
description='Handle Google OAuth completion callback and return authorization status.',
|
|
)
|
|
def router_callback_google_auth() -> ResponseBase:
|
|
"""
|
|
Handle Google OAuth completion callback and return authorization status.
|
|
|
|
Returns:
|
|
ResponseModel: A model containing the response data for the Google OAuth completion callback.
|
|
"""
|
|
pass |