Publish current ComfyUI O1Key code baseline

Replace the prior release tree with the current plugin, frontend, tests, and documentation. Document retired node IDs and the public Gitea update source.

utils/config.py

@@ -4,22 +4,24 @@
 """
 
 import os
+import threading
 from typing import Dict, Optional
 
 
 # 获取插件根目录
 PLUGIN_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 CONFIG_FILE = os.path.join(PLUGIN_ROOT, ".config")
+_CONFIG_LOCK = threading.RLock()
 
 
 # ============ API 基础配置 ============
 # 所有 API 客户端的统一基础 URL
 # 可通过环境变量 O1KEY_API_BASE_URL 覆盖
-DEFAULT_API_BASE_URL = "https://api.o1key.com"
+DEFAULT_API_BASE_URL = "https://api.o1key.cn"
 
 # 异步 API 基础 URL（用于异步提交+轮询模式）
 # 可通过环境变量 O1KEY_ASYNC_API_BASE_URL 覆盖
-DEFAULT_ASYNC_API_BASE_URL = "https://cf-api.o1key.com"
+DEFAULT_ASYNC_API_BASE_URL = "https://api.o1key.cn"
 
 # ============ 网络线路配置 ============
 NETWORK_ROUTES = {
@@ -28,6 +30,8 @@ NETWORK_ROUTES = {
     "美国直连": "https://api.o1key.com",
 }
 NETWORK_ROUTE_OPTIONS = ["全球加速", "CF加速", "美国直连"]
+DEFAULT_NETWORK_ROUTE = "全球加速"
+NETWORK_ROUTE_CONFIG_KEY = "O1KEY_NETWORK_ROUTE"
 
 
 def load_config(config_path: Optional[str] = None) -> Dict[str, str]:
@@ -76,6 +80,45 @@ def load_config(config_path: Optional[str] = None) -> Dict[str, str]:
     return config
 
 
+def save_config(config: Dict[str, str], config_path: Optional[str] = None) -> None:
+    """原子写入配置，避免界面快速切换时产生半写入文件。"""
+    if config_path is None:
+        config_path = CONFIG_FILE
+
+    temp_path = f"{config_path}.tmp"
+    with _CONFIG_LOCK:
+        with open(temp_path, "w", encoding="utf-8", newline="\n") as config_file:
+            for key, value in config.items():
+                config_file.write(f"{key}={value}\n")
+        os.replace(temp_path, config_path)
+
+
+def update_config(
+    updates: Optional[Dict[str, str]] = None,
+    remove: Optional[list[str]] = None,
+) -> Dict[str, str]:
+    """以一次原子写入更新配置，并返回更新后的配置。"""
+    with _CONFIG_LOCK:
+        config = load_config()
+        for key, value in (updates or {}).items():
+            config[key] = value
+        for key in remove or []:
+            config.pop(key, None)
+        save_config(config)
+        return config
+
+
+def get_runtime_config_signature() -> tuple[str, str, str, str]:
+    """用于复用客户端；仅当运行时 API 配置变化时才重建客户端。"""
+    config = load_config()
+    return (
+        config.get("O1KEY_API_KEY", ""),
+        config.get(NETWORK_ROUTE_CONFIG_KEY, DEFAULT_NETWORK_ROUTE),
+        config.get("O1KEY_API_BASE_URL", ""),
+        config.get("O1KEY_ASYNC_API_BASE_URL", ""),
+    )
+
+
 def get_api_key(key_name: str = "O1KEY_API_KEY") -> Optional[str]:
     """
     获取 API 密钥
@@ -112,6 +155,14 @@ def get_api_key_or_raise(key_name: str = "O1KEY_API_KEY") -> str:
     return api_key
 
 
+def get_network_route() -> str:
+    """Return the globally configured network route."""
+    route = load_config().get(NETWORK_ROUTE_CONFIG_KEY, DEFAULT_NETWORK_ROUTE)
+    if isinstance(route, (list, tuple)):
+        route = route[0] if route else DEFAULT_NETWORK_ROUTE
+    return route if route in NETWORK_ROUTES else DEFAULT_NETWORK_ROUTE
+
+
 def get_api_base_url() -> str:
     """
     获取 API 基础 URL
@@ -121,6 +172,9 @@ def get_api_base_url() -> str:
         API 基础 URL 字符串
     """
     config = load_config()
+    route = config.get(NETWORK_ROUTE_CONFIG_KEY)
+    if route in NETWORK_ROUTES:
+        return NETWORK_ROUTES[route].rstrip('/')
     base_url = config.get("O1KEY_API_BASE_URL")
 
     if base_url:
@@ -138,6 +192,9 @@ def get_async_api_base_url() -> str:
         异步 API 基础 URL 字符串
     """
     config = load_config()
+    route = config.get(NETWORK_ROUTE_CONFIG_KEY)
+    if route in NETWORK_ROUTES:
+        return NETWORK_ROUTES[route].rstrip('/')
     base_url = config.get("O1KEY_ASYNC_API_BASE_URL")
 
     if base_url:
@@ -146,8 +203,10 @@ def get_async_api_base_url() -> str:
     return DEFAULT_ASYNC_API_BASE_URL
 
 
-def get_base_url_by_route(route: str) -> str:
-    """根据网络线路选项返回对应域名，未匹配则走 config 垫底"""
+def get_base_url_by_route(route: Optional[str] = None) -> str:
+    """Resolve an explicit route or fall back to the global route setting."""
     if isinstance(route, (list, tuple)):
         route = route[0] if route else None
+    if route is None:
+        route = get_network_route()
     return NETWORK_ROUTES.get(route, get_api_base_url())