@josephyan/qingflow-cli 0.2.0-beta.55

package/src/qingflow_mcp/config.py

@@ -0,0 +1,221 @@
+from __future__ import annotations
+
+import json
+import os
+from pathlib import Path
+from typing import Any
+from urllib.parse import urlsplit, urlunsplit
+
+DEFAULT_PROFILE = "default"
+DEFAULT_TIMEOUT_SECONDS = 30.0
+DEFAULT_USER_AGENT = "qingflow-mcp/1.0"
+DEFAULT_RECORD_LIST_TYPE = 8
+ATTACHMENT_QUESTION_TYPE = 13
+DEFAULT_BASE_URL = "https://qingflow.com/api"
+DEFAULT_FEEDBACK_APP_KEY = "e0d017kju002"
+DEFAULT_FEEDBACK_QSOURCE_TOKEN = "mcp-feedback-7755d14748fc"
+
+
+def get_mcp_home() -> Path:
+    custom_home = os.getenv("QINGFLOW_MCP_HOME")
+    return Path(custom_home).expanduser() if custom_home else Path.home() / ".qingflow-mcp"
+
+
+def get_profiles_path() -> Path:
+    return get_mcp_home() / "profiles.json"
+
+
+def get_config_file_paths() -> list[Path]:
+    """
+    获取可能的配置文件路径列表，按优先级排序：
+    1. 环境变量 QINGFLOW_MCP_CONFIG_PATH 指定的路径
+    2. 当前工作目录下的 qingflow-mcp.config.json
+    3. MCP home 目录下的 config.json
+    4. 系统级配置 (Linux/Mac: /etc/qingflow-mcp/config.json)
+    """
+    paths: list[Path] = []
+
+    # 1. 环境变量
+    env_config = os.getenv("QINGFLOW_MCP_CONFIG_PATH")
+    if env_config:
+        paths.append(Path(env_config).expanduser())
+
+    # 2. 当前工作目录
+    paths.append(Path.cwd() / "qingflow-mcp.config.json")
+
+    # 3. MCP home 目录
+    paths.append(get_mcp_home() / "config.json")
+
+    # 4. 系统级配置 (仅非 Windows)
+    if os.name != "nt":
+        paths.append(Path("/etc/qingflow-mcp/config.json"))
+
+    return paths
+
+
+def load_config_file() -> dict[str, Any]:
+    """
+    加载第一个存在的配置文件
+
+    Returns:
+        配置字典，如果没有找到配置文件则返回空字典
+    """
+    for path in get_config_file_paths():
+        if path.exists():
+            try:
+                with open(path, "r", encoding="utf-8") as f:
+                    content = f.read()
+                    # 移除 JSON 注释 (简单的行注释处理)
+                    lines = []
+                    for line in content.split("\n"):
+                        stripped = line.strip()
+                        if not stripped.startswith("//") and not stripped.startswith("#"):
+                            lines.append(line)
+                    return json.loads("\n".join(lines))
+            except (json.JSONDecodeError, IOError) as e:
+                # 配置文件存在但读取失败，记录警告但不中断
+                print(f"Warning: Failed to load config from {path}: {e}")
+                continue
+    return {}
+
+
+def get_config_value(key: str, env_var: str | None = None, default: Any = None) -> Any:
+    """
+    获取配置值，优先级：环境变量 > 配置文件 > 默认值
+
+    Args:
+        key: 配置文件中的键名 (支持点号分隔的嵌套键，如 "profiles.default.name")
+        env_var: 环境变量名
+        default: 默认值
+
+    Returns:
+        配置值
+    """
+    # 1. 环境变量
+    if env_var:
+        env_value = os.getenv(env_var)
+        if env_value is not None:
+            return env_value
+
+    # 2. 配置文件
+    config = load_config_file()
+    keys = key.split(".")
+    value = config
+    for k in keys:
+        if isinstance(value, dict) and k in value:
+            value = value[k]
+        else:
+            value = None
+            break
+
+    if value is not None:
+        return value
+
+    # 3. 默认值
+    return default
+
+
+def get_default_base_url() -> str | None:
+    """获取默认的 Qingflow 后端地址"""
+    value = get_config_value(
+        "default_base_url",
+        env_var="QINGFLOW_MCP_DEFAULT_BASE_URL",
+        default=DEFAULT_BASE_URL
+    )
+    return normalize_base_url(value) if value else None
+
+
+def get_default_qf_version() -> str | None:
+    """获取默认的 qfVersion 路由值"""
+    value = get_config_value(
+        "default_qf_version",
+        env_var="QINGFLOW_MCP_DEFAULT_QF_VERSION",
+        default=None,
+    )
+    if value is None:
+        return None
+    normalized = str(value).strip()
+    return normalized or None
+
+
+def get_feedback_qsource_token() -> str | None:
+    """获取反馈 q-source 被动入口 token"""
+    value = get_config_value(
+        "feedback.qsource_token",
+        env_var="QINGFLOW_MCP_FEEDBACK_QSOURCE_TOKEN",
+        default=DEFAULT_FEEDBACK_QSOURCE_TOKEN,
+    )
+    if value is None:
+        return None
+    normalized = str(value).strip()
+    return normalized or None
+
+
+def get_feedback_base_url() -> str | None:
+    """获取反馈 q-source 使用的 base URL"""
+    value = get_config_value(
+        "feedback.base_url",
+        env_var="QINGFLOW_MCP_FEEDBACK_BASE_URL",
+        default=None,
+    )
+    if value is None:
+        return get_default_base_url()
+    normalized = normalize_base_url(value)
+    return normalized or get_default_base_url()
+
+
+def get_feedback_app_key() -> str:
+    """获取内部反馈表 app_key"""
+    value = get_config_value(
+        "feedback.app_key",
+        env_var="QINGFLOW_MCP_FEEDBACK_APP_KEY",
+        default=DEFAULT_FEEDBACK_APP_KEY,
+    )
+    normalized = str(value or "").strip()
+    return normalized or DEFAULT_FEEDBACK_APP_KEY
+
+
+def get_timeout_seconds() -> float:
+    """获取 HTTP 超时秒数"""
+    value = get_config_value(
+        "timeout_seconds",
+        env_var="QINGFLOW_MCP_TIMEOUT_SECONDS",
+        default=DEFAULT_TIMEOUT_SECONDS
+    )
+    try:
+        return float(value)
+    except (ValueError, TypeError):
+        return DEFAULT_TIMEOUT_SECONDS
+
+
+def get_log_level() -> str:
+    """获取日志级别"""
+    return get_config_value(
+        "log_level",
+        env_var="QINGFLOW_MCP_LOG_LEVEL",
+        default="INFO"
+    )
+
+
+def normalize_base_url(base_url: str | None) -> str | None:
+    """规范化 base URL"""
+    if base_url is None:
+        return None
+    normalized = base_url.strip()
+    if not normalized:
+        return None
+    normalized = normalized.rstrip("/")
+    try:
+        parsed = urlsplit(normalized)
+    except ValueError:
+        return normalized
+    if not parsed.scheme or not parsed.netloc:
+        return normalized
+
+    hostname = parsed.hostname or ""
+    if hostname.lower() == "www.qingflow.com":
+        netloc = "qingflow.com"
+        if parsed.port is not None:
+            netloc = f"{netloc}:{parsed.port}"
+        normalized = urlunsplit((parsed.scheme, netloc, parsed.path, parsed.query, parsed.fragment))
+    return normalized

package/src/qingflow_mcp/errors.py

@@ -0,0 +1,66 @@
+from __future__ import annotations
+
+import json
+from dataclasses import asdict, dataclass
+
+from .json_types import JSONObject, JSONScalar
+
+
+INVALID_TOKEN_MARKERS = (
+    "invalid token",
+    "token invalid",
+    "token失效",
+    "无效token",
+    "登录失效",
+    "login token invalid",
+    "access token invalid",
+)
+
+
+@dataclass(slots=True)
+class QingflowApiError(Exception):
+    category: str
+    message: str
+    backend_code: JSONScalar = None
+    request_id: str | None = None
+    http_status: int | None = None
+    details: JSONObject | None = None
+
+    def to_dict(self) -> JSONObject:
+        return asdict(self)
+
+    def as_json(self) -> str:
+        return json.dumps(self.to_dict(), ensure_ascii=False)
+
+    def __str__(self) -> str:
+        return self.as_json()
+
+    def looks_like_invalid_token(self) -> bool:
+        text = self.message.lower()
+        return any(marker in text for marker in INVALID_TOKEN_MARKERS)
+
+    @classmethod
+    def auth_required(cls, profile: str) -> "QingflowApiError":
+        return cls(
+            category="auth",
+            message=f"Profile '{profile}' is not logged in. Run auth_login first.",
+        )
+
+    @classmethod
+    def workspace_not_selected(cls, profile: str) -> "QingflowApiError":
+        return cls(
+            category="workspace",
+            message=f"WORKSPACE_NOT_SELECTED: profile '{profile}' has no selected workspace. Run workspace_select first.",
+        )
+
+    @classmethod
+    def config_error(cls, message: str) -> "QingflowApiError":
+        return cls(category="config", message=message)
+
+    @classmethod
+    def not_supported(cls, message: str) -> "QingflowApiError":
+        return cls(category="not_supported", message=message)
+
+
+def raise_tool_error(error: QingflowApiError) -> None:
+    raise RuntimeError(error.as_json())

package/src/qingflow_mcp/import_store.py

@@ -0,0 +1,121 @@
+from __future__ import annotations
+
+import json
+import os
+from dataclasses import dataclass
+from datetime import datetime, timedelta, timezone
+from pathlib import Path
+from typing import Any
+
+from .config import get_mcp_home
+
+
+def _utc_now() -> datetime:
+    return datetime.now(timezone.utc)
+
+
+def _parse_utc(value: Any) -> datetime | None:
+    if not isinstance(value, str) or not value.strip():
+        return None
+    normalized = value.strip().replace("Z", "+00:00")
+    try:
+        parsed = datetime.fromisoformat(normalized)
+    except ValueError:
+        return None
+    if parsed.tzinfo is None:
+        return parsed.replace(tzinfo=timezone.utc)
+    return parsed.astimezone(timezone.utc)
+
+
+def _json_safe_key(value: str) -> str:
+    keep = []
+    for char in value:
+        if char.isalnum() or char in {"-", "_"}:
+            keep.append(char)
+        else:
+            keep.append("_")
+    result = "".join(keep).strip("_")
+    return result or "entry"
+
+
+def _store_dir(env_var: str, default_name: str) -> Path:
+    custom = os.getenv(env_var)
+    if custom:
+        return Path(custom).expanduser()
+    return get_mcp_home() / default_name
+
+
+@dataclass(slots=True)
+class _JsonEntryStore:
+    base_dir: Path
+    ttl: timedelta
+
+    def __post_init__(self) -> None:
+        self.base_dir.mkdir(parents=True, exist_ok=True)
+        self.prune()
+
+    def put(self, entry_id: str, payload: dict[str, Any]) -> None:
+        data = dict(payload)
+        data["id"] = entry_id
+        data["updated_at"] = _utc_now().isoformat()
+        self._path(entry_id).write_text(json.dumps(data, ensure_ascii=False, indent=2), encoding="utf-8")
+
+    def get(self, entry_id: str) -> dict[str, Any] | None:
+        path = self._path(entry_id)
+        if not path.exists():
+            return None
+        try:
+            payload = json.loads(path.read_text(encoding="utf-8"))
+        except (OSError, json.JSONDecodeError):
+            path.unlink(missing_ok=True)
+            return None
+        created_at = _parse_utc(payload.get("created_at")) or _parse_utc(payload.get("updated_at"))
+        if created_at is None or _utc_now() - created_at > self.ttl:
+            path.unlink(missing_ok=True)
+            return None
+        return payload
+
+    def prune(self) -> None:
+        for path in self.base_dir.glob("*.json"):
+            try:
+                payload = json.loads(path.read_text(encoding="utf-8"))
+            except (OSError, json.JSONDecodeError):
+                path.unlink(missing_ok=True)
+                continue
+            created_at = _parse_utc(payload.get("created_at")) or _parse_utc(payload.get("updated_at"))
+            if created_at is None or _utc_now() - created_at > self.ttl:
+                path.unlink(missing_ok=True)
+
+    def list(self) -> list[dict[str, Any]]:
+        entries: list[dict[str, Any]] = []
+        self.prune()
+        for path in self.base_dir.glob("*.json"):
+            try:
+                payload = json.loads(path.read_text(encoding="utf-8"))
+            except (OSError, json.JSONDecodeError):
+                continue
+            created_at = _parse_utc(payload.get("created_at")) or _parse_utc(payload.get("updated_at"))
+            if created_at is None:
+                continue
+            entries.append(payload)
+        entries.sort(key=lambda item: item.get("created_at") or item.get("updated_at") or "", reverse=True)
+        return entries
+
+    def _path(self, entry_id: str) -> Path:
+        return self.base_dir / f"{_json_safe_key(entry_id)}.json"
+
+
+class ImportVerificationStore(_JsonEntryStore):
+    def __init__(self, base_dir: Path | None = None, *, ttl_seconds: int = 3600) -> None:
+        super().__init__(
+            base_dir=base_dir or _store_dir("QINGFLOW_MCP_IMPORT_VERIFY_HOME", "import-verifications"),
+            ttl=timedelta(seconds=ttl_seconds),
+        )
+
+
+class ImportJobStore(_JsonEntryStore):
+    def __init__(self, base_dir: Path | None = None, *, ttl_seconds: int = 24 * 3600) -> None:
+        super().__init__(
+            base_dir=base_dir or _store_dir("QINGFLOW_MCP_IMPORT_JOB_HOME", "import-jobs"),
+            ttl=timedelta(seconds=ttl_seconds),
+        )

package/src/qingflow_mcp/json_types.py

@@ -0,0 +1,18 @@
+from __future__ import annotations
+
+from typing import Protocol, Any
+
+# Use Any for JSON types to avoid Pydantic recursion issues
+# These are used for MCP tool signatures where exact typing is less critical
+JSONScalar = Any
+JSONValue = Any
+JSONObject = dict[str, Any]
+JSONArray = list[Any]
+
+
+class KeyringBackend(Protocol):
+    def set_password(self, service: str, key: str, value: str) -> None: ...
+
+    def get_password(self, service: str, key: str) -> str | None: ...
+
+    def delete_password(self, service: str, key: str) -> None: ...

package/src/qingflow_mcp/list_type_labels.py

@@ -0,0 +1,76 @@
+from __future__ import annotations
+
+
+RECORD_LIST_TYPE_LABELS: dict[int, str] = {
+    1: "待办",
+    2: "已办",
+    3: "我发起的-已通过",
+    4: "我发起的-已拒绝",
+    5: "我发起的-草稿",
+    6: "我发起的-待完善",
+    7: "我发起的-流程中",
+    8: "数据管理-全部数据",
+    9: "数据管理-已通过",
+    10: "数据管理-已拒绝",
+    11: "数据管理-流程中",
+    12: "抄送",
+    13: "图表分享者",
+    14: "我发起的",
+    15: "数据管理-已结束",
+    16: "我发起的-已结束",
+}
+
+SYSTEM_VIEW_DEFINITIONS: tuple[tuple[str, int, str], ...] = (
+    ("system:all", 8, "全部数据"),
+    ("system:initiated", 14, "我发起的"),
+    ("system:todo", 1, "待办"),
+    ("system:done", 2, "已办"),
+    ("system:cc", 12, "抄送我的"),
+)
+
+SYSTEM_VIEW_ID_TO_LIST_TYPE: dict[str, int] = {view_id: list_type for view_id, list_type, _ in SYSTEM_VIEW_DEFINITIONS}
+SYSTEM_VIEW_ID_TO_NAME: dict[str, str] = {view_id: name for view_id, _, name in SYSTEM_VIEW_DEFINITIONS}
+SYSTEM_LIST_TYPE_TO_VIEW_ID: dict[int, str] = {list_type: view_id for view_id, list_type, _ in SYSTEM_VIEW_DEFINITIONS}
+
+TASK_TYPE_LABELS: dict[int, str] = {
+    1: "待办",
+    2: "我发起的",
+    3: "抄送",
+    5: "已办",
+}
+
+APP_PUBLISH_STATUS_LABELS: dict[int, str] = {
+    0: "未发布",
+    1: "已发布-有修改",
+    2: "已发布",
+}
+
+
+def get_record_list_type_label(list_type: int | None) -> str | None:
+    if list_type is None:
+        return None
+    return RECORD_LIST_TYPE_LABELS.get(list_type)
+
+
+def get_system_view_id(list_type: int | None) -> str | None:
+    if list_type is None:
+        return None
+    return SYSTEM_LIST_TYPE_TO_VIEW_ID.get(list_type)
+
+
+def get_system_view_name(view_id: str | None) -> str | None:
+    if view_id is None:
+        return None
+    return SYSTEM_VIEW_ID_TO_NAME.get(view_id)
+
+
+def get_task_type_label(type_value: int | None) -> str | None:
+    if type_value is None:
+        return None
+    return TASK_TYPE_LABELS.get(type_value)
+
+
+def get_app_publish_status_label(status: int | None) -> str | None:
+    if status is None:
+        return None
+    return APP_PUBLISH_STATUS_LABELS.get(status)