@josephyan/qingflow-cli 0.2.0-beta.70 → 0.2.0-beta.72

package/src/qingflow_mcp/tools/resource_read_tools.py

@@ -0,0 +1,399 @@
+from __future__ import annotations
+
+from copy import deepcopy
+from typing import Any
+
+from ..errors import QingflowApiError, raise_tool_error
+from ..json_types import JSONObject
+from ..list_type_labels import SYSTEM_VIEW_DEFINITIONS
+from .app_tools import _analysis_supported_for_view_type
+from .app_tools import AppTools
+from .base import ToolBase
+from .qingbi_report_tools import QingbiReportTools
+
+
+class ResourceReadTools(ToolBase):
+    def __init__(self, sessions, backend) -> None:
+        super().__init__(sessions, backend)
+        self.apps = AppTools(sessions, backend)
+        self.charts = QingbiReportTools(sessions, backend)
+
+    def portal_list(self, *, profile: str) -> JSONObject:
+        def runner(session_profile, context):
+            raw_items = self.backend.request("GET", context, "/dash")
+            items = _normalize_portal_list_items(raw_items)
+            return {
+                "profile": profile,
+                "ws_id": session_profile.selected_ws_id,
+                "ok": True,
+                "warnings": [],
+                "verification": {"portal_list_loaded": True},
+                "data": {
+                    "items": items,
+                    "total": len(items),
+                },
+            }
+
+        return self._run(profile, runner)
+
+    def portal_get(self, *, profile: str, dash_key: str) -> JSONObject:
+        self._require_dash_key(dash_key)
+
+        def runner(session_profile, context):
+            result = self.backend.request("GET", context, f"/dash/{dash_key}", params={"beingDraft": False})
+            dash_name = str(result.get("dashName") or "").strip() or None
+            dash_icon = str(result.get("dashIcon") or "").strip() or None
+            package_tag_ids = [
+                tag_id
+                for tag_id in (
+                    _coerce_positive_int((tag or {}).get("tagId"))
+                    for tag in (result.get("tags") or [])
+                    if isinstance(tag, dict)
+                )
+                if tag_id is not None
+            ]
+            components = _normalize_user_portal_components(result.get("components"))
+            return {
+                "profile": profile,
+                "ws_id": session_profile.selected_ws_id,
+                "ok": True,
+                "warnings": [],
+                "verification": {"portal_exists": True},
+                "data": {
+                    "dash_key": dash_key,
+                    "dash_name": dash_name,
+                    "dash_icon": dash_icon,
+                    "package_tag_ids": package_tag_ids,
+                    "component_count": len(components),
+                    "components": components,
+                },
+            }
+
+        return self._run(profile, runner)
+
+    def view_get(self, *, profile: str, view_id: str) -> JSONObject:
+        self._require_view_id(view_id)
+        view_key = _extract_custom_view_key(view_id)
+        system_view = _lookup_system_view_descriptor(view_id)
+        if view_key is None and system_view is None:
+            raise_tool_error(
+                QingflowApiError.config_error(
+                    "view_get only accepts accessible view_id values such as `custom:VIEW_KEY` or `system:all`; use app_get or portal_get first to find a valid view_id."
+                )
+            )
+        if system_view is not None:
+            return self._run(
+                profile,
+                lambda session_profile, _context: {
+                    "profile": profile,
+                    "ws_id": session_profile.selected_ws_id,
+                    "ok": True,
+                    "warnings": [
+                        {
+                            "code": "VIEW_APP_KEY_UNRESOLVED",
+                            "message": f"view_get could not resolve app_key for system view `{view_id}`; keep using the app_key from the parent app context.",
+                        }
+                    ],
+                    "verification": {
+                        "view_exists": True,
+                        "descriptor_only": True,
+                    },
+                    "data": {
+                        "app_key": None,
+                        "view_id": view_id,
+                        "view_key": None,
+                        "view_name": system_view["view_name"],
+                        "view_type": system_view["view_type"],
+                        "visible_columns": [],
+                        "analysis_supported": system_view["analysis_supported"],
+                    },
+                },
+            )
+
+        def runner(session_profile, context):
+            warnings: list[JSONObject] = []
+            verification = {
+                "view_exists": True,
+                "questions_verified": True,
+            }
+            config = self.backend.request("GET", context, f"/view/{view_key}/viewConfig")
+            base_info = self.backend.request("GET", context, f"/view/{view_key}/viewConfig/baseInfo")
+            questions: list[dict[str, Any]] = []
+            try:
+                questions_payload = self.backend.request("GET", context, f"/view/{view_key}/question")
+                if isinstance(questions_payload, list):
+                    questions = [deepcopy(item) for item in questions_payload if isinstance(item, dict)]
+            except QingflowApiError:
+                verification["questions_verified"] = False
+                warnings.append(
+                    {
+                        "code": "VIEW_QUESTIONS_UNAVAILABLE",
+                        "message": "view_get could not load visible columns because question readback is unavailable.",
+                    }
+                )
+
+            raw_view_type = (
+                str(base_info.get("viewgraphType") or "").strip()
+                or str(config.get("viewgraphType") or config.get("viewType") or "").strip()
+            )
+            resolved_app_key = str(base_info.get("appKey") or config.get("appKey") or "").strip() or None
+            if not resolved_app_key:
+                resolved_app_key = self._resolve_app_key_from_form_id(
+                    profile=profile,
+                    form_id=_coerce_positive_int(base_info.get("formId") or config.get("formId")),
+                )
+            if not resolved_app_key:
+                warnings.append(
+                    {
+                        "code": "VIEW_APP_KEY_UNRESOLVED",
+                        "message": f"view_get could not resolve app_key for `{view_id}` from view metadata; keep using the app_key from the parent app or portal context.",
+                    }
+                )
+            return {
+                "profile": profile,
+                "ws_id": session_profile.selected_ws_id,
+                "ok": True,
+                "warnings": warnings,
+                "verification": verification,
+                "data": {
+                    "app_key": resolved_app_key,
+                    "view_id": view_id,
+                    "view_key": view_key,
+                    "view_name": str(
+                        base_info.get("viewgraphName") or config.get("viewgraphName") or config.get("viewName") or view_key
+                    ).strip()
+                    or view_key,
+                    "view_type": _normalize_view_type(raw_view_type),
+                    "visible_columns": [
+                        str(item.get("queTitle") or item.get("title") or "").strip()
+                        for item in questions
+                        if str(item.get("queTitle") or item.get("title") or "").strip()
+                    ],
+                    "analysis_supported": _analysis_supported_for_view_type(raw_view_type or None),
+                },
+            }
+
+        return self._run(profile, runner)
+
+    def chart_get(self, *, profile: str, chart_id: str) -> JSONObject:
+        self._require_chart_id(chart_id)
+
+        def runner(session_profile, _context):
+            base = self.charts.qingbi_report_get_base(profile=profile, chart_id=chart_id).get("result") or {}
+            data = self.charts.qingbi_report_get_data(profile=profile, chart_id=chart_id, payload={}).get("result") or {}
+            data_config = data.get("config") if isinstance(data, dict) and isinstance(data.get("config"), dict) else {}
+            return {
+                "profile": profile,
+                "ws_id": session_profile.selected_ws_id,
+                "ok": True,
+                "warnings": [],
+                "verification": {
+                    "chart_exists": True,
+                    "chart_data_loaded": True,
+                },
+                "data": {
+                    "chart_id": chart_id,
+                    "chart_name": str(base.get("chartName") or base.get("name") or chart_id).strip() or chart_id,
+                    "chart_type": str(base.get("chartType") or data_config.get("chartType") or "").strip() or None,
+                    "data_source_type": str(base.get("dataSourceType") or "").strip() or None,
+                    "data_source_id": str(base.get("dataSourceId") or "").strip() or None,
+                    "data": deepcopy(data) if isinstance(data, dict) else {"value": data},
+                },
+            }
+
+        return self._run(profile, runner)
+
+    def _require_dash_key(self, dash_key: str) -> None:
+        if not dash_key:
+            raise_tool_error(QingflowApiError.config_error("dash_key is required"))
+
+    def _require_view_id(self, view_id: str) -> None:
+        if not view_id:
+            raise_tool_error(QingflowApiError.config_error("view_id is required"))
+
+    def _require_chart_id(self, chart_id: str) -> None:
+        if not chart_id:
+            raise_tool_error(QingflowApiError.config_error("chart_id is required"))
+
+    def _resolve_app_key_from_form_id(self, *, profile: str, form_id: int | None) -> str | None:
+        if form_id is None:
+            return None
+        try:
+            payload = self.apps.app_list(profile=profile)
+        except QingflowApiError:
+            return None
+        items = payload.get("items")
+        if not isinstance(items, list):
+            return None
+        for item in items:
+            if not isinstance(item, dict):
+                continue
+            if _coerce_positive_int(item.get("form_id")) != form_id:
+                continue
+            app_key = str(item.get("app_key") or "").strip()
+            if app_key:
+                return app_key
+        return None
+
+
+def _normalize_portal_list_items(raw_items: Any) -> list[dict[str, Any]]:
+    if not isinstance(raw_items, list):
+        return []
+    items: list[dict[str, Any]] = []
+    for item in raw_items:
+        if not isinstance(item, dict):
+            continue
+        dash_key = str(item.get("dashKey") or "").strip()
+        dash_name = str(item.get("dashName") or "").strip()
+        dash_icon = str(item.get("dashIcon") or "").strip() or None
+        package_tag_ids = [
+            tag_id
+            for tag_id in (
+                _coerce_positive_int((tag or {}).get("tagId"))
+                for tag in (item.get("tags") or [])
+                if isinstance(tag, dict)
+            )
+            if tag_id is not None
+        ]
+        if not any((dash_key, dash_name, dash_icon, package_tag_ids)):
+            continue
+        items.append(
+            {
+                "dash_key": dash_key or None,
+                "dash_name": dash_name or None,
+                "dash_icon": dash_icon,
+                "package_tag_ids": package_tag_ids,
+            }
+        )
+    return items
+
+
+def _normalize_user_portal_components(components: Any) -> list[dict[str, Any]]:
+    if not isinstance(components, list):
+        return []
+    items: list[dict[str, Any]] = []
+    config_key_map = {
+        "grid": "gridConfig",
+        "link": "linkConfig",
+        "text": "textConfig",
+        "filter": "filterConfig",
+        "chart": "chartConfig",
+        "view": "viewgraphConfig",
+    }
+    for index, component in enumerate(components):
+        if not isinstance(component, dict):
+            continue
+        source_type = _normalize_portal_component_source_type(component.get("type"))
+        title = _extract_portal_component_title(component, source_type=source_type)
+        summary: dict[str, Any] = {
+            "order": index,
+            "source_type": source_type,
+            "title": title,
+        }
+        position = component.get("position")
+        if isinstance(position, dict):
+            summary["position"] = deepcopy(position)
+        config_key = config_key_map.get(source_type, "")
+        config = component.get(config_key) if isinstance(component.get(config_key), dict) else {}
+        if source_type == "chart":
+            summary["chart_ref"] = {
+                "chart_id": str(config.get("biChartId") or "").strip() or None,
+                "chart_name": str(config.get("chartComponentTitle") or title or "").strip() or None,
+            }
+        elif source_type == "view":
+            view_key = str(config.get("viewgraphKey") or "").strip() or None
+            summary["view_ref"] = {
+                "app_key": str(config.get("appKey") or "").strip() or None,
+                "view_id": f"custom:{view_key}" if view_key else None,
+                "view_key": view_key,
+                "view_name": str(config.get("viewgraphName") or title or "").strip() or None,
+            }
+        elif source_type in {"grid", "link", "text", "filter"} and config:
+            summary["config"] = deepcopy(config)
+        items.append(summary)
+    return items
+
+
+def _normalize_portal_component_source_type(value: Any) -> str:
+    raw = str(value or "").strip()
+    mapping = {
+        "2": "grid",
+        "4": "link",
+        "5": "text",
+        "6": "filter",
+        "9": "chart",
+        "10": "view",
+        "grid": "grid",
+        "link": "link",
+        "text": "text",
+        "filter": "filter",
+        "chart": "chart",
+        "bi": "chart",
+        "view": "view",
+        "viewgraph": "view",
+    }
+    return mapping.get(raw, raw.lower() or "unknown")
+
+
+def _extract_portal_component_title(component: dict[str, Any], *, source_type: str) -> str | None:
+    config_key_map = {
+        "grid": "gridConfig",
+        "link": "linkConfig",
+        "text": "textConfig",
+        "filter": "filterConfig",
+        "chart": "chartConfig",
+        "view": "viewgraphConfig",
+    }
+    config_key = config_key_map.get(source_type, "")
+    config = component.get(config_key) if isinstance(component.get(config_key), dict) else {}
+    title_candidates = {
+        "grid": ["gridTitle"],
+        "link": ["title", "linkTitle"],
+        "text": ["title", "textTitle"],
+        "filter": ["title", "filterTitle"],
+        "chart": ["chartComponentTitle", "componentTitle", "title"],
+        "view": ["componentTitle", "viewgraphName", "title"],
+    }
+    for key in title_candidates.get(source_type, []):
+        title = str(config.get(key) or "").strip()
+        if title:
+            return title
+    return None
+
+
+def _extract_custom_view_key(view_id: str) -> str | None:
+    value = str(view_id or "").strip()
+    if not value.startswith("custom:"):
+        return None
+    view_key = value.split(":", 1)[1].strip()
+    return view_key or None
+
+
+def _lookup_system_view_descriptor(view_id: str) -> dict[str, Any] | None:
+    normalized = str(view_id or "").strip()
+    for system_view_id, _list_type, name in SYSTEM_VIEW_DEFINITIONS:
+        if system_view_id != normalized:
+            continue
+        return {
+            "view_name": name,
+            "view_type": "system",
+            "analysis_supported": True,
+        }
+    return None
+
+
+def _normalize_view_type(view_type: Any) -> str | None:
+    value = str(view_type or "").strip()
+    if not value:
+        return None
+    if value.lower().endswith("view"):
+        value = value[:-4]
+    return value or None
+
+
+def _coerce_positive_int(value: Any) -> int | None:
+    try:
+        number = int(value)
+    except (TypeError, ValueError):
+        return None
+    return number if number > 0 else None