@melihmucuk/pi-crew 1.0.17 → 1.0.18

package/extension/runtime/owner-session-coordinator.ts

@@ -1,138 +0,0 @@
-import type { SendMessageFn } from "../message-delivery-policy.js";
-import {
-	type SteeringPayload,
-	sendRemainingNote,
-	sendSteeringMessage,
-} from "../subagent-messages.js";
-
-export interface ActiveRuntimeBinding {
-	sessionId: string;
-	isIdle: () => boolean;
-	sendMessage: SendMessageFn;
-}
-
-interface PendingMessage {
-	ownerSessionId: string;
-	payload: SteeringPayload;
-	queuedAt: number;
-}
-
-interface OwnerSessionCoordinatorDeps {
-	countRunningForOwner: (ownerSessionId: string, excludeId: string) => number;
-	onRefreshOwnerSession: (ownerSessionId: string) => void;
-	now?: () => number;
-	scheduleFlush?: (callback: () => void) => void;
-}
-
-const PENDING_MESSAGE_TTL_MS = 86_400_000;
-
-export class OwnerSessionCoordinator {
-	private binding: ActiveRuntimeBinding | undefined;
-	private pendingMessages: PendingMessage[] = [];
-	private flushScheduled = false;
-	private readonly countRunningForOwner: (ownerSessionId: string, excludeId: string) => number;
-	private readonly onRefreshOwnerSession: (ownerSessionId: string) => void;
-	private readonly now: () => number;
-	private readonly scheduleFlush: (callback: () => void) => void;
-
-	constructor(deps: OwnerSessionCoordinatorDeps) {
-		this.countRunningForOwner = deps.countRunningForOwner;
-		this.onRefreshOwnerSession = deps.onRefreshOwnerSession;
-		this.now = deps.now ?? Date.now;
-		this.scheduleFlush = deps.scheduleFlush ?? ((callback) => setTimeout(callback, 0));
-	}
-
-	activateSession(binding: ActiveRuntimeBinding): void {
-		this.binding = binding;
-
-		// Delay flush to next macrotask. session_start fires before pi-core
-		// calls _reconnectToAgent(), so synchronous delivery would emit agent
-		// events while the session listener is disconnected, losing JSONL persistence.
-		if (this.pendingMessages.some((entry) => entry.ownerSessionId === binding.sessionId)) {
-			this.flushScheduled = true;
-			this.scheduleFlush(() => {
-				this.flushScheduled = false;
-				this.flushPending();
-			});
-		}
-	}
-
-	deactivateSession(sessionId: string): void {
-		if (this.binding?.sessionId === sessionId) {
-			this.binding = undefined;
-		}
-	}
-
-	refresh(ownerSessionId: string): void {
-		this.onRefreshOwnerSession(ownerSessionId);
-	}
-
-	deliver(ownerSessionId: string, payload: SteeringPayload): void {
-		if (!this.binding || ownerSessionId !== this.binding.sessionId || this.flushScheduled) {
-			this.queue(ownerSessionId, payload);
-			return;
-		}
-
-		this.send(ownerSessionId, payload);
-	}
-
-	private queue(ownerSessionId: string, payload: SteeringPayload): void {
-		this.pendingMessages.push({ ownerSessionId, payload, queuedAt: this.now() });
-	}
-
-	private cleanStaleMessages(): void {
-		const cutoff = this.now() - PENDING_MESSAGE_TTL_MS;
-		this.pendingMessages = this.pendingMessages.filter(
-			(entry) => entry.queuedAt >= cutoff,
-		);
-	}
-
-	private flushPending(): void {
-		if (!this.binding) return;
-		const targetSessionId = this.binding.sessionId;
-
-		this.cleanStaleMessages();
-
-		const toDeliver: PendingMessage[] = [];
-		const remaining: PendingMessage[] = [];
-
-		for (const entry of this.pendingMessages) {
-			if (entry.ownerSessionId === targetSessionId) {
-				toDeliver.push(entry);
-			} else {
-				remaining.push(entry);
-			}
-		}
-
-		this.pendingMessages = remaining;
-
-		for (const entry of toDeliver) {
-			this.send(entry.ownerSessionId, entry.payload);
-		}
-	}
-
-	/**
-	 * Result messages always go first. If more subagents are still running and the
-	 * owner is idle, queue the result without triggering, then queue the separate
-	 * remaining note with triggerTurn so the next turn sees both in order.
-	 */
-	private send(ownerSessionId: string, payload: SteeringPayload): void {
-		if (!this.binding || this.binding.sessionId !== ownerSessionId) {
-			this.queue(ownerSessionId, payload);
-			return;
-		}
-
-		const remaining = this.countRunningForOwner(ownerSessionId, payload.id);
-		const isIdle = this.binding.isIdle();
-		const triggerResultTurn = !(isIdle && remaining > 0);
-
-		sendSteeringMessage(payload, this.binding.sendMessage, {
-			isIdle,
-			triggerTurn: triggerResultTurn,
-		});
-		sendRemainingNote(remaining, this.binding.sendMessage, {
-			isIdle,
-			triggerTurn: isIdle && remaining > 0,
-		});
-	}
-}

package/extension/runtime/subagent-lifecycle.ts

@@ -1,203 +0,0 @@
-import type { AgentMessage } from "@earendil-works/pi-agent-core";
-import type { AssistantMessage } from "@earendil-works/pi-ai";
-import type { AgentSession } from "@earendil-works/pi-coding-agent";
-import type { BootstrapContext } from "../bootstrap-session.js";
-import { bootstrapSession } from "../bootstrap-session.js";
-import type { SubagentStatus } from "../subagent-messages.js";
-import { runPromptWithOverflowRecovery } from "./overflow-recovery.js";
-import type { SubagentState } from "./subagent-registry.js";
-import { isAborted } from "./subagent-transitions.js";
-
-interface PromptOutcome {
-	status: Extract<SubagentStatus, "done" | "waiting" | "error" | "aborted">;
-	result?: string;
-	error?: string;
-}
-
-interface StartOptions {
-	cwd: string;
-	ctx: BootstrapContext;
-	extensionResolvedPath: string;
-	onWarning?: (message: string) => void;
-}
-
-interface SubagentLifecycleCallbacks {
-	isCurrent: (state: SubagentState) => boolean;
-	onProgress: (ownerSessionId: string) => void;
-	onSettled: (
-		state: SubagentState,
-		status: Extract<SubagentStatus, "done" | "waiting" | "error" | "aborted">,
-		outcome: { result?: string; error?: string },
-	) => void;
-}
-
-function getLastAssistantMessage(
-	messages: AgentMessage[],
-): AssistantMessage | undefined {
-	for (let i = messages.length - 1; i >= 0; i--) {
-		const msg = messages[i];
-		if (msg.role === "assistant") {
-			return msg as AssistantMessage;
-		}
-	}
-	return undefined;
-}
-
-function getAssistantText(
-	message: AssistantMessage | undefined,
-): string | undefined {
-	if (!message) return undefined;
-
-	const texts: string[] = [];
-	for (const part of message.content) {
-		if (part.type === "text") {
-			texts.push(part.text);
-		}
-	}
-
-	return texts.length > 0 ? texts.join("\n") : undefined;
-}
-
-function getPromptOutcome(state: SubagentState): PromptOutcome {
-	const lastAssistant = getLastAssistantMessage(state.session!.messages);
-	const text = getAssistantText(lastAssistant);
-
-	if (lastAssistant?.stopReason === "error") {
-		return {
-			status: "error",
-			error: lastAssistant.errorMessage ?? text ?? "(no output)",
-		};
-	}
-
-	if (lastAssistant?.stopReason === "aborted") {
-		return {
-			status: "aborted",
-			error: lastAssistant.errorMessage ?? text ?? "(no output)",
-		};
-	}
-
-	return {
-		status: state.agentConfig.interactive ? "waiting" : "done",
-		result: text ?? "(no output)",
-	};
-}
-
-export class SubagentLifecycle {
-	constructor(private readonly callbacks: SubagentLifecycleCallbacks) {}
-
-	start(state: SubagentState, opts: StartOptions): void {
-		void this.spawnSession(state, opts);
-	}
-
-	respond(state: SubagentState, message: string): void {
-		void this.runPromptCycle(state, message);
-	}
-
-	abortPrompt(state: SubagentState): void {
-		state.promptAbortController?.abort();
-		state.promptAbortController = undefined;
-		state.session?.abortCompaction();
-		state.session?.abortRetry();
-		state.session?.abort().catch(() => {});
-	}
-
-	private attachSessionListeners(
-		state: SubagentState,
-		session: AgentSession,
-	): void {
-		state.unsubscribe = session.subscribe((event) => {
-			if (event.type !== "turn_end") return;
-
-			state.turns++;
-			const msg = event.message;
-			if (msg.role === "assistant") {
-				const assistantMsg = msg as AssistantMessage;
-				state.contextTokens = assistantMsg.usage.totalTokens;
-				state.model = assistantMsg.model;
-			}
-			this.callbacks.onProgress(state.ownerSessionId);
-		});
-	}
-
-	private attachSpawnedSession(
-		state: SubagentState,
-		session: AgentSession,
-	): boolean {
-		if (!this.callbacks.isCurrent(state)) {
-			session.dispose();
-			return false;
-		}
-
-		state.session = session;
-		return true;
-	}
-
-	private async runPromptCycle(
-		state: SubagentState,
-		prompt: string,
-	): Promise<void> {
-		if (isAborted(state)) return;
-
-		const abortController = new AbortController();
-		state.promptAbortController = abortController;
-
-		try {
-			const recovery = await runPromptWithOverflowRecovery(
-				state.session!,
-				prompt,
-				abortController.signal,
-			);
-			if (isAborted(state)) return;
-
-			const outcome = getPromptOutcome(state);
-
-			if (recovery === "failed" && outcome.status !== "error") {
-				this.callbacks.onSettled(state, "error", {
-					error: "Context overflow recovery failed",
-				});
-				return;
-			}
-
-			this.callbacks.onSettled(state, outcome.status, outcome);
-		} catch (err) {
-			if (isAborted(state)) return;
-
-			const error = err instanceof Error ? err.message : String(err);
-			this.callbacks.onSettled(state, "error", { error });
-		} finally {
-			state.promptAbortController = undefined;
-		}
-	}
-
-	private async spawnSession(
-		state: SubagentState,
-		opts: StartOptions,
-	): Promise<void> {
-		try {
-			if (isAborted(state)) return;
-
-			const { session, warnings } = await bootstrapSession({
-				agentConfig: state.agentConfig,
-				cwd: opts.cwd,
-				ctx: opts.ctx,
-				extensionResolvedPath: opts.extensionResolvedPath,
-			});
-
-			for (const warning of warnings) {
-				opts.onWarning?.(warning);
-			}
-
-			if (!this.attachSpawnedSession(state, session)) return;
-
-			this.attachSessionListeners(state, session);
-			await this.runPromptCycle(state, state.task);
-		} catch (err) {
-			if (isAborted(state)) return;
-
-			if (state.status === "running") {
-				const error = err instanceof Error ? err.message : String(err);
-				this.callbacks.onSettled(state, "error", { error });
-			}
-		}
-	}
-}

package/extension/runtime/subagent-registry.ts

@@ -1,122 +0,0 @@
-import { randomBytes } from "node:crypto";
-import type { AgentSession } from "@earendil-works/pi-coding-agent";
-import type { AgentConfig } from "../agent-discovery.js";
-import type { SubagentStatus } from "../subagent-messages.js";
-import { isAbortableStatus } from "./subagent-transitions.js";
-
-export interface SubagentState {
-	id: string;
-	agentConfig: AgentConfig;
-	task: string;
-	status: SubagentStatus;
-	ownerSessionId: string;
-	session: AgentSession | null;
-	turns: number;
-	contextTokens: number;
-	model: string | undefined;
-	error?: string;
-	result?: string;
-	promptAbortController?: AbortController;
-	unsubscribe?: () => void;
-}
-
-export interface ActiveAgentSummary {
-	id: string;
-	agentName: string;
-	status: SubagentStatus;
-	turns: number;
-	contextTokens: number;
-	model: string | undefined;
-}
-
-function generateId(name: string, existingIds: Set<string>): string {
-	for (let i = 0; i < 10; i++) {
-		const id = `${name}-${randomBytes(4).toString("hex")}`;
-		if (!existingIds.has(id)) return id;
-	}
-	return `${name}-${randomBytes(8).toString("hex")}`;
-}
-
-function buildActiveAgentSummary(
-	state: SubagentState,
-): ActiveAgentSummary {
-	return {
-		id: state.id,
-		agentName: state.agentConfig.name,
-		status: state.status,
-		turns: state.turns,
-		contextTokens: state.contextTokens,
-		model: state.model,
-	};
-}
-
-export class SubagentRegistry {
-	private activeAgents = new Map<string, SubagentState>();
-
-	create(agentConfig: AgentConfig, task: string, ownerSessionId: string): SubagentState {
-		const id = generateId(agentConfig.name, new Set(this.activeAgents.keys()));
-		const state: SubagentState = {
-			id,
-			agentConfig,
-			task,
-			status: "running",
-			ownerSessionId,
-			session: null,
-			turns: 0,
-			contextTokens: 0,
-			model: undefined,
-		};
-
-		this.activeAgents.set(id, state);
-		return state;
-	}
-
-	get(id: string): SubagentState | undefined {
-		return this.activeAgents.get(id);
-	}
-
-	hasState(state: SubagentState): boolean {
-		return this.activeAgents.get(state.id) === state;
-	}
-
-	delete(id: string): void {
-		this.activeAgents.delete(id);
-	}
-
-	countRunningForOwner(ownerSessionId: string, excludeId: string): number {
-		let count = 0;
-		for (const state of this.activeAgents.values()) {
-			if (
-				state.id !== excludeId &&
-				state.ownerSessionId === ownerSessionId &&
-				state.status === "running"
-			) {
-				count++;
-			}
-		}
-		return count;
-	}
-
-	getActiveSummariesForOwner(ownerSessionId: string): ActiveAgentSummary[] {
-		return Array.from(this.activeAgents.values())
-			.filter(
-				(state) => isAbortableStatus(state.status) && state.ownerSessionId === ownerSessionId,
-			)
-			.map(buildActiveAgentSummary);
-	}
-
-	getOwnedAbortableIds(ownerSessionId: string): string[] {
-		return Array.from(this.activeAgents.values())
-			.filter(
-				(state) =>
-					state.ownerSessionId === ownerSessionId && isAbortableStatus(state.status),
-			)
-			.map((state) => state.id);
-	}
-
-	getAllAbortable(): SubagentState[] {
-		return Array.from(this.activeAgents.values()).filter((state) =>
-			isAbortableStatus(state.status),
-		);
-	}
-}

package/extension/runtime/subagent-transitions.ts

@@ -1,100 +0,0 @@
-import type { SubagentStatus } from "../subagent-messages.js";
-import type { SubagentState } from "./subagent-registry.js";
-
-export type SettledSubagentStatus = Extract<
-	SubagentStatus,
-	"done" | "waiting" | "error" | "aborted"
->;
-
-export interface SubagentTransitionOutcome {
-	result?: string;
-	error?: string;
-}
-
-type SubagentTransitionResult =
-	| { ok: true; state: SubagentState }
-	| { ok: false; error: string };
-
-export function isAborted(state: SubagentState): boolean {
-	return state.status === "aborted";
-}
-
-export function isAbortableStatus(status: SubagentStatus): boolean {
-	return status === "running" || status === "waiting";
-}
-
-export function canAbortSubagent(
-	state: SubagentState | undefined,
-): state is SubagentState {
-	return Boolean(state && isAbortableStatus(state.status));
-}
-
-function validateOwnedSubagent(
-	state: SubagentState | undefined,
-	id: string,
-	callerSessionId: string,
-	missingMessage: string,
-): SubagentTransitionResult {
-	if (!state) return { ok: false, error: missingMessage };
-	if (state.ownerSessionId !== callerSessionId) {
-		return { ok: false, error: `Subagent "${id}" belongs to a different session` };
-	}
-	return { ok: true, state };
-}
-
-export function startSubagentResponse(
-	state: SubagentState | undefined,
-	id: string,
-	callerSessionId: string,
-): SubagentTransitionResult {
-	const owned = validateOwnedSubagent(
-		state,
-		id,
-		callerSessionId,
-		`No subagent with id "${id}"`,
-	);
-	if (!owned.ok) return owned;
-
-	if (owned.state.status !== "waiting") {
-		return {
-			ok: false,
-			error: `Subagent "${id}" is not waiting for a response (status: ${owned.state.status})`,
-		};
-	}
-	if (!owned.state.session) {
-		return { ok: false, error: `Subagent "${id}" has no active session` };
-	}
-
-	owned.state.status = "running";
-	return owned;
-}
-
-export function validateSubagentDone(
-	state: SubagentState | undefined,
-	id: string,
-	callerSessionId: string,
-): SubagentTransitionResult {
-	const owned = validateOwnedSubagent(
-		state,
-		id,
-		callerSessionId,
-		`No active subagent with id "${id}"`,
-	);
-	if (!owned.ok) return owned;
-
-	if (owned.state.status !== "waiting") {
-		return { ok: false, error: `Subagent "${id}" is not in waiting state` };
-	}
-
-	return owned;
-}
-
-export function settleSubagent(
-	state: SubagentState,
-	status: SettledSubagentStatus,
-	outcome: SubagentTransitionOutcome,
-): void {
-	state.status = status;
-	state.result = outcome.result;
-	state.error = outcome.error;
-}

package/extension/status-widget.ts

@@ -1,107 +0,0 @@
-import type { ExtensionContext } from "@earendil-works/pi-coding-agent";
-import { Text } from "@earendil-works/pi-tui";
-import type { ActiveAgentSummary } from "./runtime/crew-runtime.js";
-import type { CrewRuntime } from "./runtime/crew-runtime.js";
-
-const SPINNER_FRAMES = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"];
-const SPINNER_INTERVAL_MS = 80;
-
-function formatTokens(tokens: number): string {
-	if (tokens >= 1_000_000) return `${(tokens / 1_000_000).toFixed(1)}M`;
-	if (tokens >= 1_000) return `${(tokens / 1_000).toFixed(1)}k`;
-	return String(tokens);
-}
-
-function buildLine(agent: ActiveAgentSummary, frame: string): string {
-	const model = agent.model ?? "…";
-	const icon = agent.status === "waiting" ? "⏳" : frame;
-	return `${icon} ${agent.id} (${model}) · turn ${agent.turns} · ${formatTokens(agent.contextTokens)} ctx`;
-}
-
-interface WidgetState {
-	ctx: ExtensionContext;
-	text: Text;
-	// biome-ignore lint: TUI type from factory param
-	tui: any;
-	timer: ReturnType<typeof setInterval>;
-	frameIndex: number;
-}
-
-let widget: WidgetState | undefined;
-
-function disposeWidget(state: WidgetState): void {
-	clearInterval(state.timer);
-	if (widget === state) {
-		widget = undefined;
-	}
-}
-
-function clearWidget(): void {
-	const current = widget;
-	if (!current) return;
-	disposeWidget(current);
-	current.ctx.ui.setWidget("crew-status", undefined);
-}
-
-function hasRunningAgent(agents: ActiveAgentSummary[]): boolean {
-	return agents.some((agent) => agent.status === "running");
-}
-
-function syncWidgetText(state: WidgetState, agents: ActiveAgentSummary[]): void {
-	const frame = SPINNER_FRAMES[state.frameIndex % SPINNER_FRAMES.length];
-	const lines = agents.map((agent) => buildLine(agent, frame));
-	state.text.setText(lines.join("\n"));
-	state.tui.requestRender();
-}
-
-export function updateWidget(ctx: ExtensionContext, crew: CrewRuntime): void {
-	if (!ctx.hasUI) {
-		clearWidget();
-		return;
-	}
-
-	const ownerSessionId = ctx.sessionManager.getSessionId();
-	const running = crew.getActiveSummariesForOwner(ownerSessionId);
-	if (running.length === 0) {
-		clearWidget();
-		return;
-	}
-
-	if (widget && widget.ctx !== ctx) {
-		clearWidget();
-	}
-
-	if (widget) {
-		syncWidgetText(widget, running);
-		return;
-	}
-
-	ctx.ui.setWidget("crew-status", (tui, _theme) => {
-		const text = new Text("", 1, 0);
-		const state: WidgetState = {
-			ctx,
-			text,
-			tui,
-			frameIndex: 0,
-			timer: setInterval(() => {
-				const agents = crew.getActiveSummariesForOwner(ownerSessionId);
-				if (agents.length === 0) {
-					clearWidget();
-					return;
-				}
-				if (!hasRunningAgent(agents)) return;
-				state.frameIndex++;
-				syncWidgetText(state, agents);
-			}, SPINNER_INTERVAL_MS),
-		};
-
-		widget = state;
-		syncWidgetText(state, running);
-
-		return Object.assign(text, {
-			dispose() {
-				disposeWidget(state);
-			},
-		});
-	});
-}