@arhen/pi-core-subagent 1.3.31 → 1.3.32

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@arhen/pi-core-subagent",
-	"version": "1.3.31",
+	"version": "1.3.32",
 	"type": "module",
 	"description": "pi extension: fast in-process subagents with a dependency-graph scheduler (needs edges gate tasks and carry upstream output into dependent prompts), plus background runs, intercom and agent-to-agent mailbox. Leader defines agents inline.",
 	"license": "MIT",

package/src/index.ts

@@ -30,7 +30,7 @@ import {
 	type SubagentParamsShape,
 } from "./schemas.ts";
 import { type RunDetails, type RunSnapshot, TERMINAL } from "./types.ts";
-import { cleanupMerged, reapDeadWorktrees, repoRoot, sweepStale } from "./worktree.ts";
+import { cleanupMerged, ownerAlive, reapDeadWorktrees, repoRoot, sweepStale } from "./worktree.ts";
 
 export default function (pi: ExtensionAPI) {
 	const manager = new SubagentManager(pi);
@@ -126,11 +126,11 @@ export default function (pi: ExtensionAPI) {
 		}
 		for (const root of roots) {
 			try {
-				// Nothing of ours can be live at session start: every registered subagent
-				// worktree is a crash leftover — commit its work, keep the branch, drop the
-				// dir. Then reap merged branches and dirs git no longer tracks.
-				reapDeadWorktrees(root);
-				cleanupMerged(root);
+				// A registered subagent worktree is a crash leftover UNLESS another pi
+				// session still owns it (pid marker) — commit its work, keep the branch,
+				// drop the dir. Then reap merged branches and dirs git no longer tracks.
+				reapDeadWorktrees(root, ownerAlive);
+				cleanupMerged(root, { skipBranches: manager.liveBranches() });
 				sweepStale(root);
 			} catch {
 				/* recovery is best-effort — never block session start */

package/src/manager.ts

@@ -45,6 +45,7 @@ import {
 } from "./types.ts";
 import {
 	branchDiff,
+	claimWorktree,
 	cleanupMerged,
 	commitWorktree,
 	createWorktree,
@@ -58,6 +59,8 @@ export const MAX_CONCURRENCY = 8;
 /** No default wall-clock cap: a subagent runs until its task is done, it stalls, or the user aborts. */
 const DEFAULT_RUNTIME_MS = 0;
 const DEFAULT_STALL_MS = 180_000; // 3 min: long model thinking streams emit no events, but they're not stalled.
+/** Cap on a child's wait for reply_subagent — an ignored question must not pin the run open forever. */
+const PARENT_REPLY_TIMEOUT_MS = 600_000; // 10 min
 const READONLY_TOOLS = ["read", "grep", "find", "ls"];
 const WRITE_TOOLS = ["read", "grep", "find", "ls", "bash", "edit", "write"];
 /** Tools that can mutate the tree — their presence is what earns a worktree. */
@@ -221,7 +224,10 @@ export interface ParkedMsg {
 
 export class SubagentManager {
 	private runs = new Map<string, RunSnapshot>();
-	private settlers = new Map<string, (run: RunSnapshot) => void>();
+	/** Runs that are still settleable (presence = not yet settled). */
+	private settlers = new Map<string, true>();
+	/** Everyone parked on a run — a set, so re-parking can't build a closure chain. */
+	private settleWaiters = new Map<string, Set<(run: RunSnapshot) => void>>();
 	private pendingReplies = new Map<string, PendingReply>();
 	private liveChildren = new Map<
 		string,
@@ -296,8 +302,12 @@ export class SubagentManager {
 			child.dispose();
 		}
 		this.liveChildren.clear();
+		// Ownership markers stay on disk; the next session reaps those dirs (commit,
+		// keep branch, drop dir) once this pid is gone.
+		this.liveWorktrees.clear();
 		this.runs.clear();
 		this.settlers.clear();
+		this.settleWaiters.clear();
 		this.pendingReplies.clear();
 		this.runControllers.clear();
 		this.mailboxes = createMailbox();
@@ -508,31 +518,24 @@ export class SubagentManager {
 	private makeChildHandlers(run: RunSnapshot, task: TaskSnapshot, ctx: ExtensionContext): ChildHandlers {
 		return {
 			onAskParent: async (_taskId, question) => {
+				const key = `${run.id}:${task.id}`;
 				this.updateTask(run, task, { status: "awaiting_parent" }, ctx);
-				this.liveChildren.get(`${run.id}:${task.id}`)?.touchWatchdog();
-				// While the leader is parked in await_subagent, the question rides the wait
-				// instead of the steering queue — no boundary needed, no starvation.
-				if (this.collectParked(run.id, { kind: "ask", taskId: task.id, agent: task.agent, text: question })) {
-					// The parked leader gets the question inside its await result and answers
-					// with reply_subagent — so the pending entry MUST exist here too, or the
-					// reply lands nowhere and the child waits on an answer that never comes.
-					const keepAlive = setInterval(() => this.liveChildren.get(`${run.id}:${task.id}`)?.touchWatchdog(), 30_000);
-					try {
-						const reply = await this.awaitParentReply(run.id, task.id);
-						this.updateTask(run, task, { status: "running" }, ctx);
-						this.liveChildren.get(`${run.id}:${task.id}`)?.touchWatchdog();
-						return reply;
-					} finally {
-						clearInterval(keepAlive);
-					}
+				this.liveChildren.get(key)?.touchWatchdog();
+				// While the leader is parked in await_subagent the question rides the wait
+				// (no steering queue, no turn boundary); otherwise it goes out as a notice.
+				// Either way the pending reply entry must exist, or reply_subagent has
+				// nowhere to land and the child waits on an answer that never comes.
+				if (!this.collectParked(run.id, { kind: "ask", taskId: task.id, agent: task.agent, text: question })) {
+					this.notifyParent(run, "asked", { taskId: task.id, question });
 				}
-				this.notifyParent(run, "asked", { taskId: task.id, question });
-				// M3: a waiting child is not stalled — keep the watchdog fed until the reply.
-				const keepAlive = setInterval(() => this.liveChildren.get(`${run.id}:${task.id}`)?.touchWatchdog(), 30_000);
+				// A waiting child is not stalled — keep the watchdog fed until the reply.
+				// But the wait is BOUNDED: an unanswered question would otherwise keep the
+				// run non-terminal forever (widget never clears, run never settles).
+				const keepAlive = setInterval(() => this.liveChildren.get(key)?.touchWatchdog(), 30_000);
 				try {
-					const reply = await this.awaitParentReply(run.id, task.id);
+					const reply = await this.awaitParentReply(run.id, task.id, PARENT_REPLY_TIMEOUT_MS);
 					this.updateTask(run, task, { status: "running" }, ctx);
-					this.liveChildren.get(`${run.id}:${task.id}`)?.touchWatchdog();
+					this.liveChildren.get(key)?.touchWatchdog();
 					return reply;
 				} finally {
 					clearInterval(keepAlive);
@@ -572,9 +575,24 @@ export class SubagentManager {
 			onPollMailbox: (taskId) => this.mailboxes.poll(`${run.id}:${taskId}`),
 		};
 	}
-	private awaitParentReply(runId: string, taskId: string): Promise<string> {
+	private awaitParentReply(runId: string, taskId: string, timeoutMs = 0): Promise<string> {
+		const key = `${runId}:${taskId}`;
 		return new Promise<string>((resolve) => {
-			this.pendingReplies.set(`${runId}:${taskId}`, { resolve });
+			const timer =
+				timeoutMs > 0
+					? setTimeout(() => {
+							this.pendingReplies.delete(key);
+							resolve(
+								"The parent did not answer in time. Proceed autonomously with your best judgment and state the assumption you made in your final answer.",
+							);
+						}, timeoutMs)
+					: undefined;
+			this.pendingReplies.set(key, {
+				resolve: (message) => {
+					if (timer) clearTimeout(timer);
+					resolve(message);
+				},
+			});
 		});
 	}
 	deliverReply(runId: string, taskId: string, message: string): boolean {
@@ -683,8 +701,31 @@ export class SubagentManager {
 		const canWrite = baseTools.some((t) => WRITE_CAPABLE.includes(t));
 
 		// Write agents run in an isolated git worktree (branch subagents/<run>/<task>);
-		// non-git repos fall back to in-place. The worktree is created BEFORE session
-		// start so the child's cwd + AGENTS.md context chain are the worktree's.
+		// Model + thinking resolve against the pi model registry BEFORE any worktree
+		// exists — a bad request fails the TASK with a helpful message and can't leak a
+		// checkout past this early return.
+		let model: Model<Api> | undefined;
+		try {
+			model = resolveChildModel(ctx, file?.model ?? input.model);
+			validateThinking(model, thinking);
+		} catch (err) {
+			this.updateTask(
+				run,
+				task,
+				{
+					status: "failed",
+					error: err instanceof Error ? err.message : String(err),
+					endedAt: Date.now(),
+				},
+				ctx,
+				onUpdate,
+			);
+			return;
+		}
+
+		// Write agents run in an isolated git worktree (branch subagents/<run>/<task>);
+		// non-git repos fall back to in-place. Created BEFORE session start so the
+		// child's cwd + AGENTS.md context chain are the worktree's.
 		let wt: Worktree | undefined;
 		if (canWrite) {
 			try {
@@ -694,39 +735,23 @@ export class SubagentManager {
 			}
 		}
 		// Map a per-task cwd subpath into the worktree so relative paths stay correct.
-		// If the mapping can't be trusted (cwd outside the repo via symlink), drop the
-		// worktree rather than run in the main tree while reporting a branch.
+		// Both sides go through realpath — a symlinked root would otherwise look
+		// "outside" the repo. If the mapping can't be trusted, drop the worktree AND
+		// reset the cwd (never point the child at a dir that was just removed).
 		let childCwd = wt?.path ?? task.cwd;
 		if (wt) {
-			const rel = relative(wt.root, safeRealPath(task.cwd));
+			const rel = relative(safeRealPath(wt.root), safeRealPath(task.cwd));
 			if (rel.startsWith("..")) {
 				removeWorktree(wt);
 				wt = undefined;
+				childCwd = task.cwd;
 			} else if (rel && rel !== ".") {
 				childCwd = join(wt.path, rel);
 			}
 		}
-		if (wt) this.liveWorktrees.set(`${run.id}:${task.id}`, wt);
-
-		// Model + thinking resolve against the pi model registry; a bad request
-		// fails the TASK with a helpful message, not the whole run.
-		let model: Model<Api> | undefined;
-		try {
-			model = resolveChildModel(ctx, file?.model ?? input.model);
-			validateThinking(model, thinking);
-		} catch (err) {
-			this.updateTask(
-				run,
-				task,
-				{
-					status: "failed",
-					error: err instanceof Error ? err.message : String(err),
-					endedAt: Date.now(),
-				},
-				ctx,
-				onUpdate,
-			);
-			return;
+		if (wt) {
+			claimWorktree(wt); // pid marker: another pi session must not reap this
+			this.liveWorktrees.set(`${run.id}:${task.id}`, wt);
 		}
 
 		this.updateTask(
@@ -746,6 +771,9 @@ export class SubagentManager {
 			onUpdate,
 		);
 
+		// Set once the dir must outlive this call: committed work awaiting the
+		// leader's merge, or a commit failure whose work exists ONLY in the dir.
+		let keepWorktreeDir = false;
 		let child: Awaited<ReturnType<typeof createAgentSession>>["session"] | undefined;
 		let unsubscribe: (() => void) | undefined;
 		let timeout: ReturnType<typeof setTimeout> | undefined;
@@ -865,6 +893,7 @@ export class SubagentManager {
 					// its work: the error is reported, the status stays completed.
 					try {
 						commitWorktree(wt, `subagent ${task.agent}: ${truncateText(input.task, 60)}`);
+						keepWorktreeDir = true; // committed — dir stays until the leader merges
 						const { stat, files } = branchDiff(wt);
 						this.updateTask(
 							run,
@@ -874,12 +903,15 @@ export class SubagentManager {
 							onUpdate,
 						);
 					} catch (commitErr) {
+						// Never drop a checkout whose work isn't on the branch — it would be
+						// unreachable once the base-tip branch is reaped as "merged".
+						keepWorktreeDir = true;
 						this.updateTask(
 							run,
 							task,
 							{
 								branch: wt.branch,
-								error: `Worktree commit failed (changes remain in ${wt.path}): ${commitErr instanceof Error ? commitErr.message : String(commitErr)}`,
+								error: `Worktree commit failed (uncommitted changes remain in ${wt.path}): ${commitErr instanceof Error ? commitErr.message : String(commitErr)}`,
 							},
 							ctx,
 							onUpdate,
@@ -919,20 +951,23 @@ export class SubagentManager {
 			watchdog.dispose();
 			if (timeout) clearTimeout(timeout);
 			child?.dispose();
-			this.liveWorktrees.delete(key);
 			// Failed/aborted: let the aborted child's last writes land (its tools may
 			// still be unwinding), commit whatever partial work exists so the branch
-			// really keeps it, then drop the checkout dir.
+			// really keeps it, then drop the checkout dir. A commit FAILURE keeps the
+			// dir — dropping it would make the work unreachable.
 			if (wt && task.status !== "completed") {
 				await new Promise((r) => setTimeout(r, 250));
 				try {
 					commitWorktree(wt, `subagent ${task.agent} (partial, ${task.status})`);
 				} catch {
-					/* nothing to commit */
+					keepWorktreeDir = true;
 				}
 				this.updateTask(run, task, { branch: wt.branch }, ctx, onUpdate);
-				removeWorktree(wt);
 			}
+			if (wt && !keepWorktreeDir) removeWorktree(wt);
+			// Released only after the dir is gone: while it exists, the branch must stay
+			// in liveBranches() so cleanup can't reap it.
+			this.liveWorktrees.delete(key);
 		}
 	}
 
@@ -1020,7 +1055,7 @@ export class SubagentManager {
 		});
 		this.turnActivity = true;
 		this.runs.set(run.id, run);
-		this.settlers.set(run.id, () => {});
+		this.settlers.set(run.id, true);
 		this.runControllers.set(run.id, new AbortController());
 		for (const task of run.tasks) this.mailboxes.open(`${run.id}:${task.id}`);
 		this.emit("subagent:run-created", { run: cloneRun(run) });
@@ -1195,6 +1230,8 @@ export class SubagentManager {
 		task.status = "aborted";
 		task.error = task.error || "Canceled from peek";
 		task.endedAt = Date.now();
+		this.pendingReplies.get(`${runId}:${taskId}`)?.resolve("(task canceled by the parent — stop work now)");
+		this.pendingReplies.delete(`${runId}:${taskId}`);
 		this.liveChildren.get(`${runId}:${taskId}`)?.abort();
 		this.mailboxes.close(`${runId}:${taskId}`);
 		if (ctx) this.flushWidget(run, ctx);
@@ -1208,6 +1245,14 @@ export class SubagentManager {
 		if (TERMINAL.includes(run.status)) return { aborted: 0 }; // never corrupt a finished run
 		let aborted = 0;
 		this.runControllers.get(runId)?.abort();
+		// Release children parked in ask_parent first — an unresolved wait would keep
+		// the child alive past the abort.
+		for (const [key, pending] of this.pendingReplies) {
+			if (key.startsWith(`${runId}:`)) {
+				this.pendingReplies.delete(key);
+				pending.resolve("(run canceled by the parent — stop work and return what you have)");
+			}
+		}
 		for (const [key, child] of this.liveChildren) {
 			if (key.startsWith(`${runId}:`)) {
 				child.abort();
@@ -1233,12 +1278,15 @@ export class SubagentManager {
 		return { aborted };
 	}
 
-	/** Settle-and-delete: awaiters resolve once; no leak, no closure chain. */
+	/** Settle-and-delete: every awaiter resolves once, then the set is dropped. */
 	private settleRun(runId: string, run: RunSnapshot): void {
-		const s = this.settlers.get(runId);
-		if (!s) return;
+		if (!this.settlers.has(runId)) return;
 		this.settlers.delete(runId);
-		s(cloneRun(run));
+		const waiters = this.settleWaiters.get(runId);
+		this.settleWaiters.delete(runId);
+		if (!waiters) return;
+		const snapshot = cloneRun(run);
+		for (const waiter of waiters) waiter(snapshot);
 	}
 
 	/** Child→leader messages collected while the parent is parked in await_subagent. */
@@ -1268,14 +1316,22 @@ export class SubagentManager {
 		}
 		const msgs: ParkedMsg[] = [];
 		const settled = new Promise<RunSnapshot | undefined>((resolve) => {
-			const prev = this.settlers.get(runId);
-			this.settlers.set(runId, (r) => {
-				prev?.(r);
+			// Waiters are a SET, not a chain: the autoAwait loop re-parks on every
+			// child message, and wrapping the previous settler each time grew an
+			// unbounded closure chain (each holding a snapshot clone).
+			const waiter = (r: RunSnapshot) => {
+				this.settleWaiters.get(runId)?.delete(waiter);
 				resolve(r);
-			});
+			};
+			let waiters = this.settleWaiters.get(runId);
+			if (!waiters) {
+				waiters = new Set();
+				this.settleWaiters.set(runId, waiters);
+			}
+			waiters.add(waiter);
 			// A child→leader message while parked wakes the wait: the leader gets it
 			// IN the await result, no steering queue, no turn boundary needed.
-			this.parked.set(runId, { msgs, wake: () => resolve(cloneRun(run)) });
+			this.parked.set(runId, { msgs, wake: () => waiter(cloneRun(run)) });
 		});
 		if (timeoutMs !== undefined && timeoutMs > 0) {
 			return Promise.race([

package/src/worktree.ts

@@ -10,7 +10,7 @@
  *  touching a checked-out one), `sweepStale` (dirs git no longer knows about). */
 
 import { execFileSync } from "node:child_process";
-import { existsSync, readdirSync, realpathSync, rmSync, symlinkSync } from "node:fs";
+import { existsSync, readdirSync, readFileSync, realpathSync, rmSync, symlinkSync, writeFileSync } from "node:fs";
 import { join } from "node:path";
 
 export interface Worktree {
@@ -40,15 +40,6 @@ function gitOk(root: string, args: string[]): boolean {
 	}
 }
 
-/** git C-quotes porcelain paths containing spaces/specials — undo that. */
-function unquote(path: string): string {
-	if (!path.startsWith('"') || !path.endsWith('"')) return path;
-	return path
-		.slice(1, -1)
-		.replace(/\\([0-7]{3})/g, (_, o) => String.fromCharCode(Number.parseInt(o, 8)))
-		.replace(/\\(.)/g, (_, c) => ({ n: "\n", t: "\t", r: "\r" })[c as string] ?? c);
-}
-
 /** Compare paths through realpath so /var vs /private/var can't diverge. */
 function samePath(a: string, b: string): boolean {
 	if (a === b) return true;
@@ -73,14 +64,20 @@ export function repoRoot(cwd: string): string | undefined {
 export function createWorktree(cwd: string, runId: string, taskId: string): Worktree | undefined {
 	const root = repoRoot(cwd);
 	if (!root) return undefined;
-	const path = join(root, ".git", "subagents", runId, taskId);
-	const branch = `${BRANCH_PREFIX}${runId}/${taskId}`;
+	// --git-common-dir, not "<root>/.git": inside a linked worktree or a submodule
+	// `.git` is a FILE, and joining it would make `worktree add` fail (silently
+	// dropping isolation).
+	let container: string;
 	let base: string;
 	try {
+		const common = git(root, ["rev-parse", "--path-format=absolute", "--git-common-dir"]);
+		container = join(common, "subagents");
 		base = git(root, ["rev-parse", "HEAD"]); // SHA — detached HEAD stays correct
 	} catch {
 		return undefined; // broken repo — fall back to in-place
 	}
+	const path = join(container, runId, taskId);
+	const branch = `${BRANCH_PREFIX}${runId}/${taskId}`;
 	git(root, ["worktree", "add", "-b", branch, path, "HEAD"]);
 	// Deps follow the child into the worktree; anything else the task needs is
 	// project content already checked out there.
@@ -98,16 +95,20 @@ export function createWorktree(cwd: string, runId: string, taskId: string): Work
 /**
  * Commit the child's changes. Stages first, then commits only when something is
  * actually staged — an untracked node_modules symlink must not fake "dirty" and
- * turn into a failed empty commit.
+ * turn into a failed empty commit. Returns "committed" | "empty"; a real git
+ * failure THROWS, and callers must not delete the checkout in that case (the
+ * work would become unreachable once the base-tip branch is reaped as merged).
  */
-export function commitWorktree(wt: Worktree, message: string): void {
-	commitIn(wt.path, message);
+export function commitWorktree(wt: Worktree, message: string): "committed" | "empty" {
+	return commitIn(wt.path, message);
 }
 
-function commitIn(dir: string, message: string): void {
-	gitIn(dir, ["add", "-A", "--", ".", ":(exclude)node_modules"]); // never stage the dep symlink
-	if (gitIn(dir, ["diff", "--cached", "--name-only"]).length === 0) return; // nothing to commit
+function commitIn(dir: string, message: string): "committed" | "empty" {
+	// Exclude the root dep symlink and any nested node_modules the child created.
+	gitIn(dir, ["add", "-A", "--", ".", ":(exclude)node_modules", ":(exclude,glob)**/node_modules/**"]);
+	if (gitIn(dir, ["diff", "--cached", "--name-only"]).length === 0) return "empty";
 	gitIn(dir, ["commit", "-m", message, "--no-verify"]);
+	return "committed";
 }
 
 /** Diffstat + changed files of the branch vs its base SHA. */
@@ -129,7 +130,17 @@ export function removeByBranch(cwd: string, branch: string): void {
 	if (!branch.startsWith(BRANCH_PREFIX)) return;
 	const root = repoRoot(cwd);
 	if (!root) return;
-	dropDir(root, join(root, ".git", "subagents", branch.slice(BRANCH_PREFIX.length)));
+	const container = subagentsDir(root);
+	if (container) dropDir(root, join(container, branch.slice(BRANCH_PREFIX.length)));
+}
+
+/** `<git-common-dir>/subagents` — where our worktrees live for this repo. */
+function subagentsDir(root: string): string | undefined {
+	try {
+		return join(git(root, ["rev-parse", "--path-format=absolute", "--git-common-dir"]), "subagents");
+	} catch {
+		return undefined;
+	}
 }
 
 function dropDir(root: string, path: string): void {
@@ -169,11 +180,13 @@ export function cleanupMerged(root: string, opts: { skipBranches?: Set<string>;
 		.split("\n")
 		.map((b) => b.trim().replace(/^[+*]\s*/, ""));
 	let cleaned = 0;
+	const container = subagentsDir(root);
 	for (const branch of merged) {
-		if (!branch.startsWith(BRANCH_PREFIX)) continue;
+		if (!branch.startsWith(BRANCH_PREFIX) || !container) continue;
 		if (opts.skipBranches?.has(branch)) continue; // owned by a live run
 		if (isCheckedOut(root, branch)) continue; // fresh re-check, not a stale snapshot
-		const path = join(root, ".git", "subagents", branch.slice(BRANCH_PREFIX.length));
+		const path = join(container, branch.slice(BRANCH_PREFIX.length));
+		if (existsSync(path) && ownerAlive(path)) continue; // another session's live checkout
 		if (existsSync(path)) {
 			try {
 				git(root, ["worktree", "remove", "--force", path]);
@@ -190,10 +203,10 @@ export function cleanupMerged(root: string, opts: { skipBranches?: Set<string>;
 /** Branch names currently checked out in any worktree (incl. the main one). */
 function worktreeBranches(root: string): string[] {
 	try {
-		return git(root, ["worktree", "list", "--porcelain"])
-			.split("\n")
+		return git(root, ["worktree", "list", "--porcelain", "-z"])
+			.split("\0")
 			.filter((l) => l.startsWith("branch "))
-			.map((l) => l.slice("branch refs/heads/".length).trim());
+			.map((l) => l.slice("branch refs/heads/".length));
 	} catch {
 		return [];
 	}
@@ -204,17 +217,18 @@ function worktreeBranches(root: string): string[] {
  * (nothing of ours can be live yet). Commit whatever the dead child left so the
  * branch keeps it, then drop the dir. Branches always survive.
  */
-export function reapDeadWorktrees(root: string): number {
+export function reapDeadWorktrees(root: string, isLive: (path: string) => boolean = () => false): number {
 	root = realpathSync(root);
-	const sub = join(root, ".git", "subagents");
-	if (!existsSync(sub)) return 0;
+	const sub = subagentsDir(root);
+	if (!sub || !existsSync(sub)) return 0;
 	let reaped = 0;
 	for (const path of worktreePaths(root)) {
 		if (!isInside(path, sub)) continue; // not ours
+		if (isLive(path)) continue; // another pi session owns it
 		try {
 			commitIn(path, "subagent (recovered after interrupted session)");
 		} catch {
-			/* nothing committable */
+			continue; // git failed — never drop a dir whose work isn't on the branch
 		}
 		dropDir(root, path);
 		reaped += 1;
@@ -222,6 +236,31 @@ export function reapDeadWorktrees(root: string): number {
 	return reaped;
 }
 
+/**
+ * Ownership marker: a live worktree gets `<dir>/.subagent-owner` holding the
+ * owning pid. Another pi session must not reap a checkout whose owner is alive.
+ */
+export function claimWorktree(wt: Worktree): void {
+	try {
+		writeFileSync(join(wt.path, ".subagent-owner"), String(process.pid));
+	} catch {
+		/* best-effort: worst case another session reaps it after a crash */
+	}
+}
+
+/** True when the worktree dir is claimed by a process that still exists. */
+export function ownerAlive(path: string): boolean {
+	try {
+		const pid = Number.parseInt(readFileSync(join(path, ".subagent-owner"), "utf8").trim(), 10);
+		if (!Number.isFinite(pid) || pid <= 0) return false;
+		if (pid === process.pid) return true;
+		process.kill(pid, 0); // throws ESRCH when the owner is gone
+		return true;
+	} catch {
+		return false;
+	}
+}
+
 /**
  * Remove worktree dirs that git no longer knows about (partial-crash leftovers).
  * Registered worktrees are never touched here — `reapDeadWorktrees` owns those,
@@ -229,25 +268,27 @@ export function reapDeadWorktrees(root: string): number {
  */
 export function sweepStale(root: string): void {
 	root = realpathSync(root);
-	const sub = join(root, ".git", "subagents");
-	if (!existsSync(sub)) return;
+	const sub = subagentsDir(root);
+	if (!sub || !existsSync(sub)) return;
 	const registered = worktreePaths(root);
 	for (const runDir of readDirs(sub)) {
 		for (const taskDir of readDirs(join(sub, runDir))) {
 			const dir = join(sub, runDir, taskDir);
 			if (registered.some((p) => samePath(p, dir))) continue; // live/registered worktree
+			if (ownerAlive(dir)) continue; // claimed by a running session
 			rmSync(dir, { recursive: true, force: true });
 		}
 	}
 	prune(root);
 }
 
+/** Registered worktree paths. `-z` keeps paths verbatim (no C-quoting to undo). */
 function worktreePaths(root: string): string[] {
 	try {
-		return git(root, ["worktree", "list", "--porcelain"])
-			.split("\n")
+		return git(root, ["worktree", "list", "--porcelain", "-z"])
+			.split("\0")
 			.filter((l) => l.startsWith("worktree "))
-			.map((l) => unquote(l.slice("worktree ".length).trim()));
+			.map((l) => l.slice("worktree ".length));
 	} catch {
 		return [];
 	}