118 lines
3.8 KiB
Python
118 lines
3.8 KiB
Python
|
from app.core.logging import get_logger
|
|||
|
|
from typing import List
|
|||
|
|
|
|||
|
|
from fastapi import APIRouter, Depends, HTTPException, Request
|
|||
|
|
from fastapi.responses import PlainTextResponse
|
|||
|
|
|
|||
|
|
from app.core.dependencies import get_current_user
|
|||
|
|
from app.features.payment.deps import get_payment_order_service
|
|||
|
|
from app.features.payment.order_service import PaymentOrderService
|
|||
|
|
from app.features.payment.schemas import (
|
|||
|
|
CreateOrderRequest,
|
|||
|
|
CreateOrderResponse,
|
|||
|
|
OrderListResponse,
|
|||
|
|
OrderStatusResponse,
|
|||
|
|
)
|
|||
|
|
from app.features.plan.deps import get_plan_service
|
|||
|
|
from app.features.plan.service import PlanService
|
|||
|
|
from app.features.user.models import User
|
|||
|
|
|
|||
|
|
logger = get_logger(__name__)
|
|||
|
|
|
|||
|
|
router = APIRouter(
|
|||
|
|
prefix="/api/payment",
|
|||
|
|
tags=["payment"],
|
|||
|
|
responses={
|
|||
|
|
401: {"description": "认证失败"},
|
|||
|
|
404: {"description": "订单不存在"},
|
|||
|
|
503: {"description": "支付服务暂不可用"},
|
|||
|
|
},
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.post("/create-order", response_model=CreateOrderResponse)
|
|||
|
|
async def create_order(
|
|||
|
|
request: CreateOrderRequest,
|
|||
|
|
current_user: User = Depends(get_current_user),
|
|||
|
|
service: PaymentOrderService = Depends(get_payment_order_service),
|
|||
|
|
plan_service: PlanService = Depends(get_plan_service),
|
|||
|
|
):
|
|||
|
|
plan = next((p for p in plan_service.get_plans_for_api() if p.id == request.plan_id), None)
|
|||
|
|
if plan is None:
|
|||
|
|
raise HTTPException(status_code=400, detail="无效的套餐 ID")
|
|||
|
|
return await service.create_order(
|
|||
|
|
user_id=current_user.id,
|
|||
|
|
user_subscription_type=current_user.subscription_type,
|
|||
|
|
plan_id=plan.id,
|
|||
|
|
plan_display_name=plan.display_name,
|
|||
|
|
plan_price=plan.price,
|
|||
|
|
plan_currency=plan.currency,
|
|||
|
|
payment_method=request.payment_method,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.post("/notify/wechat", include_in_schema=False)
|
|||
|
|
async def wechat_notify(
|
|||
|
|
request: Request,
|
|||
|
|
service: PaymentOrderService = Depends(get_payment_order_service),
|
|||
|
|
):
|
|||
|
|
try:
|
|||
|
|
headers = dict(request.headers)
|
|||
|
|
body = await request.body()
|
|||
|
|
return await service.handle_wechat_notify(headers=headers, body=body.decode("utf-8"))
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.exception("微信支付回调处理失败: %s", e)
|
|||
|
|
return {"code": "FAIL", "message": str(e)}
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.post("/notify/alipay", include_in_schema=False)
|
|||
|
|
async def alipay_notify(
|
|||
|
|
request: Request,
|
|||
|
|
service: PaymentOrderService = Depends(get_payment_order_service),
|
|||
|
|
):
|
|||
|
|
try:
|
|||
|
|
form_data = await request.form()
|
|||
|
|
params = {key: value for key, value in form_data.items()}
|
|||
|
|
result = await service.handle_alipay_notify(params=params)
|
|||
|
|
return PlainTextResponse(result)
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.exception("支付宝回调处理失败: %s", e)
|
|||
|
|
return PlainTextResponse("fail")
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.get("/order/{order_id}/status", response_model=OrderStatusResponse)
|
|||
|
|
async def get_order_status(
|
|||
|
|
order_id: str,
|
|||
|
|
current_user: User = Depends(get_current_user),
|
|||
|
|
service: PaymentOrderService = Depends(get_payment_order_service),
|
|||
|
|
):
|
|||
|
|
return await service.get_order_status(order_id, current_user.id)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.get("/orders", response_model=List[OrderListResponse])
|
|||
|
|
async def list_orders(
|
|||
|
|
current_user: User = Depends(get_current_user),
|
|||
|
|
service: PaymentOrderService = Depends(get_payment_order_service),
|
|||
|
|
):
|
|||
|
|
return await service.list_orders(current_user.id)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 订单列表路由（兼容旧版 /api/orders）
|
|||
|
|
orders_router = APIRouter(
|
|||
|
|
prefix="/api/orders",
|
|||
|
|
tags=["orders"],
|
|||
|
|
responses={
|
|||
|
|
401: {"description": "认证失败"},
|
|||
|
|
404: {"description": "订单不存在"},
|
|||
|
|
},
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@orders_router.get("", response_model=List[OrderListResponse])
|
|||
|
|
async def get_orders(
|
|||
|
|
current_user: User = Depends(get_current_user),
|
|||
|
|
service: PaymentOrderService = Depends(get_payment_order_service),
|
|||
|
|
):
|
|||
|
|
"""获取当前用户的订单列表"""
|
|||
|
|
return await service.list_orders(current_user.id)
|