@gotgenes/pi-subagents 17.5.0 → 18.0.0

package/src/ui/conversation-viewer.ts

@@ -1,241 +0,0 @@
-/**
- * conversation-viewer.ts — Live conversation overlay for viewing agent sessions.
- *
- * Displays a scrollable, live-updating view of an agent's conversation.
- * Subscribes to session events for real-time streaming updates.
- */
-
-import { type Component, matchesKey, type TUI, truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
-import type { AgentConfigLookup } from "#src/config/agent-types";
-import { getLifetimeTotal } from "#src/lifecycle/usage";
-import type { Subagent } from "#src/types";
-import { buildInvocationTags, formatDuration, formatSessionTokens, getDisplayName, getPromptModeLabel, type Theme } from "#src/ui/display";
-import { formatMessage, formatStreamingIndicator } from "#src/ui/message-formatters";
-
-// ─────────────────────────────────────────────────────────────────────────────
-
-/** Base lines consumed by chrome: top border + header + header sep + footer sep + footer + bottom border. */
-const CHROME_LINES_BASE = 6;
-const MIN_VIEWPORT = 3;
-/** Height ceiling shared by the overlay's `maxHeight` and the viewer's internal viewport cap. */
-export const VIEWPORT_HEIGHT_PCT = 70;
-
-export interface ConversationViewerOptions {
-  tui: TUI;
-  record: Subagent;
-  theme: Theme;
-  done: (result: undefined) => void;
-  registry: AgentConfigLookup;
-  wrapText: (text: string, width: number) => string[];
-}
-
-export class ConversationViewer implements Component {
-  private scrollOffset = 0;
-  private autoScroll = true;
-  private unsubscribe: (() => void) | undefined;
-  private lastInnerW = 0;
-  private closed = false;
-
-  private tui: TUI;
-  private record: Subagent;
-  private theme: Theme;
-  private done: (result: undefined) => void;
-  private registry: AgentConfigLookup;
-  private wrapText: (text: string, width: number) => string[];
-
-  constructor({
-    tui,
-    record,
-    theme,
-    done,
-    registry,
-    wrapText,
-  }: ConversationViewerOptions) {
-    this.tui = tui;
-    this.record = record;
-    this.theme = theme;
-    this.done = done;
-    this.registry = registry;
-    this.wrapText = wrapText;
-    this.unsubscribe = record.subscribeToUpdates(() => {
-      if (this.closed) return;
-      this.tui.requestRender();
-    });
-  }
-
-  // fallow-ignore-next-line unused-class-member
-  handleInput(data: string): void {
-    if (matchesKey(data, "escape") || matchesKey(data, "q")) {
-      this.closed = true;
-      this.done(undefined);
-      return;
-    }
-
-    const totalLines = this.buildContentLines(this.lastInnerW).length;
-    const viewportHeight = this.viewportHeight();
-    const maxScroll = Math.max(0, totalLines - viewportHeight);
-
-    if (matchesKey(data, "up") || matchesKey(data, "k")) {
-      this.scrollOffset = Math.max(0, this.scrollOffset - 1);
-      this.autoScroll = this.scrollOffset >= maxScroll;
-    } else if (matchesKey(data, "down") || matchesKey(data, "j")) {
-      this.scrollOffset = Math.min(maxScroll, this.scrollOffset + 1);
-      this.autoScroll = this.scrollOffset >= maxScroll;
-    } else if (matchesKey(data, "pageUp") || matchesKey(data, "shift+up")) {
-      this.scrollOffset = Math.max(0, this.scrollOffset - viewportHeight);
-      this.autoScroll = false;
-    } else if (matchesKey(data, "pageDown") || matchesKey(data, "shift+down")) {
-      this.scrollOffset = Math.min(maxScroll, this.scrollOffset + viewportHeight);
-      this.autoScroll = this.scrollOffset >= maxScroll;
-    } else if (matchesKey(data, "home")) {
-      this.scrollOffset = 0;
-      this.autoScroll = false;
-    } else if (matchesKey(data, "end")) {
-      this.scrollOffset = maxScroll;
-      this.autoScroll = true;
-    }
-  }
-
-  render(width: number): string[] {
-    if (width < 6) return []; // too narrow for any meaningful rendering
-    const th = this.theme;
-    const innerW = width - 4; // border + padding
-    this.lastInnerW = innerW;
-    const lines: string[] = [];
-
-    const pad = (s: string, len: number) => {
-      const vis = visibleWidth(s);
-      return s + " ".repeat(Math.max(0, len - vis));
-    };
-    const row = (content: string) =>
-      th.fg("border", "│") + " " + truncateToWidth(pad(content, innerW), innerW) + " " + th.fg("border", "│");
-    const hrTop = th.fg("border", `╭${"─".repeat(width - 2)}╮`);
-    const hrBot = th.fg("border", `╰${"─".repeat(width - 2)}╯`);
-    const hrMid = row(th.fg("dim", "─".repeat(innerW)));
-
-    // Header
-    lines.push(hrTop);
-    const name = getDisplayName(this.record.type, this.registry);
-    const modeLabel = getPromptModeLabel(this.record.type, this.registry);
-    const modeTag = modeLabel ? ` ${th.fg("dim", `(${modeLabel})`)}` : "";
-    const statusIcon = this.record.status === "running"
-      ? th.fg("accent", "●")
-      : this.record.status === "completed"
-        ? th.fg("success", "✓")
-        : this.record.status === "error"
-          ? th.fg("error", "✗")
-          : th.fg("dim", "○");
-    const duration = formatDuration(this.record.startedAt, this.record.completedAt);
-
-    const headerParts: string[] = [duration];
-    const toolUses = this.record.toolUses;
-    if (toolUses > 0) headerParts.unshift(`${toolUses} tool${toolUses === 1 ? "" : "s"}`);
-    const tokens = getLifetimeTotal(this.record.lifetimeUsage);
-    if (tokens > 0) {
-      const percent = this.record.getContextPercent();
-      headerParts.push(formatSessionTokens(tokens, percent, th, this.record.compactionCount));
-    }
-
-    lines.push(row(
-      `${statusIcon} ${th.bold(name)}${modeTag}  ${th.fg("muted", this.record.description)} ${th.fg("dim", "·")} ${th.fg("dim", headerParts.join(" · "))}`,
-    ));
-    const invocationLine = this.invocationLine();
-    if (invocationLine) lines.push(row(invocationLine));
-    lines.push(hrMid);
-
-    // Content area — rebuild every render (live data, no cache needed)
-    const contentLines = this.buildContentLines(innerW);
-    const viewportHeight = this.viewportHeight();
-    const maxScroll = Math.max(0, contentLines.length - viewportHeight);
-
-    if (this.autoScroll) {
-      this.scrollOffset = maxScroll;
-    }
-
-    const visibleStart = Math.min(this.scrollOffset, maxScroll);
-    const visible = contentLines.slice(visibleStart, visibleStart + viewportHeight);
-
-    for (let i = 0; i < viewportHeight; i++) {
-      lines.push(row(visible[i] ?? ""));
-    }
-
-    // Footer
-    lines.push(hrMid);
-    const scrollPct = contentLines.length <= viewportHeight
-      ? "100%"
-      : `${Math.round(((visibleStart + viewportHeight) / contentLines.length) * 100)}%`;
-    const footerLeft = th.fg("dim", `${contentLines.length} lines · ${scrollPct}`);
-    const footerRight = th.fg("dim", "↑↓ scroll · PgUp/PgDn or Shift+↑↓ · Esc close");
-    const footerGap = Math.max(1, innerW - visibleWidth(footerLeft) - visibleWidth(footerRight));
-    lines.push(row(footerLeft + " ".repeat(footerGap) + footerRight));
-    lines.push(hrBot);
-
-    return lines;
-  }
-
-  // fallow-ignore-next-line unused-class-member
-  invalidate(): void { /* no cached state to clear */ }
-
-  // fallow-ignore-next-line unused-class-member
-  dispose(): void {
-    this.closed = true;
-    if (this.unsubscribe) {
-      this.unsubscribe();
-      this.unsubscribe = undefined;
-    }
-  }
-
-  // ---- Private ----
-
-  private viewportHeight(): number {
-    // Cap mirrors the overlay's maxHeight — otherwise the viewer would render
-    // more lines than the overlay shows and clip the footer.
-    const maxRows = Math.floor((this.tui.terminal.rows * VIEWPORT_HEIGHT_PCT) / 100);
-    return Math.max(MIN_VIEWPORT, maxRows - this.chromeLines());
-  }
-
-  private chromeLines(): number {
-    return CHROME_LINES_BASE + (this.invocationLine() ? 1 : 0);
-  }
-
-  private invocationLine(): string | undefined {
-    const { modelName, tags } = buildInvocationTags(this.record.invocation);
-    const parts = modelName ? [modelName, ...tags] : tags;
-    if (parts.length === 0) return undefined;
-    return this.theme.fg("dim", `  ↳ ${parts.join(" · ")}`);
-  }
-
-  private buildContentLines(width: number): string[] {
-    if (width <= 0) return [];
-
-    const th = this.theme;
-    const ctx = { theme: th, wrapText: this.wrapText };
-    const messages = this.record.messages;
-
-    if (messages.length === 0) {
-      return [th.fg("dim", "(waiting for first message...)")];
-    }
-
-    const lines: string[] = [];
-    let needsSeparator = false;
-    for (const msg of messages) {
-      const formatted = formatMessage(msg as { role: string; [key: string]: unknown }, width, ctx);
-      if (!formatted) continue;
-      if (needsSeparator) lines.push(th.fg("dim", "───"));
-      lines.push(...formatted);
-      needsSeparator = true;
-    }
-
-    // Streaming indicator for running agents — read activity off the record
-    if (this.record.status === "running") {
-      lines.push(...formatStreamingIndicator(
-        this.record.activeTools,
-        this.record.responseText,
-        width,
-        th,
-      ));
-    }
-
-    return lines.map(l => truncateToWidth(l, width));
-  }
-}

package/src/ui/message-formatters.ts

@@ -1,195 +0,0 @@
-/**
- * message-formatters.ts — Pure formatting functions for each session message type.
- *
- * Each function converts a single message or content block into display lines.
- * Returns null for empty/skippable content (caller skips the separator).
- */
-
-import { truncateToWidth } from "@earendil-works/pi-tui";
-import { extractAssistantContent } from "#src/session/content-items";
-import { extractText } from "#src/session/context";
-import type { Theme } from "#src/ui/display";
-import { describeActivity } from "#src/ui/display";
-
-// ── Types ────────────────────────────────────────────────────────────────────
-
-/** Narrow context shared by all message formatters. */
-export interface FormatterContext {
-  theme: Theme;
-  wrapText: (text: string, width: number) => string[];
-}
-
-// ── File-local types and guards ─────────────────────────────────────────────
-
-/** Bash execution message — 'bashExecution' role is not in the SDK's AgentSession message role union. */
-export interface BashExecutionMessage {
-  role: "bashExecution";
-  command: string;
-  output?: string;
-}
-
-/** Type guard for bash execution messages. */
-export function isBashExecution(msg: { role: string }): msg is BashExecutionMessage {
-  return msg.role === "bashExecution";
-}
-
-// ── formatUserMessage ─────────────────────────────────────────────────────────
-
-/**
- * Format a user message into display lines.
- * Returns null when the message text is empty (caller should skip separator).
- */
-export function formatUserMessage(
-  content: string | unknown[],
-  width: number,
-  ctx: FormatterContext,
-): string[] | null {
-  const { theme, wrapText } = ctx;
-  const text = typeof content === "string" ? content : extractText(content);
-  if (!text.trim()) return null;
-  return [
-    theme.fg("accent", "[User]"),
-    ...wrapText(text.trim(), width),
-  ];
-}
-
-// ── formatBashExecution ───────────────────────────────────────────────────────
-
-/**
- * Format a bash execution message into display lines.
- * Always returns at least the command line.
- */
-export function formatBashExecution(
-  msg: BashExecutionMessage,
-  width: number,
-  ctx: FormatterContext,
-): string[] {
-  const { theme, wrapText } = ctx;
-  const lines: string[] = [
-    truncateToWidth(theme.fg("muted", `  $ ${msg.command}`), width),
-  ];
-  const output = msg.output ?? "";
-  if (output.trim()) {
-    const out = output.length > 500 ? output.slice(0, 500) + "... (truncated)" : output;
-    lines.push(...wrapText(out.trim(), width).map(l => theme.fg("dim", l)));
-  }
-  return lines;
-}
-
-// ── formatStreamingIndicator ──────────────────────────────────────────────
-
-/**
- * Format the streaming activity indicator for a running agent.
- * Returns exactly two lines: an empty spacer and the indicator line.
- */
-export function formatStreamingIndicator(
-  activeTools: ReadonlyMap<string, string>,
-  responseText: string | undefined,
-  width: number,
-  theme: Theme,
-): string[] {
-  const act = describeActivity(activeTools, responseText);
-  return [
-    "",
-    truncateToWidth(theme.fg("accent", "\u25cd ") + theme.fg("dim", act), width),
-  ];
-}
-
-// ── formatToolResult ────────────────────────────────────────────────────────────
-
-/**
- * Format a tool result message into display lines.
- * Returns null when the result text is empty (caller should skip separator).
- */
-export function formatToolResult(
-  content: unknown[],
-  width: number,
-  ctx: FormatterContext,
-): string[] | null {
-  const { theme, wrapText } = ctx;
-  const text = extractText(content);
-  const truncated = text.length > 500 ? text.slice(0, 500) + "... (truncated)" : text;
-  if (!truncated.trim()) return null;
-  return [
-    theme.fg("dim", "[Result]"),
-    ...wrapText(truncated.trim(), width).map(l => theme.fg("dim", l)),
-  ];
-}
-
-// ── Types ─────────────────────────────────────────────────────────────────────
-
-/** Model/provider attribution for assistant messages. */
-export interface MessageAttribution {
-  provider?: string;
-  model?: string;
-}
-
-// ── formatAssistantMessage ───────────────────────────────────────────────────
-
-/**
- * Format an assistant message into display lines.
- * Always returns at least the [Assistant] header line.
- * When attribution is provided, renders as `[Assistant (provider/model)]`.
- */
-export function formatAssistantMessage(
-  content: { type: string; [key: string]: unknown }[],
-  width: number,
-  ctx: FormatterContext,
-  attribution?: MessageAttribution,
-): string[] {
-  const { theme, wrapText } = ctx;
-  const { textParts, toolNames } = extractAssistantContent(content);
-  const label = formatAttributionLabel(attribution);
-  const lines: string[] = [theme.bold(`[Assistant${label}]`)];
-  if (textParts.length > 0) {
-    lines.push(...wrapText(textParts.join("\n").trim(), width));
-  }
-  for (const name of toolNames) {
-    lines.push(truncateToWidth(theme.fg("muted", `  [Tool: ${name}]`), width));
-  }
-  return lines;
-}
-
-// ── formatMessage dispatcher ───────────────────────────────────────────────
-
-/**
- * Dispatch a session message to the appropriate formatter.
- * Returns null for empty/skippable messages and unknown roles.
- */
-export function formatMessage(
-  msg: { role: string; [key: string]: unknown },
-  width: number,
-  ctx: FormatterContext,
-): string[] | null {
-  if (msg.role === "user") {
-    return formatUserMessage(msg.content as string | unknown[], width, ctx);
-  }
-  if (msg.role === "assistant") {
-    const attribution: MessageAttribution = {
-      provider: msg.provider as string | undefined,
-      model: msg.model as string | undefined,
-    };
-    return formatAssistantMessage(
-      msg.content as { type: string; [key: string]: unknown }[],
-      width,
-      ctx,
-      attribution,
-    );
-  }
-  if (msg.role === "toolResult") {
-    return formatToolResult(msg.content as unknown[], width, ctx);
-  }
-  if (isBashExecution(msg)) {
-    return formatBashExecution(msg, width, ctx);
-  }
-  return null;
-}
-
-// ── Helpers ──────────────────────────────────────────────────────────────────
-
-/** Build a `(provider/model)` attribution suffix, or empty string when absent. */
-function formatAttributionLabel(attr?: MessageAttribution): string {
-  if (!attr?.provider && !attr?.model) return "";
-  if (attr.provider && attr.model) return ` (${attr.provider}/${attr.model})`;
-  return ` (${attr.provider ?? attr.model})`;
-}