npm - @ashx-j/lunr - Versions diffs - 0.1.0 - Mend

@ashx-j/lunr 0.1.0

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 (2269) hide show

package/dist/builtin-extensions/pi-mcp-adapter/host-html-template.js ADDED Viewed

@@ -0,0 +1,413 @@
+// Use locally bundled AppBridge to avoid CDN Zod bundling issues
+const DEFAULT_APP_BRIDGE_MODULE_URL = "/app-bridge.bundle.js";
+export function buildHostHtmlTemplate(input) {
+    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) {
+    if (!csp)
+        return undefined;
+    const directives = [];
+    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, domains) {
+    if (!domains || domains.length === 0)
+        return null;
+    return `${name} ${domains.join(" ")}`;
+}
+export function applyCspMeta(html, cspContent) {
+    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) {
+    return JSON.stringify(value)
+        .replace(/</g, "\\u003c")
+        .replace(/>/g, "\\u003e")
+        .replace(/&/g, "\\u0026")
+        .replace(/\u2028/g, "\\u2028")
+        .replace(/\u2029/g, "\\u2029");
+}
+function escapeHtml(value) {
+    return value
+        .replace(/&/g, "&amp;")
+        .replace(/</g, "&lt;")
+        .replace(/>/g, "&gt;")
+        .replace(/"/g, "&quot;")
+        .replace(/'/g, "&#39;");
+}
+function escapeHtmlAttribute(value) {
+    return value
+        .replace(/&/g, "&amp;")
+        .replace(/"/g, "&quot;")
+        .replace(/</g, "&lt;")
+        .replace(/>/g, "&gt;");
+}
+//# sourceMappingURL=host-html-template.js.map

package/dist/builtin-extensions/pi-mcp-adapter/host-html-template.js.map ADDED Viewed

@@ -0,0 +1 @@

+ {"version":3,"file":"host-html-template.js","sourceRoot":"","sources":["../../../src/builtin-extensions/pi-mcp-adapter/host-html-template.ts"],"names":[],"mappings":"AAGA,iEAAiE;AACjE,MAAM,6BAA6B,GAAG,uBAAuB,CAAC;AAe9D,MAAM,UAAU,qBAAqB,CAAC,KAA4B,EAAU;IAC1E,MAAM,UAAU,GAAG,mBAAmB,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAChE,MAAM,YAAY,GAAG,YAAY,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;IACnE,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,IAAI,EAAE,CAAC;IAE5C,MAAM,YAAY,GAAG,cAAc,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IACxD,MAAM,QAAQ,GAAG,cAAc,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IAChD,MAAM,MAAM,GAAG,cAAc,CAAC,YAAY,CAAC,CAAC;IAC5C,MAAM,UAAU,GAAG,cAAc,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IACpD,MAAM,QAAQ,GAAG,cAAc,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IAChD,MAAM,eAAe,GAAG,cAAc,CAAC,WAAW,CAAC,CAAC;IACpD,MAAM,cAAc,GAAG,cAAc,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IAC5D,MAAM,kBAAkB,GAAG,cAAc,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;IACpE,MAAM,gBAAgB,GAAG,cAAc,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;IAChE,MAAM,SAAS,GAAG,cAAc,CAAC,KAAK,CAAC,kBAAkB,IAAI,6BAA6B,CAAC,CAAC;IAE5F,OAAO;;;;;oBAKW,UAAU,CAAC,KAAK,CAAC,UAAU,CAAC,MAAM,UAAU,CAAC,KAAK,CAAC,QAAQ,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;sDAyE1B,SAAS;;4BAEnC,YAAY;0BACd,UAAU;wBACZ,QAAQ;wBACR,QAAQ;2BACL,eAAe;8BACZ,cAAc;mCACT,kBAAkB;iCACpB,gBAAgB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;QA0OzC,CAAC;AAAA,CACR;AAED,MAAM,UAAU,mBAAmB,CAAC,GAA8B,EAAsB;IACtF,IAAI,CAAC,GAAG;QAAE,OAAO,SAAS,CAAC;IAE3B,MAAM,UAAU,GAAa,EAAE,CAAC;IAChC,UAAU,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;IAEtC,MAAM,SAAS,GAAG,WAAW,CAAC,YAAY,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC;IAC/D,MAAM,QAAQ,GAAG,WAAW,CAAC,WAAW,EAAE,GAAG,CAAC,YAAY,CAAC,CAAC;IAC5D,MAAM,OAAO,GAAG,WAAW,CAAC,UAAU,EAAE,GAAG,CAAC,WAAW,CAAC,CAAC;IACzD,MAAM,MAAM,GAAG,WAAW,CAAC,SAAS,EAAE,GAAG,CAAC,UAAU,CAAC,CAAC;IACtD,MAAM,QAAQ,GAAG,WAAW,CAAC,WAAW,EAAE,GAAG,CAAC,YAAY,CAAC,CAAC;IAC5D,MAAM,UAAU,GAAG,WAAW,CAAC,aAAa,EAAE,GAAG,CAAC,cAAc,CAAC,CAAC;IAClE,MAAM,QAAQ,GAAG,WAAW,CAAC,WAAW,EAAE,GAAG,CAAC,YAAY,CAAC,CAAC;IAC5D,MAAM,SAAS,GAAG,WAAW,CAAC,YAAY,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC;IAC/D,MAAM,OAAO,GAAG,WAAW,CAAC,UAAU,EAAE,GAAG,CAAC,cAAc,CAAC,CAAC;IAE5D,IAAI,SAAS;QAAE,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC1C,IAAI,QAAQ;QAAE,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACxC,IAAI,OAAO;QAAE,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACtC,IAAI,MAAM;QAAE,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACpC,IAAI,QAAQ;QAAE,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACxC,IAAI,UAAU;QAAE,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC5C,IAAI,QAAQ;QAAE,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACxC,IAAI,SAAS;QAAE,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC1C,IAAI,OAAO;QAAE,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAEtC,OAAO,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAAA,CAC9B;AAED,SAAS,WAAW,CAAC,IAAY,EAAE,OAA6B,EAAiB;IAC/E,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IAClD,OAAO,GAAG,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;AAAA,CACvC;AAED,MAAM,UAAU,YAAY,CAAC,IAAY,EAAE,UAA8B,EAAU;IACjF,IAAI,CAAC,UAAU;QAAE,OAAO,IAAI,CAAC;IAC7B,IAAI,6CAA6C,CAAC,IAAI,CAAC,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC;IAC1E,MAAM,OAAO,GAAG,uDAAuD,mBAAmB,CAAC,UAAU,CAAC,IAAI,CAAC;IAC3G,IAAI,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC9B,OAAO,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,GAAG,KAAK,KAAK,OAAO,EAAE,CAAC,CAAC;IACzE,CAAC;IACD,OAAO,GAAG,OAAO,KAAK,IAAI,EAAE,CAAC;AAAA,CAC9B;AAED,SAAS,cAAc,CAAC,KAAc,EAAU;IAC9C,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;SACzB,OAAO,CAAC,IAAI,EAAE,SAAS,CAAC;SACxB,OAAO,CAAC,IAAI,EAAE,SAAS,CAAC;SACxB,OAAO,CAAC,IAAI,EAAE,SAAS,CAAC;SACxB,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC;SAC7B,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;AAAA,CAClC;AAED,SAAS,UAAU,CAAC,KAAa,EAAU;IACzC,OAAO,KAAK;SACT,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC;SACtB,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC;SACrB,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC;SACrB,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC;SACvB,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;AAAA,CAC3B;AAED,SAAS,mBAAmB,CAAC,KAAa,EAAU;IAClD,OAAO,KAAK;SACT,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC;SACtB,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC;SACvB,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC;SACrB,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;AAAA,CAC1B","sourcesContent":["// @ts-nocheck\nimport type { UiHostContext, UiResourceContent, UiResourceCsp } from \"./types.ts\";\n\n// Use locally bundled AppBridge to avoid CDN Zod bundling issues\nconst DEFAULT_APP_BRIDGE_MODULE_URL = \"/app-bridge.bundle.js\";\n\nexport interface HostHtmlTemplateInput {\n sessionToken: string;\n serverName: string;\n toolName: string;\n toolArgs: Record<string, unknown>;\n resource: UiResourceContent;\n allowAttribute: string;\n requireToolConsent: boolean;\n cacheToolConsent: boolean;\n hostContext?: UiHostContext;\n appBridgeModuleUrl?: string;\n}\n\nexport function buildHostHtmlTemplate(input: HostHtmlTemplateInput): string {\n const cspContent = buildCspMetaContent(input.resource.meta.csp);\n const resourceHtml = applyCspMeta(input.resource.html, cspContent);\n const hostContext = input.hostContext ?? {};\n\n const sessionToken = safeInlineJSON(input.sessionToken);\n const toolArgs = safeInlineJSON(input.toolArgs);\n const uiHtml = safeInlineJSON(resourceHtml);\n const serverName = safeInlineJSON(input.serverName);\n const toolName = safeInlineJSON(input.toolName);\n const hostContextJson = safeInlineJSON(hostContext);\n const allowAttribute = safeInlineJSON(input.allowAttribute);\n const requireToolConsent = safeInlineJSON(input.requireToolConsent);\n const cacheToolConsent = safeInlineJSON(input.cacheToolConsent);\n const moduleUrl = safeInlineJSON(input.appBridgeModuleUrl ?? DEFAULT_APP_BRIDGE_MODULE_URL);\n\n return `<!doctype html>\n<html lang=\"en\">\n<head>\n <meta charset=\"utf-8\" />\n <meta name=\"viewport\" content=\"width=device-width, initial-scale=1\" />\n <title>MCP UI - ${escapeHtml(input.serverName)} / ${escapeHtml(input.toolName)}</title>\n <style>\n :root {\n color-scheme: light dark;\n --bg: #0f1115;\n --surface: #181c22;\n --text: #ecf0f5;\n --muted: #a9b2bf;\n --accent: #43c0ff;\n --border: rgba(255, 255, 255, 0.12);\n --good: #34d399;\n --warn: #fbbf24;\n --bad: #f87171;\n }\n @media (prefers-color-scheme: light) {\n :root {\n --bg: #f6f7fb;\n --surface: #ffffff;\n --text: #1d2939;\n --muted: #667085;\n --accent: #0ea5e9;\n --border: rgba(15, 23, 42, 0.14);\n --good: #059669;\n --warn: #b45309;\n --bad: #b91c1c;\n }\n }\n * { box-sizing: border-box; }\n 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); }\n body { display: flex; flex-direction: column; min-height: 100vh; }\n header { background: var(--surface); border-bottom: 1px solid var(--border); padding: 10px 14px; display: flex; align-items: center; justify-content: space-between; gap: 10px; }\n .title { display: flex; gap: 8px; align-items: baseline; min-width: 0; }\n .server { font-size: 12px; color: var(--muted); text-transform: uppercase; letter-spacing: 0.08em; white-space: nowrap; }\n .tool { font-size: 14px; font-weight: 600; overflow: hidden; text-overflow: ellipsis; white-space: nowrap; }\n .badge { border: 1px solid var(--border); border-radius: 999px; padding: 2px 8px; font-size: 11px; color: var(--muted); white-space: nowrap; }\n .controls { display: flex; gap: 8px; align-items: center; }\n .status { font-size: 12px; color: var(--muted); white-space: nowrap; }\n button { border: 1px solid var(--border); background: transparent; color: var(--text); border-radius: 8px; padding: 6px 10px; cursor: pointer; font-size: 12px; }\n button.primary { border-color: color-mix(in srgb, var(--good) 40%, var(--border) 60%); color: var(--good); }\n button.danger { border-color: color-mix(in srgb, var(--bad) 40%, var(--border) 60%); color: var(--bad); }\n button:hover { background: color-mix(in srgb, var(--surface) 75%, var(--accent) 25%); }\n main { flex: 1; min-height: 0; padding: 10px; display: flex; }\n iframe { width: 100%; height: 100%; border: 1px solid var(--border); border-radius: 10px; background: white; }\n .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; }\n .overlay.visible { display: flex; }\n .panel { width: min(680px, calc(100vw - 40px)); background: var(--surface); border: 1px solid var(--border); border-radius: 12px; padding: 18px; }\n .panel h2 { margin: 0 0 8px; font-size: 16px; }\n .panel p { margin: 0; color: var(--muted); line-height: 1.4; font-size: 14px; white-space: pre-wrap; }\n </style>\n</head>\n<body>\n <header>\n <div class=\"title\">\n <span class=\"server\">MCP · <span id=\"server-name\"></span></span>\n <span class=\"tool\" id=\"tool-name\"></span>\n <span class=\"badge\">Sandboxed</span>\n </div>\n <div class=\"controls\">\n <span class=\"status\" id=\"status\">Loading UI...</span>\n <button class=\"primary\" id=\"done-btn\" title=\"Cmd/Ctrl+Enter\">Done</button>\n <button class=\"danger\" id=\"cancel-btn\" title=\"Escape\">Cancel</button>\n </div>\n </header>\n <main>\n <iframe id=\"mcp-app\" referrerpolicy=\"no-referrer\"></iframe>\n </main>\n <div class=\"overlay\" id=\"error-overlay\">\n <div class=\"panel\">\n <h2>UI Error</h2>\n <p id=\"error-message\"></p>\n </div>\n </div>\n <script type=\"module\">\n import { AppBridge, PostMessageTransport } from ${moduleUrl};\n\n const SESSION_TOKEN = ${sessionToken};\n const SERVER_NAME = ${serverName};\n const TOOL_NAME = ${toolName};\n const TOOL_ARGS = ${toolArgs};\n const HOST_CONTEXT = ${hostContextJson};\n const ALLOW_ATTRIBUTE = ${allowAttribute};\n const REQUIRE_TOOL_CONSENT = ${requireToolConsent};\n const CACHE_TOOL_CONSENT = ${cacheToolConsent};\n const STREAM_CONTEXT_KEY = \"pi-mcp-adapter/stream\";\n const STREAM_PATCH_METHOD = \"notifications/pi-mcp-adapter/ui-result-patch\";\n\n const iframe = document.getElementById(\"mcp-app\");\n const statusNode = document.getElementById(\"status\");\n const doneBtn = document.getElementById(\"done-btn\");\n const cancelBtn = document.getElementById(\"cancel-btn\");\n const errorOverlay = document.getElementById(\"error-overlay\");\n const errorMessage = document.getElementById(\"error-message\");\n\n document.getElementById(\"server-name\").textContent = SERVER_NAME;\n document.getElementById(\"tool-name\").textContent = TOOL_NAME;\n\n const setStatus = (text, isError = false) => {\n statusNode.textContent = text;\n statusNode.style.color = isError ? \"var(--bad)\" : \"var(--muted)\";\n };\n\n const showError = (message) => {\n errorMessage.textContent = message;\n errorOverlay.classList.add(\"visible\");\n setStatus(\"Error\", true);\n };\n\n const post = async (endpoint, params) => {\n const response = await fetch(endpoint, {\n method: \"POST\",\n headers: { \"Content-Type\": \"application/json\" },\n body: JSON.stringify({ token: SESSION_TOKEN, params }),\n });\n\n const body = await response.json().catch(() => ({ ok: false, error: \"Invalid JSON response\" }));\n if (!response.ok || !body.ok) {\n const message = body.error || (\"HTTP \" + response.status);\n throw new Error(message);\n }\n return body.result ?? {};\n };\n\n let consentGranted = !REQUIRE_TOOL_CONSENT;\n const initialStreamContext = HOST_CONTEXT?.[STREAM_CONTEXT_KEY];\n const streamMode = initialStreamContext?.mode === \"stream-first\" ? \"stream-first\" : \"eager\";\n\n const bridge = new AppBridge(\n null,\n { name: \"pi\", version: \"1.0.0\" },\n { serverTools: {}, openLinks: {}, logging: {}, updateModelContext: {}, message: {} },\n { hostContext: HOST_CONTEXT }\n );\n\n bridge.oncalltool = async (params) => {\n if (!consentGranted) {\n const accepted = window.confirm(\"Allow this UI to call server tools for this session?\");\n if (!accepted) {\n await post(\"/proxy/ui/consent\", { approved: false }).catch(() => {});\n return {\n isError: true,\n content: [{ type: \"text\", text: \"Tool call denied by user.\" }],\n };\n }\n await post(\"/proxy/ui/consent\", { approved: true });\n if (CACHE_TOOL_CONSENT) {\n consentGranted = true;\n }\n }\n const result = await post(\"/proxy/tools/call\", params);\n // Notify agent about the tool call\n await post(\"/proxy/ui/message\", {\n type: \"intent\",\n intent: \"call_tool\",\n params: { tool: params.name, arguments: params.arguments, isError: result.isError }\n }).catch(() => {});\n return result;\n };\n\n bridge.onmessage = async (params) => post(\"/proxy/ui/message\", params);\n bridge.onupdatemodelcontext = async (params) => post(\"/proxy/ui/context\", params);\n \n // Also listen for raw postMessage events with custom types (notify, prompt, intent, etc.)\n // These bypass the AppBridge protocol but are used by some MCP UI implementations\n window.addEventListener(\"message\", async (event) => {\n const data = event.data;\n if (!data || typeof data !== \"object\") return;\n \n // Skip AppBridge protocol messages (handled by bridge)\n if (data.jsonrpc || (typeof data.method === \"string\" && (data.method.startsWith(\"app/\") || data.method.startsWith(\"host/\")))) return;\n \n // Handle raw UI action messages\n const msgType = data.type;\n if (typeof msgType !== \"string\") return;\n \n if (msgType === \"notify\" || msgType === \"prompt\" || msgType === \"intent\" || msgType === \"message\") {\n // Standard MCP-UI types - preserve their semantics\n // Support both { type, payload: {...} } and { type, field: value } formats\n const { type: _, payload, ...directFields } = data;\n await post(\"/proxy/ui/message\", { type: msgType, ...directFields, ...(payload || {}) }).catch(() => {});\n } else if (!msgType.startsWith(\"ui-lifecycle-\") && !msgType.startsWith(\"ui-message-\")) {\n // Any other custom type - forward as intent with type as intent name\n // (Skip internal lifecycle/ack messages)\n const payload = data.payload || {};\n await post(\"/proxy/ui/message\", {\n type: \"intent\",\n intent: msgType,\n params: payload,\n }).catch(() => {});\n }\n });\n bridge.ondownloadfile = async (params) => post(\"/proxy/ui/download-file\", params);\n bridge.onrequestdisplaymode = async (params) => post(\"/proxy/ui/request-display-mode\", params);\n bridge.onopenlink = async (params) => {\n const result = await post(\"/proxy/ui/open-link\", params);\n if (!result.isError) {\n window.open(params.url, \"_blank\", \"noopener,noreferrer\");\n // Notify agent about the link open\n await post(\"/proxy/ui/message\", {\n type: \"intent\",\n intent: \"open_link\",\n params: { url: params.url }\n }).catch(() => {});\n }\n return result;\n };\n\n bridge.oninitialized = () => {\n if (streamMode !== \"stream-first\") {\n bridge.sendToolInput({ arguments: TOOL_ARGS });\n }\n setStatus(streamMode === \"stream-first\" ? \"Streaming…\" : \"Connected\");\n };\n\n bridge.onsizechange = ({ width, height }) => {\n if (typeof width === \"number\" && width > 0) {\n iframe.style.minWidth = Math.min(width, window.innerWidth - 24) + \"px\";\n }\n if (typeof height === \"number\" && height > 0) {\n iframe.style.height = Math.max(height, 320) + \"px\";\n }\n };\n\n if (ALLOW_ATTRIBUTE) {\n iframe.setAttribute(\"allow\", ALLOW_ATTRIBUTE);\n }\n\n // Connect bridge BEFORE loading iframe to ensure we're listening when the app sends ui/initialize\n try {\n const transport = new PostMessageTransport(iframe.contentWindow, null);\n await bridge.connect(transport);\n } catch (error) {\n console.error(\"[host] Bridge connection failed:\", error);\n showError(\"Failed to initialize AppBridge: \" + String(error));\n }\n\n const iframeLoaded = new Promise((resolve) => {\n iframe.onload = resolve;\n });\n iframe.src = \"/ui-app?session=\" + encodeURIComponent(SESSION_TOKEN);\n await iframeLoaded;\n\n const eventSource = new EventSource(\"/events?session=\" + encodeURIComponent(SESSION_TOKEN));\n eventSource.addEventListener(\"tool-input\", (event) => {\n try {\n bridge.sendToolInput(JSON.parse(event.data));\n } catch (error) {\n showError(\"Failed to forward tool input: \" + String(error));\n }\n });\n eventSource.addEventListener(\"tool-result\", (event) => {\n try {\n bridge.sendToolResult(JSON.parse(event.data));\n } catch (error) {\n showError(\"Failed to forward tool result: \" + String(error));\n }\n });\n eventSource.addEventListener(\"tool-cancelled\", (event) => {\n try {\n bridge.sendToolCancelled(JSON.parse(event.data));\n } catch (error) {\n showError(\"Failed to forward cancellation: \" + String(error));\n }\n });\n eventSource.addEventListener(\"result-patch\", async (event) => {\n try {\n await bridge.notification({\n method: STREAM_PATCH_METHOD,\n params: JSON.parse(event.data),\n });\n } catch (error) {\n showError(\"Failed to forward stream patch: \" + String(error));\n }\n });\n eventSource.addEventListener(\"host-context\", (event) => {\n try {\n bridge.setHostContext(JSON.parse(event.data));\n } catch {}\n });\n eventSource.addEventListener(\"session-complete\", async () => {\n await bridge.teardownResource({}).catch(() => {});\n eventSource.close();\n window.close();\n });\n eventSource.onerror = () => {\n setStatus(\"Connection lost\", true);\n };\n\n const heartbeat = setInterval(() => {\n post(\"/proxy/ui/heartbeat\", {}).catch(() => {});\n }, 10000);\n\n const complete = async (reason) => {\n try {\n await post(\"/proxy/ui/complete\", { reason });\n } catch {}\n try {\n await bridge.teardownResource({});\n } catch {}\n clearInterval(heartbeat);\n eventSource.close();\n window.close();\n };\n\n doneBtn.addEventListener(\"click\", () => complete(\"done\"));\n cancelBtn.addEventListener(\"click\", () => complete(\"cancel\"));\n window.addEventListener(\"keydown\", (event) => {\n if (event.key === \"Escape\") {\n event.preventDefault();\n complete(\"cancel\");\n } else if ((event.metaKey || event.ctrlKey) && event.key === \"Enter\") {\n event.preventDefault();\n complete(\"done\");\n }\n });\n </script>\n</body>\n</html>`;\n}\n\nexport function buildCspMetaContent(csp: UiResourceCsp | undefined): string | undefined {\n if (!csp) return undefined;\n\n const directives: string[] = [];\n directives.push(\"default-src 'none'\");\n\n const scriptSrc = toDirective(\"script-src\", csp.scriptDomains);\n const styleSrc = toDirective(\"style-src\", csp.styleDomains);\n const fontSrc = toDirective(\"font-src\", csp.fontDomains);\n const imgSrc = toDirective(\"img-src\", csp.imgDomains);\n const mediaSrc = toDirective(\"media-src\", csp.mediaDomains);\n const connectSrc = toDirective(\"connect-src\", csp.connectDomains);\n const frameSrc = toDirective(\"frame-src\", csp.frameDomains);\n const workerSrc = toDirective(\"worker-src\", csp.workerDomains);\n const baseUri = toDirective(\"base-uri\", csp.baseUriDomains);\n\n if (scriptSrc) directives.push(scriptSrc);\n if (styleSrc) directives.push(styleSrc);\n if (fontSrc) directives.push(fontSrc);\n if (imgSrc) directives.push(imgSrc);\n if (mediaSrc) directives.push(mediaSrc);\n if (connectSrc) directives.push(connectSrc);\n if (frameSrc) directives.push(frameSrc);\n if (workerSrc) directives.push(workerSrc);\n if (baseUri) directives.push(baseUri);\n\n return directives.join(\"; \");\n}\n\nfunction toDirective(name: string, domains: string[] | undefined): string | null {\n if (!domains || domains.length === 0) return null;\n return `${name} ${domains.join(\" \")}`;\n}\n\nexport function applyCspMeta(html: string, cspContent: string | undefined): string {\n if (!cspContent) return html;\n if (/http-equiv=[\"']Content-Security-Policy[\"']/i.test(html)) return html;\n const metaTag = `<meta http-equiv=\"Content-Security-Policy\" content=\"${escapeHtmlAttribute(cspContent)}\">`;\n if (/<head[^>]*>/i.test(html)) {\n return html.replace(/<head[^>]*>/i, (match) => `${match}\\n${metaTag}`);\n }\n return `${metaTag}\\n${html}`;\n}\n\nfunction safeInlineJSON(value: unknown): string {\n return JSON.stringify(value)\n .replace(/</g, \"\\\\u003c\")\n .replace(/>/g, \"\\\\u003e\")\n .replace(/&/g, \"\\\\u0026\")\n .replace(/\\u2028/g, \"\\\\u2028\")\n .replace(/\\u2029/g, \"\\\\u2029\");\n}\n\nfunction escapeHtml(value: string): string {\n return value\n .replace(/&/g, \"&\")\n .replace(/</g, \"<\")\n .replace(/>/g, \">\")\n .replace(/\"/g, \""\")\n .replace(/'/g, \"'\");\n}\n\nfunction escapeHtmlAttribute(value: string): string {\n return value\n .replace(/&/g, \"&\")\n .replace(/\"/g, \""\")\n .replace(/</g, \"<\")\n .replace(/>/g, \">\");\n}\n"]}

package/dist/builtin-extensions/pi-mcp-adapter/index.d.ts ADDED Viewed

@@ -0,0 +1,3 @@
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+export default function mcpAdapter(pi: ExtensionAPI): void;
+//# sourceMappingURL=index.d.ts.map

package/dist/builtin-extensions/pi-mcp-adapter/index.d.ts.map ADDED Viewed

@@ -0,0 +1 @@

+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/builtin-extensions/pi-mcp-adapter/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,YAAY,EAAY,MAAM,iCAAiC,CAAC;AAc9E,MAAM,CAAC,OAAO,UAAU,UAAU,CAAC,EAAE,EAAE,YAAY,QA8VlD","sourcesContent":["// @ts-nocheck\nimport type { ExtensionAPI, ToolInfo } from \"@earendil-works/pi-coding-agent\";\nimport type { McpExtensionState } from \"./state.ts\";\nimport { Type } from \"typebox\";\nimport { showStatus, showTools, reconnectServers, authenticateServer, logoutServer, openMcpAuthPanel, openMcpPanel, openMcpSetup } from \"./commands.ts\";\nimport { loadMcpConfig } from \"./config.ts\";\nimport { buildProxyDescription, createDirectToolExecutor, getMissingConfiguredDirectToolServers, resolveDirectTools } from \"./direct-tools.ts\";\nimport { flushMetadataCache, initializeMcp, updateStatusBar } from \"./init.ts\";\nimport { loadMetadataCache } from \"./metadata-cache.ts\";\nimport { executeAuthComplete, executeAuthStart, executeCall, executeConnect, executeDescribe, executeList, executeSearch, executeStatus, executeUiMessages } from \"./proxy-modes.ts\";\nimport { getConfigPathFromArgv, normalizeDirectToolInputSchema, truncateAtWord } from \"./utils.ts\";\nimport { initializeOAuth, shutdownOAuth } from \"./mcp-auth-flow.ts\";\nimport { createMcpDirectToolCallRenderer, renderMcpProxyToolCall, renderMcpToolResult } from \"./tool-result-renderer.ts\";\nimport { toolErrorOverride } from \"./error-signal.ts\";\n\nexport default function mcpAdapter(pi: ExtensionAPI) {\n let state: McpExtensionState | null = null;\n let initPromise: Promise<McpExtensionState> | null = null;\n let lifecycleGeneration = 0;\n\n async function shutdownState(currentState: McpExtensionState | null, reason: string): Promise<void> {\n if (!currentState) return;\n\n if (currentState.uiServer) {\n currentState.uiServer.close(reason);\n currentState.uiServer = null;\n }\n\n let flushError: unknown;\n try {\n flushMetadataCache(currentState);\n } catch (error) {\n flushError = error;\n }\n\n try {\n await currentState.lifecycle.gracefulShutdown();\n } catch (error) {\n if (flushError) {\n console.error(\"MCP: graceful shutdown failed after metadata flush error\", error);\n } else {\n throw error;\n }\n }\n\n if (flushError) {\n throw flushError;\n }\n }\n\n const earlyConfigPath = getConfigPathFromArgv();\n const earlyConfig = loadMcpConfig(earlyConfigPath);\n const earlyCache = loadMetadataCache();\n const prefix = earlyConfig.settings?.toolPrefix ?? \"server\";\n\n const envRaw = process.env.MCP_DIRECT_TOOLS;\n const directSpecs = envRaw === \"__none__\"\n ? []\n : resolveDirectTools(\n earlyConfig,\n earlyCache,\n prefix,\n envRaw?.split(\",\").map(s => s.trim()).filter(Boolean),\n );\n const missingConfiguredDirectToolServers = getMissingConfiguredDirectToolServers(earlyConfig, earlyCache);\n const shouldRegisterProxyTool =\n earlyConfig.settings?.disableProxyTool !== true\n || directSpecs.length === 0\n || missingConfiguredDirectToolServers.length > 0;\n\n for (const spec of directSpecs) {\n (pi.registerTool as (tool: unknown) => unknown)({\n name: spec.prefixedName,\n label: `MCP: ${spec.originalName}`,\n description: spec.description || \"(no description)\",\n promptSnippet: truncateAtWord(spec.description, 100) || `MCP tool from ${spec.serverName}`,\n parameters: Type.Unsafe(normalizeDirectToolInputSchema(spec.inputSchema) as never),\n execute: createDirectToolExecutor(() => state, () => initPromise, spec),\n renderCall: createMcpDirectToolCallRenderer(spec.prefixedName),\n renderResult: renderMcpToolResult,\n });\n }\n\n const getPiTools = (): ToolInfo[] => pi.getAllTools();\n\n pi.registerFlag(\"mcp-config\", {\n description: \"Path to MCP config file\",\n type: \"string\",\n });\n\n pi.on(\"session_start\", async (_event, ctx) => {\n const generation = ++lifecycleGeneration;\n const previousState = state;\n state = null;\n initPromise = null;\n\n try {\n await Promise.all([\n shutdownState(previousState, \"session_restart\"),\n shutdownOAuth(),\n ]);\n } catch (error) {\n console.error(\"MCP: failed to shut down previous session state\", error);\n }\n\n if (generation !== lifecycleGeneration) {\n return;\n }\n\n await initializeOAuth().catch(err => {\n console.error(\"MCP OAuth initialization failed:\", err);\n });\n\n const promise = initializeMcp(pi, ctx);\n initPromise = promise;\n\n promise.then(async (nextState) => {\n if (generation !== lifecycleGeneration || initPromise !== promise) {\n try {\n await shutdownState(nextState, \"stale_session_start\");\n } catch (error) {\n console.error(\"MCP: failed to clean stale session state\", error);\n }\n return;\n }\n\n state = nextState;\n updateStatusBar(nextState);\n initPromise = null;\n }).catch(err => {\n if (generation !== lifecycleGeneration) {\n return;\n }\n if (initPromise !== promise && initPromise !== null) {\n return;\n }\n console.error(\"MCP initialization failed:\", err);\n initPromise = null;\n });\n });\n\n pi.on(\"session_shutdown\", async () => {\n ++lifecycleGeneration;\n const currentState = state;\n state = null;\n initPromise = null;\n\n try {\n await Promise.all([\n shutdownState(currentState, \"session_shutdown\"),\n shutdownOAuth(),\n ]);\n } catch (error) {\n console.error(\"MCP: session shutdown cleanup failed\", error);\n }\n });\n\n // Re-flag returned MCP tool failures so pi registers them as errors (see toolErrorOverride).\n pi.on(\"tool_result\", (event) => toolErrorOverride(event.details));\n\n pi.registerCommand(\"mcp\", {\n description: \"Show MCP server status\",\n handler: async (args, ctx) => {\n if (!state && initPromise) {\n try {\n state = await initPromise;\n } catch (error) {\n const message = error instanceof Error ? error.message : String(error);\n if (ctx.hasUI) ctx.ui.notify(`MCP initialization failed: ${message}`, \"error\");\n return;\n }\n }\n if (!state) {\n if (ctx.hasUI) ctx.ui.notify(\"MCP not initialized\", \"error\");\n return;\n }\n\n const parts = args?.trim()?.split(/\\s+/) ?? [];\n const subcommand = parts[0] ?? \"\";\n const targetServer = parts[1];\n const rest = parts.slice(1).join(\" \");\n\n switch (subcommand) {\n case \"reconnect\":\n await reconnectServers(state, ctx, targetServer);\n break;\n case \"tools\":\n await showTools(state, ctx);\n break;\n case \"setup\": {\n const result = await openMcpSetup(state, pi, ctx, earlyConfigPath, \"setup\");\n if (result?.configChanged) {\n await ctx.reload();\n return;\n }\n break;\n }\n case \"logout\": {\n const serverName = rest;\n if (!serverName) {\n if (ctx.hasUI) ctx.ui.notify(\"Usage: /mcp logout <server>\", \"error\");\n return;\n }\n await logoutServer(serverName, state, ctx);\n break;\n }\n case \"status\":\n case \"\":\n default:\n if (ctx.hasUI) {\n const result = await openMcpPanel(state, pi, ctx, earlyConfigPath);\n if (result?.configChanged) {\n await ctx.reload();\n return;\n }\n } else {\n await showStatus(state, ctx);\n }\n break;\n }\n },\n });\n\n pi.registerCommand(\"mcp-auth\", {\n description: \"Authenticate with an MCP server (OAuth)\",\n handler: async (args, ctx) => {\n const serverName = args?.trim();\n if (!serverName && !ctx.hasUI) {\n return;\n }\n\n if (!state && initPromise) {\n try {\n state = await initPromise;\n } catch (error) {\n const message = error instanceof Error ? error.message : String(error);\n if (ctx.hasUI) ctx.ui.notify(`MCP initialization failed: ${message}`, \"error\");\n return;\n }\n }\n if (!state) {\n if (ctx.hasUI) ctx.ui.notify(\"MCP not initialized\", \"error\");\n return;\n }\n\n if (!serverName) {\n await openMcpAuthPanel(state, pi, ctx, earlyConfigPath);\n return;\n }\n\n await authenticateServer(serverName, state.config, ctx);\n },\n });\n\n if (shouldRegisterProxyTool) {\n (pi.registerTool as (tool: unknown) => unknown)({\n name: \"mcp\",\n label: \"MCP\",\n description: buildProxyDescription(earlyConfig, earlyCache, directSpecs),\n promptSnippet: \"MCP gateway - connect to MCP servers and call their tools\",\n renderCall: renderMcpProxyToolCall,\n parameters: Type.Object({\n tool: Type.Optional(Type.String({ description: \"Tool name to call (e.g., 'xcodebuild_list_sims')\" })),\n args: Type.Optional(Type.String({ description: \"Arguments as JSON string (e.g., '{\\\"key\\\": \\\"value\\\"}')\" })),\n connect: Type.Optional(Type.String({ description: \"Server name to connect (lazy connect + metadata refresh)\" })),\n describe: Type.Optional(Type.String({ description: \"Tool name to describe (shows parameters)\" })),\n search: Type.Optional(Type.String({ description: \"Search tools by name/description\" })),\n regex: Type.Optional(Type.Boolean({ description: \"Treat search as regex (default: substring match)\" })),\n includeSchemas: Type.Optional(Type.Boolean({ description: \"Include parameter schemas in search results (default: true)\" })),\n server: Type.Optional(Type.String({ description: \"Filter to specific server (also disambiguates tool calls)\" })),\n action: Type.Optional(Type.String({ description: \"Action: 'ui-messages', 'auth-start', or 'auth-complete'\" })),\n }),\n renderResult: renderMcpToolResult,\n async execute(_toolCallId, params: {\n tool?: string;\n args?: string;\n connect?: string;\n describe?: string;\n search?: string;\n regex?: boolean;\n includeSchemas?: boolean;\n server?: string;\n action?: string;\n }, signal, _onUpdate, _ctx) {\n let parsedArgs: Record<string, unknown> | undefined;\n if (params.args) {\n try {\n parsedArgs = JSON.parse(params.args);\n if (typeof parsedArgs !== \"object\" || parsedArgs === null || Array.isArray(parsedArgs)) {\n const gotType = Array.isArray(parsedArgs) ? \"array\" : parsedArgs === null ? \"null\" : typeof parsedArgs;\n throw new Error(`Invalid args: expected a JSON object, got ${gotType}`);\n }\n } catch (error) {\n if (error instanceof SyntaxError) {\n throw new Error(`Invalid args JSON: ${error.message}`, { cause: error });\n }\n throw error;\n }\n }\n\n if (!state && initPromise) {\n try {\n state = await initPromise;\n } catch (error) {\n const message = error instanceof Error ? error.message : String(error);\n return {\n content: [{ type: \"text\" as const, text: `MCP initialization failed: ${message}` }],\n details: { error: \"init_failed\", message },\n };\n }\n }\n if (!state) {\n return {\n content: [{ type: \"text\" as const, text: \"MCP not initialized\" }],\n details: { error: \"not_initialized\" },\n };\n }\n\n if (params.action === \"ui-messages\") {\n return executeUiMessages(state);\n }\n if (params.action === \"auth-start\") {\n if (!params.server) {\n return {\n content: [{ type: \"text\" as const, text: \"auth-start requires `server`. Example: mcp({ action: \\\"auth-start\\\", server: \\\"linear-server\\\" })\" }],\n details: { mode: \"auth-start\", error: \"missing_server\" },\n };\n }\n return executeAuthStart(state, params.server);\n }\n if (params.action === \"auth-complete\") {\n if (!params.server) {\n return {\n content: [{ type: \"text\" as const, text: \"auth-complete requires `server`.\" }],\n details: { mode: \"auth-complete\", error: \"missing_server\" },\n };\n }\n const input = parsedArgs?.redirectUrl ?? parsedArgs?.code ?? parsedArgs?.input;\n if (typeof input !== \"string\" || input.trim().length === 0) {\n return {\n content: [{ type: \"text\" as const, text: \"auth-complete requires args with `redirectUrl`, `code`, or `input`.\" }],\n details: { mode: \"auth-complete\", error: \"missing_input\" },\n };\n }\n return executeAuthComplete(state, params.server, input);\n }\n if (params.tool) {\n return executeCall(state, params.tool, parsedArgs, params.server, getPiTools, signal);\n }\n if (params.connect) {\n return executeConnect(state, params.connect, signal);\n }\n if (params.describe) {\n return executeDescribe(state, params.describe);\n }\n if (params.search) {\n return executeSearch(state, params.search, params.regex, params.server, params.includeSchemas);\n }\n if (params.server) {\n return executeList(state, params.server);\n }\n return executeStatus(state);\n },\n });\n }\n}\n"]}