pi-crew 0.1.12 → 0.1.13

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-crew",
-  "version": "0.1.12",
+  "version": "0.1.13",
   "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

@@ -108,9 +108,18 @@ 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 `⚙ pi-crew · ${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[] {
@@ -119,15 +128,12 @@ export function buildCrewWidgetLines(cwd: string, frame = 0, maxLines = 8): stri
 	const runningGlyph = SPINNER[frame % SPINNER.length] ?? "⠋";
 	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");
+		const completed = agents.filter((agent) => agent.status === "completed").length;
+		lines.push(`${glyph(run.status, runningGlyph)} ${shortRunLabel(run)} · ${completed}/${agents.length} done · ${run.runId.slice(-8)}`);
 		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}` : ""}`);
+			lines.push(`  ${glyph(agent.status, runningGlyph)} ${agent.agent} (${agent.role}) · ${agentActivity(agent)}${stats ? ` · ${stats}` : ""}`);
 		}
 		if (activeAgents.length > 3) lines.push(`  … +${activeAgents.length - 3} more agents`);
 		if (lines.length >= maxLines) break;
@@ -153,7 +159,7 @@ class CrewWidgetComponent implements WidgetComponent {
 		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("⚙ 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);
 		});