@qingflow-tech/qingflow-app-user-mcp 1.0.2 → 1.0.4

package/src/qingflow_mcp/cli/commands/workspace.py

@@ -3,6 +3,9 @@ from __future__ import annotations
 import argparse
 
 from ..context import CliContext
+from ..interaction import cancelled_result, resolve_interactive_selection
+from ..terminal_ui import SelectionOption
+from .common import raise_config_error
 
 
 def register(subparsers: argparse._SubParsersAction[argparse.ArgumentParser]) -> None:
@@ -15,6 +18,14 @@ def register(subparsers: argparse._SubParsersAction[argparse.ArgumentParser]) ->
     list_parser.add_argument("--include-external", action="store_true")
     list_parser.set_defaults(handler=_handle_list, format_hint="workspace_list")
 
+    get_parser = workspace_subparsers.add_parser("get", help="读取工作区详情")
+    get_parser.add_argument("--ws-id", type=int, default=0)
+    get_parser.set_defaults(handler=_handle_get, format_hint="workspace_get")
+
+    select_parser = workspace_subparsers.add_parser("select", help="切换当前工作区")
+    select_parser.add_argument("--ws-id", type=int, default=0, help="不传时在交互终端中选择工作区")
+    select_parser.set_defaults(handler=_handle_select, format_hint="workspace_select")
+
 
 def _handle_list(args: argparse.Namespace, context: CliContext) -> dict:
     return context.workspace.workspace_list(
@@ -23,3 +34,76 @@ def _handle_list(args: argparse.Namespace, context: CliContext) -> dict:
         page_size=args.page_size,
         include_external=bool(args.include_external),
     )
+
+
+def _handle_get(args: argparse.Namespace, context: CliContext) -> dict:
+    return context.workspace.workspace_get(
+        profile=args.profile,
+        ws_id=args.ws_id if int(args.ws_id or 0) > 0 else None,
+    )
+
+
+def _handle_select(args: argparse.Namespace, context: CliContext) -> dict:
+    if int(args.ws_id or 0) <= 0:
+        selection = _choose_workspace_interactively(args, context)
+        if selection.status == "unavailable":
+            raise_config_error(
+                "workspace select requires --ws-id, or an interactive terminal to choose a workspace",
+                fix_hint="Retry in an interactive terminal, or pass `--ws-id WS_ID` explicitly.",
+            )
+        if selection.status == "empty":
+            raise_config_error(
+                selection.message or "workspace select could not open a selector because no workspaces are available.",
+                fix_hint="Run `workspace list` to confirm visible workspaces, or retry with `--ws-id WS_ID`.",
+            )
+        if selection.status == "cancelled":
+            return cancelled_result(selection.message or "已取消")
+        args.ws_id = int(selection.value or 0)
+    return context.workspace.workspace_select(
+        profile=args.profile,
+        ws_id=int(args.ws_id),
+    )
+
+
+def _choose_workspace_interactively(args: argparse.Namespace, context: CliContext):
+    current_ws_id = None
+    sessions = getattr(context, "sessions", None)
+    if sessions is not None and hasattr(sessions, "get_profile"):
+        try:
+            session_profile = sessions.get_profile(args.profile)
+        except Exception:
+            session_profile = None
+        current_ws_id = getattr(session_profile, "selected_ws_id", None) if session_profile is not None else None
+    def load_options() -> list[SelectionOption[int]]:
+        page = context.workspace.workspace_list(
+            profile=args.profile,
+            page_num=1,
+            page_size=100,
+            include_external=False,
+        ).get("page")
+        items = page.get("list") if isinstance(page, dict) and isinstance(page.get("list"), list) else []
+        options: list[SelectionOption[int]] = []
+        for item in items:
+            if not isinstance(item, dict):
+                continue
+            ws_id = int(item.get("wsId") or 0)
+            if ws_id <= 0:
+                continue
+            workspace_name = str(item.get("workspaceName") or item.get("wsName") or f"Workspace {ws_id}")
+            remark = str(item.get("remark") or "").strip()
+            label = workspace_name
+            if remark:
+                label = f"{workspace_name} - {remark}"
+            hint = f"ws_id={ws_id}"
+            if current_ws_id == ws_id:
+                hint += " · 当前"
+            options.append(SelectionOption(value=ws_id, label=label, hint=hint))
+        return options
+
+    return resolve_interactive_selection(
+        args,
+        title="选择工作区",
+        unavailable_message="workspace select requires --ws-id, or an interactive terminal to choose a workspace",
+        empty_message="workspace select could not open a selector because no visible workspaces were returned.",
+        load_options=load_options,
+    )

package/src/qingflow_mcp/cli/context.py

@@ -8,6 +8,7 @@ from ..tools.ai_builder_tools import AiBuilderTools
 from ..tools.app_tools import AppTools
 from ..tools.auth_tools import AuthTools
 from ..tools.code_block_tools import CodeBlockTools
+from ..tools.export_tools import ExportTools
 from ..tools.feedback_tools import FeedbackTools
 from ..tools.file_tools import FileTools
 from ..tools.import_tools import ImportTools
@@ -29,6 +30,7 @@ class CliContext:
     record: RecordTools
     code_block: CodeBlockTools
     imports: ImportTools
+    exports: ExportTools
     task: TaskContextTools
     files: FileTools
     builder_feedback: FeedbackTools
@@ -52,6 +54,7 @@ def build_cli_context() -> CliContext:
         record=RecordTools(sessions, backend),
         code_block=CodeBlockTools(sessions, backend),
         imports=ImportTools(sessions, backend),
+        exports=ExportTools(sessions, backend),
         task=TaskContextTools(sessions, backend),
         files=FileTools(sessions, backend),
         builder_feedback=FeedbackTools(backend, mcp_side="App Builder MCP"),