@agentunion/kite 1.4.0 → 1.5.0

package/extensions/services/proxy/evol/presenter/apikeyPresenter.py

@@ -0,0 +1,106 @@
+
+import hashlib
+import secrets
+import string
+import threading
+from typing import Optional, List, Dict
+
+from ..config import API_BASE_URL
+from ..http.async_http import AsyncHttpClient
+from .configPresenter import configPresenter
+
+
+class apikeyPresenter:
+    """API Key管理Presenter"""
+    _lock = threading.RLock()
+
+    @classmethod
+    def _mask_api_key(cls, api_key: str) -> str:
+        """
+        脱敏显示API Key
+        显示规则：前12位 + ... + 后12位
+        """
+        if not api_key or len(api_key) <= 24:
+            return api_key
+        prefix = api_key[:12]
+        suffix = api_key[-12:]
+        return f"{prefix}...{suffix}"
+
+    @classmethod
+    async def get_api_key_list(cls) -> Dict:
+        """
+        查询我的API Key列表
+
+        Returns:
+            dict: 响应结果
+            {
+                "status": "success",
+                "data": {
+                    "code": 200,
+                    "msg": "查询成功",
+                    "data": [
+                        {
+                            "id": 1,
+                            "apiKey": "sk_live_abc123...mno345pqr",
+                            "name": "生产环境密钥",
+                            "status": 1,
+                            "createTime": "2025-10-13 10:00:00"
+                        }
+                    ]
+                }
+            }
+        """
+        try:
+            if not cls._is_logged_in():
+                return {
+                    "status": "error",
+                    "error": "用户未登录，请先登录"
+                }
+
+            url = f"{API_BASE_URL}/api/ai-key/list"
+            response = await AsyncHttpClient.get(url, include_token=True)
+
+            return {
+                "status": "success",
+                "data": response
+            }
+
+        except Exception as e:
+            error_msg = str(e)
+            if "Token已失效" in error_msg or "401" in error_msg:
+                return {
+                    "status": "error",
+                    "error": "Token已失效，请重新登录",
+                    "need_relogin": True
+                }
+            return {
+                "status": "error",
+                "error": f"查询API Key列表失败: {error_msg}"
+            }
+
+    @classmethod
+    async def get_first_api_key(cls) -> Optional[str]:
+        """
+        获取用户的第一个可用 API Key（脱敏显示）
+
+        Returns:
+            str or None: 第一个 API Key（脱敏），无则返回 None
+        """
+        try:
+            result = await cls.get_api_key_list()
+            if result.get("status") == "success":
+                api_data = result.get("data", {})
+                key_list = api_data.get("data", []) if isinstance(api_data, dict) else []
+                for item in key_list:
+                    api_key = item.get("apiKey")
+                    if api_key:
+                        return api_key
+        except Exception:
+            pass
+        return None
+
+    @classmethod
+    def _is_logged_in(cls) -> bool:
+        """判断用户是否已登录"""
+        token = configPresenter.get_token()
+        return token is not None and token != ""