pi-extended-teams 2.1.16

package/extensions/ui/read-agent-status.ts

@@ -0,0 +1,125 @@
+import type { RunningReadAgent } from "../runtime/types";
+import { formatElapsed } from "./renderers.js";
+
+export const READ_AGENT_IDLE_NUDGE_MS = 5 * 60_000;
+export const READ_AGENT_HANGING_NUDGE_MS = 15 * 60_000;
+
+export type ReadAgentStatusLabel = RunningReadAgent["status"] | "idle" | "hanging";
+export type ReadAgentIdleLevel = "none" | "soft" | "hard";
+
+export interface ReadAgentStatusDescription {
+  label: ReadAgentStatusLabel;
+  detail: string;
+  idleLevel: ReadAgentIdleLevel;
+  idleMs: number;
+}
+
+export type ReadAgentStatusCounts = Partial<Record<ReadAgentStatusLabel, number>>;
+
+export interface ReadAgentStatusSample {
+  agent: RunningReadAgent;
+  status: ReadAgentStatusDescription;
+}
+
+export interface ReadAgentStatusSummary {
+  total: number;
+  counts: ReadAgentStatusCounts;
+  samples: ReadAgentStatusSample[];
+}
+
+export interface ReadAgentStatusSummaryOptions {
+  now?: number;
+  maxDetailed?: number;
+}
+
+interface ReadAgentStatusClassification {
+  label: ReadAgentStatusLabel;
+  idleLevel: ReadAgentIdleLevel;
+  idleMs: number;
+}
+
+export function buildReadAgentIdleNudgeMessage(agent: RunningReadAgent, status: ReadAgentStatusDescription): string {
+  const severity = status.idleLevel === "hard" ? "appears hung" : "has gone quiet";
+  return `Read agent ${agent.name} on team ${agent.teamName} ${severity}: no response or token change for ${formatElapsed(status.idleMs)}. Status is visible in the live agent view; do not ping/check repeatedly.`;
+}
+
+export function shouldNudgeReadAgentIdle(_previousLevel: "soft" | "hard" | undefined, _currentLevel: ReadAgentIdleLevel): boolean {
+  return false;
+}
+
+export function describeReadAgentStatus(agent: RunningReadAgent, now = Date.now()): ReadAgentStatusDescription {
+  return describeReadAgentStatusFromClassification(agent, classifyReadAgentStatus(agent, now));
+}
+
+export function summarizeReadAgentStatuses(
+  agents: Iterable<RunningReadAgent>,
+  options: ReadAgentStatusSummaryOptions = {}
+): ReadAgentStatusSummary {
+  const now = options.now ?? Date.now();
+  const maxDetailed = Math.max(0, options.maxDetailed ?? Number.POSITIVE_INFINITY);
+  const counts: ReadAgentStatusCounts = {};
+  const samples: ReadAgentStatusSample[] = [];
+  let total = 0;
+
+  for (const agent of agents) {
+    const classification = classifyReadAgentStatus(agent, now);
+    counts[classification.label] = (counts[classification.label] ?? 0) + 1;
+    if (samples.length < maxDetailed) {
+      samples.push({
+        agent,
+        status: describeReadAgentStatusFromClassification(agent, classification),
+      });
+    }
+    total++;
+  }
+
+  return { total, counts, samples };
+}
+
+function classifyReadAgentStatus(agent: RunningReadAgent, now: number): ReadAgentStatusClassification {
+  const lastActivityAt = agent.lastActivityAt || agent.startedAt;
+  const idleMs = Math.max(0, now - lastActivityAt);
+
+  if (idleMs >= READ_AGENT_HANGING_NUDGE_MS) {
+    return { label: "hanging", idleLevel: "hard", idleMs };
+  }
+
+  if (idleMs >= READ_AGENT_IDLE_NUDGE_MS) {
+    return { label: "idle", idleLevel: "soft", idleMs };
+  }
+
+  return { label: agent.status, idleLevel: "none", idleMs };
+}
+
+function describeReadAgentStatusFromClassification(
+  agent: RunningReadAgent,
+  classification: ReadAgentStatusClassification
+): ReadAgentStatusDescription {
+  if (classification.idleLevel === "hard") {
+    return {
+      ...classification,
+      detail: `no response/token change for ${formatElapsed(classification.idleMs)} · visible in live agent view`,
+    };
+  }
+
+  if (classification.idleLevel === "soft") {
+    return {
+      ...classification,
+      detail: `no response/token change for ${formatElapsed(classification.idleMs)} · visible in live agent view`,
+    };
+  }
+
+  return {
+    ...classification,
+    detail: describeActiveReadAgentWork(agent),
+  };
+}
+
+function describeActiveReadAgentWork(agent: RunningReadAgent): string {
+  if (agent.status === "starting") return "starting session";
+  if (agent.status === "finishing") return "sending report";
+  if (agent.status === "working") {
+    return agent.activeToolName ? `using ${agent.activeToolName}` : "using tools";
+  }
+  return "waiting for model response";
+}

package/extensions/ui/renderers.ts

@@ -0,0 +1,243 @@
+import { keyText } from "@mariozechner/pi-coding-agent";
+import { Text, truncateToWidth } from "@mariozechner/pi-tui";
+import type { ContextUsageSnapshot } from "../../src/utils/runtime";
+import { dimAnsi, pink, purple } from "./ansi";
+
+const ANSI_ESCAPE_SEQUENCE = /\x1b(?:\[[0-?]*[ -/]*[@-~]|\][^\x07\x1b]*(?:\x07|\x1b\\)|_[^\x07\x1b]*(?:\x07|\x1b\\)|[PX^][^\x1b]*(?:\x1b\\|\x07)|[@-Z\\-_])/g;
+const SAFE_SGR_SEQUENCE = /^\x1b\[[0-9;]*m$/;
+const ANIMATED_PROGRESS_DOTS = [".", "..", "..."] as const;
+
+export function sanitizeTuiText(text: string): string {
+  return text
+    .replace(ANSI_ESCAPE_SEQUENCE, (sequence) => SAFE_SGR_SEQUENCE.test(sequence) ? sequence : "")
+    .replace(/\r\n?/g, "\n")
+    .replace(/\t/g, "   ")
+    .replace(/[\u0000-\u0008\u000B\u000C\u000E-\u001A\u001C-\u001F\u007F-\u009F]/g, "")
+    .replace(/\x1b(?!\[[0-9;]*m)/g, "");
+}
+
+export function sanitizeTuiLine(text: string): string {
+  return sanitizeTuiText(text).replace(/\n/g, " ");
+}
+
+export function sanitizePlainTuiLine(text: string): string {
+  return sanitizeTuiLine(text).replace(/\x1b\[[0-9;]*m/g, "");
+}
+
+export function formatModelLabel(model?: string, thinking?: string): string {
+  const shortModel = model ? model.split("/").pop() || model : "";
+  const t = thinking && thinking !== "off" ? ` · ${thinking}` : "";
+  return shortModel ? `${shortModel}${t}` : "";
+}
+
+export function formatAnimatedProgress(text: string, now: number): string {
+  const base = sanitizePlainTuiLine(text).replace(/\.+$/, "");
+  const phase = Math.floor(now / 1000) % ANIMATED_PROGRESS_DOTS.length;
+  return `${base}${phase === 0 ? ANIMATED_PROGRESS_DOTS[0] : phase === 1 ? ANIMATED_PROGRESS_DOTS[1] : ANIMATED_PROGRESS_DOTS[2]}`;
+}
+
+export function formatElapsed(ms: number): string {
+  const totalSeconds = Math.max(0, Math.floor(ms / 1000));
+  const minutes = Math.floor(totalSeconds / 60);
+  const seconds = totalSeconds % 60;
+  return minutes > 0 ? `${minutes}m${seconds.toString().padStart(2, "0")}s` : `${seconds}s`;
+}
+
+export function formatTokenCount(count: number): string {
+  if (count < 1000) return `${count}`;
+  if (count < 1000000) return `${(count / 1000).toFixed(count < 10000 ? 1 : 0)}k`;
+  return `${(count / 1000000).toFixed(1)}M`;
+}
+
+export function formatContextUsage(usage: ContextUsageSnapshot | undefined): string {
+  if (typeof usage?.tokens !== "number"
+    || !Number.isFinite(usage.tokens)
+    || typeof usage.percent !== "number"
+    || !Number.isFinite(usage.percent)) {
+    return "? tok (?%)";
+  }
+  return `${formatTokenCount(usage.tokens)} tok (${Math.round(usage.percent)}%)`;
+}
+
+export function extractTextParts(content: any): string {
+  if (typeof content === "string") return content.trim();
+  if (Array.isArray(content)) {
+    return content
+      .filter((part: any) => part?.type === "text" && typeof part.text === "string")
+      .map((part: any) => part.text)
+      .join("\n")
+      .trim();
+  }
+  return "";
+}
+
+export function getLastAssistantText(messages: any[]): string {
+  for (let i = messages.length - 1; i >= 0; i--) {
+    if (messages[i]?.role !== "assistant") continue;
+    const text = extractTextParts(messages[i].content);
+    if (text) return text;
+  }
+  return "";
+}
+
+export function formatTranscriptLines(messages: any[]): string[] {
+  const out: string[] = [];
+  for (const message of messages || []) {
+    if (message?.role === "user") {
+      const text = sanitizeTuiText(extractTextParts(message.content));
+      if (text) out.push(`${pink("user ▸")} ${text}`);
+    } else if (message?.role === "assistant") {
+      for (const part of Array.isArray(message.content) ? message.content : []) {
+        if (part?.type === "text" && part.text?.trim()) out.push(sanitizeTuiText(part.text.trim()));
+        else if (part?.type === "toolCall") out.push(purple(`⚙ ${sanitizeTuiLine(String(part.name || "tool"))}`));
+      }
+    } else if (message?.role === "toolResult") {
+      const text = sanitizeTuiText(extractTextParts(message.content));
+      const head = text.split("\n").find((line: string) => line.trim()) || "(no output)";
+      out.push(dimAnsi(`  ↳ ${sanitizeTuiLine(String(message.toolName || "tool"))}: ${head}`));
+    }
+  }
+  return out;
+}
+
+export function summarizeInboxMessage(message: any): string {
+  const from = message?.from || "unknown";
+  const summary = message?.summary || "message";
+  const timestamp = message?.timestamp ? new Date(message.timestamp).toLocaleTimeString() : "";
+  return `${from}${timestamp ? ` ${timestamp}` : ""}: ${summary}`;
+}
+
+export function formatInboxMessagesForModel(messages: any[]): string {
+  if (messages.length === 0) return "No inbox messages.";
+
+  return messages.map((message, index) => {
+    const header = `[${index + 1}] ${summarizeInboxMessage(message)}`;
+    const text = String(message?.text || "").trim() || "(empty message)";
+    return `${header}\n${text}`;
+  }).join("\n\n");
+}
+
+export function renderInboxMessages(result: any, expanded: boolean, theme: any) {
+  const messages = result.details?.messages || [];
+  if (messages.length === 0) {
+    return new Text(theme.fg("muted", "inbox: no messages"), 0, 0);
+  }
+
+  const lines = [theme.fg("toolTitle", theme.bold(`inbox: ${messages.length} message${messages.length === 1 ? "" : "s"}`))];
+  if (expanded) {
+    for (const [index, message] of messages.entries()) {
+      lines.push(theme.fg("muted", `\n${index + 1}. ${summarizeInboxMessage(message)}`));
+      lines.push(String(message?.text || "").trim() || theme.fg("dim", "(empty message)"));
+    }
+  } else {
+    for (const [index, message] of messages.slice(0, 5).entries()) {
+      lines.push(truncateToWidth(`${index + 1}. ${summarizeInboxMessage(message)}`, 120));
+    }
+    if (messages.length > 5) lines.push(theme.fg("dim", `… ${messages.length - 5} more`));
+    lines.push(theme.fg("dim", "expand tool output to read full message text"));
+  }
+
+  return new Text(lines.join("\n"), 0, 0);
+}
+
+export function formatTeammateStatusForModel(agentName: string, details: any): string {
+  const heartbeat = details.hasRecentHeartbeat ? "fresh heartbeat" : "stale/no heartbeat";
+  const ready = details.agentLoopReady ? "ready" : "not ready";
+  const unread = `${details.unreadCount ?? 0} unread`;
+  const released = details.releasedClaims?.length ? `; released claims: ${details.releasedClaims.join(", ")}` : "";
+  return `${agentName}: ${details.health}; ${unread}; ${ready}; ${heartbeat}${released}`;
+}
+
+export function renderTeammateStatus(result: any, expanded: boolean, theme: any) {
+  const details = result.details || {};
+  const agentName = details.agentName || "teammate";
+  const health = details.health || (details.alive ? "alive" : "dead");
+  const color = health === "dead" || health === "stalled" ? "warning" : "success";
+  const headline = `${agentName}: ${health} • ${details.unreadCount ?? 0} unread • ${details.agentLoopReady ? "ready" : "not ready"}`;
+
+  if (!expanded) {
+    return new Text(theme.fg(color, headline), 0, 0);
+  }
+
+  const runtimeStatus = details.runtime || {};
+  const lines = [theme.fg(color, theme.bold(headline))];
+  lines.push(`alive: ${details.alive ? "yes" : "no"}`);
+  lines.push(`heartbeat: ${details.hasRecentHeartbeat ? "fresh" : "stale/missing"}`);
+  if (runtimeStatus.startedAt) lines.push(`started: ${new Date(runtimeStatus.startedAt).toLocaleString()}`);
+  if (runtimeStatus.lastHeartbeatAt) lines.push(`last heartbeat: ${new Date(runtimeStatus.lastHeartbeatAt).toLocaleString()}`);
+  if (details.startupStalled) lines.push(theme.fg("warning", "startup appears stalled"));
+  if (details.releasedClaims?.length) lines.push(`released claims: ${details.releasedClaims.join(", ")}`);
+  return new Text(lines.join("\n"), 0, 0);
+}
+
+function expandKeyLabel(): string {
+  try {
+    return keyText("app.tools.expand") || "ctrl+o";
+  } catch {
+    return "ctrl+o";
+  }
+}
+
+function plural(count: number, label: string): string {
+  return `${count} ${label}${count === 1 ? "" : "s"}`;
+}
+
+function memberStatusLine(member: any): string {
+  const runtimeStatus = member.runtime || {};
+  const bits = [
+    member.role,
+    member.health,
+    `${member.unreadCount ?? 0} unread`,
+    member.agentLoopReady ? "ready" : "not ready",
+    member.hasRecentHeartbeat ? "fresh heartbeat" : "stale/no heartbeat",
+    runtimeStatus.currentAction,
+    runtimeStatus.activeToolName ? `tool: ${runtimeStatus.activeToolName}` : "",
+  ].filter(Boolean);
+  return `${member.agentName || member.member?.name || "teammate"}: ${bits.join(" · ")}`;
+}
+
+export function renderTeamObservation(result: any, expanded: boolean, theme: any) {
+  const details = result.details || {};
+  const members = Array.isArray(details.members) ? details.members : [];
+  const tasks = Array.isArray(details.tasks) ? details.tasks : [];
+  const claims = Array.isArray(details.claims) ? details.claims : [];
+  const writeQueue = Array.isArray(details.writeQueue) ? details.writeQueue : [];
+  const reports = Array.isArray(details.reports) ? details.reports : [];
+  const activeMembers = members.filter((member: any) => member.agentName !== "team-lead" && member.alive !== false);
+  const unread = members.reduce((sum: number, member: any) => sum + (Number(member.unreadCount) || 0), 0);
+  const troubled = members.filter((member: any) => member.health === "dead" || member.health === "stalled");
+  const teamName = details.teamName || "team";
+  const summary = [
+    plural(members.length, "member"),
+    plural(activeMembers.length, "active agent"),
+    unread ? plural(unread, "unread message") : "0 unread messages",
+    tasks.length ? plural(tasks.length, "task") : "0 tasks",
+    claims.length ? plural(claims.length, "claim") : "0 claims",
+    writeQueue.length ? `${writeQueue.length} queued writer${writeQueue.length === 1 ? "" : "s"}` : "0 queued writers",
+    reports.length ? plural(reports.length, "report") : "0 reports",
+  ].join(" · ");
+  const color = troubled.length > 0 ? "warning" : "success";
+  const headline = `${teamName}: ${summary}`;
+
+  if (!expanded) {
+    return new Text(`${theme.fg(color, headline)}  ${theme.fg("dim", `(${expandKeyLabel()} details)`)}`, 0, 0);
+  }
+
+  const lines = [theme.fg(color, theme.bold(headline))];
+  if (troubled.length > 0) lines.push(theme.fg("warning", `${troubled.length} member${troubled.length === 1 ? "" : "s"} need attention`));
+
+  if (members.length > 0) {
+    lines.push(theme.fg("muted", "members:"));
+    for (const member of members.slice(0, 12)) lines.push(`  ${memberStatusLine(member)}`);
+    if (members.length > 12) lines.push(theme.fg("dim", `  … ${members.length - 12} more`));
+  }
+
+  const sections = [
+    tasks.length ? `tasks: ${tasks.map((task: any) => task.id || task.subject || task.path || "task").slice(0, 5).join(", ")}${tasks.length > 5 ? `, … ${tasks.length - 5} more` : ""}` : "tasks: none",
+    claims.length ? `claims: ${claims.map((claim: any) => claim.path || claim.file || "claim").slice(0, 5).join(", ")}${claims.length > 5 ? `, … ${claims.length - 5} more` : ""}` : "claims: none",
+    writeQueue.length ? `write queue: ${writeQueue.map((item: any) => item.name || item.agentName || "writer").slice(0, 5).join(", ")}${writeQueue.length > 5 ? `, … ${writeQueue.length - 5} more` : ""}` : "write queue: empty",
+    reports.length ? `reports: ${reports.map((report: any) => report.name || report.agentName || report.summary || "report").slice(0, 5).join(", ")}${reports.length > 5 ? `, … ${reports.length - 5} more` : ""}` : "reports: none",
+  ];
+  lines.push(...sections.map((line) => theme.fg("muted", line)));
+  return new Text(lines.join("\n"), 0, 0);
+}

package/extensions/ui/status-widget.test.ts

@@ -0,0 +1,143 @@
+import { describe, expect, it, vi } from "vitest";
+import { teamActivityStatusWidget, type TeamActivityStatusEntry, type TeamActivityStatusSnapshot } from "./status-widget.js";
+
+function makeSnapshot(overrides: Partial<TeamActivityStatusSnapshot> = {}): TeamActivityStatusSnapshot {
+  return {
+    activeCount: 0,
+    readCount: 0,
+    writeCount: 0,
+    unreadCount: 0,
+    entries: [],
+    updatedAt: Date.UTC(2026, 0, 1, 0, 0, 0),
+    ...overrides,
+  };
+}
+
+describe("agent activity status widget", () => {
+  it("renders open snapshots without a collapse hint", () => {
+    const entries: TeamActivityStatusEntry[] = [{
+      name: "reader",
+      role: "read",
+      status: "thinking",
+      detail: "provider/model · 12 tok · waiting for model response",
+    }];
+    const snapshot = makeSnapshot({
+      activeCount: 1,
+      readCount: 1,
+      entries,
+      statusCounts: { thinking: 1 },
+    });
+
+    const lines = teamActivityStatusWidget(() => snapshot, () => false).render(120);
+    const rendered = lines.join("\n");
+
+    expect(lines[0]).toContain("agent activity");
+    expect(lines.at(-1)).toContain("─");
+    expect(rendered).toContain("1 active");
+    expect(rendered).toContain("1 read");
+    expect(rendered).toContain("↓ navigate");
+    expect(rendered).toContain("reader");
+    expect(rendered).toContain("thinking");
+    expect(rendered).not.toContain("collapse");
+    expect(rendered).not.toContain("ctrl+o");
+  });
+
+  it("fades the old progress and reveals the new phrase within one second", () => {
+    vi.useFakeTimers();
+    vi.setSystemTime(0);
+    try {
+      let snapshot = makeSnapshot({
+        activeCount: 1,
+        readCount: 1,
+        entries: [{
+          name: "reader",
+          role: "read",
+          displayText: "(reader) model/high · reading-fast · 1s · 10 tok · Collecting files.",
+        }],
+      });
+      const requestRender = vi.fn();
+      const widget = teamActivityStatusWidget(() => snapshot, () => false, requestRender);
+      expect(widget.render(120).join("\n")).toContain("Collecting files.");
+
+      snapshot = makeSnapshot({
+        activeCount: 1,
+        readCount: 1,
+        entries: [{
+          name: "reader",
+          role: "read",
+          displayText: "(reader) model/high · reading-fast · 2s · 12 tok · Reviewing results.",
+        }],
+      });
+      const fading = widget.render(120).join("\n");
+      expect(fading).toContain("Collecting files");
+      expect(fading).not.toContain("Reviewing results");
+
+      vi.advanceTimersByTime(650);
+      const revealing = widget.render(120).join("\n");
+      expect(revealing).toContain("Reviewing");
+      expect(revealing).not.toContain("Reviewing results.");
+
+      vi.advanceTimersByTime(350);
+      expect(widget.render(120).join("\n")).toMatch(/Reviewing results\.{1,3}/);
+      expect(requestRender).toHaveBeenCalled();
+      widget.dispose();
+    } finally {
+      vi.useRealTimers();
+    }
+  });
+
+  it("updates a context-only suffix immediately instead of animating it as progress", () => {
+    let snapshot = makeSnapshot({
+      activeCount: 1,
+      readCount: 1,
+      entries: [{
+        name: "reader",
+        role: "read",
+        displayText: "(reader) model/high · read-review · 1s · 46k tok (23%)",
+      }],
+    });
+    const widget = teamActivityStatusWidget(() => snapshot, () => false);
+
+    expect(widget.render(120).join("\n")).toContain("46k tok (23%)");
+    snapshot = makeSnapshot({
+      activeCount: 1,
+      readCount: 1,
+      updatedAt: snapshot.updatedAt + 1,
+      entries: [{
+        name: "reader",
+        role: "read",
+        displayText: "(reader) model/high · read-review · 2s · 80k tok (40%)",
+      }],
+    });
+    const updated = widget.render(120).join("\n");
+
+    expect(updated).toContain("80k tok (40%)");
+    expect(updated).not.toContain("46k tok (23%)");
+    widget.dispose();
+  });
+
+  it("keeps expanded rendering bounded while surfacing the aggregate summary", () => {
+    const entries = Array.from({ length: 12 }, (_, index) => ({
+      name: `reader-${index}`,
+      role: "read" as const,
+      status: index < 8 ? "thinking" : "working",
+      detail: `detail-${index}`,
+    }));
+    const snapshot = makeSnapshot({
+      activeCount: 12,
+      readCount: 12,
+      entries,
+      statusCounts: { thinking: 8, working: 4 },
+    });
+
+    const rendered = teamActivityStatusWidget(() => snapshot, () => true).render(160).join("\n");
+
+    expect(rendered).toContain("8 thinking");
+    expect(rendered).toContain("4 working");
+    expect(rendered).toContain("reader-0");
+    expect(rendered).toContain("reader-9");
+    expect(rendered).not.toContain("reader-10");
+    expect(rendered).not.toContain("detail-10");
+    expect(rendered).toContain("2 more active agents");
+  });
+});