初步增加, 扫码登录成功

2026-08-12 16:22:55 +08:00
parent 7db3b2cd4f
commit d19c0902f2
60 changed files with 7413 additions and 315 deletions
@@ -0,0 +1,124 @@
+"""应用宝和平精英充值工作台 API。"""
+
+from __future__ import annotations
+
+import uuid
+
+from fastapi import APIRouter, Depends, HTTPException, Query
+from sqlalchemy.orm import Session
+
+from ..database import get_db
+from ..deps import get_current_user, require_permission
+from ..models import User, YybRechargeTask
+from ..permissions import user_has_permission
+from ..schemas import YybLoginRequest, YybSelectionRequest, YybTaskCreateRequest
+from ..services.yyb_service import public_task, sync_task
+from ..services.yyb_worker_client import YybWorkerClient, YybWorkerError
+
+
+router = APIRouter(prefix="/api/yyb", tags=["应用宝充值"])
+
+
+def _get_task(db: Session, task_id: int, current: User) -> YybRechargeTask:
+    task = db.query(YybRechargeTask).filter(YybRechargeTask.id == task_id).first()
+    if not task:
+        raise HTTPException(404, "充值任务不存在")
+    if task.created_by != current.id and not user_has_permission(current, "yyb:history"):
+        raise HTTPException(403, "无权查看该充值任务")
+    return task
+
+
+def _worker_call(call):
+    try:
+        return call()
+    except YybWorkerError as exc:
+        raise HTTPException(502, str(exc)) from exc
+
+
+@router.post("/tasks")
+def create_task(payload: YybTaskCreateRequest, db: Session = Depends(get_db), current: User = Depends(require_permission("yyb:session"))):
+    data = _worker_call(YybWorkerClient().create_job)
+    task = YybRechargeTask(task_id=uuid.uuid4().hex[:16], created_by=current.id,
+                           worker_job_id=str(data["job_id"]), status=str(data.get("status", "created")),
+                           phase="login", message="请选择登录方式")
+    db.add(task)
+    db.commit()
+    db.refresh(task)
+    return public_task(task, include_qr=False)
+
+
+@router.post("/tasks/{task_id}/login")
+def login(task_id: int, payload: YybLoginRequest, db: Session = Depends(get_db), current: User = Depends(require_permission("yyb:session"))):
+    task = _get_task(db, task_id, current)
+    data = _worker_call(lambda: YybWorkerClient().login(task.worker_job_id, payload.provider, payload.timeout))
+    task.provider = payload.provider
+    task.status = str(data.get("status", "waiting_login"))
+    task.phase = "login"
+    task.message = "请扫码登录并在手机确认"
+    if data.get("qr_data"):
+        task.login_qr_data = data["qr_data"]
+    db.commit()
+    return public_task(task)
+
+
+@router.get("/tasks/{task_id}")
+def get_task(task_id: int, db: Session = Depends(get_db), current: User = Depends(get_current_user)):
+    task = _get_task(db, task_id, current)
+    try:
+        sync_task(db, task, YybWorkerClient())
+    except YybWorkerError:
+        # Worker 暂时重启时仍返回最近一次持久化状态。
+        pass
+    return public_task(task, include_qr=user_has_permission(current, "yyb:session"),
+                       include_payment_qr=user_has_permission(current, "yyb:recharge"))
+
+
+@router.get("/tasks")
+def list_tasks(limit: int = Query(50, ge=1, le=200), db: Session = Depends(get_db), current: User = Depends(require_permission("yyb:history"))):
+    tasks = db.query(YybRechargeTask).order_by(YybRechargeTask.id.desc()).limit(limit).all()
+    return [public_task(task, include_qr=False) for task in tasks]
+
+
+@router.get("/tasks/{task_id}/selection-options")
+def selection_options(task_id: int, platform: str = Query("android"), points: int | None = Query(None), zone_id: str | None = Query(None), db: Session = Depends(get_db), current: User = Depends(require_permission("yyb:session"))):
+    task = _get_task(db, task_id, current)
+    data = _worker_call(lambda: YybWorkerClient().selection_options(task.worker_job_id, platform, points, zone_id))
+    task.platform = platform
+    db.commit()
+    return data
+
+
+@router.post("/tasks/{task_id}/selection")
+def selection(task_id: int, payload: YybSelectionRequest, db: Session = Depends(get_db), current: User = Depends(require_permission("yyb:session"))):
+    task = _get_task(db, task_id, current)
+    data = _worker_call(lambda: YybWorkerClient().selection(task.worker_job_id, payload.model_dump()))
+    selected = data.get("selection", payload.model_dump())
+    for field in ("platform", "points", "product_id", "zone_id", "zone_name", "role_id", "role_name"):
+        setattr(task, field, selected[field])
+    task.phase, task.status, task.message = "payment", "ready", "选择已保存，可以创建付款码"
+    db.commit()
+    return public_task(task)
+
+
+@router.post("/tasks/{task_id}/payment")
+def payment(task_id: int, db: Session = Depends(get_db), current: User = Depends(require_permission("yyb:recharge"))):
+    task = _get_task(db, task_id, current)
+    if not task.product_id or not task.role_id:
+        raise HTTPException(400, "请先完成商品、区服和角色选择")
+    data = _worker_call(lambda: YybWorkerClient().payment(task.worker_job_id))
+    task.status = str(data.get("status", "running"))
+    task.phase = "payment"
+    task.message = "正在创建订单和付款码"
+    db.commit()
+    return public_task(task)
+
+
+@router.post("/tasks/{task_id}/stop")
+def stop(task_id: int, db: Session = Depends(get_db), current: User = Depends(require_permission("yyb:session"))):
+    task = _get_task(db, task_id, current)
+    data = _worker_call(lambda: YybWorkerClient().stop(task.worker_job_id))
+    task.status = "failed"
+    task.phase = "stopped"
+    task.message = "任务已停止"
+    db.commit()
+    return public_task(task)