pi-crew 0.1.30 → 0.1.32

package/src/state/jsonl-writer.ts

@@ -1,77 +1,77 @@
-import * as fs from "node:fs";
-
-export interface DrainableSource {
-	pause(): void;
-	resume(): void;
-}
-
-export interface JsonlWriteStream {
-	write(chunk: string): boolean;
-	once(event: "drain", listener: () => void): JsonlWriteStream;
-	end(callback?: () => void): void;
-}
-
-const DEFAULT_MAX_JSONL_BYTES = 50 * 1024 * 1024;
-
-export interface JsonlWriterDeps {
-	createWriteStream?: (filePath: string) => JsonlWriteStream;
-	maxBytes?: number;
-}
-
-export interface JsonlWriter {
-	writeLine(line: string): void;
-	close(): Promise<void>;
-}
-
-export function createJsonlWriter(filePath: string | undefined, source: DrainableSource, deps: JsonlWriterDeps = {}): JsonlWriter {
-	if (!filePath) {
-		return {
-			writeLine() {},
-			async close() {},
-		};
-	}
-
-	const createWriteStream = deps.createWriteStream ?? ((targetPath: string) => fs.createWriteStream(targetPath, { flags: "a" }));
-	let stream: JsonlWriteStream | undefined;
-	try {
-		stream = createWriteStream(filePath);
-	} catch {
-		return {
-			writeLine() {},
-			async close() {},
-		};
-	}
-
-	let backpressured = false;
-	let closed = false;
-	let bytesWritten = 0;
-	const maxBytes = deps.maxBytes ?? DEFAULT_MAX_JSONL_BYTES;
-
-	return {
-		writeLine(line: string) {
-			if (!stream || closed || !line.trim()) return;
-			const chunk = `${line}\n`;
-			const chunkBytes = Buffer.byteLength(chunk, "utf-8");
-			if (bytesWritten + chunkBytes > maxBytes) return;
-			try {
-				const ok = stream.write(chunk);
-				bytesWritten += chunkBytes;
-				if (!ok && !backpressured) {
-					backpressured = true;
-					source.pause();
-					stream.once("drain", () => {
-						backpressured = false;
-						if (!closed) source.resume();
-					});
-				}
-			} catch {}
-		},
-		async close() {
-			if (!stream || closed) return;
-			closed = true;
-			const current = stream;
-			stream = undefined;
-			await new Promise<void>((resolve) => current.end(() => resolve()));
-		},
-	};
-}
+import * as fs from "node:fs";
+
+export interface DrainableSource {
+	pause(): void;
+	resume(): void;
+}
+
+export interface JsonlWriteStream {
+	write(chunk: string): boolean;
+	once(event: "drain", listener: () => void): JsonlWriteStream;
+	end(callback?: () => void): void;
+}
+
+const DEFAULT_MAX_JSONL_BYTES = 50 * 1024 * 1024;
+
+export interface JsonlWriterDeps {
+	createWriteStream?: (filePath: string) => JsonlWriteStream;
+	maxBytes?: number;
+}
+
+export interface JsonlWriter {
+	writeLine(line: string): void;
+	close(): Promise<void>;
+}
+
+export function createJsonlWriter(filePath: string | undefined, source: DrainableSource, deps: JsonlWriterDeps = {}): JsonlWriter {
+	if (!filePath) {
+		return {
+			writeLine() {},
+			async close() {},
+		};
+	}
+
+	const createWriteStream = deps.createWriteStream ?? ((targetPath: string) => fs.createWriteStream(targetPath, { flags: "a" }));
+	let stream: JsonlWriteStream | undefined;
+	try {
+		stream = createWriteStream(filePath);
+	} catch {
+		return {
+			writeLine() {},
+			async close() {},
+		};
+	}
+
+	let backpressured = false;
+	let closed = false;
+	let bytesWritten = 0;
+	const maxBytes = deps.maxBytes ?? DEFAULT_MAX_JSONL_BYTES;
+
+	return {
+		writeLine(line: string) {
+			if (!stream || closed || !line.trim()) return;
+			const chunk = `${line}\n`;
+			const chunkBytes = Buffer.byteLength(chunk, "utf-8");
+			if (bytesWritten + chunkBytes > maxBytes) return;
+			try {
+				const ok = stream.write(chunk);
+				bytesWritten += chunkBytes;
+				if (!ok && !backpressured) {
+					backpressured = true;
+					source.pause();
+					stream.once("drain", () => {
+						backpressured = false;
+						if (!closed) source.resume();
+					});
+				}
+			} catch {}
+		},
+		async close() {
+			if (!stream || closed) return;
+			closed = true;
+			const current = stream;
+			stream = undefined;
+			await new Promise<void>((resolve) => current.end(() => resolve()));
+		},
+	};
+}

package/src/state/state-store.ts

@@ -6,7 +6,7 @@ import { atomicWriteJson, atomicWriteJsonAsync, readJsonFile } from "./atomic-wr
 import { appendEvent } from "./event-log.ts";
 import { DEFAULT_CACHE, DEFAULT_PATHS } from "../config/defaults.ts";
 import { createRunId, createTaskId } from "../utils/ids.ts";
-import { findRepoRoot, projectPiRoot, userPiRoot } from "../utils/paths.ts";
+import { findRepoRoot, projectCrewRoot, userCrewRoot } from "../utils/paths.ts";
 import type { TeamConfig } from "../teams/team-config.ts";
 import type { WorkflowConfig } from "../workflows/workflow-config.ts";
 
@@ -48,18 +48,17 @@ function invalidateRunCache(stateRoot: string): void {
 	manifestCache.delete(stateRoot);
 }
 
+function scopeBaseRoot(cwd: string): string {
+	return useProjectState(cwd) ? projectCrewRoot(cwd) : userCrewRoot();
+}
+
 function resolveRunStateRoot(cwd: string, runId: string): string | undefined {
-	const projectPath = path.join(projectPiRoot(cwd), DEFAULT_PATHS.state.projectBase, DEFAULT_PATHS.state.runsSubdir, runId);
-	const userPath = path.join(userPiRoot(), "extensions", "pi-crew", DEFAULT_PATHS.state.userBase, DEFAULT_PATHS.state.runsSubdir, runId);
-	if (fs.existsSync(projectPath)) return projectPath;
-	if (fs.existsSync(userPath)) return userPath;
-	return undefined;
+	const scopedPath = path.join(scopeBaseRoot(cwd), DEFAULT_PATHS.state.runsSubdir, runId);
+	return fs.existsSync(scopedPath) ? scopedPath : undefined;
 }
 
 export function createRunPaths(cwd: string, runId = createRunId()): RunPaths {
-	const baseRoot = useProjectState(cwd)
-		? path.join(projectPiRoot(cwd), DEFAULT_PATHS.state.projectBase)
-		: path.join(userPiRoot(), "extensions", "pi-crew", DEFAULT_PATHS.state.userBase);
+	const baseRoot = scopeBaseRoot(cwd);
 	const stateRoot = path.join(baseRoot, DEFAULT_PATHS.state.runsSubdir, runId);
 	const artifactsRoot = path.join(baseRoot, DEFAULT_PATHS.state.artifactsSubdir, runId);
 	return {

package/src/state/task-claims.ts

@@ -1,42 +1,42 @@
-import { randomUUID } from "node:crypto";
-import type { TeamTaskState } from "./types.ts";
-
-export interface TaskClaimState {
-	owner: string;
-	token: string;
-	leasedUntil: string;
-}
-
-export function createTaskClaim(owner: string, leaseMs = 5 * 60_000, now = new Date()): TaskClaimState {
-	return { owner, token: randomUUID(), leasedUntil: new Date(now.getTime() + leaseMs).toISOString() };
-}
-
-export function isTaskClaimExpired(claim: TaskClaimState | undefined, now = new Date()): boolean {
-	if (!claim) return false;
-	return Date.parse(claim.leasedUntil) <= now.getTime();
-}
-
-export function canUseTaskClaim(task: Pick<TeamTaskState, "claim">, owner: string, token: string, now = new Date()): boolean {
-	return task.claim?.owner === owner && task.claim.token === token && !isTaskClaimExpired(task.claim, now);
-}
-
-export function claimTask<T extends TeamTaskState>(task: T, owner: string, leaseMs?: number, now = new Date()): T {
-	if (task.claim && !isTaskClaimExpired(task.claim, now)) {
-		throw new Error(`Task '${task.id}' is already claimed by '${task.claim.owner}'.`);
-	}
-	return { ...task, claim: createTaskClaim(owner, leaseMs, now) };
-}
-
-export function releaseTaskClaim<T extends TeamTaskState>(task: T, owner: string, token: string, now = new Date()): T {
-	if (!canUseTaskClaim(task, owner, token, now)) {
-		throw new Error(`Task '${task.id}' claim is not held by '${owner}' or has expired.`);
-	}
-	return { ...task, claim: undefined };
-}
-
-export function transitionClaimedTaskStatus<T extends TeamTaskState>(task: T, owner: string, token: string, status: T["status"], now = new Date()): T {
-	if (!canUseTaskClaim(task, owner, token, now)) {
-		throw new Error(`Task '${task.id}' claim is not held by '${owner}' or has expired.`);
-	}
-	return { ...task, status };
-}
+import { randomUUID } from "node:crypto";
+import type { TeamTaskState } from "./types.ts";
+
+export interface TaskClaimState {
+	owner: string;
+	token: string;
+	leasedUntil: string;
+}
+
+export function createTaskClaim(owner: string, leaseMs = 5 * 60_000, now = new Date()): TaskClaimState {
+	return { owner, token: randomUUID(), leasedUntil: new Date(now.getTime() + leaseMs).toISOString() };
+}
+
+export function isTaskClaimExpired(claim: TaskClaimState | undefined, now = new Date()): boolean {
+	if (!claim) return false;
+	return Date.parse(claim.leasedUntil) <= now.getTime();
+}
+
+export function canUseTaskClaim(task: Pick<TeamTaskState, "claim">, owner: string, token: string, now = new Date()): boolean {
+	return task.claim?.owner === owner && task.claim.token === token && !isTaskClaimExpired(task.claim, now);
+}
+
+export function claimTask<T extends TeamTaskState>(task: T, owner: string, leaseMs?: number, now = new Date()): T {
+	if (task.claim && !isTaskClaimExpired(task.claim, now)) {
+		throw new Error(`Task '${task.id}' is already claimed by '${task.claim.owner}'.`);
+	}
+	return { ...task, claim: createTaskClaim(owner, leaseMs, now) };
+}
+
+export function releaseTaskClaim<T extends TeamTaskState>(task: T, owner: string, token: string, now = new Date()): T {
+	if (!canUseTaskClaim(task, owner, token, now)) {
+		throw new Error(`Task '${task.id}' claim is not held by '${owner}' or has expired.`);
+	}
+	return { ...task, claim: undefined };
+}
+
+export function transitionClaimedTaskStatus<T extends TeamTaskState>(task: T, owner: string, token: string, status: T["status"], now = new Date()): T {
+	if (!canUseTaskClaim(task, owner, token, now)) {
+		throw new Error(`Task '${task.id}' claim is not held by '${owner}' or has expired.`);
+	}
+	return { ...task, status };
+}

package/src/state/types.ts

@@ -1,180 +1,180 @@
-import type { TeamRunStatus, TeamTaskStatus } from "./contracts.ts";
-import type { TaskClaimState } from "./task-claims.ts";
-import type { WorkerHeartbeatState } from "../runtime/worker-heartbeat.ts";
-import type { CrewAgentProgress } from "../runtime/crew-agent-runtime.ts";
-export type { TeamRunStatus, TeamTaskStatus } from "./contracts.ts";
-
-export interface ArtifactDescriptor {
-	kind: "plan" | "prompt" | "result" | "summary" | "log" | "diff" | "patch" | "progress" | "notepad" | "metadata";
-	path: string;
-	createdAt: string;
-	producer: string;
-	sizeBytes?: number;
-	contentHash?: string;
-	retention: "run" | "project" | "temporary";
-	expiresAt?: string;
-}
-
-export type TaskScope = "workspace" | "module" | "single_file" | "custom";
-export type GreenLevel = "none" | "targeted" | "package" | "workspace" | "merge_ready";
-
-export interface VerificationCommandResult {
-	cmd: string;
-	status: "passed" | "failed" | "not_run";
-	exitCode?: number | null;
-	outputArtifact?: ArtifactDescriptor;
-}
-
-export interface VerificationContract {
-	requiredGreenLevel: GreenLevel;
-	commands: string[];
-	allowManualEvidence: boolean;
-}
-
-export interface VerificationEvidence {
-	requiredGreenLevel: GreenLevel;
-	observedGreenLevel: GreenLevel;
-	satisfied: boolean;
-	commands: VerificationCommandResult[];
-	notes?: string;
-}
-
-export interface TaskPacket {
-	objective: string;
-	scope: TaskScope;
-	scopePath?: string;
-	repo: string;
-	worktree?: string;
-	branchPolicy: string;
-	acceptanceTests: string[];
-	commitPolicy: string;
-	reportingContract: string;
-	escalationPolicy: string;
-	constraints: string[];
-	expectedArtifacts: string[];
-	verification: VerificationContract;
-}
-
-export type PolicyDecisionAction = "retry" | "reassign" | "escalate" | "block" | "notify" | "cleanup" | "closeout";
-export type PolicyDecisionReason = "task_failed" | "worker_stale" | "green_unsatisfied" | "limit_exceeded" | "run_complete" | "mailbox_timeout" | "review_rejected" | "branch_stale" | "scope_mismatch";
-
-export interface PolicyDecision {
-	action: PolicyDecisionAction;
-	reason: PolicyDecisionReason;
-	message: string;
-	taskId?: string;
-	createdAt: string;
-}
-
-export interface TaskGraphNode {
-	taskId: string;
-	parentId?: string;
-	children: string[];
-	dependencies: string[];
-	queue: "ready" | "blocked" | "running" | "done";
-	sessionForkFrom?: string;
-}
-
-export interface AsyncRunState {
-	pid?: number;
-	logPath: string;
-	spawnedAt: string;
-}
-
-export interface TeamRunManifest {
-	schemaVersion: 1;
-	runId: string;
-	team: string;
-	workflow?: string;
-	goal: string;
-	status: TeamRunStatus;
-	workspaceMode: "single" | "worktree";
-	createdAt: string;
-	updatedAt: string;
-	cwd: string;
-	stateRoot: string;
-	artifactsRoot: string;
-	tasksPath: string;
-	eventsPath: string;
-	artifacts: ArtifactDescriptor[];
-	async?: AsyncRunState;
-	summary?: string;
-	policyDecisions?: PolicyDecision[];
-}
-
-export interface UsageState {
-	input?: number;
-	output?: number;
-	cacheRead?: number;
-	cacheWrite?: number;
-	cost?: number;
-	turns?: number;
-}
-
-export interface ModelAttemptState {
-	model: string;
-	success: boolean;
-	exitCode?: number | null;
-	error?: string;
-}
-
-export interface ModelRoutingState {
-	requested?: string;
-	resolved: string;
-	fallbackChain: string[];
-	reason?: string;
-	usedAttempt: number;
-}
-
-export interface TaskWorktreeState {
-	path: string;
-	branch: string;
-	reused: boolean;
-}
-
-export interface TaskCheckpointState {
-	phase: "started" | "child-spawned" | "child-stdout-final" | "artifact-written";
-	updatedAt: string;
-	childPid?: number;
-}
-
-export interface TeamTaskState {
-	id: string;
-	runId: string;
-	stepId?: string;
-	role: string;
-	agent: string;
-	title: string;
-	status: TeamTaskStatus;
-	dependsOn: string[];
-	cwd: string;
-	worktree?: TaskWorktreeState;
-	promptArtifact?: ArtifactDescriptor;
-	resultArtifact?: ArtifactDescriptor;
-	logArtifact?: ArtifactDescriptor;
-	transcriptArtifact?: ArtifactDescriptor;
-	startedAt?: string;
-	finishedAt?: string;
-	exitCode?: number | null;
-	model?: string;
-	modelAttempts?: ModelAttemptState[];
-	modelRouting?: ModelRoutingState;
-	usage?: UsageState;
-	jsonEvents?: number;
-	agentProgress?: CrewAgentProgress;
-	error?: string;
-	claim?: TaskClaimState;
-	heartbeat?: WorkerHeartbeatState;
-	checkpoint?: TaskCheckpointState;
-	taskPacket?: TaskPacket;
-	verification?: VerificationEvidence;
-	graph?: TaskGraphNode;
-	adaptive?: {
-		phase: string;
-		task: string;
-	};
-	policy?: {
-		retryCount?: number;
-		lastDecision?: PolicyDecision;
-	};
-}
+import type { TeamRunStatus, TeamTaskStatus } from "./contracts.ts";
+import type { TaskClaimState } from "./task-claims.ts";
+import type { WorkerHeartbeatState } from "../runtime/worker-heartbeat.ts";
+import type { CrewAgentProgress } from "../runtime/crew-agent-runtime.ts";
+export type { TeamRunStatus, TeamTaskStatus } from "./contracts.ts";
+
+export interface ArtifactDescriptor {
+	kind: "plan" | "prompt" | "result" | "summary" | "log" | "diff" | "patch" | "progress" | "notepad" | "metadata";
+	path: string;
+	createdAt: string;
+	producer: string;
+	sizeBytes?: number;
+	contentHash?: string;
+	retention: "run" | "project" | "temporary";
+	expiresAt?: string;
+}
+
+export type TaskScope = "workspace" | "module" | "single_file" | "custom";
+export type GreenLevel = "none" | "targeted" | "package" | "workspace" | "merge_ready";
+
+export interface VerificationCommandResult {
+	cmd: string;
+	status: "passed" | "failed" | "not_run";
+	exitCode?: number | null;
+	outputArtifact?: ArtifactDescriptor;
+}
+
+export interface VerificationContract {
+	requiredGreenLevel: GreenLevel;
+	commands: string[];
+	allowManualEvidence: boolean;
+}
+
+export interface VerificationEvidence {
+	requiredGreenLevel: GreenLevel;
+	observedGreenLevel: GreenLevel;
+	satisfied: boolean;
+	commands: VerificationCommandResult[];
+	notes?: string;
+}
+
+export interface TaskPacket {
+	objective: string;
+	scope: TaskScope;
+	scopePath?: string;
+	repo: string;
+	worktree?: string;
+	branchPolicy: string;
+	acceptanceTests: string[];
+	commitPolicy: string;
+	reportingContract: string;
+	escalationPolicy: string;
+	constraints: string[];
+	expectedArtifacts: string[];
+	verification: VerificationContract;
+}
+
+export type PolicyDecisionAction = "retry" | "reassign" | "escalate" | "block" | "notify" | "cleanup" | "closeout";
+export type PolicyDecisionReason = "task_failed" | "worker_stale" | "green_unsatisfied" | "limit_exceeded" | "run_complete" | "mailbox_timeout" | "review_rejected" | "branch_stale" | "scope_mismatch";
+
+export interface PolicyDecision {
+	action: PolicyDecisionAction;
+	reason: PolicyDecisionReason;
+	message: string;
+	taskId?: string;
+	createdAt: string;
+}
+
+export interface TaskGraphNode {
+	taskId: string;
+	parentId?: string;
+	children: string[];
+	dependencies: string[];
+	queue: "ready" | "blocked" | "running" | "done";
+	sessionForkFrom?: string;
+}
+
+export interface AsyncRunState {
+	pid?: number;
+	logPath: string;
+	spawnedAt: string;
+}
+
+export interface TeamRunManifest {
+	schemaVersion: 1;
+	runId: string;
+	team: string;
+	workflow?: string;
+	goal: string;
+	status: TeamRunStatus;
+	workspaceMode: "single" | "worktree";
+	createdAt: string;
+	updatedAt: string;
+	cwd: string;
+	stateRoot: string;
+	artifactsRoot: string;
+	tasksPath: string;
+	eventsPath: string;
+	artifacts: ArtifactDescriptor[];
+	async?: AsyncRunState;
+	summary?: string;
+	policyDecisions?: PolicyDecision[];
+}
+
+export interface UsageState {
+	input?: number;
+	output?: number;
+	cacheRead?: number;
+	cacheWrite?: number;
+	cost?: number;
+	turns?: number;
+}
+
+export interface ModelAttemptState {
+	model: string;
+	success: boolean;
+	exitCode?: number | null;
+	error?: string;
+}
+
+export interface ModelRoutingState {
+	requested?: string;
+	resolved: string;
+	fallbackChain: string[];
+	reason?: string;
+	usedAttempt: number;
+}
+
+export interface TaskWorktreeState {
+	path: string;
+	branch: string;
+	reused: boolean;
+}
+
+export interface TaskCheckpointState {
+	phase: "started" | "child-spawned" | "child-stdout-final" | "artifact-written";
+	updatedAt: string;
+	childPid?: number;
+}
+
+export interface TeamTaskState {
+	id: string;
+	runId: string;
+	stepId?: string;
+	role: string;
+	agent: string;
+	title: string;
+	status: TeamTaskStatus;
+	dependsOn: string[];
+	cwd: string;
+	worktree?: TaskWorktreeState;
+	promptArtifact?: ArtifactDescriptor;
+	resultArtifact?: ArtifactDescriptor;
+	logArtifact?: ArtifactDescriptor;
+	transcriptArtifact?: ArtifactDescriptor;
+	startedAt?: string;
+	finishedAt?: string;
+	exitCode?: number | null;
+	model?: string;
+	modelAttempts?: ModelAttemptState[];
+	modelRouting?: ModelRoutingState;
+	usage?: UsageState;
+	jsonEvents?: number;
+	agentProgress?: CrewAgentProgress;
+	error?: string;
+	claim?: TaskClaimState;
+	heartbeat?: WorkerHeartbeatState;
+	checkpoint?: TaskCheckpointState;
+	taskPacket?: TaskPacket;
+	verification?: VerificationEvidence;
+	graph?: TaskGraphNode;
+	adaptive?: {
+		phase: string;
+		task: string;
+	};
+	policy?: {
+		retryCount?: number;
+		lastDecision?: PolicyDecision;
+	};
+}