feat(cookies): auto-detect JSON or KV cookie formats from sources

utils/cookie_handler.py

@@ -94,4 +94,54 @@ def convert_kv_to_playwright(kv_string, default_domain=".google.com", logger=Non
         if logger:
             logger.debug(f"成功转换 Cookie: {name} -> domain={default_domain}")
 
-    return playwright_cookies
+    return playwright_cookies
+
+
+def auto_convert_to_playwright(cookie_data, default_domain=".google.com", logger=None):
+    """
+    自动识别 Cookie 数据格式并转换为 Playwright 兼容格式。
+    支持两种输入格式:
+    1. JSON 数组 (Cookie-Editor 导出格式)
+    2. KV 字符串 (键值对格式: "name1=value1; name2=value2; ...")
+
+    Args:
+        cookie_data: Cookie 数据，可以是 list (JSON格式) 或 str (KV格式)
+        default_domain (str): KV格式使用的默认域名，默认为".google.com"
+        logger: 日志记录器
+
+    Returns:
+        list: Playwright 兼容的 Cookie 列表
+
+    Raises:
+        ValueError: 当格式无法识别时抛出异常
+    """
+    # 格式1: JSON 数组格式 (Cookie-Editor 导出格式)
+    if isinstance(cookie_data, list):
+        if logger:
+            logger.debug(f"检测到 JSON 数组格式的 Cookie 数据，共 {len(cookie_data)} 个条目")
+        return convert_cookie_editor_to_playwright(cookie_data, logger=logger)
+
+    # 格式2: KV 字符串格式
+    if isinstance(cookie_data, str):
+        # 去除首尾空白字符
+        cookie_str = cookie_data.strip()
+
+        if not cookie_str:
+            if logger:
+                logger.warning("收到空的 Cookie 字符串")
+            return []
+
+        if logger:
+            logger.debug(f"检测到 KV 字符串格式的 Cookie 数据")
+
+        return convert_kv_to_playwright(
+            cookie_str,
+            default_domain=default_domain,
+            logger=logger
+        )
+
+    # 无法识别的格式
+    error_msg = f"无法识别的 Cookie 数据格式: {type(cookie_data).__name__}"
+    if logger:
+        logger.error(error_msg)
+    raise ValueError(error_msg)

utils/cookie_manager.py

@@ -8,7 +8,7 @@ import json
 from dataclasses import dataclass
 from typing import List, Dict, Optional
 from utils.paths import cookies_dir
-from utils.cookie_handler import convert_cookie_editor_to_playwright, convert_kv_to_playwright
+from utils.cookie_handler import auto_convert_to_playwright
 from utils.common import clean_env_value
 
 @dataclass
@@ -144,29 +144,59 @@ class CookieManager:
         return cookies
 
     def _load_from_file(self, filename: str) -> List[Dict]:
-        """从JSON文件加载 Cookie"""
+        """从文件加载 Cookie，自动识别 JSON 或 KV 格式"""
         cookie_path = cookies_dir() / filename
 
         if not os.path.exists(cookie_path):
             raise FileNotFoundError(f"Cookie 文件不存在: {cookie_path}")
 
         with open(cookie_path, 'r', encoding='utf-8') as f:
-            cookies_from_file = json.load(f)
+            file_content = f.read().strip()
 
-        return convert_cookie_editor_to_playwright(cookies_from_file, logger=self.logger)
+        # 尝试解析为 JSON
+        try:
+            cookies_from_file = json.loads(file_content)
+            # JSON 解析成功，使用自动转换函数
+            return auto_convert_to_playwright(
+                cookies_from_file,
+                default_domain=".google.com",
+                logger=self.logger
+            )
+        except json.JSONDecodeError:
+            # JSON 解析失败，当作 KV 格式处理
+            if self.logger:
+                self.logger.info(f"文件 {filename} 不是有效的 JSON 格式，尝试作为 KV 格式解析")
+            return auto_convert_to_playwright(
+                file_content,
+                default_domain=".google.com",
+                logger=self.logger
+            )
 
     def _load_from_env(self, env_var_name: str) -> List[Dict]:
-        """从环境变量加载 Cookie"""
+        """从环境变量加载 Cookie，自动识别 JSON 或 KV 格式"""
         env_value = clean_env_value(os.getenv(env_var_name))
 
         if not env_value:
             raise ValueError(f"环境变量 {env_var_name} 不存在或为空")
 
-        return convert_kv_to_playwright(
-            env_value,
-            default_domain=".google.com",
-            logger=self.logger
-        )
+        # 尝试解析为 JSON
+        try:
+            cookies_from_env = json.loads(env_value)
+            # JSON 解析成功，使用自动转换函数
+            return auto_convert_to_playwright(
+                cookies_from_env,
+                default_domain=".google.com",
+                logger=self.logger
+            )
+        except json.JSONDecodeError:
+            # JSON 解析失败，当作 KV 格式处理
+            if self.logger:
+                self.logger.debug(f"环境变量 {env_var_name} 不是有效的 JSON 格式，作为 KV 格式解析")
+            return auto_convert_to_playwright(
+                env_value,
+                default_domain=".google.com",
+                logger=self.logger
+            )
 
     def get_all_sources(self) -> List[CookieSource]:
         """获取所有检测到的 Cookie 来源"""