@gotgenes/pi-subagents 17.4.0 → 18.0.0

package/src/ui/agent-menu.ts

@@ -1,331 +0,0 @@
-/* eslint-disable @typescript-eslint/no-unsafe-assignment -- Pi SDK types are not fully exported; see upstream Pi SDK for type improvements */
-import { wrapTextWithAnsi } from "@earendil-works/pi-tui";
-import { AgentTypeRegistry } from "#src/config/agent-types";
-import type { ParentSnapshot } from "#src/lifecycle/parent-snapshot";
-import { type ModelRegistry, resolveModel } from "#src/session/model-resolver";
-import { getModelLabelFromConfig } from "#src/tools/helpers";
-import type { AgentConfig, Subagent } from "#src/types";
-import { AgentConfigEditor } from "#src/ui/agent-config-editor";
-import { AgentCreationWizard } from "#src/ui/agent-creation-wizard";
-import type { AgentFileOps } from "#src/ui/agent-file-ops";
-import { formatDuration, getDisplayName } from "#src/ui/display";
-
-// ---- Narrow interfaces ----
-
-/** Narrow manager interface for menu operations. */
-export interface AgentMenuManager {
-  listAgents: () => Subagent[];
-  getRecord: (id: string) => Subagent | undefined;
-  /** Used by generate wizard to spawn an agent that writes the .md file. */
-  spawnAndWait: (
-    parentSnapshot: ParentSnapshot,
-    type: string,
-    prompt: string,
-    opts: { description: string; maxTurns: number },
-  ) => Promise<Subagent>;
-}
-
-/** Narrow settings interface required by the agent menu. */
-export interface AgentMenuSettings {
-  readonly maxConcurrent: number;
-  readonly defaultMaxTurns: number | undefined;
-  readonly graceTurns: number;
-  applyMaxConcurrent(n: number): { message: string; level: "info" | "warning" };
-  applyDefaultMaxTurns(n: number): { message: string; level: "info" | "warning" };
-  applyGraceTurns(n: number): { message: string; level: "info" | "warning" };
-}
-
-// ---- Narrow UI context types ----
-
-/** Narrow UI interface — only the ctx.ui methods menu handlers actually call. */
-export interface MenuUI {
-  select(title: string, options: string[]): Promise<string | undefined>;
-  confirm(title: string, message: string): Promise<boolean>;
-  input(title: string, defaultValue?: string): Promise<string | undefined>;
-  notify(message: string, level: "info" | "warning" | "error"): void;
-  editor(title: string, content: string): Promise<string | undefined>;
-  custom<R>(component: any, options?: any): Promise<R>;
-}
-
-// ---- Class ----
-
-/**
- * Handler for the `/agents` slash command.
- *
- * Call `handle(ctx)` from the Pi command registration to open the interactive menu.
- */
-export class AgentsMenuHandler {
-  private readonly editor: AgentConfigEditor;
-  private readonly wizard: AgentCreationWizard;
-
-  constructor(
-    private readonly manager: AgentMenuManager,
-    private readonly registry: AgentTypeRegistry,
-    private readonly settings: AgentMenuSettings,
-    fileOps: AgentFileOps,
-    personalAgentsDir: string,
-    projectAgentsDir: string,
-  ) {
-    this.editor = new AgentConfigEditor(
-      fileOps,
-      registry,
-      personalAgentsDir,
-      projectAgentsDir,
-    );
-    this.wizard = new AgentCreationWizard(
-      fileOps,
-      manager,
-      registry,
-      personalAgentsDir,
-      projectAgentsDir,
-    );
-  }
-
-  async handle({
-    ui,
-    modelRegistry,
-    parentSnapshot,
-  }: {
-    ui: MenuUI;
-    modelRegistry: ModelRegistry;
-    parentSnapshot: ParentSnapshot;
-  }): Promise<void> {
-    await this.showAgentsMenu(ui, modelRegistry, parentSnapshot);
-  }
-
-  private getModelLabel(type: string, modelRegistry?: ModelRegistry): string {
-    const cfg = this.registry.resolveAgentConfig(type);
-    if (!cfg.model) return "inherit";
-    if (modelRegistry) {
-      const resolved = resolveModel(cfg.model, modelRegistry);
-      if (typeof resolved === "string") return "inherit";
-    }
-    return getModelLabelFromConfig(cfg.model);
-  }
-
-  private async showAgentsMenu(
-    ui: MenuUI,
-    modelRegistry: ModelRegistry,
-    parentSnapshot: ParentSnapshot,
-  ): Promise<void> {
-    this.registry.reload();
-    const allNames = this.registry.getAllTypes();
-
-    const options: string[] = [];
-
-    const agents = this.manager.listAgents();
-    if (agents.length > 0) {
-      const running = agents.filter(
-        (a) => a.status === "running" || a.status === "queued",
-      ).length;
-      const done = agents.filter(
-        (a) => a.status === "completed" || a.status === "steered",
-      ).length;
-      options.push(
-        `Running agents (${agents.length}) — ${running} running, ${done} done`,
-      );
-    }
-
-    if (allNames.length > 0) {
-      options.push(`Agent types (${allNames.length})`);
-    }
-
-    options.push("Create new agent");
-    options.push("Settings");
-
-    const noAgentsMsg =
-      allNames.length === 0 && agents.length === 0
-        ? "No agents found. Create specialized subagents that can be delegated to.\n\n" +
-          "Each subagent has its own context window, custom system prompt, and specific tools.\n\n" +
-          "Try creating: Code Reviewer, Security Auditor, Test Writer, or Documentation Writer.\n\n"
-        : "";
-
-    if (noAgentsMsg) {
-      ui.notify(noAgentsMsg, "info");
-    }
-
-    const choice = await ui.select("Agents", options);
-    if (!choice) return;
-
-    if (choice.startsWith("Running agents (")) {
-      await this.showRunningAgents(ui);
-      await this.showAgentsMenu(ui, modelRegistry, parentSnapshot);
-    } else if (choice.startsWith("Agent types (")) {
-      await this.showAllAgentsList(ui, modelRegistry);
-      await this.showAgentsMenu(ui, modelRegistry, parentSnapshot);
-    } else if (choice === "Create new agent") {
-      await this.wizard.showCreateWizard(ui, parentSnapshot);
-    } else if (choice === "Settings") {
-      await this.showSettings(ui);
-      await this.showAgentsMenu(ui, modelRegistry, parentSnapshot);
-    }
-  }
-
-  private async showAllAgentsList(ui: MenuUI, modelRegistry: ModelRegistry): Promise<void> {
-    const allNames = this.registry.getAllTypes();
-    if (allNames.length === 0) {
-      ui.notify("No agents.", "info");
-      return;
-    }
-
-    const sourceIndicator = (cfg: AgentConfig | undefined) => {
-      const disabled = cfg?.enabled === false;
-      if (cfg?.source === "project") return disabled ? "✕• " : "•  ";
-      if (cfg?.source === "global") return disabled ? "✕◦ " : "◦  ";
-      if (disabled) return "✕  ";
-      return "   ";
-    };
-
-    const entries = allNames.map((name) => {
-      const cfg = this.registry.resolveAgentConfig(name);
-      const disabled = cfg.enabled === false;
-      const model = this.getModelLabel(name, modelRegistry);
-      const indicator = sourceIndicator(cfg);
-      const prefix = `${indicator}${name} · ${model}`;
-      const desc = disabled ? "(disabled)" : cfg.description;
-      return { name, prefix, desc };
-    });
-    const maxPrefix = Math.max(...entries.map((e) => e.prefix.length));
-
-    const hasCustom = allNames.some((n) => {
-      const c = this.registry.resolveAgentConfig(n);
-      return !c.isDefault && c.enabled !== false;
-    });
-    const hasDisabled = allNames.some(
-      (n) => this.registry.resolveAgentConfig(n).enabled === false,
-    );
-    const legendParts: string[] = [];
-    if (hasCustom) legendParts.push("• = project  ◦ = global");
-    if (hasDisabled) legendParts.push("✕ = disabled");
-    const legend = legendParts.length ? "\n" + legendParts.join("  ") : "";
-
-    const options = entries.map(
-      ({ prefix, desc }) => `${prefix.padEnd(maxPrefix)} — ${desc}`,
-    );
-    if (legend) options.push(legend);
-
-    const choice = await ui.select("Agent types", options);
-    if (!choice) return;
-
-    const agentName = choice
-      .split(" · ")[0]
-      .replace(/^[•◦✕\s]+/, "")
-      .trim();
-    if (this.registry.resolveType(agentName) != null) {
-      await this.editor.showAgentDetail(ui, agentName);
-      await this.showAllAgentsList(ui, modelRegistry);
-    }
-  }
-
-  private async showRunningAgents(ui: MenuUI): Promise<void> {
-    const agents = this.manager.listAgents();
-    if (agents.length === 0) {
-      ui.notify("No agents.", "info");
-      return;
-    }
-
-    const options = agents.map((a) => {
-      const dn = getDisplayName(a.type, this.registry);
-      const dur = formatDuration(a.startedAt, a.completedAt);
-      return `${dn} (${a.description}) · ${a.toolUses} tools · ${a.status} · ${dur}`;
-    });
-
-    const choice = await ui.select("Running agents", options);
-    if (!choice) return;
-
-    const idx = options.indexOf(choice);
-    if (idx < 0) return;
-    const record = agents[idx];
-
-    await this.viewAgentConversation(ui, record);
-    await this.showRunningAgents(ui);
-  }
-
-  private async viewAgentConversation(ui: MenuUI, record: Subagent): Promise<void> {
-    if (!record.isSessionReady()) {
-      ui.notify(
-        `Agent is ${record.status === "queued" ? "queued" : "expired"} — no session available.`,
-        "info",
-      );
-      return;
-    }
-
-    const { ConversationViewer, VIEWPORT_HEIGHT_PCT } = await import(
-      "./conversation-viewer"
-    );
-
-    await ui.custom<undefined>(
-      (tui: any, theme: any, _keybindings: any, done: any) => {
-        return new ConversationViewer({
-          tui,
-          record,
-          theme,
-          done,
-          registry: this.registry,
-          wrapText: wrapTextWithAnsi,
-        });
-      },
-      {
-        overlay: true,
-        overlayOptions: {
-          anchor: "center",
-          width: "90%",
-          maxHeight: `${VIEWPORT_HEIGHT_PCT}%`,
-        },
-      },
-    );
-  }
-
-  private async showSettings(ui: MenuUI): Promise<void> {
-    const choice = await ui.select("Settings", [
-      `Max concurrency (current: ${this.settings.maxConcurrent})`,
-      `Default max turns (current: ${this.settings.defaultMaxTurns ?? "unlimited"})`,
-      `Grace turns (current: ${this.settings.graceTurns})`,
-    ]);
-    if (!choice) return;
-
-    if (choice.startsWith("Max concurrency")) {
-      const val = await ui.input(
-        "Max concurrent background agents",
-        String(this.settings.maxConcurrent),
-      );
-      if (val) {
-        const n = parseInt(val, 10);
-        if (n >= 1) {
-          const toast = this.settings.applyMaxConcurrent(n);
-          ui.notify(toast.message, toast.level);
-        } else {
-          ui.notify("Must be a positive integer.", "warning");
-        }
-      }
-    } else if (choice.startsWith("Default max turns")) {
-      const val = await ui.input(
-        "Default max turns before wrap-up (0 = unlimited)",
-        String(this.settings.defaultMaxTurns ?? 0),
-      );
-      if (val) {
-        const n = parseInt(val, 10);
-        if (n >= 0) {
-          const toast = this.settings.applyDefaultMaxTurns(n);
-          ui.notify(toast.message, toast.level);
-        } else {
-          ui.notify("Must be 0 (unlimited) or a positive integer.", "warning");
-        }
-      }
-    } else if (choice.startsWith("Grace turns")) {
-      const val = await ui.input(
-        "Grace turns after wrap-up steer",
-        String(this.settings.graceTurns),
-      );
-      if (val) {
-        const n = parseInt(val, 10);
-        if (n >= 1) {
-          const toast = this.settings.applyGraceTurns(n);
-          ui.notify(toast.message, toast.level);
-        } else {
-          ui.notify("Must be a positive integer.", "warning");
-        }
-      }
-    }
-  }
-}

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));
-  }
-}