npm - @docyrus/docyrus - Versions diffs - 0.0.19 → 0.0.21 - Mend

@docyrus/docyrus 0.0.19 → 0.0.21

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (111) hide show

package/resources/pi-agent/extensions/pi-mcp-adapter/host-html-template.ts ADDED Viewed

@@ -0,0 +1,427 @@
+import type { UiHostContext, UiResourceContent, UiResourceCsp } from "./types.js";
+// Use locally bundled AppBridge to avoid CDN Zod bundling issues
+const DEFAULT_APP_BRIDGE_MODULE_URL = "/app-bridge.bundle.js";
+export interface HostHtmlTemplateInput {
+  sessionToken: string;
+  serverName: string;
+  toolName: string;
+  toolArgs: Record<string, unknown>;
+  resource: UiResourceContent;
+  allowAttribute: string;
+  requireToolConsent: boolean;
+  cacheToolConsent: boolean;
+  hostContext?: UiHostContext;
+  appBridgeModuleUrl?: string;
+}
+export function buildHostHtmlTemplate(input: HostHtmlTemplateInput): string {
+  const cspContent = buildCspMetaContent(input.resource.meta.csp);
+  const resourceHtml = applyCspMeta(input.resource.html, cspContent);
+  const hostContext = input.hostContext ?? {};
+  const sessionToken = safeInlineJSON(input.sessionToken);
+  const toolArgs = safeInlineJSON(input.toolArgs);
+  const uiHtml = safeInlineJSON(resourceHtml);
+  const serverName = safeInlineJSON(input.serverName);
+  const toolName = safeInlineJSON(input.toolName);
+  const hostContextJson = safeInlineJSON(hostContext);
+  const allowAttribute = safeInlineJSON(input.allowAttribute);
+  const requireToolConsent = safeInlineJSON(input.requireToolConsent);
+  const cacheToolConsent = safeInlineJSON(input.cacheToolConsent);
+  const moduleUrl = safeInlineJSON(input.appBridgeModuleUrl ?? DEFAULT_APP_BRIDGE_MODULE_URL);
+  return `<!doctype html>
+<html lang="en">
+<head>
+  <meta charset="utf-8" />
+  <meta name="viewport" content="width=device-width, initial-scale=1" />
+  <title>MCP UI - ${escapeHtml(input.serverName)} / ${escapeHtml(input.toolName)}</title>
+  <style>
+    :root {
+      color-scheme: light dark;
+      --bg: #0f1115;
+      --surface: #181c22;
+      --text: #ecf0f5;
+      --muted: #a9b2bf;
+      --accent: #43c0ff;
+      --border: rgba(255, 255, 255, 0.12);
+      --good: #34d399;
+      --warn: #fbbf24;
+      --bad: #f87171;
+    }
+    @media (prefers-color-scheme: light) {
+      :root {
+        --bg: #f6f7fb;
+        --surface: #ffffff;
+        --text: #1d2939;
+        --muted: #667085;
+        --accent: #0ea5e9;
+        --border: rgba(15, 23, 42, 0.14);
+        --good: #059669;
+        --warn: #b45309;
+        --bad: #b91c1c;
+      }
+    }
+    * { box-sizing: border-box; }
+    html, body { margin: 0; padding: 0; height: 100%; font-family: ui-sans-serif, -apple-system, BlinkMacSystemFont, "Segoe UI", sans-serif; background: var(--bg); color: var(--text); }
+    body { display: flex; flex-direction: column; min-height: 100vh; }
+    header { background: var(--surface); border-bottom: 1px solid var(--border); padding: 10px 14px; display: flex; align-items: center; justify-content: space-between; gap: 10px; }
+    .title { display: flex; gap: 8px; align-items: baseline; min-width: 0; }
+    .server { font-size: 12px; color: var(--muted); text-transform: uppercase; letter-spacing: 0.08em; white-space: nowrap; }
+    .tool { font-size: 14px; font-weight: 600; overflow: hidden; text-overflow: ellipsis; white-space: nowrap; }
+    .badge { border: 1px solid var(--border); border-radius: 999px; padding: 2px 8px; font-size: 11px; color: var(--muted); white-space: nowrap; }
+    .controls { display: flex; gap: 8px; align-items: center; }
+    .status { font-size: 12px; color: var(--muted); white-space: nowrap; }
+    button { border: 1px solid var(--border); background: transparent; color: var(--text); border-radius: 8px; padding: 6px 10px; cursor: pointer; font-size: 12px; }
+    button.primary { border-color: color-mix(in srgb, var(--good) 40%, var(--border) 60%); color: var(--good); }
+    button.danger { border-color: color-mix(in srgb, var(--bad) 40%, var(--border) 60%); color: var(--bad); }
+    button:hover { background: color-mix(in srgb, var(--surface) 75%, var(--accent) 25%); }
+    main { flex: 1; min-height: 0; padding: 10px; display: flex; }
+    iframe { width: 100%; height: 100%; border: 1px solid var(--border); border-radius: 10px; background: white; }
+    .overlay { position: fixed; inset: 0; background: color-mix(in srgb, var(--bg) 90%, black 10%); display: none; align-items: center; justify-content: center; z-index: 2; }
+    .overlay.visible { display: flex; }
+    .panel { width: min(680px, calc(100vw - 40px)); background: var(--surface); border: 1px solid var(--border); border-radius: 12px; padding: 18px; }
+    .panel h2 { margin: 0 0 8px; font-size: 16px; }
+    .panel p { margin: 0; color: var(--muted); line-height: 1.4; font-size: 14px; white-space: pre-wrap; }
+  </style>
+</head>
+<body>
+  <header>
+    <div class="title">
+      <span class="server">MCP · <span id="server-name"></span></span>
+      <span class="tool" id="tool-name"></span>
+      <span class="badge">Sandboxed</span>
+    </div>
+    <div class="controls">
+      <span class="status" id="status">Loading UI...</span>
+      <button class="primary" id="done-btn" title="Cmd/Ctrl+Enter">Done</button>
+      <button class="danger" id="cancel-btn" title="Escape">Cancel</button>
+    </div>
+  </header>
+  <main>
+    <iframe id="mcp-app" referrerpolicy="no-referrer"></iframe>
+  </main>
+  <div class="overlay" id="error-overlay">
+    <div class="panel">
+      <h2>UI Error</h2>
+      <p id="error-message"></p>
+    </div>
+  </div>
+  <script type="module">
+    import { AppBridge, PostMessageTransport } from ${moduleUrl};
+    const SESSION_TOKEN = ${sessionToken};
+    const SERVER_NAME = ${serverName};
+    const TOOL_NAME = ${toolName};
+    const TOOL_ARGS = ${toolArgs};
+    const HOST_CONTEXT = ${hostContextJson};
+    const ALLOW_ATTRIBUTE = ${allowAttribute};
+    const REQUIRE_TOOL_CONSENT = ${requireToolConsent};
+    const CACHE_TOOL_CONSENT = ${cacheToolConsent};
+    const STREAM_CONTEXT_KEY = "pi-mcp-adapter/stream";
+    const STREAM_PATCH_METHOD = "notifications/pi-mcp-adapter/ui-result-patch";
+    const iframe = document.getElementById("mcp-app");
+    const statusNode = document.getElementById("status");
+    const doneBtn = document.getElementById("done-btn");
+    const cancelBtn = document.getElementById("cancel-btn");
+    const errorOverlay = document.getElementById("error-overlay");
+    const errorMessage = document.getElementById("error-message");
+    document.getElementById("server-name").textContent = SERVER_NAME;
+    document.getElementById("tool-name").textContent = TOOL_NAME;
+    const setStatus = (text, isError = false) => {
+      statusNode.textContent = text;
+      statusNode.style.color = isError ? "var(--bad)" : "var(--muted)";
+    };
+    const showError = (message) => {
+      errorMessage.textContent = message;
+      errorOverlay.classList.add("visible");
+      setStatus("Error", true);
+    };
+    const post = async (endpoint, params) => {
+      const response = await fetch(endpoint, {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ token: SESSION_TOKEN, params }),
+      });
+      const body = await response.json().catch(() => ({ ok: false, error: "Invalid JSON response" }));
+      if (!response.ok || !body.ok) {
+        const message = body.error || ("HTTP " + response.status);
+        throw new Error(message);
+      }
+      return body.result ?? {};
+    };
+    let consentGranted = !REQUIRE_TOOL_CONSENT;
+    const initialStreamContext = HOST_CONTEXT?.[STREAM_CONTEXT_KEY];
+    const streamMode = initialStreamContext?.mode === "stream-first" ? "stream-first" : "eager";
+    const bridge = new AppBridge(
+      null,
+      { name: "pi", version: "1.0.0" },
+      { serverTools: {}, openLinks: {}, logging: {}, updateModelContext: {}, message: {} },
+      { hostContext: HOST_CONTEXT }
+    );
+    bridge.oncalltool = async (params) => {
+      if (!consentGranted) {
+        const accepted = window.confirm("Allow this UI to call server tools for this session?");
+        if (!accepted) {
+          await post("/proxy/ui/consent", { approved: false }).catch(() => {});
+          return {
+            isError: true,
+            content: [{ type: "text", text: "Tool call denied by user." }],
+          };
+        }
+        await post("/proxy/ui/consent", { approved: true });
+        if (CACHE_TOOL_CONSENT) {
+          consentGranted = true;
+        }
+      }
+      const result = await post("/proxy/tools/call", params);
+      // Notify agent about the tool call
+      await post("/proxy/ui/message", {
+        type: "intent",
+        intent: "call_tool",
+        params: { tool: params.name, arguments: params.arguments, isError: result.isError }
+      }).catch(() => {});
+      return result;
+    };
+    bridge.onmessage = async (params) => post("/proxy/ui/message", params);
+    bridge.onupdatemodelcontext = async (params) => post("/proxy/ui/context", params);
+    // Also listen for raw postMessage events with custom types (notify, prompt, intent, etc.)
+    // These bypass the AppBridge protocol but are used by some MCP UI implementations
+    window.addEventListener("message", async (event) => {
+      const data = event.data;
+      if (!data || typeof data !== "object") return;
+      // Skip AppBridge protocol messages (handled by bridge)
+      if (data.jsonrpc || (typeof data.method === "string" && (data.method.startsWith("app/") || data.method.startsWith("host/")))) return;
+      // Handle raw UI action messages
+      const msgType = data.type;
+      if (typeof msgType !== "string") return;
+      if (msgType === "notify" || msgType === "prompt" || msgType === "intent" || msgType === "message") {
+        // Standard MCP-UI types - preserve their semantics
+        // Support both { type, payload: {...} } and { type, field: value } formats
+        const { type: _, payload, ...directFields } = data;
+        await post("/proxy/ui/message", { type: msgType, ...directFields, ...(payload || {}) }).catch(() => {});
+      } else if (!msgType.startsWith("ui-lifecycle-") && !msgType.startsWith("ui-message-")) {
+        // Any other custom type - forward as intent with type as intent name
+        // (Skip internal lifecycle/ack messages)
+        const payload = data.payload || {};
+        await post("/proxy/ui/message", {
+          type: "intent",
+          intent: msgType,
+          params: payload,
+        }).catch(() => {});
+      }
+    });
+    bridge.ondownloadfile = async (params) => post("/proxy/ui/download-file", params);
+    bridge.onrequestdisplaymode = async (params) => post("/proxy/ui/request-display-mode", params);
+    bridge.onopenlink = async (params) => {
+      const result = await post("/proxy/ui/open-link", params);
+      if (!result.isError) {
+        window.open(params.url, "_blank", "noopener,noreferrer");
+        // Notify agent about the link open
+        await post("/proxy/ui/message", {
+          type: "intent",
+          intent: "open_link",
+          params: { url: params.url }
+        }).catch(() => {});
+      }
+      return result;
+    };
+    bridge.oninitialized = () => {
+      if (streamMode !== "stream-first") {
+        bridge.sendToolInput({ arguments: TOOL_ARGS });
+      }
+      setStatus(streamMode === "stream-first" ? "Streaming…" : "Connected");
+    };
+    bridge.onsizechange = ({ width, height }) => {
+      if (typeof width === "number" && width > 0) {
+        iframe.style.minWidth = Math.min(width, window.innerWidth - 24) + "px";
+      }
+      if (typeof height === "number" && height > 0) {
+        iframe.style.height = Math.max(height, 320) + "px";
+      }
+    };
+    if (ALLOW_ATTRIBUTE) {
+      iframe.setAttribute("allow", ALLOW_ATTRIBUTE);
+    }
+    // Connect bridge BEFORE loading iframe to ensure we're listening when the app sends ui/initialize
+    try {
+      const transport = new PostMessageTransport(iframe.contentWindow, null);
+      await bridge.connect(transport);
+    } catch (error) {
+      console.error("[host] Bridge connection failed:", error);
+      showError("Failed to initialize AppBridge: " + String(error));
+    }
+    const iframeLoaded = new Promise((resolve) => {
+      iframe.onload = resolve;
+    });
+    iframe.src = "/ui-app?session=" + encodeURIComponent(SESSION_TOKEN);
+    await iframeLoaded;
+    const eventSource = new EventSource("/events?session=" + encodeURIComponent(SESSION_TOKEN));
+    eventSource.addEventListener("tool-input", (event) => {
+      try {
+        bridge.sendToolInput(JSON.parse(event.data));
+      } catch (error) {
+        showError("Failed to forward tool input: " + String(error));
+      }
+    });
+    eventSource.addEventListener("tool-result", (event) => {
+      try {
+        bridge.sendToolResult(JSON.parse(event.data));
+      } catch (error) {
+        showError("Failed to forward tool result: " + String(error));
+      }
+    });
+    eventSource.addEventListener("tool-cancelled", (event) => {
+      try {
+        bridge.sendToolCancelled(JSON.parse(event.data));
+      } catch (error) {
+        showError("Failed to forward cancellation: " + String(error));
+      }
+    });
+    eventSource.addEventListener("result-patch", async (event) => {
+      try {
+        await bridge.notification({
+          method: STREAM_PATCH_METHOD,
+          params: JSON.parse(event.data),
+        });
+      } catch (error) {
+        showError("Failed to forward stream patch: " + String(error));
+      }
+    });
+    eventSource.addEventListener("host-context", (event) => {
+      try {
+        bridge.setHostContext(JSON.parse(event.data));
+      } catch {}
+    });
+    eventSource.addEventListener("session-complete", async () => {
+      await bridge.teardownResource({}).catch(() => {});
+      eventSource.close();
+      window.close();
+    });
+    eventSource.onerror = () => {
+      setStatus("Connection lost", true);
+    };
+    const heartbeat = setInterval(() => {
+      post("/proxy/ui/heartbeat", {}).catch(() => {});
+    }, 10000);
+    const complete = async (reason) => {
+      try {
+        await post("/proxy/ui/complete", { reason });
+      } catch {}
+      try {
+        await bridge.teardownResource({});
+      } catch {}
+      clearInterval(heartbeat);
+      eventSource.close();
+      window.close();
+    };
+    doneBtn.addEventListener("click", () => complete("done"));
+    cancelBtn.addEventListener("click", () => complete("cancel"));
+    window.addEventListener("keydown", (event) => {
+      if (event.key === "Escape") {
+        event.preventDefault();
+        complete("cancel");
+      } else if ((event.metaKey || event.ctrlKey) && event.key === "Enter") {
+        event.preventDefault();
+        complete("done");
+      }
+    });
+  </script>
+</body>
+</html>`;
+}
+export function buildCspMetaContent(csp: UiResourceCsp | undefined): string | undefined {
+  if (!csp) return undefined;
+  const directives: string[] = [];
+  directives.push("default-src 'none'");
+  const scriptSrc = toDirective("script-src", csp.scriptDomains);
+  const styleSrc = toDirective("style-src", csp.styleDomains);
+  const fontSrc = toDirective("font-src", csp.fontDomains);
+  const imgSrc = toDirective("img-src", csp.imgDomains);
+  const mediaSrc = toDirective("media-src", csp.mediaDomains);
+  const connectSrc = toDirective("connect-src", csp.connectDomains);
+  const frameSrc = toDirective("frame-src", csp.frameDomains);
+  const workerSrc = toDirective("worker-src", csp.workerDomains);
+  const baseUri = toDirective("base-uri", csp.baseUriDomains);
+  if (scriptSrc) directives.push(scriptSrc);
+  if (styleSrc) directives.push(styleSrc);
+  if (fontSrc) directives.push(fontSrc);
+  if (imgSrc) directives.push(imgSrc);
+  if (mediaSrc) directives.push(mediaSrc);
+  if (connectSrc) directives.push(connectSrc);
+  if (frameSrc) directives.push(frameSrc);
+  if (workerSrc) directives.push(workerSrc);
+  if (baseUri) directives.push(baseUri);
+  return directives.join("; ");
+}
+function toDirective(name: string, domains: string[] | undefined): string | null {
+  if (!domains || domains.length === 0) return null;
+  return `${name} ${domains.join(" ")}`;
+}
+export function applyCspMeta(html: string, cspContent: string | undefined): string {
+  if (!cspContent) return html;
+  if (/http-equiv=["']Content-Security-Policy["']/i.test(html)) return html;
+  const metaTag = `<meta http-equiv="Content-Security-Policy" content="${escapeHtmlAttribute(cspContent)}">`;
+  if (/<head[^>]*>/i.test(html)) {
+    return html.replace(/<head[^>]*>/i, (match) => `${match}\n${metaTag}`);
+  }
+  return `${metaTag}\n${html}`;
+}
+function safeInlineJSON(value: unknown): string {
+  return JSON.stringify(value)
+    .replace(/</g, "\\u003c")
+    .replace(/>/g, "\\u003e")
+    .replace(/&/g, "\\u0026")
+    .replace(/\u2028/g, "\\u2028")
+    .replace(/\u2029/g, "\\u2029");
+}
+function escapeHtml(value: string): string {
+  return value
+    .replace(/&/g, "&amp;")
+    .replace(/</g, "&lt;")
+    .replace(/>/g, "&gt;")
+    .replace(/"/g, "&quot;")
+    .replace(/'/g, "&#39;");
+}
+function escapeHtmlAttribute(value: string): string {
+  return value
+    .replace(/&/g, "&amp;")
+    .replace(/"/g, "&quot;")
+    .replace(/</g, "&lt;")
+    .replace(/>/g, "&gt;");
+}

package/resources/pi-agent/extensions/pi-mcp-adapter/index.ts ADDED Viewed

@@ -0,0 +1,232 @@
+import type { ExtensionAPI, ExtensionContext, ToolInfo } from "@mariozechner/pi-coding-agent";
+import type { McpExtensionState } from "./state.js";
+import { Type } from "@sinclair/typebox";
+import { showStatus, showTools, reconnectServers, authenticateServer, openMcpPanel } from "./commands.js";
+import { loadMcpConfig } from "./config.js";
+import { buildProxyDescription, createDirectToolExecutor, resolveDirectTools } from "./direct-tools.js";
+import { flushMetadataCache, initializeMcp, updateStatusBar } from "./init.js";
+import { loadMetadataCache } from "./metadata-cache.js";
+import { executeCall, executeConnect, executeDescribe, executeList, executeSearch, executeStatus, executeUiMessages } from "./proxy-modes.js";
+import { getConfigPathFromArgv } from "./utils.js";
+export default function mcpAdapter(pi: ExtensionAPI) {
+  let state: McpExtensionState | null = null;
+  let initPromise: Promise<McpExtensionState> | null = null;
+  const earlyConfigPath = getConfigPathFromArgv();
+  const earlyConfig = loadMcpConfig(earlyConfigPath);
+  const earlyCache = loadMetadataCache();
+  const prefix = earlyConfig.settings?.toolPrefix ?? "server";
+  const envRaw = process.env.MCP_DIRECT_TOOLS;
+  const directSpecs = envRaw === "__none__"
+    ? []
+    : resolveDirectTools(
+        earlyConfig,
+        earlyCache,
+        prefix,
+        envRaw?.split(",").map(s => s.trim()).filter(Boolean),
+      );
+  for (const spec of directSpecs) {
+    pi.registerTool({
+      name: spec.prefixedName,
+      label: `MCP: ${spec.originalName}`,
+      description: spec.description || "(no description)",
+      parameters: Type.Unsafe<Record<string, unknown>>(spec.inputSchema || { type: "object", properties: {} }),
+      execute: createDirectToolExecutor(() => state, () => initPromise, spec),
+    });
+  }
+  const getPiTools = (): ToolInfo[] => pi.getAllTools();
+  pi.registerFlag("mcp-config", {
+    description: "Path to MCP config file",
+    type: "string",
+  });
+  pi.on("session_start", async (_event, ctx) => {
+    initPromise = initializeMcp(pi, ctx);
+    initPromise.then(s => {
+      state = s;
+      initPromise = null;
+      updateStatusBar(s);
+    }).catch(err => {
+      console.error("MCP initialization failed:", err);
+      initPromise = null;
+    });
+  });
+  pi.on("session_shutdown", async () => {
+    if (initPromise) {
+      try {
+        state = await initPromise;
+      } catch {
+        // Initialization failed, nothing to clean up
+      }
+    }
+    if (state) {
+      if (state.uiServer) {
+        state.uiServer.close("session_shutdown");
+        state.uiServer = null;
+      }
+      flushMetadataCache(state);
+      await state.lifecycle.gracefulShutdown();
+      state = null;
+    }
+  });
+  pi.registerCommand("mcp", {
+    description: "Show MCP server status",
+    handler: async (args, ctx) => {
+      if (!state && initPromise) {
+        try {
+          state = await initPromise;
+        } catch {
+          if (ctx.hasUI) ctx.ui.notify("MCP initialization failed", "error");
+          return;
+        }
+      }
+      if (!state) {
+        if (ctx.hasUI) ctx.ui.notify("MCP not initialized", "error");
+        return;
+      }
+      const parts = args?.trim()?.split(/\s+/) ?? [];
+      const subcommand = parts[0] ?? "";
+      const targetServer = parts[1];
+      switch (subcommand) {
+        case "reconnect":
+          await reconnectServers(state, ctx, targetServer);
+          break;
+        case "tools":
+          await showTools(state, ctx);
+          break;
+        case "status":
+        case "":
+        default:
+          if (ctx.hasUI) {
+            await openMcpPanel(state, pi, ctx, earlyConfigPath);
+          } else {
+            await showStatus(state, ctx);
+          }
+          break;
+      }
+    },
+  });
+  pi.registerCommand("mcp-auth", {
+    description: "Authenticate with an MCP server (OAuth)",
+    handler: async (args, ctx) => {
+      const serverName = args?.trim();
+      if (!serverName) {
+        if (ctx.hasUI) ctx.ui.notify("Usage: /mcp-auth <server-name>", "error");
+        return;
+      }
+      if (!state && initPromise) {
+        try {
+          state = await initPromise;
+        } catch {
+          if (ctx.hasUI) ctx.ui.notify("MCP initialization failed", "error");
+          return;
+        }
+      }
+      if (!state) {
+        if (ctx.hasUI) ctx.ui.notify("MCP not initialized", "error");
+        return;
+      }
+      await authenticateServer(serverName, state.config, ctx);
+    },
+  });
+  pi.registerTool({
+    name: "mcp",
+    label: "MCP",
+    description: buildProxyDescription(earlyConfig, earlyCache, directSpecs),
+    parameters: Type.Object({
+      tool: Type.Optional(Type.String({ description: "Tool name to call (e.g., 'xcodebuild_list_sims')" })),
+      args: Type.Optional(Type.String({ description: "Arguments as JSON string (e.g., '{\"key\": \"value\"}')" })),
+      connect: Type.Optional(Type.String({ description: "Server name to connect (lazy connect + metadata refresh)" })),
+      describe: Type.Optional(Type.String({ description: "Tool name to describe (shows parameters)" })),
+      search: Type.Optional(Type.String({ description: "Search tools by name/description" })),
+      regex: Type.Optional(Type.Boolean({ description: "Treat search as regex (default: substring match)" })),
+      includeSchemas: Type.Optional(Type.Boolean({ description: "Include parameter schemas in search results (default: true)" })),
+      server: Type.Optional(Type.String({ description: "Filter to specific server (also disambiguates tool calls)" })),
+      action: Type.Optional(Type.String({ description: "Action: 'ui-messages' to retrieve prompts/intents from UI sessions" })),
+    }),
+    async execute(_toolCallId, params: {
+      tool?: string;
+      args?: string;
+      connect?: string;
+      describe?: string;
+      search?: string;
+      regex?: boolean;
+      includeSchemas?: boolean;
+      server?: string;
+      action?: string;
+    }, _signal, _onUpdate, _ctx) {
+      let parsedArgs: Record<string, unknown> | undefined;
+      if (params.args) {
+        try {
+          parsedArgs = JSON.parse(params.args);
+          if (typeof parsedArgs !== "object" || parsedArgs === null || Array.isArray(parsedArgs)) {
+            const gotType = Array.isArray(parsedArgs) ? "array" : parsedArgs === null ? "null" : typeof parsedArgs;
+            return {
+              content: [{ type: "text" as const, text: `Invalid args: expected a JSON object, got ${gotType}` }],
+              isError: true,
+              details: { error: "invalid_args_type" },
+            };
+          }
+        } catch (e) {
+          return {
+            content: [{ type: "text" as const, text: `Invalid args JSON: ${e instanceof Error ? e.message : e}` }],
+            isError: true,
+            details: { error: "invalid_args" },
+          };
+        }
+      }
+      if (!state && initPromise) {
+        try {
+          state = await initPromise;
+        } catch {
+          return {
+            content: [{ type: "text" as const, text: "MCP initialization failed" }],
+            details: { error: "init_failed" },
+          };
+        }
+      }
+      if (!state) {
+        return {
+          content: [{ type: "text" as const, text: "MCP not initialized" }],
+          details: { error: "not_initialized" },
+        };
+      }
+      if (params.action === "ui-messages") {
+        return executeUiMessages(state);
+      }
+      if (params.tool) {
+        return executeCall(state, params.tool, parsedArgs, params.server);
+      }
+      if (params.connect) {
+        return executeConnect(state, params.connect);
+      }
+      if (params.describe) {
+        return executeDescribe(state, params.describe);
+      }
+      if (params.search) {
+        return executeSearch(state, params.search, params.regex, params.server, params.includeSchemas, getPiTools);
+      }
+      if (params.server) {
+        return executeList(state, params.server);
+      }
+      return executeStatus(state);
+    },
+  });
+}