pi-crew 0.1.10 → 0.1.12

package/src/ui/crew-widget.ts

@@ -1,8 +1,9 @@
 import type { ExtensionContext } from "@mariozechner/pi-coding-agent";
+import type { CrewUiConfig } from "../config/config.ts";
 import { listRuns } from "../extension/run-index.ts";
-import { isDisplayActiveRun, isLikelyOrphanedActiveRun } from "../runtime/process-status.ts";
 import { readCrewAgents } from "../runtime/crew-agent-records.ts";
 import type { CrewAgentRecord } from "../runtime/crew-agent-runtime.ts";
+import { isDisplayActiveRun } from "../runtime/process-status.ts";
 import type { TeamRunManifest } from "../state/types.ts";
 
 const SPINNER = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"];
@@ -15,12 +16,48 @@ const TOOL_LABELS: Record<string, string> = {
 	find: "finding files",
 	ls: "listing",
 };
+const ANSI_PATTERN = /\u001b\[[0-?]*[ -/]*[@-~]/g;
+
+type ThemeLike = { fg?: (color: string, text: string) => string; bold?: (text: string) => string };
+type WidgetComponent = { render(width: number): string[]; invalidate(): void };
 
 export interface CrewWidgetState {
 	frame: number;
 	interval?: ReturnType<typeof setInterval>;
 }
 
+interface WidgetRun {
+	run: TeamRunManifest;
+	agents: CrewAgentRecord[];
+}
+
+function visibleWidth(value: string): number {
+	return value.replace(ANSI_PATTERN, "").length;
+}
+
+function truncate(value: string, width: number): string {
+	if (width <= 0) return "";
+	if (visibleWidth(value) <= width) return value;
+	if (width <= 1) return "…";
+	let output = "";
+	let visible = 0;
+	for (let index = 0; index < value.length;) {
+		const slice = value.slice(index);
+		const ansi = slice.match(/^\u001b\[[0-?]*[ -/]*[@-~]/);
+		if (ansi?.[0]) {
+			output += ansi[0];
+			index += ansi[0].length;
+			continue;
+		}
+		const char = value[index]!;
+		if (visible >= width - 1) break;
+		output += char;
+		visible += 1;
+		index += char.length;
+	}
+	return `${output}\u001b[0m…`;
+}
+
 function elapsed(iso: string | undefined, now = Date.now()): string | undefined {
 	if (!iso) return undefined;
 	const ms = Math.max(0, now - new Date(iso).getTime());
@@ -45,7 +82,7 @@ function agentActivity(agent: CrewAgentRecord): string {
 	const recent = agent.progress?.recentOutput?.at(-1);
 	if (recent) return recent.replace(/\s+/g, " ").trim();
 	if (agent.progress?.activityState === "needs_attention") return "needs attention";
-	if (agent.status === "queued") return "queued…";
+	if (agent.status === "queued") return "queued";
 	if (agent.status === "running") return "thinking…";
 	if (agent.status === "failed") return agent.error ?? "failed";
 	return "done";
@@ -61,59 +98,86 @@ function agentStats(agent: CrewAgentRecord): string {
 	return parts.join(" · ");
 }
 
-function runStep(run: TeamRunManifest, agents: CrewAgentRecord[]): string {
-	if (isLikelyOrphanedActiveRun(run, agents)) return "stale";
-	const running = agents.find((agent) => agent.status === "running");
-	if (running) return running.taskId;
-	const queued = agents.find((agent) => agent.status === "queued");
-	if (queued) return queued.taskId;
-	return run.status;
-}
-
 function agentsFor(run: TeamRunManifest): CrewAgentRecord[] {
 	try { return readCrewAgents(run); } catch { return []; }
 }
 
-export function buildCrewWidgetLines(cwd: string, frame = 0, maxLines = 8): string[] {
+function activeWidgetRuns(cwd: string): WidgetRun[] {
 	const runs = listRuns(cwd).slice(0, 20);
-	const runAgents = new Map(runs.map((run) => [run.runId, agentsFor(run)]));
-	const activeRuns = runs.filter((run) => isDisplayActiveRun(run, runAgents.get(run.runId) ?? []));
-	const recentRuns = runs.filter((run) => !isDisplayActiveRun(run, runAgents.get(run.runId) ?? [])).slice(0, 3);
-	const shownRuns = [...activeRuns, ...recentRuns];
-	if (!shownRuns.length) return [];
+	return runs.map((run) => ({ run, agents: agentsFor(run) })).filter((item) => isDisplayActiveRun(item.run, item.agents));
+}
+
+function statusSummary(runs: WidgetRun[]): string {
+	const runningAgents = runs.flatMap((item) => item.agents).filter((agent) => agent.status === "running").length;
+	const queuedAgents = runs.flatMap((item) => item.agents).filter((agent) => agent.status === "queued").length;
+	return `● pi-crew · runs=${runs.length} agents=${runningAgents} running${queuedAgents ? ` · ${queuedAgents} queued` : ""} · /team-dashboard`;
+}
+
+export function buildCrewWidgetLines(cwd: string, frame = 0, maxLines = 8): string[] {
+	const runs = activeWidgetRuns(cwd);
+	if (!runs.length) return [];
 	const runningGlyph = SPINNER[frame % SPINNER.length] ?? "⠋";
-	const lines: string[] = [];
-	const activeCount = activeRuns.length;
-	lines.push(`${activeCount ? "●" : "○"} pi-crew · active=${activeCount} recent=${recentRuns.length} · /team-dashboard`);
-	for (const run of shownRuns) {
-		const agents = runAgents.get(run.runId) ?? [];
-		const stale = isLikelyOrphanedActiveRun(run, agents);
-		const counts = new Map<string, number>();
-		for (const agent of agents) counts.set(agent.status, (counts.get(agent.status) ?? 0) + 1);
-		const countText = stale ? "stale queued run" : [...counts.entries()].map(([status, count]) => `${status}:${count}`).join(" ") || run.status;
-		lines.push(`${glyph(stale ? "failed" : run.status, runningGlyph)} ${run.runId.slice(-8)} ${run.team}/${run.workflow ?? "none"} · ${runStep(run, agents)} · ${countText}`);
-		for (const agent of (stale ? [] : agents.filter((item) => item.status === "running" || item.status === "queued").slice(0, 2))) {
+	const lines: string[] = [statusSummary(runs)];
+	for (const { run, agents } of runs) {
+		const running = agents.find((agent) => agent.status === "running");
+		const queued = agents.find((agent) => agent.status === "queued");
+		const step = running?.taskId ?? queued?.taskId ?? run.status;
+		const completed = agents.filter((agent) => agent.status === "completed").length;
+		lines.push(`${glyph(run.status, runningGlyph)} ${run.runId.slice(-8)} ${run.team}/${run.workflow ?? "none"} · ${step} · ${completed}/${agents.length} done`);
+		const activeAgents = agents.filter((item) => item.status === "running" || item.status === "queued");
+		for (const agent of activeAgents.slice(0, 3)) {
 			const stats = agentStats(agent);
 			lines.push(`  ${glyph(agent.status, runningGlyph)} ${agent.taskId} ${agent.role}→${agent.agent} · ${agentActivity(agent)}${stats ? ` · ${stats}` : ""}`);
 		}
+		if (activeAgents.length > 3) lines.push(`  … +${activeAgents.length - 3} more agents`);
 		if (lines.length >= maxLines) break;
 	}
 	return lines.slice(0, maxLines);
 }
 
-export function updateCrewWidget(ctx: Pick<ExtensionContext, "cwd" | "hasUI" | "ui">, state: CrewWidgetState): void {
+class CrewWidgetComponent implements WidgetComponent {
+	private cwd: string;
+	private frame: number;
+	private maxLines: number;
+	private theme: ThemeLike;
+
+	constructor(cwd: string, frame: number, maxLines: number, theme: ThemeLike) {
+		this.cwd = cwd;
+		this.frame = frame;
+		this.maxLines = maxLines;
+		this.theme = theme;
+	}
+	invalidate(): void {}
+	render(width: number): string[] {
+		const fg = this.theme.fg?.bind(this.theme) ?? ((_color: string, text: string) => text);
+		const bold = this.theme.bold?.bind(this.theme) ?? ((text: string) => text);
+		return buildCrewWidgetLines(this.cwd, this.frame, this.maxLines).map((line, index) => {
+			const colored = index === 0
+				? line.replace("● pi-crew", `${fg("accent", "●")} ${bold("pi-crew")}`)
+				: line.replace(/^\s*([⠋⠙⠹⠸⠼⠴⠦⠧⠇⠏▶◦✓✗■·])/, (match, icon: string) => match.replace(icon, fg(icon === "✓" ? "success" : icon === "✗" ? "error" : icon === "◦" ? "dim" : "accent", icon)));
+			return truncate(colored, width);
+		});
+	}
+}
+
+export function updateCrewWidget(ctx: Pick<ExtensionContext, "cwd" | "hasUI" | "ui">, state: CrewWidgetState, config?: CrewUiConfig): void {
 	if (!ctx.hasUI) return;
 	state.frame += 1;
-	const lines = buildCrewWidgetLines(ctx.cwd, state.frame);
+	const maxLines = config?.widgetMaxLines ?? 8;
+	const lines = buildCrewWidgetLines(ctx.cwd, state.frame, maxLines);
 	ctx.ui.setStatus("pi-crew", lines.length ? lines[0] : undefined);
-	ctx.ui.setWidget("pi-crew", lines.length ? lines : undefined, { placement: "aboveEditor" });
+	if (!lines.length) {
+		ctx.ui.setWidget("pi-crew", undefined, { placement: config?.widgetPlacement ?? "aboveEditor" });
+		return;
+	}
+	ctx.ui.setWidget("pi-crew", ((_tui: unknown, theme: unknown) => new CrewWidgetComponent(ctx.cwd, state.frame, maxLines, theme as ThemeLike)) as never, { placement: config?.widgetPlacement ?? "aboveEditor" });
 }
 
-export function stopCrewWidget(ctx: Pick<ExtensionContext, "hasUI" | "ui"> | undefined, state: CrewWidgetState): void {
+export function stopCrewWidget(ctx: Pick<ExtensionContext, "hasUI" | "ui"> | undefined, state: CrewWidgetState, config?: CrewUiConfig): void {
 	if (state.interval) clearInterval(state.interval);
 	state.interval = undefined;
 	if (ctx?.hasUI) {
 		ctx.ui.setStatus("pi-crew", undefined);
-		ctx.ui.setWidget("pi-crew", undefined, { placement: "aboveEditor" });
+		ctx.ui.setWidget("pi-crew", undefined, { placement: config?.widgetPlacement ?? "aboveEditor" });
 	}
 }

package/src/ui/powerbar-publisher.ts

@@ -0,0 +1,55 @@
+import type { CrewUiConfig } from "../config/config.ts";
+import { listRuns } from "../extension/run-index.ts";
+import { readCrewAgents } from "../runtime/crew-agent-records.ts";
+import { isDisplayActiveRun } from "../runtime/process-status.ts";
+
+type EventBus = { emit?: (event: string, data: unknown) => void } | undefined;
+
+function safeEmit(events: EventBus, event: string, data: unknown): void {
+	try { events?.emit?.(event, data); } catch {}
+}
+
+export function registerPiCrewPowerbarSegments(events: EventBus, config?: CrewUiConfig): void {
+	if (config?.powerbar === false) return;
+	safeEmit(events, "powerbar:register-segment", { id: "pi-crew-active", label: "pi-crew active agents" });
+	safeEmit(events, "powerbar:register-segment", { id: "pi-crew-progress", label: "pi-crew run progress" });
+}
+
+export function updatePiCrewPowerbar(events: EventBus, cwd: string, config?: CrewUiConfig): void {
+	if (config?.powerbar === false) return;
+	const active = listRuns(cwd).slice(0, 20).map((run) => {
+		let agents = [] as ReturnType<typeof readCrewAgents>;
+		try { agents = readCrewAgents(run); } catch {}
+		return { run, agents };
+	}).filter((item) => isDisplayActiveRun(item.run, item.agents));
+	if (!active.length) {
+		safeEmit(events, "powerbar:update", { id: "pi-crew-active" });
+		safeEmit(events, "powerbar:update", { id: "pi-crew-progress" });
+		return;
+	}
+	const agents = active.flatMap((item) => item.agents);
+	const running = agents.filter((agent) => agent.status === "running").length;
+	const queued = agents.filter((agent) => agent.status === "queued").length;
+	const completed = agents.filter((agent) => agent.status === "completed").length;
+	const total = Math.max(1, agents.length);
+	safeEmit(events, "powerbar:update", {
+		id: "pi-crew-active",
+		icon: "⚙",
+		text: `crew ${running || active.length}`,
+		suffix: queued ? `${queued}q` : undefined,
+		color: running ? "accent" : "warning",
+	});
+	safeEmit(events, "powerbar:update", {
+		id: "pi-crew-progress",
+		text: active[0]?.run.team ?? "crew",
+		bar: Math.round((completed / total) * 100),
+		suffix: `${completed}/${total}`,
+		color: completed === total ? "success" : "accent",
+		barSegments: 8,
+	});
+}
+
+export function clearPiCrewPowerbar(events: EventBus): void {
+	safeEmit(events, "powerbar:update", { id: "pi-crew-active" });
+	safeEmit(events, "powerbar:update", { id: "pi-crew-progress" });
+}

package/src/ui/run-dashboard.ts

@@ -10,6 +10,8 @@ interface DashboardComponent {
 	handleInput(data: string): void;
 }
 
+type DashboardTheme = { fg?: (color: string, text: string) => string; bold?: (text: string) => string };
+
 export type RunDashboardAction = "status" | "summary" | "artifacts" | "api" | "events" | "agents" | "agent-events" | "agent-output" | "agent-transcript" | "reload";
 export interface RunDashboardSelection {
 	runId: string;
@@ -49,6 +51,14 @@ function padVisible(value: string, width: number): string {
 	return `${value}${" ".repeat(Math.max(0, width - visibleLength(value)))}`;
 }
 
+function colorForStatus(status: string): string {
+	if (status === "completed") return "success";
+	if (status === "failed" || status === "stale") return "error";
+	if (status === "cancelled" || status === "blocked") return "warning";
+	if (status === "running") return "accent";
+	return "dim";
+}
+
 function statusIcon(status: string): string {
 	if (status === "completed") return "✓";
 	if (status === "failed" || status === "stale") return "✗";
@@ -143,23 +153,28 @@ export class RunDashboard implements DashboardComponent {
 	private showFullProgress = false;
 	private readonly runs: TeamRunManifest[];
 	private readonly done: (selection: RunDashboardSelection | undefined) => void;
+	private readonly theme: DashboardTheme;
 
-	constructor(runs: TeamRunManifest[], done: (selection: RunDashboardSelection | undefined) => void) {
+	constructor(runs: TeamRunManifest[], done: (selection: RunDashboardSelection | undefined) => void, theme: unknown = {}) {
 		this.runs = runs;
 		this.done = done;
+		this.theme = theme as DashboardTheme;
 	}
 
 	invalidate(): void {}
 
 	render(width: number): string[] {
+		const fg = this.theme.fg?.bind(this.theme) ?? ((_color: string, text: string) => text);
+		const bold = this.theme.bold?.bind(this.theme) ?? ((text: string) => text);
 		const innerWidth = Math.max(20, width - 4);
 		const borderWidth = Math.min(innerWidth, Math.max(0, width - 2));
+		const border = (text: string) => fg("border", text);
 		const lines = [
-			`╭${"─".repeat(borderWidth)}╮`,
-			`│ ${padVisible(truncate("pi-crew dashboard", innerWidth - 1), innerWidth - 1)}│`,
-			`│ ${padVisible(truncate("↑/↓/j/k select • r reload • p progress • s/u/a/i actions • d agents • e/v/o viewers • q close", innerWidth - 1), innerWidth - 1)}│`,
+			border(`╭${"─".repeat(borderWidth)}╮`),
+			`│ ${padVisible(truncate(`${fg("accent", "●")} ${bold("pi-crew dashboard")}`, innerWidth - 1), innerWidth - 1)}│`,
+			`│ ${padVisible(truncate(fg("dim", "↑/↓/j/k select • r reload • p progress • s/u/a/i actions • d agents • e/v/o viewers • q close"), innerWidth - 1), innerWidth - 1)}│`,
 			`│ ${padVisible(truncate(`Runs: ${this.runs.length} • ${countByStatus(this.runs)}`, innerWidth - 1), innerWidth - 1)}│`,
-			`├${"─".repeat(borderWidth)}┤`,
+			border(`├${"─".repeat(borderWidth)}┤`),
 		];
 		if (this.runs.length === 0) {
 			lines.push(`│ ${padVisible(truncate("No runs found.", innerWidth - 1), innerWidth - 1)}│`);
@@ -168,15 +183,17 @@ export class RunDashboard implements DashboardComponent {
 			const runRows = rows.filter((row) => row.run);
 			for (const row of rows) {
 				if (!row.run) {
-					lines.push(`│ ${padVisible(truncate(row.label, innerWidth - 1), innerWidth - 1)}│`);
+					lines.push(`│ ${padVisible(truncate(fg("accent", row.label), innerWidth - 1), innerWidth - 1)}│`);
 					continue;
 				}
 				const index = runRows.findIndex((candidate) => candidate.run?.runId === row.run?.runId);
-				lines.push(`│ ${padVisible(truncate(runLabel(row.run, index === this.selected), innerWidth - 1), innerWidth - 1)}│`);
+				const label = runLabel(row.run, index === this.selected);
+				const status = isLikelyOrphanedActiveRun(row.run, agentsFor(row.run)) ? "stale" : row.run.status;
+				lines.push(`│ ${padVisible(truncate(fg(colorForStatus(status), label), innerWidth - 1), innerWidth - 1)}│`);
 			}
 			const selectedRun = selectedRunFromGrouped(this.runs, this.selected);
 			if (selectedRun) {
-				lines.push(`├${"─".repeat(borderWidth)}┤`);
+				lines.push(border(`├${"─".repeat(borderWidth)}┤`));
 				const details = [
 					`Selected: ${selectedRun.runId}`,
 					`Status: ${selectedRun.status} | Team: ${selectedRun.team} | Workflow: ${selectedRun.workflow ?? "none"}`,
@@ -191,7 +208,7 @@ export class RunDashboard implements DashboardComponent {
 				}
 			}
 		}
-		lines.push(`╰${"─".repeat(borderWidth)}╯`);
+		lines.push(border(`╰${"─".repeat(borderWidth)}╯`));
 		return lines.map((line) => truncate(line, width));
 	}