@herbertgao/pi-extensions 2026.9.17 → 2026.9.19

package/node_modules/pi-mcp-adapter/tool-result-renderer.ts

@@ -1,573 +0,0 @@
-import type { AgentToolResult, ToolRenderResultOptions } from "@earendil-works/pi-coding-agent";
-import { type Component, Text, truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
-
-type McpToolResultDetails = Record<string, unknown> & { error?: unknown };
-type McpToolContentBlock = AgentToolResult<McpToolResultDetails>["content"][number];
-
-export interface RenderTheme {
-  fg: (name: string, text: string) => string;
-  bold?: (text: string) => string;
-}
-
-const plainTheme: RenderTheme = { fg: (_name, text) => text };
-
-export interface McpProxyToolCallInput {
-  tool?: string;
-  args?: string | Record<string, unknown>;
-  connect?: string;
-  describe?: string;
-  search?: string;
-  regex?: boolean;
-  includeSchemas?: boolean;
-  server?: string;
-  action?: string;
-}
-
-interface McpToolRenderState {
-  compactTitle?: string;
-  compactInputPreview?: string;
-}
-
-export interface McpToolRenderContext {
-  isError: boolean;
-  isPartial?: boolean;
-  expanded?: boolean;
-  state?: McpToolRenderState;
-}
-
-export type McpToolResultRendering = "compact" | "boxed";
-
-export interface McpToolRenderOptions {
-  resultRendering: McpToolResultRendering;
-  collapsedResultLines: 1 | 2 | 3;
-}
-
-export interface McpToolRenderSettings {
-  toolResultRendering?: unknown;
-  collapsedResultLines?: unknown;
-}
-
-export interface McpToolResultDisplay {
-  lines: string[];
-  truncated: boolean;
-}
-
-const DEFAULT_MAX_CALL_INPUT_CHARS = 1500;
-const DEFAULT_MAX_COMPACT_INPUT_CHARS = 240;
-const DEFAULT_BOXED_COLLAPSED_LINES = 3;
-const DEFAULT_COMPACT_COLLAPSED_LINES = 1;
-const DEFAULT_MAX_COLLAPSED_CHARS = 8000;
-const COLLAPSED_RENDER_CHAR_SLACK = 8;
-
-class EmptyComponent implements Component {
-  render(): string[] {
-    return [];
-  }
-
-  invalidate(): void {}
-}
-
-class CompactMcpToolResult implements Component {
-  private rendered: { width: number; lines: string[] } | null = null;
-
-  constructor(
-    private readonly title: string,
-    private readonly inputPreview: string,
-    private readonly display: McpToolResultDisplay,
-    private readonly theme: RenderTheme,
-    /** Shown after the title and never truncated; the title shrinks instead (e.g. `✗3`). */
-    private readonly status = "",
-  ) {}
-
-  render(width: number): string[] {
-    const safeWidth = Math.max(1, Math.floor(width));
-    if (this.rendered?.width === safeWidth) return this.rendered.lines;
-
-    const resultLines = this.display.lines.filter((line, index, lines) => {
-      return !(this.display.truncated && index === lines.length - 1 && line === "…");
-    });
-    const lines = resultLines.length > 0 ? resultLines : [""];
-    const bodies = lines.map((line, index) => {
-      const prefix = index === 0 ? this.renderPrefix(safeWidth) : "";
-      return `${prefix}${this.theme.fg("toolOutput", line)}`;
-    });
-    const hiddenText = this.display.truncated || bodies.some((body) => visibleWidth(body) > safeWidth);
-    const rendered = bodies.map((body, index) => {
-      const suffix = hiddenText && index === bodies.length - 1 ? " … (Ctrl+O to expand)" : "";
-      if (!suffix) return truncateToWidth(body, safeWidth, "…");
-      if (safeWidth >= suffix.length + 20) {
-        return `${truncateToWidth(body, safeWidth - suffix.length, "…")}${this.theme.fg("muted", suffix)}`;
-      }
-      const shortSuffix = " (Ctrl+O)";
-      if (safeWidth >= shortSuffix.length + 5) {
-        return `${truncateToWidth(body, safeWidth - shortSuffix.length, "…")}${this.theme.fg("muted", shortSuffix)}`;
-      }
-      return truncateToWidth(this.theme.fg("muted", shortSuffix.trim()), safeWidth, "…");
-    });
-    this.rendered = { width: safeWidth, lines: rendered };
-    return rendered;
-  }
-
-  invalidate(): void {
-    this.rendered = null;
-  }
-
-  private renderPrefix(width: number): string {
-    if (!this.title) return "";
-    const arrow = " → ";
-    const statusWidth = this.status ? visibleWidth(this.status) + 1 : 0;
-    const head = (maxWidth?: number) => {
-      const title = maxWidth === undefined
-        ? this.title
-        : truncateToWidth(this.title, Math.max(1, maxWidth - statusWidth), "…");
-      const status = this.status ? ` ${this.theme.fg("error", this.status)}` : "";
-      return `${this.theme.fg("toolTitle", title)}${status}`;
-    };
-    if (!this.inputPreview) return `${head()}${arrow}`;
-
-    const maxPrefixWidth = Math.max(12, Math.floor(width * 0.55));
-    const titleWidth = visibleWidth(this.title) + statusWidth;
-    const inputWidth = Math.max(0, maxPrefixWidth - titleWidth - 1);
-    if (inputWidth <= 3) return `${head(maxPrefixWidth)}${arrow}`;
-
-    const input = truncateToWidth(this.inputPreview, inputWidth, "…");
-    return `${head()} ${this.theme.fg("muted", input)}${arrow}`;
-  }
-}
-
-class CollapsibleText implements Component {
-  private readonly fullText: Text;
-  private readonly footerText: Text;
-  private collapsedText: { charBudget: number; fullyIncluded: boolean; text: Text } | null = null;
-  private collapsedRender: { width: number; charBudget: number; lines: string[] } | null = null;
-
-  constructor(
-    private readonly text: string,
-    private readonly expanded: boolean,
-    private readonly maxCollapsedLines: number,
-    ellipsis: string,
-    expandHint: string,
-    private readonly preTruncated = false,
-  ) {
-    this.fullText = new Text(text, 0, 0);
-    this.footerText = new Text(`${ellipsis}\n${expandHint}`, 0, 0);
-  }
-
-  render(width: number): string[] {
-    if (this.expanded) {
-      return this.fullText.render(width);
-    }
-
-    const safeWidth = Math.max(1, Math.floor(width));
-    const charBudget = safeWidth * (this.maxCollapsedLines + 1) * COLLAPSED_RENDER_CHAR_SLACK;
-    if (!this.collapsedText || this.collapsedText.charBudget !== charBudget) {
-      const prefix = this.text.length > charBudget
-        ? this.text.slice(0, charBudget)
-        : this.text;
-      this.collapsedText = {
-        charBudget,
-        fullyIncluded: prefix === this.text,
-        text: new Text(prefix, 0, 0),
-      };
-      this.collapsedRender = null;
-    }
-
-    const lines = this.collapsedText.text.render(width);
-    if (!this.preTruncated && this.collapsedText.fullyIncluded && lines.length <= this.maxCollapsedLines) return lines;
-    if (this.collapsedRender?.width === width && this.collapsedRender.charBudget === charBudget) {
-      return this.collapsedRender.lines;
-    }
-
-    const rendered = [
-      ...lines.slice(0, this.maxCollapsedLines),
-      ...this.footerText.render(width),
-    ];
-    this.collapsedRender = { width, charBudget, lines: rendered };
-    return rendered;
-  }
-
-  invalidate(): void {
-    this.fullText.invalidate();
-    this.footerText.invalidate();
-    this.collapsedText?.text.invalidate();
-    this.collapsedRender = null;
-  }
-}
-
-function truncateText(value: string, maxChars: number): string {
-  if (value.length <= maxChars) return value;
-  return `${value.slice(0, Math.max(0, maxChars - 1))}…`;
-}
-
-function formatJsonish(value: unknown, maxChars: number): string {
-  if (typeof value === "string") {
-    try {
-      return truncateText(JSON.stringify(JSON.parse(value), null, 2), maxChars);
-    } catch {
-      return truncateText(value, maxChars);
-    }
-  }
-
-  try {
-    return truncateText(JSON.stringify(value, null, 2), maxChars);
-  } catch {
-    return truncateText(String(value), maxChars);
-  }
-}
-
-function hasUsefulObjectContent(value: unknown): boolean {
-  return typeof value === "object" && value !== null && !Array.isArray(value) && Object.keys(value).length > 0;
-}
-
-export function formatMcpProxyToolCallLines(
-  args: McpProxyToolCallInput,
-  maxInputChars = DEFAULT_MAX_CALL_INPUT_CHARS,
-): string[] {
-  if (args.action === "ui-messages") return [`mcp ${args.action}`];
-
-  if (args.tool) {
-    const target = args.server ? `${args.tool} @ ${args.server}` : args.tool;
-    const lines = [`mcp call ${target}`];
-    if (args.args) lines.push(formatJsonish(args.args, maxInputChars));
-    return lines;
-  }
-
-  if (args.connect) return [`mcp connect ${args.connect}`];
-  if (args.describe) {
-    const target = args.server ? `${args.describe} @ ${args.server}` : args.describe;
-    return [`mcp describe ${target}`];
-  }
-
-  if (args.search) {
-    let line = `mcp search ${args.search}`;
-    if (args.server) line += ` @ ${args.server}`;
-    if (args.regex === true) line += " (regex)";
-    if (args.includeSchemas === false) line += " (schemas hidden)";
-    return [line];
-  }
-
-  if (args.server) return [`mcp list ${args.server}`];
-  if (args.action) return [`mcp ${args.action}`];
-
-  return ["mcp status"];
-}
-
-export function formatMcpDirectToolCallLines(
-  displayName: string,
-  args: Record<string, unknown>,
-  maxInputChars = DEFAULT_MAX_CALL_INPUT_CHARS,
-): string[] {
-  if (!hasUsefulObjectContent(args)) return [displayName];
-  return [displayName, formatJsonish(args, maxInputChars)];
-}
-
-export function formatMcpScriptToolCallLines(
-  args: { code: string },
-  maxInputChars = DEFAULT_MAX_CALL_INPUT_CHARS,
-): string[] {
-  return ["mcpScript", truncateText(args.code, maxInputChars)];
-}
-
-function renderToolCallLines(lines: string[], theme?: RenderTheme) {
-  const activeTheme = theme ?? plainTheme;
-  const [title = "mcp", ...rest] = lines;
-  const styledTitle = activeTheme.fg("toolTitle", activeTheme.bold ? activeTheme.bold(title) : title);
-  const styledRest = rest.map(line => activeTheme.fg("muted", line));
-  return new Text([styledTitle, ...styledRest].join("\n"), 0, 0);
-}
-
-function formatCompactInputPreview(lines: string[], maxChars = DEFAULT_MAX_COMPACT_INPUT_CHARS): string {
-  return truncateText(lines.slice(1).join(" ").replace(/\s+/g, " ").trim(), maxChars);
-}
-
-export function resolveMcpToolRenderOptions(settings?: McpToolRenderSettings): McpToolRenderOptions {
-  const resultRendering = settings?.toolResultRendering === "boxed" ? "boxed" : "compact";
-  const collapsedLines = settings?.collapsedResultLines;
-  const defaultLines = resultRendering === "boxed" ? DEFAULT_BOXED_COLLAPSED_LINES : DEFAULT_COMPACT_COLLAPSED_LINES;
-  return {
-    resultRendering,
-    collapsedResultLines: collapsedLines === 1 || collapsedLines === 2 || collapsedLines === 3 ? collapsedLines : defaultLines,
-  };
-}
-
-function shouldUseCompactFinalRender(options: McpToolRenderOptions, context?: McpToolRenderContext): boolean {
-  return options.resultRendering === "compact"
-    && context !== undefined
-    && context.isPartial === false
-    && context.expanded !== true
-    && context.isError !== true;
-}
-
-function renderToolCall(
-  lines: string[],
-  theme: RenderTheme | undefined,
-  context: McpToolRenderContext | undefined,
-  options: McpToolRenderOptions,
-) {
-  if (context?.state) {
-    context.state.compactTitle = lines[0] ?? "mcp";
-    context.state.compactInputPreview = formatCompactInputPreview(lines);
-  }
-  if (shouldUseCompactFinalRender(options, context)) return new EmptyComponent();
-  return renderToolCallLines(lines, theme);
-}
-
-export function renderMcpProxyToolCall(
-  args: McpProxyToolCallInput,
-  theme?: RenderTheme,
-  context?: McpToolRenderContext,
-) {
-  return renderToolCall(formatMcpProxyToolCallLines(args), theme, context, resolveMcpToolRenderOptions());
-}
-
-export function createMcpProxyToolCallRenderer(options: McpToolRenderOptions) {
-  return (args: McpProxyToolCallInput, theme?: RenderTheme, context?: McpToolRenderContext) => {
-    return renderToolCall(formatMcpProxyToolCallLines(args), theme, context, options);
-  };
-}
-
-export function createMcpDirectToolCallRenderer(displayName: string, options = resolveMcpToolRenderOptions()) {
-  return (args: Record<string, unknown>, theme?: RenderTheme, context?: McpToolRenderContext) => {
-    return renderToolCall(formatMcpDirectToolCallLines(displayName, args), theme, context, options);
-  };
-}
-
-export function createMcpScriptToolCallRenderer(options = resolveMcpToolRenderOptions()) {
-  return (args: { code: string }, theme?: RenderTheme, context?: McpToolRenderContext) => {
-    // Compact mode hides this call row once the result lands, so leave a title for the result row.
-    // The tool list comes from the result's `details.calls` trace (see formatMcpScriptCallSummary);
-    // the script code itself is never copied into the compact row.
-    if (context?.state) context.state.compactTitle = MCP_SCRIPT_TITLE;
-    if (shouldUseCompactFinalRender(options, context)) return new EmptyComponent();
-    return renderToolCallLines(formatMcpScriptToolCallLines(args), theme);
-  };
-}
-
-const MCP_SCRIPT_TITLE = "mcpScript";
-const MCP_SCRIPT_SUMMARY_MAX_TOOLS = 4;
-
-export interface McpScriptCallSummary {
-  /** Traced operations that failed; the compact row shows it as `mcpScript ✗N`. */
-  failed: number;
-  /** What the script touched; "" when it made no MCP operations. */
-  preview: string;
-}
-
-const PLAIN_TOOL_PATH = /^[A-Za-z0-9_.\/:@-]+$/;
-
-/**
- * Traced paths are script-supplied. Only identifier-like paths (what MCP tool names look like)
- * are shown bare. Anything else is shown as a quoted, ASCII-escaped JSON string, so it cannot
- * break the row, cannot pass for summary syntax (`foo×2`, `a, b`, ` · `, `+1 more`), and stays
- * distinct from the clean path it may resemble (`demo_red` vs `"\u001b[31mdemo_red"`).
- */
-function formatTracedPath(path: string): string {
-  if (PLAIN_TOOL_PATH.test(path)) return path;
-  // JSON.stringify quotes and escapes C0 controls, quotes and backslashes; escaping all other
-  // non-ASCII too rules out DEL, C1 controls, bidi overrides and invisible characters.
-  return JSON.stringify(path).replace(/[^\x20-\x7e]/g, (char) => `\\u${char.charCodeAt(0).toString(16).padStart(4, "0")}`);
-}
-
-/**
- * Failure count and compact-row preview for an mcpScript run, from the `details.calls` trace.
- * The preview lists distinct tool paths in first-call order with repeat counts, then other
- * operation counts, e.g. `github_search_issues×6, slack_post_message · search`.
- * Scripts that only describe tools list the described paths instead (`describe a, b`).
- * The failure count is rendered as a title status, not in the preview: narrow rows drop the
- * preview first, and the status is never truncated (the title shrinks instead).
- * Returns null for non-script results.
- */
-export function formatMcpScriptCallSummary(
-  details: McpToolResultDetails | undefined,
-  maxTools = MCP_SCRIPT_SUMMARY_MAX_TOOLS,
-): McpScriptCallSummary | null {
-  if (details?.mode !== "script") return null;
-  const operations = Array.isArray(details.calls) ? details.calls : [];
-  const callCounts = new Map<string, number>();
-  const describeCounts = new Map<string, number>();
-  const metaCounts = new Map<string, number>();
-  let failed = 0;
-  for (const operation of operations) {
-    if (typeof operation !== "object" || operation === null) continue;
-    const { operation: kind, path, ok } = operation as { operation?: unknown; path?: unknown; ok?: unknown };
-    if (typeof kind !== "string") continue;
-    if (ok === false) failed += 1;
-    const target = kind === "call" ? callCounts : kind === "describe" ? describeCounts : null;
-    if (target && typeof path === "string") {
-      target.set(path, (target.get(path) ?? 0) + 1);
-    } else {
-      metaCounts.set(kind, (metaCounts.get(kind) ?? 0) + 1);
-    }
-  }
-
-  const withCount = (name: string, count: number) => (count > 1 ? `${name}×${count}` : name);
-  const listPaths = (counts: Map<string, number>) => {
-    const names = [...counts].map(([path, count]) => withCount(formatTracedPath(path), count));
-    return names.length > maxTools
-      ? [...names.slice(0, maxTools), `+${names.length - maxTools} more`].join(", ")
-      : names.join(", ");
-  };
-
-  let tools = listPaths(callCounts);
-  if (!tools && describeCounts.size > 0) {
-    tools = `describe ${listPaths(describeCounts)}`;
-  } else {
-    const describes = [...describeCounts.values()].reduce((sum, count) => sum + count, 0);
-    if (describes > 0) metaCounts.set("describe", describes);
-  }
-  const meta = [...metaCounts].map(([kind, count]) => withCount(kind, count)).join(", ");
-  return { failed, preview: [tools, meta].filter(Boolean).join(" · ") };
-}
-
-function blockToLines(block: McpToolContentBlock): string[] {
-  if (block.type === "text") {
-    return block.text.split("\n");
-  }
-  return [`[image: ${block.mimeType}]`];
-}
-
-function collectCollapsedResultLines(
-  content: AgentToolResult<McpToolResultDetails>["content"],
-  maxLines: number,
-  maxChars: number,
-): McpToolResultDisplay {
-  if (content.length === 0) return { lines: ["(empty result)"], truncated: false };
-
-  const lines: string[] = [];
-  let remainingChars = maxChars;
-  let truncated = false;
-
-  const appendLine = (line: string) => {
-    if (lines.length === 0) {
-      const previewWidth = Math.min(line.length, remainingChars);
-      if (line.slice(0, previewWidth).trim() === "") {
-        if (line.length >= remainingChars) {
-          truncated = true;
-          remainingChars = 0;
-          return false;
-        }
-        remainingChars -= line.length + 1;
-        return true;
-      }
-    }
-
-    if (lines.length >= maxLines || remainingChars <= 0) {
-      truncated = true;
-      return false;
-    }
-
-    if (line.length > remainingChars) {
-      lines.push(line.slice(0, remainingChars));
-      truncated = true;
-      remainingChars = 0;
-      return false;
-    }
-
-    lines.push(line);
-    remainingChars -= line.length + 1;
-    return true;
-  };
-
-  for (const block of content) {
-    if (block.type !== "text") {
-      if (!appendLine(`[image: ${block.mimeType}]`)) break;
-      continue;
-    }
-
-    let start = 0;
-    while (start <= block.text.length) {
-      const newline = block.text.indexOf("\n", start);
-      const line = newline === -1 ? block.text.slice(start) : block.text.slice(start, newline);
-      if (!appendLine(line)) break;
-      if (newline === -1) break;
-      start = newline + 1;
-    }
-
-    if (truncated) break;
-  }
-
-  if (lines.length === 0) lines.push(truncated ? "(leading blank output omitted)" : "");
-  if (truncated && lines.length >= maxLines) lines.push("…");
-  return { lines, truncated };
-}
-
-export function formatMcpToolResultIdentity(details: McpToolResultDetails | undefined): string | null {
-  if (details?.mode !== "call") return null;
-  const server = typeof details.server === "string"
-    ? details.server
-    : typeof details.hintServer === "string"
-      ? details.hintServer
-      : null;
-  if (!server) return null;
-  if (typeof details.tool === "string") return `MCP ${server}/${details.tool}`;
-  if (typeof details.resourceUri === "string") return `MCP ${server} resource ${details.resourceUri}`;
-  if (typeof details.requestedTool === "string") return `MCP ${server}/${details.requestedTool}`;
-  return null;
-}
-
-export function formatMcpToolResultLines(
-  result: Pick<AgentToolResult<McpToolResultDetails>, "content">,
-  expanded: boolean,
-  maxCollapsedLines = DEFAULT_BOXED_COLLAPSED_LINES,
-  maxCollapsedChars = DEFAULT_MAX_COLLAPSED_CHARS,
-): McpToolResultDisplay {
-  if (!expanded) {
-    return collectCollapsedResultLines(result.content, maxCollapsedLines, maxCollapsedChars);
-  }
-
-  const allLines = result.content.flatMap(blockToLines);
-  const lines = allLines.length > 0 ? allLines : ["(empty result)"];
-  return { lines, truncated: false };
-}
-
-export function renderMcpToolResult(
-  result: AgentToolResult<McpToolResultDetails>,
-  options: ToolRenderResultOptions,
-  theme?: RenderTheme,
-  context?: McpToolRenderContext,
-  renderOptions = resolveMcpToolRenderOptions(),
-) {
-  const activeTheme = theme ?? plainTheme;
-  if (options.isPartial) {
-    return new Text(activeTheme.fg("warning", "Running MCP tool..."), 0, 0);
-  }
-
-  const hasErrorDetails = Boolean(result.details.error);
-  const expanded = options.expanded || context?.isError === true || hasErrorDetails;
-  if (!expanded && renderOptions.resultRendering === "compact") {
-    const display = formatMcpToolResultLines(result, false, renderOptions.collapsedResultLines);
-    const scriptSummary = formatMcpScriptCallSummary(result.details);
-    if (scriptSummary) {
-      const status = scriptSummary.failed > 0 ? `✗${scriptSummary.failed}` : "";
-      return new CompactMcpToolResult(MCP_SCRIPT_TITLE, scriptSummary.preview, display, activeTheme, status);
-    }
-    const title = context?.state?.compactTitle ?? formatMcpToolResultIdentity(result.details) ?? "";
-    const inputPreview = context?.state?.compactInputPreview ?? "";
-    return new CompactMcpToolResult(title, inputPreview, display, activeTheme);
-  }
-
-  const display = formatMcpToolResultLines(result, expanded, renderOptions.collapsedResultLines);
-  const identity = formatMcpToolResultIdentity(result.details);
-  const output = [
-    ...(identity ? [activeTheme.fg("muted", identity)] : []),
-    ...display.lines.map((line) => activeTheme.fg("toolOutput", line)),
-  ].join("\n");
-
-  return new CollapsibleText(
-    output,
-    expanded,
-    renderOptions.collapsedResultLines + (identity ? 1 : 0),
-    activeTheme.fg("muted", "…"),
-    activeTheme.fg("muted", "(Ctrl+O to expand)"),
-    display.truncated,
-  );
-}
-
-export function createMcpToolResultRenderer(renderOptions: McpToolRenderOptions) {
-  return (
-    result: AgentToolResult<McpToolResultDetails>,
-    options: ToolRenderResultOptions,
-    theme?: RenderTheme,
-    context?: McpToolRenderContext,
-  ) => renderMcpToolResult(result, options, theme, context, renderOptions);
-}