@comate/zulu 1.1.0 → 1.2.0

package/comate-engine/assets/skills/smart-commit/scripts/icafe/client.py

@@ -0,0 +1,444 @@
+"""iCafe API 客户端
+
+包含常量、配置类和 API 客户端。
+"""
+
+import os
+from dataclasses import dataclass
+from pathlib import Path
+from typing import Dict, List, Any, Optional
+
+import requests
+
+from logger import get_logger
+
+logger = get_logger()
+
+
+# ========================================
+# 常量配置
+# ========================================
+
+# iCafe API base URL
+ICAFE_BASE_URL = "http://10.11.152.208:8701/api/process/icafe"
+
+# Farseer API base URL
+FARSEER_BASE_URL = "https://farseer.baidu.com/api/v1"
+
+# 默认超时时间（秒）
+DEFAULT_TIMEOUT = 30
+
+# 默认查询参数
+DEFAULT_LOOKBACK_DAYS = 14
+DEFAULT_MAX_RECORDS = 20
+
+# Git log 提取时的最大提交数
+DEFAULT_MAX_COMMITS = 200
+
+# Git diff 输出的最大行数
+DEFAULT_MAX_DIFF_LINES = 500
+
+# 不是 iCafe 空间 ID 的常见前缀（小写）
+GIT_LOG_FALSE_POSITIVE_PREFIXES = {
+    'v', 'sha', 'utf', 'iso', 'rfc', 'p', 'rc', 'pr', 'mr',
+    'http', 'https', 'feat', 'fix', 'chore', 'docs', 'style',
+    'refactor', 'perf', 'test', 'build', 'ci', 'revert',
+}
+
+# 卡片类型 ID -> 名称映射表
+ISSUE_TYPE_MAP = {
+    56075: "Epic",
+    54443: "Feature",
+    5007: "Story",
+    54444: "Task",
+    54621: "Tech Feature",
+    54622: "Tech Task",
+    5009: "Bug",
+    5811: "Bug(线上)",
+    5011: "Case",
+    65085: "非研发任务",
+    49460: "项目",
+    5008: "需求",
+    5010: "任务",
+    88805: "Prompt Story",
+    88806: "Prompt Task",
+    88807: "GenAI Data Story",
+    88808: "GenAI Data Task",
+}
+
+# 反向映射：名称 -> issueTypeId
+ISSUE_TYPE_NAME_MAP = {v: k for k, v in ISSUE_TYPE_MAP.items()}
+
+
+# ========================================
+# 配置类
+# ========================================
+
+@dataclass
+class ICafeQueryConfig:
+    """iCafe 查询 API 配置"""
+
+    base_url: str = ICAFE_BASE_URL
+    timeout: int = DEFAULT_TIMEOUT
+
+    @staticmethod
+    def _get_auth_token() -> Optional[str]:
+        """获取认证 token
+
+        优先从环境变量 COMATE_AUTH_TOKEN 读取，
+        如果没有则从 ~/.comate/login 文件读取
+
+        Returns:
+            认证 token 字符串，如果都未找到则返回 None
+        """
+        token = os.environ.get("COMATE_AUTH_TOKEN")
+        if token:
+            logger.info("Token 来源: env")
+            return token
+
+        login_file = Path.home() / ".comate" / "login"
+        if login_file.exists():
+            try:
+                content = login_file.read_text().strip()
+                if content:
+                    logger.info("Token 来源: file (~/.comate/login)")
+                    return content
+            except Exception:
+                pass
+
+        logger.warning("Token 来源: none, 未找到认证 token")
+        return None
+
+
+# ========================================
+# API 客户端
+# ========================================
+
+class ICafeQueryClient:
+    """iCafe 卡片查询客户端
+
+    专注于卡片查询场景，支持：
+    - 通过 IQL 条件表达式查询卡片
+    - 通过卡片 ID 获取卡片详情
+    - 查询空间列表和空间计划
+    - 分页、排序、展示选项等高级查询能力
+    """
+
+    def __init__(self, config: Optional[ICafeQueryConfig] = None):
+        self.config = config or ICafeQueryConfig()
+        self.session = requests.Session()
+
+        self.session.headers.update({
+            "Content-Type": "application/json"
+        })
+
+        auth_token = self.config._get_auth_token()
+        if auth_token:
+            self.session.headers["x-ac-Authorization"] = auth_token
+
+    def _get_headers(self) -> Dict[str, str]:
+        """获取通用请求 headers"""
+        return {
+            'User-Agent': 'iAPI/1.0.0 (http://iapi.baidu-int.com)'
+        }
+
+    def _get_with_retry(self, url: str, params: dict, max_retries: int = 3) -> requests.Response:
+        """带重试的 GET 请求"""
+        import time
+        last_exc = None
+        for attempt in range(max_retries + 1):
+            try:
+                response = self.session.get(
+                    url, params=params, headers=self._get_headers(),
+                    timeout=self.config.timeout
+                )
+                response.raise_for_status()
+                return response
+            except Exception as e:
+                last_exc = e
+                if attempt < max_retries:
+                    logger.warning("API 请求失败 (attempt %d/%d): %s, error=%s", attempt + 1, max_retries, url, e)
+                    time.sleep(0.5 * (attempt + 1))
+        logger.error("API 请求最终失败: %s, error=%s", url, last_exc)
+        raise last_exc
+
+    # ========================================
+    # 1. 通过 IQL 条件查询卡片
+    # ========================================
+    def query_cards(self, space_id: str, iql: str,
+                    page: Optional[int] = None,
+                    show_detail: bool = False,
+                    show_associations: bool = False,
+                    is_desc: bool = False,
+                    order: Optional[str] = None,
+                    show_children: bool = False,
+                    max_records: Optional[int] = None,
+                    show_okr: bool = False,
+                    show_accumulate: bool = False) -> Dict[str, Any]:
+        """通过 IQL 条件表达式查询卡片
+
+        Args:
+            space_id: 空间 ID，如 "joytest"
+            iql: IQL 查询表达式，如 "类型 = Bug AND 负责人 = currentUser"
+            page: 页码（从 1 开始），可选
+            show_detail: 是否显示卡片详情内容，默认 False
+            show_associations: 是否显示关联信息，默认 False
+            is_desc: 是否降序排列，默认 False
+            order: 排序字段（如 "创建时间"、"更新时间"），可选
+            show_children: 是否显示子卡片，默认 False
+            max_records: 最大返回记录数，可选
+            show_okr: 是否显示 OKR 信息，默认 False
+            show_accumulate: 是否显示累计信息，默认 False
+
+        Returns:
+            查询结果字典，包含卡片列表和分页信息
+
+        Raises:
+            ValueError: 当 space_id 或 iql 为空时
+            requests.RequestException: 请求失败时
+
+        IQL 表达式示例：
+            - "类型 = Bug"
+            - "负责人 = currentUser"
+            - "负责人 = dongkexin01"
+            - "类型 = Bug AND 负责人 = currentUser"
+            - "创建时间 > \\"2025-01-01\\""
+            - "类型 in (Bug, Epic, Story)"
+            - "标题 ~ 测试"
+            - "流程状态 in (新建, 开发中)"
+        """
+        if not space_id:
+            raise ValueError("space_id 不能为空")
+        if not iql:
+            raise ValueError("iql 查询表达式不能为空")
+
+        params = {'iql': iql}
+        if page is not None:
+            params['page'] = page
+        if show_detail:
+            params['showDetail'] = ''
+        if show_associations:
+            params['showAssociations'] = ''
+        if is_desc:
+            params['isDesc'] = ''
+        if order:
+            params['order'] = order
+        if show_children:
+            params['showChildren'] = ''
+        if max_records:
+            params['maxRecords'] = max_records
+        if show_okr:
+            params['showOkr'] = ''
+        if show_accumulate:
+            params['showAccumulate'] = ''
+
+        url = f"{self.config.base_url}/api/spaces/{space_id}/cards"
+        logger.info("查询卡片: space=%s, iql=%s", space_id, iql)
+        response = self._get_with_retry(url, params)
+        result = response.json()
+        cards_count = len(result.get("cards", []))
+        logger.info("查询卡片返回: %d 张", cards_count)
+        return result
+
+    # ========================================
+    # 2. 通过卡片 ID 获取卡片详情
+    # ========================================
+    def get_card_by_id(self, space_id: str, card_id: str,
+                       show_associations: bool = False,
+                       show_children: bool = False,
+                       show_okr: bool = False,
+                       show_accumulate: bool = False) -> Dict[str, Any]:
+        """根据空间 ID 和卡片 ID 获取单张卡片详情
+
+        Args:
+            space_id: 空间 ID
+            card_id: 卡片 ID
+            show_associations: 是否显示关联信息
+            show_children: 是否显示子卡片
+            show_okr: 是否显示 OKR 信息
+            show_accumulate: 是否显示累计信息
+
+        Returns:
+            卡片信息字典，包含 id, title, description, status, assignee 等
+
+        Raises:
+            ValueError: 当 space_id 或 card_id 为空时
+            requests.RequestException: 请求失败时
+        """
+        if not space_id or not card_id:
+            raise ValueError("space_id 和 card_id 不能为空")
+
+        params = {}
+        if show_associations:
+            params['showAssociations'] = ''
+        if show_children:
+            params['showChildren'] = ''
+        if show_okr:
+            params['showOkr'] = ''
+        if show_accumulate:
+            params['showAccumulate'] = ''
+
+        url = f"{self.config.base_url}/api/spaces/{space_id}/cards/{card_id}"
+        response = self._get_with_retry(url, params)
+        return response.json()
+
+    # ========================================
+    # 3. 查询最近访问的空间列表
+    # ========================================
+    def get_latest_spaces(self, current_user: str, limit: int = 3) -> List[Dict[str, Any]]:
+        """查询用户最近访问的空间列表
+
+        Args:
+            current_user: 用户 ID，如 "dongkexin01"
+            limit: 最多返回的空间数量，默认 3
+
+        Returns:
+            空间列表，每个空间包含 id, name, prefix_code, access_time 等
+
+        Raises:
+            ValueError: 当 current_user 为空时
+            requests.RequestException: 请求失败时
+        """
+        if not current_user:
+            raise ValueError("current_user 不能为空")
+
+        logger.info("查询最近访问空间: user=%s, limit=%d", current_user, limit)
+        url = f"{self.config.base_url}/api/v2/space/latest"
+        params = {'currentUser': current_user}
+        response = self._get_with_retry(url, params)
+        data = response.json()
+
+        # 对返回的空间列表截取前 limit 个
+        if isinstance(data, dict) and 'result' in data:
+            data['result'] = data['result'][:limit]
+        elif isinstance(data, list):
+            data = data[:limit]
+
+        return data
+
+    # ========================================
+    # 4. 获取空间内所有计划
+    # ========================================
+    def get_space_plans(self, space_id: str) -> List[Dict[str, Any]]:
+        """获取空间内所有计划
+
+        Args:
+            space_id: 空间 ID
+
+        Returns:
+            计划列表，每个计划包含 id, name, start_date, end_date, status 等
+
+        Raises:
+            ValueError: 当 space_id 为空时
+            requests.RequestException: 请求失败时
+        """
+        if not space_id:
+            raise ValueError("space_id 不能为空")
+
+        url = f"{self.config.base_url}/api/v2/space/{space_id}/plans"
+        response = self._get_with_retry(url, {})
+        return response.json()
+
+    # ========================================
+    # 5. 获取空间卡片类型列表
+    # ========================================
+    def get_space_issue_types(self, space_prefix: str) -> List[Dict[str, Any]]:
+        """获取空间的卡片类型列表
+
+        Args:
+            space_prefix: 空间前缀（如 "dkx"）
+
+        Returns:
+            卡片类型列表，每个类型包含 id, name 等
+
+        Raises:
+            ValueError: 当 space_prefix 为空时
+            requests.RequestException: 请求失败时
+        """
+        if not space_prefix:
+            raise ValueError("space_prefix 不能为空")
+
+        url = f"{self.config.base_url}/api/v2/space/{space_prefix}/issueTypes"
+        response = self._get_with_retry(url, {})
+        return response.json()
+
+    # ========================================
+    # 6. 创建卡片
+    # ========================================
+    def create_card(
+        self,
+        title: str,
+        space_id: str,
+        card_type: str,
+        description: str = "",
+        assignee_id: Optional[str] = None,
+    ) -> Optional[Dict[str, Any]]:
+        """创建 iCafe 卡片
+
+        Args:
+            title: 卡片标题
+            space_id: 空间前缀（字符串类型，如 "dkx"），注意不是数字 ID
+            card_type: 卡片类型名称（如 "Bug", "Story"）
+            description: 卡片描述，默认为空
+            assignee_id: 负责人 ID，默认为当前用户
+
+        Returns:
+            创建成功的卡片信息，包含：id, title, type, sequence 等。
+            失败时返回 None。
+
+        Note:
+            此实现完全参考 icafe-card-assistant 的 icafe_client.py create_card 方法
+            重要: space_id 参数应该是空间前缀（如 "dkx"），而不是数字 ID
+        """
+        try:
+            if not title or not space_id:
+                return None
+
+            logger.info("创建卡片: title=%s, space=%s, type=%s, assignee=%s", title, space_id, card_type, assignee_id)
+            issue = {
+                'title': title,
+                'detail': description,  # 简化处理，不做 HTML 转换
+            }
+
+            if card_type:
+                issue['type'] = card_type
+
+            # 构建自定义字段
+            fields = {}
+            if assignee_id:
+                fields['负责人'] = assignee_id
+
+            if fields:
+                issue['fields'] = fields
+
+            data = {
+                'issues': [issue]
+            }
+
+            url = f"{self.config.base_url}/api/v2/space/{space_id}/issue/new"
+            headers = {
+                'User-Agent': 'iAPI/1.0.0 (http://iapi.baidu-int.com)',
+                'Content-Type': 'application/json'
+            }
+
+            # 使用 self.session.post 发送请求
+            response = self.session.post(url, json=data, headers=headers, timeout=self.config.timeout)
+            response.raise_for_status()
+            result = response.json()
+
+            # 检查返回的状态
+            if isinstance(result, dict):
+                response_status = result.get('status')
+                if response_status != 200:
+                    logger.error("创建卡片失败: API 返回业务状态码 %s, 消息: %s", response_status, result.get('message', 'Unknown'))
+                    print(f"错误: API 返回业务状态码 {response_status}, 消息: {result.get('message', 'Unknown')}", file=__import__('sys').stderr)
+                    return None
+
+            issues = result.get("issues", [])
+            seq = issues[0].get("sequence") if issues else "unknown"
+            logger.info("创建卡片成功: sequence=%s", seq)
+            return result
+        except Exception as e:
+            logger.error("创建卡片异常: %s: %s", type(e).__name__, e)
+            print(f"创建卡片异常: {type(e).__name__}: {e}", file=__import__('sys').stderr)
+            return None

package/comate-engine/assets/skills/smart-commit/scripts/icafe/farseer.py

@@ -0,0 +1,53 @@
+"""Farseer API 客户端"""
+
+import requests
+from typing import Dict, List, Any, Optional
+
+from .client import FARSEER_BASE_URL, DEFAULT_TIMEOUT, ISSUE_TYPE_MAP
+from logger import get_logger
+
+logger = get_logger()
+
+
+def get_binding_card_types(space_id: int, timeout: int = DEFAULT_TIMEOUT) -> List[Dict[str, Any]]:
+    """查询空间可绑定的卡片类型列表（含 ID 和名称）
+
+    调用 farseer storyRule 接口获取空间的 engineeringBindingTypes，
+    返回包含 issueTypeId 和类型名称的完整列表。
+
+    Args:
+        space_id: 空间数字 ID（如 56355）
+        timeout: 请求超时时间（秒）
+
+    Returns:
+        卡片类型列表，如 [{"id": "5009", "name": "Bug"}, {"id": "5007", "name": "Story"}]。
+        接口调用失败或无配置时返回空列表。
+    """
+    logger.info("查询 farseer 可绑定类型: space_id=%s", space_id)
+    try:
+        url = f"{FARSEER_BASE_URL}/storyRule/getRule"
+        response = requests.get(
+            url,
+            params={'icafeSpaceId': space_id},
+            timeout=timeout
+        )
+        response.raise_for_status()
+        data = response.json()
+
+        rule_data = data.get('data', {}).get('data', {})
+        binding_types = rule_data.get('engineeringBindingTypes', [])
+
+        result = []
+        for item in binding_types:
+            type_id = item.get('issueTypeId')
+            if type_id and type_id in ISSUE_TYPE_MAP:
+                result.append({
+                    'id': str(type_id),
+                    'name': ISSUE_TYPE_MAP[type_id]
+                })
+
+        logger.info("获取到 %d 种可绑定卡片类型", len(result))
+        return result
+    except Exception as e:
+        logger.warning("farseer 查询失败: %s", e)
+        return []