@josephyan/qingflow-cli 0.2.0-beta.1000

package/src/qingflow_mcp/tools/custom_button_tools.py

@@ -0,0 +1,202 @@
+from __future__ import annotations
+
+from copy import deepcopy
+
+from ..config import DEFAULT_PROFILE
+from ..errors import QingflowApiError, raise_tool_error
+from ..json_types import JSONObject
+from .base import ToolBase, tool_cn_name
+
+
+class CustomButtonTools(ToolBase):
+    """自定义按钮工具（中文名：按钮配置管理）。
+
+    类型：应用配置工具。
+    主要职责：
+    1. 查询应用自定义按钮；
+    2. 创建、更新、删除按钮配置；
+    3. 支持草稿与发布态按钮配置维护。
+    """
+
+    @tool_cn_name("自定义按钮列表")
+    def custom_button_list(
+        self,
+        *,
+        profile: str,
+        app_key: str,
+        being_draft: bool = True,
+        include_raw: bool = False,
+    ) -> JSONObject:
+        """执行工具方法逻辑。"""
+        self._require_app_key(app_key)
+
+        def runner(session_profile, context):
+            result = self.backend.request(
+                "GET",
+                context,
+                f"/app/{app_key}/customButton",
+                params={"beingDraft": being_draft},
+            )
+            items = []
+            raw_items = result.get("result") if isinstance(result, dict) and isinstance(result.get("result"), list) else []
+            for item in raw_items:
+                if not isinstance(item, dict):
+                    continue
+                items.append(self._compact_button_base_info(item))
+            response = {
+                "profile": profile,
+                "ws_id": session_profile.selected_ws_id,
+                "app_key": app_key,
+                "being_draft": being_draft,
+                "items": result if include_raw else items,
+                "count": len(items),
+                "compact": not include_raw,
+            }
+            if include_raw:
+                response["summary"] = items
+            return response
+
+        return self._run(profile, runner)
+
+    @tool_cn_name("自定义按钮详情")
+    def custom_button_get(
+        self,
+        *,
+        profile: str,
+        app_key: str,
+        button_id: int,
+        being_draft: bool = True,
+        include_raw: bool = False,
+    ) -> JSONObject:
+        """执行工具方法逻辑。"""
+        self._require_app_key(app_key)
+        self._require_button_id(button_id)
+
+        def runner(session_profile, context):
+            params = {"beingDraft": being_draft}
+            result = self.backend.request("GET", context, f"/app/{app_key}/customButton/{button_id}", params=params)
+            response = {
+                "profile": profile,
+                "ws_id": session_profile.selected_ws_id,
+                "app_key": app_key,
+                "button_id": button_id,
+                "being_draft": being_draft,
+                "result": result if include_raw else self._compact_button_detail(result if isinstance(result, dict) else {}),
+                "compact": not include_raw,
+            }
+            if include_raw:
+                response["summary"] = self._compact_button_detail(result if isinstance(result, dict) else {})
+            return response
+
+        return self._run(profile, runner)
+
+    @tool_cn_name("创建自定义按钮")
+    def custom_button_create(self, *, profile: str, app_key: str, payload: JSONObject) -> JSONObject:
+        """执行工具方法逻辑。"""
+        self._require_app_key(app_key)
+        body = self._require_dict(payload)
+
+        def runner(session_profile, context):
+            result = self.backend.request("POST", context, f"/app/{app_key}/customButton", json_body=deepcopy(body))
+            return {"profile": profile, "ws_id": session_profile.selected_ws_id, "app_key": app_key, "result": result}
+
+        return self._run(profile, runner)
+
+    @tool_cn_name("更新自定义按钮")
+    def custom_button_update(self, *, profile: str, app_key: str, button_id: int, payload: JSONObject) -> JSONObject:
+        """执行工具方法逻辑。"""
+        self._require_app_key(app_key)
+        self._require_button_id(button_id)
+        body = self._require_dict(payload)
+
+        def runner(session_profile, context):
+            result = self.backend.request(
+                "POST",
+                context,
+                f"/app/{app_key}/customButton/{button_id}",
+                json_body=deepcopy(body),
+            )
+            return self._attach_human_review_notice(
+                {
+                    "profile": profile,
+                    "ws_id": session_profile.selected_ws_id,
+                    "app_key": app_key,
+                    "button_id": button_id,
+                    "result": result,
+                },
+                operation="update",
+                target="custom button configuration",
+            )
+
+        return self._run(profile, runner)
+
+    @tool_cn_name("删除自定义按钮")
+    def custom_button_delete(self, *, profile: str, app_key: str, button_id: int) -> JSONObject:
+        """执行工具方法逻辑。"""
+        self._require_app_key(app_key)
+        self._require_button_id(button_id)
+
+        def runner(session_profile, context):
+            result = self.backend.request("DELETE", context, f"/app/{app_key}/customButton/{button_id}")
+            return self._attach_human_review_notice(
+                {
+                    "profile": profile,
+                    "ws_id": session_profile.selected_ws_id,
+                    "app_key": app_key,
+                    "button_id": button_id,
+                    "result": result,
+                },
+                operation="delete",
+                target="custom button configuration",
+            )
+
+        return self._run(profile, runner)
+
+    def _require_app_key(self, app_key: str) -> None:
+        """执行内部辅助逻辑。"""
+        if not str(app_key or "").strip():
+            raise_tool_error(QingflowApiError.config_error("app_key is required"))
+
+    def _require_button_id(self, button_id: int) -> None:
+        """执行内部辅助逻辑。"""
+        if not isinstance(button_id, int) or isinstance(button_id, bool) or button_id <= 0:
+            raise_tool_error(QingflowApiError.config_error("button_id must be a positive integer"))
+
+    def _compact_button_base_info(self, item: dict[str, object]) -> JSONObject:
+        """执行内部辅助逻辑。"""
+        creator = item.get("creatorUserInfo") if isinstance(item.get("creatorUserInfo"), dict) else {}
+        return {
+            "button_id": item.get("buttonId"),
+            "button_text": item.get("buttonText"),
+            "button_icon": item.get("buttonIcon"),
+            "icon_color": item.get("iconColor"),
+            "background_color": item.get("backgroundColor"),
+            "text_color": item.get("textColor"),
+            "creator_user_info": {
+                "uid": creator.get("uid"),
+                "name": creator.get("name"),
+                "email": creator.get("email"),
+            }
+            if creator
+            else None,
+            "used_in_chart_count": item.get("userInChartCount"),
+            "being_effective_external_qrobot": item.get("beingEffectiveExternalQRobot"),
+        }
+
+    def _compact_button_detail(self, item: dict[str, object]) -> JSONObject:
+        """执行内部辅助逻辑。"""
+        return {
+            "button_id": item.get("buttonId"),
+            "button_text": item.get("buttonText"),
+            "button_icon": item.get("buttonIcon"),
+            "icon_color": item.get("iconColor"),
+            "background_color": item.get("backgroundColor"),
+            "text_color": item.get("textColor"),
+            "trigger_action": item.get("triggerAction"),
+            "trigger_link_url": item.get("triggerLinkUrl"),
+            "trigger_add_data_config": deepcopy(item.get("triggerAddDataConfig")) if isinstance(item.get("triggerAddDataConfig"), dict) else None,
+            "external_qrobot_config": deepcopy(item.get("customButtonExternalQRobotRelationVO"))
+            if isinstance(item.get("customButtonExternalQRobotRelationVO"), dict)
+            else None,
+            "trigger_wings_config": deepcopy(item.get("triggerWingsConfig")) if isinstance(item.get("triggerWingsConfig"), dict) else None,
+        }