codex-register/src/web/routes/upload/sub2api_services.py

"""
Sub2API 服务管理 API 路由
"""

from typing import List, Optional
from fastapi import APIRouter, HTTPException
from pydantic import BaseModel

from ....database import crud
from ....database.session import get_db
from ....core.upload.sub2api_upload import test_sub2api_connection, batch_upload_to_sub2api

router = APIRouter()


# ============== Pydantic Models ==============

class Sub2ApiServiceCreate(BaseModel):
    name: str
    api_url: str
    api_key: str
    enabled: bool = True
    priority: int = 0


class Sub2ApiServiceUpdate(BaseModel):
    name: Optional[str] = None
    api_url: Optional[str] = None
    api_key: Optional[str] = None
    enabled: Optional[bool] = None
    priority: Optional[int] = None


class Sub2ApiServiceResponse(BaseModel):
    id: int
    name: str
    api_url: str
    has_key: bool
    enabled: bool
    priority: int
    created_at: Optional[str] = None
    updated_at: Optional[str] = None

    class Config:
        from_attributes = True


class Sub2ApiTestRequest(BaseModel):
    api_url: Optional[str] = None
    api_key: Optional[str] = None


class Sub2ApiUploadRequest(BaseModel):
    account_ids: List[int]
    service_id: Optional[int] = None
    concurrency: int = 3
    priority: int = 50


def _to_response(svc) -> Sub2ApiServiceResponse:
    return Sub2ApiServiceResponse(
        id=svc.id,
        name=svc.name,
        api_url=svc.api_url,
        has_key=bool(svc.api_key),
        enabled=svc.enabled,
        priority=svc.priority,
        created_at=svc.created_at.isoformat() if svc.created_at else None,
        updated_at=svc.updated_at.isoformat() if svc.updated_at else None,
    )


# ============== API Endpoints ==============

@router.get("", response_model=List[Sub2ApiServiceResponse])
async def list_sub2api_services(enabled: Optional[bool] = None):
    """获取 Sub2API 服务列表"""
    with get_db() as db:
        services = crud.get_sub2api_services(db, enabled=enabled)
        return [_to_response(s) for s in services]


@router.post("", response_model=Sub2ApiServiceResponse)
async def create_sub2api_service(request: Sub2ApiServiceCreate):
    """新增 Sub2API 服务"""
    with get_db() as db:
        svc = crud.create_sub2api_service(
            db,
            name=request.name,
            api_url=request.api_url,
            api_key=request.api_key,
            enabled=request.enabled,
            priority=request.priority,
        )
        return _to_response(svc)


@router.get("/{service_id}", response_model=Sub2ApiServiceResponse)
async def get_sub2api_service(service_id: int):
    """获取单个 Sub2API 服务详情"""
    with get_db() as db:
        svc = crud.get_sub2api_service_by_id(db, service_id)
        if not svc:
            raise HTTPException(status_code=404, detail="Sub2API 服务不存在")
        return _to_response(svc)


@router.get("/{service_id}/full")
async def get_sub2api_service_full(service_id: int):
    """获取 Sub2API 服务完整配置（含 API Key）"""
    with get_db() as db:
        svc = crud.get_sub2api_service_by_id(db, service_id)
        if not svc:
            raise HTTPException(status_code=404, detail="Sub2API 服务不存在")
        return {
            "id": svc.id,
            "name": svc.name,
            "api_url": svc.api_url,
            "api_key": svc.api_key,
            "enabled": svc.enabled,
            "priority": svc.priority,
        }


@router.patch("/{service_id}", response_model=Sub2ApiServiceResponse)
async def update_sub2api_service(service_id: int, request: Sub2ApiServiceUpdate):
    """更新 Sub2API 服务配置"""
    with get_db() as db:
        svc = crud.get_sub2api_service_by_id(db, service_id)
        if not svc:
            raise HTTPException(status_code=404, detail="Sub2API 服务不存在")

        update_data = {}
        if request.name is not None:
            update_data["name"] = request.name
        if request.api_url is not None:
            update_data["api_url"] = request.api_url
        # api_key 留空则保持原值
        if request.api_key:
            update_data["api_key"] = request.api_key
        if request.enabled is not None:
            update_data["enabled"] = request.enabled
        if request.priority is not None:
            update_data["priority"] = request.priority

        svc = crud.update_sub2api_service(db, service_id, **update_data)
        return _to_response(svc)


@router.delete("/{service_id}")
async def delete_sub2api_service(service_id: int):
    """删除 Sub2API 服务"""
    with get_db() as db:
        svc = crud.get_sub2api_service_by_id(db, service_id)
        if not svc:
            raise HTTPException(status_code=404, detail="Sub2API 服务不存在")
        crud.delete_sub2api_service(db, service_id)
        return {"success": True, "message": f"Sub2API 服务 {svc.name} 已删除"}


@router.post("/{service_id}/test")
async def test_sub2api_service(service_id: int):
    """测试 Sub2API 服务连接"""
    with get_db() as db:
        svc = crud.get_sub2api_service_by_id(db, service_id)
        if not svc:
            raise HTTPException(status_code=404, detail="Sub2API 服务不存在")
        success, message = test_sub2api_connection(svc.api_url, svc.api_key)
        return {"success": success, "message": message}


@router.post("/test-connection")
async def test_sub2api_connection_direct(request: Sub2ApiTestRequest):
    """直接测试 Sub2API 连接（用于添加前验证）"""
    if not request.api_url or not request.api_key:
        raise HTTPException(status_code=400, detail="api_url 和 api_key 不能为空")
    success, message = test_sub2api_connection(request.api_url, request.api_key)
    return {"success": success, "message": message}


@router.post("/upload")
async def upload_accounts_to_sub2api(request: Sub2ApiUploadRequest):
    """批量上传账号到 Sub2API 平台"""
    if not request.account_ids:
        raise HTTPException(status_code=400, detail="账号 ID 列表不能为空")

    with get_db() as db:
        if request.service_id:
            svc = crud.get_sub2api_service_by_id(db, request.service_id)
        else:
            svcs = crud.get_sub2api_services(db, enabled=True)
            svc = svcs[0] if svcs else None

        if not svc:
            raise HTTPException(status_code=400, detail="未找到可用的 Sub2API 服务")

        api_url = svc.api_url
        api_key = svc.api_key

    results = batch_upload_to_sub2api(
        request.account_ids,
        api_url,
        api_key,
        concurrency=request.concurrency,
        priority=request.priority,
    )
    return results