feat(service): 拆分服务

src/core/sub2api_upload.py

@@ -0,0 +1,202 @@
+"""
+Sub2API 账号上传功能
+将账号以 sub2api-data 格式批量导入到 Sub2API 平台
+"""
+
+import json
+import logging
+from datetime import datetime, timezone
+from typing import List, Tuple, Optional
+
+from curl_cffi import requests as cffi_requests
+
+from ..database.session import get_db
+from ..database.models import Account
+
+logger = logging.getLogger(__name__)
+
+
+def upload_to_sub2api(
+    accounts: List[Account],
+    api_url: str,
+    api_key: str,
+    concurrency: int = 3,
+    priority: int = 50,
+) -> Tuple[bool, str]:
+    """
+    上传账号列表到 Sub2API 平台（不走代理）
+
+    Args:
+        accounts: 账号模型实例列表
+        api_url: Sub2API 地址，如 http://host
+        api_key: Admin API Key（x-api-key header）
+        concurrency: 账号并发数，默认 3
+        priority: 账号优先级，默认 50
+
+    Returns:
+        (成功标志, 消息)
+    """
+    if not accounts:
+        return False, "无可上传的账号"
+
+    if not api_url:
+        return False, "Sub2API URL 未配置"
+
+    if not api_key:
+        return False, "Sub2API API Key 未配置"
+
+    exported_at = datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ")
+
+    account_items = []
+    for acc in accounts:
+        if not acc.access_token:
+            continue
+        account_items.append({
+            "name": acc.email,
+            "platform": "openai",
+            "type": "oauth",
+            "credentials": {
+                "access_token": acc.access_token,
+            },
+            "concurrency": concurrency,
+            "priority": priority,
+        })
+
+    if not account_items:
+        return False, "所有账号均缺少 access_token，无法上传"
+
+    payload = {
+        "data": {
+            "type": "sub2api-data",
+            "version": 1,
+            "exported_at": exported_at,
+            "proxies": [],
+            "accounts": account_items,
+        },
+        "skip_default_group_bind": True,
+    }
+
+    url = api_url.rstrip("/") + "/api/v1/admin/accounts/data"
+    headers = {
+        "Content-Type": "application/json",
+        "x-api-key": api_key,
+        "Idempotency-Key": f"import-{exported_at}",
+    }
+
+    try:
+        response = cffi_requests.post(
+            url,
+            json=payload,
+            headers=headers,
+            proxies=None,
+            timeout=30,
+            impersonate="chrome110",
+        )
+
+        if response.status_code in (200, 201):
+            return True, f"成功上传 {len(account_items)} 个账号"
+
+        error_msg = f"上传失败: HTTP {response.status_code}"
+        try:
+            detail = response.json()
+            if isinstance(detail, dict):
+                error_msg = detail.get("message", error_msg)
+        except Exception:
+            error_msg = f"{error_msg} - {response.text[:200]}"
+        return False, error_msg
+
+    except Exception as e:
+        logger.error(f"Sub2API 上传异常: {e}")
+        return False, f"上传异常: {str(e)}"
+
+
+def batch_upload_to_sub2api(
+    account_ids: List[int],
+    api_url: str,
+    api_key: str,
+    concurrency: int = 3,
+    priority: int = 50,
+) -> dict:
+    """
+    批量上传指定 ID 的账号到 Sub2API 平台
+
+    Returns:
+        包含成功/失败/跳过统计和详情的字典
+    """
+    results = {
+        "success_count": 0,
+        "failed_count": 0,
+        "skipped_count": 0,
+        "details": []
+    }
+
+    with get_db() as db:
+        accounts = []
+        for account_id in account_ids:
+            acc = db.query(Account).filter(Account.id == account_id).first()
+            if not acc:
+                results["failed_count"] += 1
+                results["details"].append({"id": account_id, "email": None, "success": False, "error": "账号不存在"})
+                continue
+            if not acc.access_token:
+                results["skipped_count"] += 1
+                results["details"].append({"id": account_id, "email": acc.email, "success": False, "error": "缺少 access_token"})
+                continue
+            accounts.append(acc)
+
+        if not accounts:
+            return results
+
+        success, message = upload_to_sub2api(accounts, api_url, api_key, concurrency, priority)
+
+        if success:
+            for acc in accounts:
+                results["success_count"] += 1
+                results["details"].append({"id": acc.id, "email": acc.email, "success": True, "message": message})
+        else:
+            for acc in accounts:
+                results["failed_count"] += 1
+                results["details"].append({"id": acc.id, "email": acc.email, "success": False, "error": message})
+
+    return results
+
+
+def test_sub2api_connection(api_url: str, api_key: str) -> Tuple[bool, str]:
+    """
+    测试 Sub2API 连接（GET /api/v1/admin/accounts/data 探活）
+
+    Returns:
+        (成功标志, 消息)
+    """
+    if not api_url:
+        return False, "API URL 不能为空"
+    if not api_key:
+        return False, "API Key 不能为空"
+
+    url = api_url.rstrip("/") + "/api/v1/admin/accounts/data"
+    headers = {"x-api-key": api_key}
+
+    try:
+        response = cffi_requests.get(
+            url,
+            headers=headers,
+            proxies=None,
+            timeout=10,
+            impersonate="chrome110",
+        )
+
+        if response.status_code in (200, 201, 204, 405):
+            return True, "Sub2API 连接测试成功"
+        if response.status_code == 401:
+            return False, "连接成功，但 API Key 无效"
+        if response.status_code == 403:
+            return False, "连接成功，但权限不足"
+
+        return False, f"服务器返回异常状态码: {response.status_code}"
+
+    except cffi_requests.exceptions.ConnectionError as e:
+        return False, f"无法连接到服务器: {str(e)}"
+    except cffi_requests.exceptions.Timeout:
+        return False, "连接超时，请检查网络配置"
+    except Exception as e:
+        return False, f"连接测试失败: {str(e)}"