pi-crew 0.1.12 → 0.1.14

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-crew",
-  "version": "0.1.12",
+  "version": "0.1.14",
   "description": "Pi extension for coordinated AI teams, workflows, worktrees, and async task orchestration",
   "author": "baphuongna",
   "license": "MIT",

package/src/extension/register.ts

@@ -100,15 +100,38 @@ function setNestedConfig(config: Record<string, unknown>, key: string, value: un
 }
 
 export function registerPiTeams(pi: ExtensionAPI): void {
+	const globalStore = globalThis as Record<string, unknown>;
+	const runtimeCleanupStoreKey = "__piCrewRuntimeCleanup";
+	const previousRuntimeCleanup = globalStore[runtimeCleanupStoreKey];
+	if (typeof previousRuntimeCleanup === "function") {
+		try { previousRuntimeCleanup(); } catch {}
+	}
 	const notifierState: AsyncNotifierState = { seenFinishedRunIds: new Set() };
 	let currentCtx: ExtensionContext | undefined;
 	let rpcHandle: PiCrewRpcHandle | undefined;
+	let cleanedUp = false;
 	const widgetState: CrewWidgetState = { frame: 0 };
 	const foregroundControllers = new Set<AbortController>();
 	registerAutonomousPolicy(pi);
 	rpcHandle = registerPiCrewRpc((pi as unknown as { events?: Parameters<typeof registerPiCrewRpc>[0] }).events, () => currentCtx);
+	const cleanupRuntime = (): void => {
+		if (cleanedUp) return;
+		cleanedUp = true;
+		for (const controller of foregroundControllers) controller.abort();
+		foregroundControllers.clear();
+		terminateActiveChildPiProcesses();
+		stopAsyncRunNotifier(notifierState);
+		stopCrewWidget(currentCtx, widgetState, currentCtx ? loadConfig(currentCtx.cwd).config.ui : undefined);
+		clearPiCrewPowerbar(pi.events);
+		rpcHandle?.unsubscribe();
+		rpcHandle = undefined;
+		currentCtx = undefined;
+		if (globalStore[runtimeCleanupStoreKey] === cleanupRuntime) delete globalStore[runtimeCleanupStoreKey];
+	};
+	globalStore[runtimeCleanupStoreKey] = cleanupRuntime;
 
 	pi.on("session_start", (_event, ctx) => {
+		cleanedUp = false;
 		currentCtx = ctx;
 		notifyActiveRuns(ctx);
 		const loadedConfig = loadConfig(ctx.cwd);
@@ -125,14 +148,7 @@ export function registerPiTeams(pi: ExtensionAPI): void {
 		widgetState.interval.unref?.();
 	});
 	pi.on("session_shutdown", () => {
-		for (const controller of foregroundControllers) controller.abort();
-		terminateActiveChildPiProcesses();
-		stopAsyncRunNotifier(notifierState);
-		stopCrewWidget(currentCtx, widgetState, currentCtx ? loadConfig(currentCtx.cwd).config.ui : undefined);
-		clearPiCrewPowerbar(pi.events);
-		currentCtx = undefined;
-		rpcHandle?.unsubscribe();
-		rpcHandle = undefined;
+		cleanupRuntime();
 	});
 
 	const tool: ToolDefinition = {

package/src/runtime/team-runner.ts

@@ -160,6 +160,7 @@ export async function executeTeamRun(input: ExecuteTeamRunInput): Promise<{ mani
 		if (failed) {
 			tasks = markBlocked(tasks, `Blocked by failed task '${failed.id}'.`);
 			saveRunTasks(manifest, tasks);
+			saveCrewAgents(manifest, tasks.map((task) => recordFromTask(manifest, task, runtimeKind)));
 			manifest = updateRunStatus(manifest, "failed", `Failed at task '${failed.id}'.`);
 			return { manifest, tasks };
 		}
@@ -169,6 +170,7 @@ export async function executeTeamRun(input: ExecuteTeamRunInput): Promise<{ mani
 		if (readyBatch.length === 0) {
 			tasks = markBlocked(tasks, "No ready queued task; dependency graph may be invalid.");
 			saveRunTasks(manifest, tasks);
+			saveCrewAgents(manifest, tasks.map((task) => recordFromTask(manifest, task, runtimeKind)));
 			manifest = updateRunStatus(manifest, "blocked", "No ready queued task.");
 			return { manifest, tasks };
 		}

package/src/ui/crew-widget.ts

@@ -17,6 +17,9 @@ const TOOL_LABELS: Record<string, string> = {
 	ls: "listing",
 };
 const ANSI_PATTERN = /\u001b\[[0-?]*[ -/]*[@-~]/g;
+const LEGACY_WIDGET_KEY = "pi-crew";
+const WIDGET_KEY = "pi-crew-active";
+const STATUS_KEY = "pi-crew";
 
 type ThemeLike = { fg?: (color: string, text: string) => string; bold?: (text: string) => string };
 type WidgetComponent = { render(width: number): string[]; invalidate(): void };
@@ -108,33 +111,62 @@ function activeWidgetRuns(cwd: string): WidgetRun[] {
 }
 
 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`;
+	const agents = runs.flatMap((item) => item.agents);
+	const runningAgents = agents.filter((agent) => agent.status === "running").length;
+	const queuedAgents = agents.filter((agent) => agent.status === "queued").length;
+	const completedAgents = agents.filter((agent) => agent.status === "completed").length;
+	const parts = [`${runningAgents} running`];
+	if (queuedAgents) parts.push(`${queuedAgents} queued`);
+	if (completedAgents) parts.push(`${completedAgents}/${agents.length} done`);
+	return `Crew: ${parts.join(", ")}`;
+}
+
+function widgetHeader(runs: WidgetRun[], runningGlyph: string): string {
+	const agents = runs.flatMap((item) => item.agents);
+	const runningAgents = agents.filter((agent) => agent.status === "running").length;
+	const queuedAgents = agents.filter((agent) => agent.status === "queued").length;
+	const completedAgents = agents.filter((agent) => agent.status === "completed").length;
+	const parts = [`${runningAgents} running`];
+	if (queuedAgents) parts.push(`${queuedAgents} queued`);
+	if (completedAgents) parts.push(`${completedAgents}/${agents.length} done`);
+	return `${runningGlyph} Crew agents · ${parts.join(" · ")} · /team-dashboard`;
+}
+
+function shortRunLabel(run: TeamRunManifest): string {
+	return `${run.team}/${run.workflow ?? "none"}`;
 }
 
 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[] = [statusSummary(runs)];
+	const lines: string[] = [widgetHeader(runs, runningGlyph)];
 	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 completed = agents.filter((agent) => agent.status === "completed").length;
+		const runGlyph = glyph(run.status, runningGlyph);
+		lines.push(`├─ ${runGlyph} ${shortRunLabel(run)} · ${completed}/${agents.length} done · ${run.runId.slice(-8)}`);
+		const visibleAgents = activeAgents.slice(0, 3);
+		for (const [index, agent] of visibleAgents.entries()) {
+			const last = index === visibleAgents.length - 1 && activeAgents.length <= 3;
+			const branch = last ? "└─" : "├─";
 			const stats = agentStats(agent);
-			lines.push(`  ${glyph(agent.status, runningGlyph)} ${agent.taskId} ${agent.role}→${agent.agent} · ${agentActivity(agent)}${stats ? ` · ${stats}` : ""}`);
+			lines.push(`│  ${branch} ${glyph(agent.status, runningGlyph)} ${agent.agent} · ${agent.role}`);
+			lines.push(`│     ⎿ ${agentActivity(agent)}${stats ? ` · ${stats}` : ""}`);
 		}
-		if (activeAgents.length > 3) lines.push(`  … +${activeAgents.length - 3} more agents`);
+		if (activeAgents.length > 3) lines.push(`│  └─ … +${activeAgents.length - 3} more agents`);
 		if (lines.length >= maxLines) break;
 	}
 	return lines.slice(0, maxLines);
 }
 
+function colorWidgetLine(line: string, index: number, theme: ThemeLike): string {
+	const fg = theme.fg?.bind(theme) ?? ((_color: string, text: string) => text);
+	const bold = theme.bold?.bind(theme) ?? ((text: string) => text);
+	if (index === 0) return line.replace("Crew agents", bold(fg("accent", "Crew agents")));
+	return line.replace(/([⠋⠙⠹⠸⠼⠴⠦⠧⠇⠏▶◦✓✗■·])/, (icon: string) => fg(icon === "✓" ? "success" : icon === "✗" ? "error" : icon === "◦" ? "dim" : "accent", icon));
+}
+
 class CrewWidgetComponent implements WidgetComponent {
 	private cwd: string;
 	private frame: number;
@@ -149,35 +181,39 @@ class CrewWidgetComponent implements WidgetComponent {
 	}
 	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);
-		});
+		return buildCrewWidgetLines(this.cwd, this.frame, this.maxLines).map((line, index) => truncate(colorWidgetLine(line, index, this.theme), width));
 	}
 }
 
+function requestRender(ctx: Pick<ExtensionContext, "ui">): void {
+	(ctx.ui as { requestRender?: () => void }).requestRender?.();
+}
+
 export function updateCrewWidget(ctx: Pick<ExtensionContext, "cwd" | "hasUI" | "ui">, state: CrewWidgetState, config?: CrewUiConfig): void {
 	if (!ctx.hasUI) return;
 	state.frame += 1;
-	const maxLines = config?.widgetMaxLines ?? 8;
+	const maxLines = config?.widgetMaxLines ?? 10;
 	const lines = buildCrewWidgetLines(ctx.cwd, state.frame, maxLines);
-	ctx.ui.setStatus("pi-crew", lines.length ? lines[0] : undefined);
+	const placement = config?.widgetPlacement ?? "aboveEditor";
+	ctx.ui.setStatus(STATUS_KEY, lines.length ? statusSummary(activeWidgetRuns(ctx.cwd)) : undefined);
+	ctx.ui.setWidget(LEGACY_WIDGET_KEY, undefined, { placement });
 	if (!lines.length) {
-		ctx.ui.setWidget("pi-crew", undefined, { placement: config?.widgetPlacement ?? "aboveEditor" });
+		ctx.ui.setWidget(WIDGET_KEY, undefined, { placement });
+		requestRender(ctx);
 		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" });
+	ctx.ui.setWidget(WIDGET_KEY, ((_tui: unknown, theme: unknown) => new CrewWidgetComponent(ctx.cwd, state.frame, maxLines, theme as ThemeLike)) as never, { placement });
+	requestRender(ctx);
 }
 
 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: config?.widgetPlacement ?? "aboveEditor" });
+		const placement = config?.widgetPlacement ?? "aboveEditor";
+		ctx.ui.setStatus(STATUS_KEY, undefined);
+		ctx.ui.setWidget(LEGACY_WIDGET_KEY, undefined, { placement });
+		ctx.ui.setWidget(WIDGET_KEY, undefined, { placement });
+		requestRender(ctx);
 	}
 }