@fitchmultz/pi-mcp-adapter 6.0.0 → 7.1.0

package/mcp-code.ts

@@ -1,412 +0,0 @@
-import type { ToolInfo } from "@earendil-works/pi-coding-agent";
-import { formatWithOptions } from "node:util";
-import { Worker } from "node:worker_threads";
-import { guardMcpOutput, guardedMcpDetails, resolveMcpOutputGuardOptions, readMcpResult, formatMcpResultReference, formatMcpPayloadFile, type McpPayloadFile, type ReadMcpResultInput } from "./mcp-output-guard.ts";
-import { executeCall, executeConnect } from "./proxy-modes.ts";
-import { executeResourceList, executeResourceRead } from "./resource-tools.ts";
-import { combineAbortSignals } from "./runtime-owner.ts";
-import { paginate, rankSuggestions, rankToolMatches } from "./search-ranking.ts";
-import type { McpExtensionState } from "./state.ts";
-import { catalogCoverage, toToolDescriptor } from "./tool-metadata.ts";
-import { isServerDisabled, type ContentBlock, type McpOperationContext } from "./types.ts";
-
-export const DEFAULT_MCP_SCRIPT_TIMEOUT_MS = 30_000;
-
-class McpScriptTimeoutError extends Error {
-  constructor(timeoutMs: number) {
-    super(`mcp_script timed out after ${timeoutMs}ms`);
-    this.name = "McpScriptTimeoutError";
-  }
-}
-
-type SearchInput = { query?: unknown; server?: unknown; limit?: unknown; offset?: unknown };
-type DescribeInput = { path?: unknown; server?: unknown };
-type WorkerMessage =
-  | { type: "emit"; block: unknown }
-  | { type: "call"; id: number; path: string; args?: unknown; server?: string }
-  | { type: "search"; id: number; input?: unknown }
-  | { type: "describe" | "resources" | "readResource" | "readResult"; id: number; input?: unknown }
-  | { type: "done"; returnBlock?: unknown }
-  | { type: "error"; message: string };
-
-type WorkerResultMessage = { type: "result"; id: number; envelope: unknown };
-
-function needsInspectableFormatting(value: unknown, stack = new WeakSet<object>()): boolean {
-  if (value === undefined || typeof value === "bigint" || typeof value === "function" || typeof value === "symbol") return true;
-  if (typeof value !== "object" || value === null) return false;
-  if (stack.has(value)) return true;
-  if (value instanceof Map || value instanceof Set || value instanceof WeakMap || value instanceof WeakSet) return true;
-  stack.add(value);
-  try {
-    return Object.values(value).some((entry) => needsInspectableFormatting(entry, stack));
-  } finally {
-    stack.delete(value);
-  }
-}
-
-function formatValue(value: unknown): string {
-  if (typeof value === "string") return value;
-  try {
-    if (!needsInspectableFormatting(value)) {
-      const json = JSON.stringify(value, null, 2);
-      if (json !== undefined) return json;
-    }
-    return formatWithOptions({ colors: false, depth: 6 }, value);
-  } catch {
-    return "[unserializable value]";
-  }
-}
-
-function toContentBlock(value: unknown): ContentBlock {
-  if (typeof value === "object" && value !== null) {
-    const block = value as Record<string, unknown>;
-    if (block.type === "text" && typeof block.text === "string") {
-      return { type: "text", text: block.text };
-    }
-    if (block.type === "image" && typeof block.data === "string" && typeof block.mimeType === "string") {
-      return { type: "image", data: block.data, mimeType: block.mimeType };
-    }
-  }
-  return { type: "text", text: formatValue(value) };
-}
-
-function textFromContent(content: ContentBlock[]): string {
-  return content
-    .filter((block) => block.type === "text")
-    .map((block) => block.text)
-    .join("\n");
-}
-
-function abortReasonError(reason: unknown): Error {
-  return reason instanceof Error ? reason : new Error(String(reason ?? "MCP request aborted"));
-}
-
-function parseWorkerMessage(value: unknown): WorkerMessage | null {
-  if (typeof value !== "object" || value === null) return null;
-  const message = value as Record<string, unknown>;
-  if (message.type === "emit" && "block" in message) return { type: "emit", block: message.block };
-  if (message.type === "call" && typeof message.id === "number" && typeof message.path === "string") {
-    return { type: "call", id: message.id, path: message.path, ...(message.args !== undefined ? { args: message.args } : {}), ...(typeof message.server === "string" ? { server: message.server } : {}) };
-  }
-  if ((message.type === "search" || message.type === "describe" || message.type === "resources" || message.type === "readResource" || message.type === "readResult") && typeof message.id === "number") {
-    return "input" in message
-      ? { type: message.type, id: message.id, input: message.input }
-      : { type: message.type, id: message.id };
-  }
-  if (message.type === "done") {
-    return "returnBlock" in message ? { type: "done", returnBlock: message.returnBlock } : { type: "done" };
-  }
-  if (message.type === "error" && typeof message.message === "string") {
-    return { type: "error", message: message.message };
-  }
-  return null;
-}
-
-class McpScriptCaptureError extends Error {
-  constructor(readonly recovery: unknown, message: string) {
-    super(message);
-  }
-}
-
-export const runMcpScript: typeof runMcpScriptOperation = async (...args) => {
-  const state = args[0];
-  state.owner?.beforeActivity();
-  state.activeScripts = (state.activeScripts ?? 0) + 1;
-  try { return await runMcpScriptOperation(...args); }
-  finally { state.activeScripts--; }
-};
-
-async function runMcpScriptOperation(
-  state: McpExtensionState,
-  code: string,
-  timeoutMs: number | null = DEFAULT_MCP_SCRIPT_TIMEOUT_MS,
-  getPiTools?: () => ToolInfo[],
-  signal?: AbortSignal,
-  toolCallId?: string,
-  beforeDispatch?: (signal: AbortSignal | undefined, operation: McpOperationContext) => Promise<void>,
-) {
-  const resolvedTimeoutMs = timeoutMs === null ? null : Number.isFinite(timeoutMs) && timeoutMs > 0
-    ? Math.floor(timeoutMs)
-    : DEFAULT_MCP_SCRIPT_TIMEOUT_MS;
-  const output: ContentBlock[] = [];
-  const externalSignal = combineAbortSignals(state.owner?.signal, signal);
-  const timeoutController = new AbortController();
-  const callSignal = combineAbortSignals(externalSignal, timeoutController.signal);
-
-  type ScriptOperation =
-    | { operation: "call"; path: string; ok: true; durationMs: number }
-    | { operation: "call"; path: string; ok: false; error: string; durationMs: number; recovery?: unknown }
-    | { operation: "search"; query: string; ok: true; durationMs: number }
-    | { operation: "search"; query: string; ok: false; error: string; durationMs: number }
-    | { operation: "describe"; path: string; ok: true; durationMs: number }
-    | { operation: "describe"; path: string; ok: false; error: string; durationMs: number };
-  type TrackedScriptOperation = ScriptOperation & { startedAt: number };
-  const calls: TrackedScriptOperation[] = [];
-  const snapshotCalls = (): ScriptOperation[] => calls.map(({ startedAt, ...operation }) => ({
-    ...operation,
-    durationMs: "error" in operation && operation.error === "incomplete"
-      ? Math.max(0, Date.now() - startedAt)
-      : operation.durationMs,
-  }));
-  let callsSnapshot: ScriptOperation[] | undefined;
-  const resultRefs = new Set<string>();
-  const payloadNotices = new Set<string>();
-  const recordCall = async (path: string, dispatch: () => ReturnType<typeof executeCall>) => {
-    // Record before dispatch so calls still in flight at timeout/abort appear in the trace.
-    const startedAt = Date.now();
-    const index = calls.push({ operation: "call", path, ok: false, error: "incomplete", durationMs: 0, startedAt }) - 1;
-    const result = await dispatch();
-    const details = result.details;
-    if (typeof details.resultRef === "string") resultRefs.add(details.resultRef);
-    if (typeof details.resultWriteError === "string") payloadNotices.add(`[MCP result could not be saved: ${details.resultWriteError}]`);
-    for (const file of (details.payloadFiles ?? []) as McpPayloadFile[]) payloadNotices.add(formatMcpPayloadFile(file));
-    if (details.error !== undefined) {
-      const errorCode = String(details.error);
-      const suggestions = Array.isArray(details.suggestions)
-        ? details.suggestions.filter((suggestion): suggestion is string => typeof suggestion === "string")
-        : [];
-      const message = errorCode === "tool_not_found"
-        ? `Tool "${path}" not found. Use await tools.search({ query: "..." }) inside mcp_script.${suggestions.length > 0 ? ` Did you mean: ${suggestions.join(", ")}` : ""}`
-        : typeof details.message === "string"
-          ? details.message
-          : textFromContent(result.content);
-      calls[index] = {
-        operation: "call", path, ok: false, error: errorCode, durationMs: Date.now() - startedAt, startedAt,
-        ...(errorCode === "ambiguous_outcome" ? { recovery: details.recovery } : {}),
-      };
-      if (errorCode === "call_capture_failed") throw new McpScriptCaptureError(details.recovery, message);
-      if (errorCode === "ambiguous_outcome") output.push({ type: "text", text: message });
-      return {
-        ok: false as const,
-        ...(details.mcpResult !== undefined ? { data: details.mcpResult } : {}),
-        ...(details.resultRef ? { resultRef: details.resultRef } : {}),
-        error: { code: errorCode, message, ...(details.recovery ? { recovery: details.recovery } : {}) },
-      };
-    }
-    calls[index] = { operation: "call", path, ok: true, durationMs: Date.now() - startedAt, startedAt };
-    return {
-      ok: true as const,
-      data: details.mcpResult !== undefined ? details.mcpResult : textFromContent(result.content),
-      ...(details.resultRef ? { resultRef: details.resultRef } : {}),
-    };
-  };
-
-  const callIdentity = (innerCallId: number) => ({ ...(toolCallId !== undefined ? { toolCallId } : {}), innerCallId });
-  const callTool = (innerCallId: number, path: string, args?: Record<string, unknown>, server?: string) => recordCall(path,
-    () => executeCall(state, path, args, server, getPiTools, callSignal, callIdentity(innerCallId), beforeDispatch, { raw: true }));
-
-  const discoverServer = async (server?: string) => {
-    if (!server) return undefined;
-    if (isServerDisabled(state.config.mcpServers[server])) return { code: "server_disabled", message: `Server "${server}" is disabled.` };
-    if (state.config.mcpServers[server] && state.toolMetadata.has(server)) return undefined;
-    const result = await executeConnect(state, server, callSignal);
-    return result.details.error ? { code: result.details.error, message: textFromContent(result.content) } : undefined;
-  };
-  const searchTools = async (input?: SearchInput) => {
-    const startedAt = Date.now();
-    const query = typeof input?.query === "string" ? input.query : "";
-    let error: unknown;
-    try {
-      const server = typeof input?.server === "string" ? input.server : undefined;
-      const discoveryError = await discoverServer(server);
-      const coverage = catalogCoverage(state, server);
-      if (discoveryError) { error = discoveryError.code; return { error: discoveryError, coverage }; }
-      if (query.trim() === "") return { items: [], total: 0, hasMore: false, nextOffset: null, coverage };
-      const limit = typeof input?.limit === "number" ? input.limit : 12;
-      const offset = typeof input?.offset === "number" ? input.offset : 0;
-      const page = paginate(rankToolMatches(state, query, server).filter(match => !match.tool.resourceUri), offset, limit);
-      return {
-        ...page, coverage,
-        items: page.items.map(({ server: matchServer, tool, score }) => ({
-          path: tool.name,
-          name: tool.originalName,
-          server: matchServer,
-          ...(tool.description ? { description: tool.description } : {}),
-          score,
-        })),
-      };
-    } catch (caught) {
-      error = caught;
-      throw caught;
-    } finally {
-      calls.push(error === undefined
-        ? { operation: "search", query, ok: true, durationMs: Date.now() - startedAt, startedAt }
-        : { operation: "search", query, ok: false, error: error instanceof Error ? error.message : String(error), durationMs: Date.now() - startedAt, startedAt });
-    }
-  };
-
-  const describeTool = async (input?: DescribeInput) => {
-    const startedAt = Date.now();
-    const path = typeof input?.path === "string" ? input.path : "";
-    let error: unknown;
-    try {
-      const selectedServer = typeof input?.server === "string" ? input.server : undefined;
-      const discoveryError = await discoverServer(selectedServer);
-      if (discoveryError) { error = discoveryError.code; return { path, error: discoveryError }; }
-      const matches = [...state.toolMetadata].flatMap(([server, metadata]) => {
-        if ((selectedServer && server !== selectedServer) || !state.config.mcpServers[server] || isServerDisabled(state.config.mcpServers[server])) return [];
-        const tools = metadata.filter(tool => !tool.resourceUri);
-        const originals = selectedServer ? tools.filter(tool => tool.originalName === path) : [];
-        return (originals.length ? originals : tools.filter(tool => tool.name === path)).map(tool => toToolDescriptor(server, tool));
-      });
-      if (matches.length === 1) return matches[0];
-      if (matches.length > 1) { error = "ambiguous_tool"; return { path, error: { code: "ambiguous_tool", message: "Specify server and exact tool name." } }; }
-      const suggestions = path ? rankSuggestions(state, path, 5, selectedServer) : [];
-      error = "tool_not_found";
-      return {
-        path,
-        error: {
-          code: "tool_not_found",
-          message: `Tool not found: ${path}`,
-          suggestions,
-        },
-      };
-    } catch (caught) {
-      error = caught;
-      throw caught;
-    } finally {
-      calls.push(error === undefined
-        ? { operation: "describe", path, ok: true, durationMs: Date.now() - startedAt, startedAt }
-        : { operation: "describe", path, ok: false, error: error instanceof Error ? error.message : String(error), durationMs: Date.now() - startedAt, startedAt });
-    }
-  };
-
-  let worker: Worker | undefined;
-  let timer: ReturnType<typeof setTimeout> | undefined;
-  let removeAbortListener = () => {};
-  let errorCode: "timeout" | "aborted" | "script_error" | "call_capture_failed" | undefined;
-  let recovery: unknown;
-  let errorMessage: string | undefined;
-
-  try {
-    if (externalSignal?.aborted) {
-      throw abortReasonError(externalSignal.reason);
-    }
-
-    worker = new Worker(new URL("./mcp-script-worker.mjs", import.meta.url), {
-      workerData: { code },
-      env: {},
-    });
-    const activeWorker = worker;
-    const execution = new Promise<void>((resolve, reject) => {
-      let completed = false;
-      activeWorker.on("message", (value: unknown) => {
-        const message = parseWorkerMessage(value);
-        if (!message || completed) return;
-        if (message.type === "emit") {
-          output.push(toContentBlock(message.block));
-          return;
-        }
-        if (message.type === "done") {
-          completed = true;
-          if ("returnBlock" in message) output.push(toContentBlock(message.returnBlock));
-          resolve();
-          return;
-        }
-        if (message.type === "error") {
-          completed = true;
-          reject(new Error(message.message));
-          return;
-        }
-
-        void (async () => {
-          let envelope: unknown;
-          if (message.type === "call") {
-            envelope = await callTool(message.id, message.path, message.args as Record<string, unknown> | undefined, message.server);
-          } else if (message.type === "search") {
-            envelope = await searchTools(message.input as SearchInput | undefined);
-          } else if (message.type === "describe") {
-            envelope = await describeTool(message.input as DescribeInput | undefined);
-          } else if (message.type === "readResult") {
-            envelope = await readMcpResult(message.input as ReadMcpResultInput, resolveMcpOutputGuardOptions(state.config.settings, state.outputDirectory));
-          } else {
-            const input = message.input as { server?: unknown; uri?: unknown; limit?: unknown; offset?: unknown } | undefined;
-            if (typeof input?.server !== "string" || !input.server || (message.type === "readResource" && (typeof input.uri !== "string" || !input.uri))) {
-              envelope = { error: { code: "invalid_arguments", message: "Specify server and, for reads, resource uri." } };
-            } else if (message.type === "resources") {
-              const result = await executeResourceList(state, input.server, typeof input.limit === "number" ? input.limit : 12, typeof input.offset === "number" ? input.offset : 0, callSignal);
-              envelope = result.details.error ? { error: { code: result.details.error, message: textFromContent(result.content) } } : result.details;
-            } else {
-              const server = input.server;
-              const uri = input.uri as string;
-              envelope = await recordCall(`${server}:${uri}`, () => executeResourceRead(state, server, uri, callSignal, callIdentity(message.id), beforeDispatch, true));
-            }
-          }
-          const response: WorkerResultMessage = { type: "result", id: message.id, envelope };
-          activeWorker.postMessage(response);
-        })().catch(reject);
-      });
-      activeWorker.once("error", reject);
-      activeWorker.once("exit", (code) => {
-        if (!completed && code !== 0) reject(new Error(`mcp_script worker exited with code ${code}`));
-      });
-    });
-    const timeout = new Promise<never>((_resolve, reject) => {
-      if (resolvedTimeoutMs === null) return;
-      const timeoutError = new McpScriptTimeoutError(resolvedTimeoutMs);
-      timer = setTimeout(() => {
-        callsSnapshot = snapshotCalls();
-        timeoutController.abort(timeoutError);
-        void activeWorker.terminate();
-        reject(timeoutError);
-      }, resolvedTimeoutMs);
-    });
-    const aborted = externalSignal
-      ? new Promise<never>((_resolve, reject) => {
-          const onAbort = () => {
-            callsSnapshot = snapshotCalls();
-            void activeWorker.terminate();
-            reject(abortReasonError(externalSignal.reason));
-          };
-          externalSignal.addEventListener("abort", onAbort, { once: true });
-          removeAbortListener = () => externalSignal.removeEventListener("abort", onAbort);
-        })
-      : new Promise<never>(() => {});
-
-    await Promise.race([execution, timeout, aborted]);
-  } catch (error) {
-    if (error instanceof McpScriptCaptureError) {
-      errorCode = "call_capture_failed";
-      errorMessage = error.message;
-      recovery = error.recovery;
-    } else if (error instanceof McpScriptTimeoutError) {
-      errorCode = "timeout";
-      errorMessage = `mcp_script timed out after ${resolvedTimeoutMs}ms`;
-    } else if (externalSignal?.aborted) {
-      errorCode = "aborted";
-      errorMessage = error instanceof Error ? error.message : String(error);
-    } else {
-      errorCode = "script_error";
-      errorMessage = error instanceof Error ? error.message : String(error);
-    }
-    output.push({ type: "text", text: errorMessage });
-  } finally {
-    clearTimeout(timer);
-    removeAbortListener();
-    // "incomplete" means the call had not settled when the script finished
-    // (deadline, abort, or early return). Snapshot before aborting stragglers.
-    callsSnapshot ??= snapshotCalls();
-    // A script may finish without awaiting every call; abort leftovers so
-    // parent-side dispatches do not outlive the script.
-    timeoutController.abort(new Error("mcp_script finished"));
-    await worker?.terminate();
-  }
-
-  // Snapshot before the asynchronous output guard; the terminated worker can no longer emit.
-  const guarded = await guardMcpOutput(
-    [...(output.length > 0 ? output : [{ type: "text" as const, text: "(no output)" }]), ...[...payloadNotices].map(text => ({ type: "text" as const, text })), ...[...resultRefs].map(ref => ({ type: "text" as const, text: formatMcpResultReference(ref) }))],
-    resolveMcpOutputGuardOptions(state.config.settings, state.outputDirectory),
-  );
-  return {
-    content: guarded.content,
-    details: {
-      mode: "script",
-      ...(resultRefs.size ? { resultRefs: [...resultRefs] } : {}),
-      ...(errorCode ? { error: errorCode, message: errorMessage } : {}),
-      timeoutMs: resolvedTimeoutMs,
-      ...(recovery !== undefined ? { recovery } : {}),
-      ...(callsSnapshot.length > 0 ? { calls: callsSnapshot } : {}),
-      ...guardedMcpDetails(guarded),
-    },
-  };
-}

package/mcp-keyring-helper.cjs

@@ -1,89 +0,0 @@
-#!/usr/bin/env node
-'use strict';
-
-const { createRequire } = require('node:module');
-const { dirname, join } = require('node:path');
-
-const requireFromHere = createRequire(__filename);
-
-function loadKeyringEntryClass() {
-  try {
-    return requireFromHere('@napi-rs/keyring').Entry;
-  } catch (loaderError) {
-    const suffixes = getNativeBindingSuffixes(process.platform, process.arch);
-    let lastError;
-    for (const suffix of suffixes) {
-      try {
-        const packageJsonPath = requireFromHere.resolve(`@napi-rs/keyring-${suffix}/package.json`);
-        return requireFromHere(join(dirname(packageJsonPath), `keyring.${suffix}.node`)).Entry;
-      } catch (error) {
-        lastError = error;
-      }
-    }
-    const error = new Error(`Failed to load @napi-rs/keyring in recovery helper: ${lastError?.message ?? loaderError.message}`);
-    error.cause = loaderError;
-    throw error;
-  }
-}
-
-function getNativeBindingSuffixes(platform, arch) {
-  if (platform === 'linux') {
-    if (arch === 'arm64') return ['linux-arm64-gnu', 'linux-arm64-musl'];
-    if (arch === 'arm') return ['linux-arm-gnueabihf'];
-    if (arch === 'riscv64') return ['linux-riscv64-gnu'];
-    if (arch === 'x64') return ['linux-x64-gnu', 'linux-x64-musl'];
-  }
-  return [];
-}
-
-function readStdin() {
-  return new Promise((resolve, reject) => {
-    let input = '';
-    process.stdin.setEncoding('utf8');
-    process.stdin.on('data', chunk => {
-      input += chunk;
-      if (input.length > 1024 * 1024) {
-        reject(new Error('request too large'));
-        process.stdin.destroy();
-      }
-    });
-    process.stdin.on('error', reject);
-    process.stdin.on('end', () => resolve(input));
-  });
-}
-
-function writeResponse(response) {
-  process.stdout.write(`${JSON.stringify(response)}\n`);
-}
-
-(async () => {
-  try {
-    const request = JSON.parse(await readStdin());
-    if (!request || typeof request !== 'object') throw new Error('invalid request');
-    const { operation, service, account, payload } = request;
-    if (!['read', 'write', 'remove'].includes(operation)) throw new Error('invalid operation');
-    if (typeof service !== 'string' || !service) throw new Error('invalid service');
-    if (typeof account !== 'string' || !account) throw new Error('invalid account');
-
-    const Entry = loadKeyringEntryClass();
-    const entry = new Entry(service, account);
-
-    if (operation === 'read') {
-      const value = entry.getPassword();
-      writeResponse(value === null ? { ok: true, found: false } : { ok: true, found: true, value });
-      return;
-    }
-    if (operation === 'write') {
-      if (typeof payload !== 'string') throw new Error('invalid payload');
-      entry.setPassword(payload);
-      writeResponse({ ok: true });
-      return;
-    }
-
-    entry.deleteCredential();
-    writeResponse({ ok: true });
-  } catch (error) {
-    writeResponse({ ok: false, error: error instanceof Error ? error.message : String(error) });
-    process.exitCode = 1;
-  }
-})();