实现虎牙密码登录

web/backend/routers/huya.py

@@ -7,6 +7,8 @@ from datetime import datetime, timezone
 from fastapi import APIRouter, Depends, HTTPException, Query, WebSocket, WebSocketDisconnect
 from sqlalchemy.orm import Session, joinedload
 
+from core.huya import HuyaCredentialError, HuyaLoginError, login_huya_password
+
 from ..database import SessionLocal, get_db
 from ..deps import authenticate_websocket, require_permission
 from ..models import HuyaAccount, HuyaConfig, HuyaGoodsSnapshot, HuyaRechargeGoodsSnapshot, HuyaTask, User
@@ -16,6 +18,7 @@ from ..schemas import (
     HuyaConfigUpdate,
     HuyaCookieImport,
     HuyaGoodsOut,
+    HuyaPasswordLoginRequest,
     HuyaRechargeGoodsOut,
     HuyaTaskBatchRequest,
     HuyaTaskOut,
@@ -28,6 +31,7 @@ from ..services.huya_service import (
     ensure_huya_config,
     huya_config_value,
     import_huya_cookies,
+    upsert_huya_cookie,
 )
 from ..services.huya_runner import HuyaBatchRunner, huya_batch_registry
 
@@ -130,6 +134,41 @@ def import_cookies(
     }
 
 
+@router.post("/accounts/password-login")
+def password_login_account(
+    req: HuyaPasswordLoginRequest,
+    db: Session = Depends(get_db),
+    current: User = Depends(require_permission("huya:account")),
+):
+    """使用账号密码登录虎牙，成功后保存 Cookie。"""
+    try:
+        result = login_huya_password(
+            username=req.username.strip(),
+            password=req.password,
+            cookie=req.cookie.strip() or None,
+        )
+    except HuyaCredentialError as exc:
+        raise HTTPException(status_code=400, detail=str(exc)) from exc
+    except HuyaLoginError as exc:
+        raise HTTPException(status_code=502, detail=str(exc)) from exc
+    except Exception as exc:
+        raise HTTPException(status_code=502, detail=f"虎牙密码登录失败: {exc}") from exc
+
+    if not result.success or not result.cookie:
+        raise HTTPException(status_code=502, detail=result.message or "虎牙密码登录失败")
+
+    try:
+        account = upsert_huya_cookie(db, result.cookie, tag=req.tag, username_hint=req.username)
+    except ValueError as exc:
+        raise HTTPException(status_code=502, detail=str(exc)) from exc
+    return {
+        "message": "登录成功，Cookie 已保存",
+        "success": True,
+        "account": _account_out(account),
+        "sdid": result.sdid,
+    }
+
+
 @router.delete("/accounts/batch")
 def delete_accounts_batch(
     account_ids: str = Query(..., description="逗号分隔的虎牙账号ID"),