@herbertgao/pi-extensions 2026.9.17 → 2026.9.19

package/node_modules/pi-mcp-adapter/elicitation-handler.ts

@@ -1,358 +0,0 @@
-import type { ExtensionUIContext } from "@earendil-works/pi-coding-agent";
-import type { Client } from "@modelcontextprotocol/client";
-import {
-  ProtocolError,
-  ProtocolErrorCode,
-  type ElicitRequest,
-  type ElicitRequestFormParams,
-  type ElicitRequestURLParams,
-  type ElicitResult,
-} from "@modelcontextprotocol/client";
-import { AjvJsonSchemaValidator } from "@modelcontextprotocol/client/validators/ajv";
-import type { JsonSchemaType } from "@modelcontextprotocol/client";
-import open from "open";
-import { abortable, throwIfAborted } from "./abort.ts";
-
-export type ElicitationValue = string | number | boolean | string[] | undefined;
-type FormProperty = ElicitRequestFormParams["requestedSchema"]["properties"][string];
-type FieldCollectionResult = { status: "cancelled" } | { status: "collected"; value: ElicitationValue };
-
-export type ElicitationUIContext = Pick<ExtensionUIContext, "select" | "input" | "notify">;
-
-export interface ElicitationHandlerOptions {
-  serverName: string;
-  ui: ElicitationUIContext;
-  allowUrl: boolean;
-  onUrlAccepted?: (elicitationId: string) => void;
-}
-
-export type ServerElicitationConfig = Omit<ElicitationHandlerOptions, "serverName" | "onUrlAccepted">;
-
-export function registerElicitationHandler(client: Client, options: ElicitationHandlerOptions): void {
-  client.setRequestHandler("elicitation/create", request =>
-    handleElicitationRequest(options, request));
-}
-
-export async function handleElicitationRequest(
-  options: ElicitationHandlerOptions,
-  request: ElicitRequest,
-  signal?: AbortSignal,
-): Promise<ElicitResult> {
-  throwIfAborted(signal);
-  return request.params.mode === "url"
-    ? handleUrlElicitation(options, request.params, signal)
-    : handleFormElicitation(options, request.params, signal);
-}
-
-export async function handleFormElicitation(
-  options: ElicitationHandlerOptions,
-  params: ElicitRequestFormParams,
-  signal?: AbortSignal,
-): Promise<ElicitResult> {
-  const properties = Object.entries(params.requestedSchema.properties);
-  const decision = await abortable(options.ui.select(
-    `MCP Input Request\nServer: ${options.serverName}\n\n${params.message}`,
-    ["Continue", "Decline"],
-  ), signal);
-  if (decision === undefined) return { action: "cancel" };
-  if (decision === "Decline") return { action: "decline" };
-  if (properties.length === 0) return { action: "accept", content: {} };
-
-  const values: Record<string, ElicitationValue> = {};
-  for (const [name, schema] of properties) {
-    const value = await collectValidField(options.ui, params, name, schema, undefined, signal);
-    if (value.status === "cancelled") return { action: "cancel" };
-    values[name] = value.value;
-  }
-
-  while (true) {
-    const content = coerceAndValidateFormValues(params, values);
-    const action = await abortable(options.ui.select(
-      formatReview(options.serverName, properties, content),
-      ["Submit", "Edit", "Decline"],
-    ), signal);
-    if (action === undefined) return { action: "cancel" };
-    if (action === "Decline") return { action: "decline" };
-    if (action === "Submit") return { action: "accept", content };
-
-    const labels = properties.map(([name, schema]) => `${schema.title ?? humanizeName(name)} (${name})`);
-    const selected = await abortable(options.ui.select("Choose a field to edit", labels), signal);
-    if (selected === undefined) return { action: "cancel" };
-    const property = properties[labels.indexOf(selected)];
-    if (!property) continue;
-    const [name, schema] = property;
-    const value = await collectValidField(options.ui, params, name, schema, values[name], signal);
-    if (value.status === "cancelled") return { action: "cancel" };
-    values[name] = value.value;
-  }
-}
-
-async function collectValidField(
-  ui: ElicitationUIContext,
-  params: ElicitRequestFormParams,
-  name: string,
-  schema: FormProperty,
-  current?: ElicitationValue,
-  signal?: AbortSignal,
-): Promise<FieldCollectionResult> {
-  const required = params.requestedSchema.required?.includes(name) === true;
-  while (true) {
-    const result = await collectField(ui, params, name, schema, current, signal);
-    if (result.status === "cancelled") return result;
-    try {
-      coerceAndValidateFormValues({
-        ...params,
-        requestedSchema: {
-          type: "object",
-          properties: { [name]: schema },
-          ...(required ? { required: [name] } : {}),
-        },
-      }, { [name]: result.value });
-      return result;
-    } catch (error) {
-      ui.notify(error instanceof Error ? error.message : String(error), "error");
-      current = result.value;
-    }
-  }
-}
-
-async function collectField(
-  ui: ElicitationUIContext,
-  params: ElicitRequestFormParams,
-  name: string,
-  schema: FormProperty,
-  current?: ElicitationValue,
-  signal?: AbortSignal,
-): Promise<FieldCollectionResult> {
-  const required = params.requestedSchema.required?.includes(name) === true;
-  const title = [schema.title ?? humanizeName(name), required ? "(required)" : "", schema.description]
-    .filter(Boolean)
-    .join(" ");
-
-  if (schema.type === "string" && ("enum" in schema || "oneOf" in schema)) {
-    const choices = "oneOf" in schema
-      ? schema.oneOf.map(option => ({ value: option.const, display: formatChoice(option.const, option.title) }))
-      : schema.enum.map((value, index) => ({
-          value,
-          display: formatChoice(value, "enumNames" in schema ? schema.enumNames?.[index] : undefined),
-        }));
-    const displays = uniqueLabels(choices.map(choice => choice.display));
-    const actions = [...displays];
-    const useDefault = schema.default === undefined ? undefined : uniqueAction("Use default", actions);
-    if (useDefault) actions.push(useDefault);
-    const omit = required ? undefined : uniqueAction("Omit", actions);
-    if (omit) actions.push(omit);
-    const action = await abortable(ui.select(title, actions), signal);
-    if (action === undefined) return { status: "cancelled" };
-    if (action === useDefault) return { status: "collected", value: schema.default };
-    if (action === omit) return { status: "collected", value: undefined };
-    return { status: "collected", value: choices[displays.indexOf(action)]?.value };
-  }
-
-  if (schema.type === "boolean") {
-    const actions = ["Yes", "No"];
-    if (schema.default !== undefined) actions.push("Use default");
-    if (!required) actions.push("Omit");
-    const action = await abortable(ui.select(title, actions), signal);
-    if (action === undefined) return { status: "cancelled" };
-    if (action === "Use default") return { status: "collected", value: schema.default };
-    if (action === "Omit") return { status: "collected", value: undefined };
-    return { status: "collected", value: action === "Yes" };
-  }
-
-  if (schema.type === "array") {
-    const actions = ["Choose values"];
-    if (schema.default !== undefined) actions.push("Use default");
-    if (!required) actions.push("Omit");
-    const action = await abortable(ui.select(title, actions), signal);
-    if (action === undefined) return { status: "cancelled" };
-    if (action === "Use default") return { status: "collected", value: schema.default };
-    if (action === "Omit") return { status: "collected", value: undefined };
-
-    const choices = extractMultiSelectOptions(schema);
-    const selected = new Set(Array.isArray(current) ? current : []);
-    while (true) {
-      const displays = uniqueLabels(choices.map(choice => selected.has(choice.value) ? `✓ ${choice.display}` : choice.display));
-      const done = uniqueAction("Done", displays);
-      const picked = await abortable(ui.select(title, [...displays, done]), signal);
-      if (picked === undefined) return { status: "cancelled" };
-      if (picked === done) return { status: "collected", value: [...selected] };
-      const choice = choices[displays.indexOf(picked)];
-      if (!choice) continue;
-      if (selected.has(choice.value)) selected.delete(choice.value);
-      else selected.add(choice.value);
-    }
-  }
-
-  const actions = ["Enter value"];
-  if (schema.default !== undefined) actions.push("Use default");
-  if (!required) actions.push("Omit");
-  const action = await abortable(ui.select(title, actions), signal);
-  if (action === undefined) return { status: "cancelled" };
-  if (action === "Use default") return { status: "collected", value: schema.default };
-  if (action === "Omit") return { status: "collected", value: undefined };
-  const entered = await abortable(ui.input(title, current === undefined ? undefined : String(current)), signal);
-  return entered === undefined ? { status: "cancelled" } : { status: "collected", value: entered };
-}
-
-export function coerceAndValidateFormValues(
-  params: ElicitRequestFormParams,
-  values: Record<string, ElicitationValue>,
-): Record<string, string | number | boolean | string[]> {
-  const output: Record<string, string | number | boolean | string[]> = {};
-  const required = new Set(params.requestedSchema.required ?? []);
-  for (const [name, schema] of Object.entries(params.requestedSchema.properties)) {
-    const value = values[name];
-    if (value === undefined) {
-      if (required.has(name)) throw new Error(`Missing required elicitation field: ${name}`);
-      continue;
-    }
-    if (schema.type === "string") {
-      const stringValue = String(value);
-      const limits = schema as typeof schema & { minLength?: number; maxLength?: number };
-      if (limits.minLength !== undefined && stringValue.length < limits.minLength) {
-        throw new Error(`Elicitation field ${name} is shorter than minimum length ${limits.minLength}`);
-      }
-      if (limits.maxLength !== undefined && stringValue.length > limits.maxLength) {
-        throw new Error(`Elicitation field ${name} is longer than maximum length ${limits.maxLength}`);
-      }
-      if ("enum" in schema && !schema.enum.includes(stringValue)) {
-        throw new Error(`Elicitation field ${name} is not an allowed value`);
-      }
-      if ("oneOf" in schema && !schema.oneOf.some(option => option.const === stringValue)) {
-        throw new Error(`Elicitation field ${name} is not an allowed value`);
-      }
-      output[name] = stringValue;
-      continue;
-    }
-    if (schema.type === "number" || schema.type === "integer") {
-      if (typeof value === "string" && value.trim() === "") {
-        throw new Error(`Elicitation field ${name} must be a number`);
-      }
-      const numberValue = typeof value === "number" ? value : Number(value);
-      if (!Number.isFinite(numberValue)) throw new Error(`Elicitation field ${name} must be a number`);
-      if (schema.type === "integer" && !Number.isInteger(numberValue)) {
-        throw new Error(`Elicitation field ${name} must be an integer`);
-      }
-      if (schema.minimum !== undefined && numberValue < schema.minimum) {
-        throw new Error(`Elicitation field ${name} is below minimum ${schema.minimum}`);
-      }
-      if (schema.maximum !== undefined && numberValue > schema.maximum) {
-        throw new Error(`Elicitation field ${name} is above maximum ${schema.maximum}`);
-      }
-      output[name] = numberValue;
-      continue;
-    }
-    if (schema.type === "boolean") {
-      output[name] = typeof value === "boolean" ? value : value === "true";
-      continue;
-    }
-    if (schema.type === "array") {
-      if (!Array.isArray(value)) throw new Error(`Elicitation field ${name} must be a list`);
-      const allowed = new Set(extractMultiSelectOptions(schema).map(option => option.value));
-      const arrayValue = value.map(String);
-      if (schema.minItems !== undefined && arrayValue.length < schema.minItems) {
-        throw new Error(`Elicitation field ${name} has fewer than ${schema.minItems} selections`);
-      }
-      if (schema.maxItems !== undefined && arrayValue.length > schema.maxItems) {
-        throw new Error(`Elicitation field ${name} has more than ${schema.maxItems} selections`);
-      }
-      if (arrayValue.some(item => !allowed.has(item))) {
-        throw new Error(`Elicitation field ${name} contains an invalid selection`);
-      }
-      output[name] = arrayValue;
-    }
-  }
-  const validation = new AjvJsonSchemaValidator()
-    .getValidator(params.requestedSchema as JsonSchemaType)(output);
-  if (!validation.valid) {
-    throw new Error(`Invalid elicitation response: ${validation.errorMessage}`);
-  }
-  return output;
-}
-
-function formatChoice(value: string, title?: string): string {
-  return title && title !== value ? `${title} (${value})` : value;
-}
-
-function uniqueLabels(labels: string[]): string[] {
-  const used = new Set<string>();
-  return labels.map(label => {
-    let unique = label;
-    while (used.has(unique)) unique += "…";
-    used.add(unique);
-    return unique;
-  });
-}
-
-function uniqueAction(label: string, choices: string[]): string {
-  let unique = label;
-  while (choices.includes(unique)) unique += "…";
-  return unique;
-}
-
-function extractMultiSelectOptions(schema: Extract<FormProperty, { type: "array" }>): Array<{ value: string; display: string }> {
-  const items = schema.items as { enum?: string[]; anyOf?: Array<{ const: string; title: string }> };
-  return items.anyOf
-    ? items.anyOf.map(option => ({ value: option.const, display: formatChoice(option.const, option.title) }))
-    : (items.enum ?? []).map(value => ({ value, display: value }));
-}
-
-function formatReview(
-  serverName: string,
-  properties: Array<[string, FormProperty]>,
-  content: Record<string, string | number | boolean | string[]>,
-): string {
-  const rows = properties.map(([name, schema]) =>
-    `${schema.title ?? humanizeName(name)}: ${content[name] === undefined ? "(omitted)" : String(content[name])}`);
-  return [`Review input for ${serverName}`, "", ...rows].join("\n");
-}
-
-export async function handleUrlElicitation(
-  options: ElicitationHandlerOptions,
-  params: ElicitRequestURLParams,
-  signal?: AbortSignal,
-): Promise<ElicitResult> {
-  throwIfAborted(signal);
-  if (!options.allowUrl) throw new ProtocolError(ProtocolErrorCode.InvalidParams, "URL elicitation is not supported");
-
-  let parsed: URL;
-  try {
-    parsed = new URL(params.url);
-  } catch {
-    throw new ProtocolError(ProtocolErrorCode.InvalidParams, "URL elicitation supplied an invalid URL");
-  }
-  if (parsed.protocol !== "http:" && parsed.protocol !== "https:") {
-    throw new ProtocolError(ProtocolErrorCode.InvalidParams, "URL elicitation only supports HTTP and HTTPS URLs");
-  }
-
-  const decision = await abortable(options.ui.select([
-    "MCP Browser Request",
-    `Server: ${options.serverName}`,
-    "",
-    params.message,
-    "",
-    `Host: ${parsed.host}`,
-    `Full URL: ${params.url}`,
-    "",
-    "Open this URL in your browser?",
-  ].join("\n"), ["Open", "Decline"]), signal);
-  if (decision === undefined) return { action: "cancel" };
-  if (decision === "Decline") return { action: "decline" };
-
-  try {
-    await abortable(open(params.url), signal);
-  } catch (error) {
-    throwIfAborted(signal);
-    options.ui.notify(`Could not open MCP elicitation URL: ${error instanceof Error ? error.message : String(error)}`, "error");
-    return { action: "cancel" };
-  }
-  throwIfAborted(signal);
-  options.onUrlAccepted?.(params.elicitationId);
-  options.ui.notify("Opened browser for MCP elicitation.", "info");
-  return { action: "accept" };
-}
-
-function humanizeName(name: string): string {
-  return name.replace(/[_-]+/g, " ").replace(/([a-z0-9])([A-Z])/g, "$1 $2").replace(/^./, char => char.toUpperCase());
-}

package/node_modules/pi-mcp-adapter/error-signal.ts

@@ -1,32 +0,0 @@
-import { INPUT_REQUIRED_NEEDS_UI } from "./errors.ts";
-
-/**
- * Decide the `isError` override for a finished tool result in the `tool_result` hook.
- *
- * A failed MCP tool call is *returned* (not thrown), tagged `details.error: "tool_error"` (the server
- * returned an error result), `"call_failed"` (the call itself threw and was caught), or a typed
- * input-required failure that could not run without UI. pi never reads a result-level `isError`,
- * so without this such a call is recorded as a success. Returning
- * `{ isError: true }` (and nothing else) flips the flag; pi's field-by-field merge keeps the original
- * `content` and `details` intact.
- *
- * Limited to execution-failure codes: the adapter's other `details.error` values (`auth_required`, connection
- * states, search/validation feedback, ...) are not failed tool calls, so they get no override.
- */
-export function toolErrorOverride(details: unknown): { isError: true } | undefined {
-  if (details && typeof details === "object" && "error" in details) {
-    const code = (details as { error?: unknown }).error;
-    if (code === "tool_error" || code === "call_failed" || code === INPUT_REQUIRED_NEEDS_UI) {
-      return { isError: true };
-    }
-  }
-  if (details && typeof details === "object" && (details as { mode?: unknown }).mode === "script") {
-    const calls = (details as { calls?: unknown }).calls;
-    if (Array.isArray(calls) && calls.some((call) => (
-      !!call && typeof call === "object" && (call as { error?: unknown }).error === INPUT_REQUIRED_NEEDS_UI
-    ))) {
-      return { isError: true };
-    }
-  }
-  return undefined;
-}