@arhen/pi-core-subagent 1.3.30 → 1.3.31

package/README.md

@@ -147,7 +147,17 @@ On completion the extension commits the child's changes (the child is told not t
 git merge --no-ff subagents/<run>/<task>
 ```
 
-Merged branches + their worktree dirs are cleaned automatically after the run. Failed/canceled tasks keep the branch (partial work survives for manual merging) but drop the worktree dir. Crash leftovers are swept at session start — worktree dirs are removed, branches are kept. Non-git repos fall back to in-place edits.
+Isolation follows the toolset the child actually receives: explicit `tools: ["bash", "edit", "write"]` earns a worktree even without `write: true`, and an agent file that narrows the child to read-only gets no branch at all.
+
+Cleanup, in order of trust:
+
+| When | What |
+|---|---|
+| Session start | Registered worktrees can't be live yet — an interrupted child's uncommitted work is committed, the branch kept, the dir dropped. Then merged branches are reaped and dirs git no longer tracks are removed. |
+| After a run | Merged branches + their dirs, never touching a branch a live run owns (checkout state re-checked per branch). |
+| Task failed/canceled | Partial work is committed first, so the branch keeps it; the dir is dropped. |
+
+Non-git repos fall back to in-place edits.
 
 ## Graph mode — `needs`
 

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@arhen/pi-core-subagent",
-	"version": "1.3.30",
+	"version": "1.3.31",
 	"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, repoRoot, sweepStale } from "./worktree.ts";
+import { cleanupMerged, reapDeadWorktrees, repoRoot, sweepStale } from "./worktree.ts";
 
 export default function (pi: ExtensionAPI) {
 	const manager = new SubagentManager(pi);
@@ -125,8 +125,16 @@ export default function (pi: ExtensionAPI) {
 			}
 		}
 		for (const root of roots) {
-			sweepStale(root);
-			cleanupMerged(root);
+			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);
+				sweepStale(root);
+			} catch {
+				/* recovery is best-effort — never block session start */
+			}
 		}
 	});
 	pi.on("session_shutdown", async (_event, ctx) => {
@@ -170,18 +178,29 @@ export default function (pi: ExtensionAPI) {
 				// until terminal — but surface an ask_parent: the child is waiting on the
 				// leader, so break out, reply via reply_subagent, then await again.
 				let run = details.run;
-				let intercom: ParkedMsg[] = [];
+				// Each park gets a FRESH msgs array — accumulate, or every wake but the
+				// last is lost (they were consumed by the park, never sent as followUp).
+				const intercom: ParkedMsg[] = [];
 				while (!TERMINAL.includes(run.status)) {
 					const awaited = await manager.awaitRun(details.run.id);
 					if (!awaited) break; // run gone (session shutdown) — stop, no busy-spin
 					if (awaited.run) run = awaited.run;
-					intercom = awaited.intercom;
-					if (intercom.some((m) => m.kind === "ask")) break;
+					intercom.push(...awaited.intercom);
+					if (awaited.intercom.some((m) => m.kind === "ask")) break;
 				}
 				const asked = intercom.find((m) => m.kind === "ask");
-				const text = asked
-					? `${makeSummary(run)}\n\nA child is waiting for your answer (${asked.agent}, ${asked.taskId}): ${asked.text}\nReply with reply_subagent(runId: "${run.id}", taskId: "${asked.taskId}", message: ...), then await_subagent again for the result.`
-					: makeSummary(run);
+				const heard = intercom.filter((m) => m.kind !== "ask");
+				const text = [
+					makeSummary(run),
+					heard.length > 0
+						? `\nIntercom while waiting:\n${heard.map((m) => `- [${m.kind}] ${m.agent} (${m.taskId}): ${truncateText(m.text)}`).join("\n")}`
+						: "",
+					asked
+						? `\nA child is waiting for your answer (${asked.agent}, ${asked.taskId}): ${asked.text}\nReply with reply_subagent(runId: "${run.id}", taskId: "${asked.taskId}", message: ...), then await_subagent again for the result.`
+						: "",
+				]
+					.filter(Boolean)
+					.join("\n");
 				return { content: [{ type: "text", text }], details: { run } };
 			}
 			return {

package/src/manager.ts

@@ -1,5 +1,5 @@
 /** SubagentManager: run lifecycle, child sessions, intercom, persistence, widget plumbing. */
-import { existsSync, readFileSync } from "node:fs";
+import { existsSync, readFileSync, realpathSync } from "node:fs";
 import { writeFile } from "node:fs/promises";
 import { join, relative } from "node:path";
 import type { ThinkingLevel } from "@earendil-works/pi-agent-core";
@@ -48,10 +48,8 @@ import {
 	cleanupMerged,
 	commitWorktree,
 	createWorktree,
-	removeByBranch,
 	removeWorktree,
 	repoRoot,
-	sweepStale,
 	type Worktree,
 } from "./worktree.ts";
 
@@ -62,6 +60,8 @@ 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.
 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. */
+const WRITE_CAPABLE = ["bash", "edit", "write"];
 /** Task ids become git refs + filesystem paths. */
 const SAFE_TASK_ID = /^[A-Za-z0-9_-]{1,64}$/;
 const WIDGET_THROTTLE_MS = 150;
@@ -86,6 +86,14 @@ function aggregateUsage(tasks: TaskSnapshot[]): UsageStats {
 	}
 	return total;
 }
+/** realpath when possible; the raw path otherwise (cwd may not exist yet). */
+function safeRealPath(path: string): string {
+	try {
+		return realpathSync(path);
+	} catch {
+		return path;
+	}
+}
 function getParentSessionFile(ctx: ExtensionContext): string | undefined {
 	try {
 		return ctx.sessionManager.getSessionFile?.();
@@ -220,6 +228,9 @@ export class SubagentManager {
 		{ abort: () => void; dispose: () => void; touchWatchdog: () => void; steer: (message: string) => void }
 	>();
 	private mailboxes: Mailbox = createMailbox();
+	/** Live worktrees by `${runId}:${taskId}` — lets cancel drop dirs and keeps
+	 *  cleanup from touching a branch that a running child owns. */
+	private liveWorktrees = new Map<string, Worktree>();
 	private runControllers = new Map<string, AbortController>();
 	private widgetTimers = new Map<string, ReturnType<typeof setTimeout>>(); // per-run stream throttle
 	private widgetRuns: RunSnapshot[] = [];
@@ -502,7 +513,18 @@ export class SubagentManager {
 				// 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 })) {
-					return "Your question was delivered to the parent (they're waiting on this run). Keep working; the answer arrives via the pending reply.";
+					// 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.notifyParent(run, "asked", { taskId: task.id, question });
 				// M3: a waiting child is not stalled — keep the watchdog fed until the reply.
@@ -518,13 +540,13 @@ export class SubagentManager {
 			},
 			onNotifyParent: (_taskId, message, level) => {
 				this.emit("subagent:intercom", { runId: run.id, taskId: task.id, kind: "notify", level, message });
+				// Parked leader gets it through the wait; otherwise queue it. `awaited` must
+				// NOT gate this — between two parks the leader is awaited but listening.
 				if (this.collectParked(run.id, { kind: "notify", taskId: task.id, agent: task.agent, text: message })) return;
-				if (!run.awaited) {
-					try {
-						this.pi.sendUserMessage(`[Subagent ${task.agent}] ${message}`, { deliverAs: "followUp" });
-					} catch {
-						/* parent mid-stream */
-					}
+				try {
+					this.pi.sendUserMessage(`[Subagent ${task.agent}] ${message}`, { deliverAs: "followUp" });
+				} catch {
+					/* parent mid-stream */
 				}
 			},
 			onSendMessage: (_taskId, to, text) => {
@@ -537,12 +559,10 @@ export class SubagentManager {
 						message: text,
 					});
 					if (this.collectParked(run.id, { kind: "notify", taskId: task.id, agent: task.agent, text })) return true;
-					if (!run.awaited) {
-						try {
-							this.pi.sendUserMessage(`[Subagent ${task.agent}] ${text}`, { deliverAs: "followUp" });
-						} catch {
-							/* parent mid-stream */
-						}
+					try {
+						this.pi.sendUserMessage(`[Subagent ${task.agent}] ${text}`, { deliverAs: "followUp" });
+					} catch {
+						/* parent mid-stream */
 					}
 					return true;
 				}
@@ -657,9 +677,10 @@ export class SubagentManager {
 		const fileTools = file?.tools?.filter((t) => allowedTools.includes(t));
 		const baseTools = fileTools?.length ? fileTools : (input.tools ?? allowedTools);
 		const tools = [...baseTools, ...(run.allowIntercom ? CHILD_TALK_TOOLS : [])];
-		// Worktree whenever the child can write — whether the leader said write:true
-		// or a file granted write-capable tools.
-		const canWrite = input.write || (file?.tools?.some((t) => WRITE_TOOLS.includes(t)) ?? false);
+		// Isolation follows the DELIVERED toolset, never the raw request: explicit
+		// tools: [bash] without write:true still gets a worktree, and a file that
+		// narrowed the child to read-only never gets the commit/merge ceremony.
+		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
@@ -673,11 +694,19 @@ 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.
 		let childCwd = wt?.path ?? task.cwd;
 		if (wt) {
-			const rel = relative(wt.root, task.cwd);
-			if (rel && !rel.startsWith("..") && rel !== ".") childCwd = join(wt.path, rel);
+			const rel = relative(wt.root, safeRealPath(task.cwd));
+			if (rel.startsWith("..")) {
+				removeWorktree(wt);
+				wt = undefined;
+			} 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.
@@ -890,9 +919,12 @@ export class SubagentManager {
 			watchdog.dispose();
 			if (timeout) clearTimeout(timeout);
 			child?.dispose();
-			// Failed/aborted: commit whatever partial work exists FIRST (so the branch
-			// really keeps it), then drop the checkout dir.
+			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.
 			if (wt && task.status !== "completed") {
+				await new Promise((r) => setTimeout(r, 250));
 				try {
 					commitWorktree(wt, `subagent ${task.agent} (partial, ${task.status})`);
 				} catch {
@@ -1026,7 +1058,18 @@ export class SubagentManager {
 				// The scheduler passes the index into the FILTERED list — never use it
 				// against the unfiltered inputs. Look the input up by task id instead.
 				const input = inputById.get(task.id);
-				if (!input) return;
+				if (!input) {
+					// Impossible unless ids drift from inputs — fail loudly instead of
+					// leaving the task queued forever (hasActiveRun would never clear).
+					this.updateTask(
+						run,
+						task,
+						{ status: "failed", error: `No input for task ${task.id}`, endedAt: Date.now() },
+						ctx,
+						onUpdate,
+					);
+					return;
+				}
 				await this.runChild(
 					run,
 					task,
@@ -1082,14 +1125,26 @@ export class SubagentManager {
 		for (const task of run.tasks) this.mailboxes.close(`${run.id}:${task.id}`);
 		this.persist(ctx);
 		// Branches merged by the leader since the run ended: drop worktree dir + branch.
-		for (const task of run.tasks) {
-			if (task.branch) {
+		// Once per repo, never for a branch another live run owns, never fatal — a
+		// throw here would re-settle an already-finished run as failed.
+		try {
+			const roots = new Set<string>();
+			for (const task of run.tasks) {
+				if (!task.branch) continue;
 				const root = repoRoot(task.cwd);
-				if (root) cleanupMerged(root);
+				if (root) roots.add(root);
 			}
+			for (const root of roots) cleanupMerged(root, { skipBranches: this.liveBranches() });
+		} catch {
+			/* cleanup is best-effort; the run outcome must stand */
 		}
 	}
 
+	/** Branches owned by worktrees of still-running children. */
+	liveBranches(): Set<string> {
+		return new Set(Array.from(this.liveWorktrees.values(), (wt) => wt.branch));
+	}
+
 	/** Spawn a run that keeps executing after this call returns. Every run is background. */
 	startInBackground(params: SubagentParamsShape, ctx: ExtensionContext): RunDetails {
 		const { run, inputs } = this.createRun(params, ctx);
@@ -1164,7 +1219,10 @@ export class SubagentManager {
 			task.error = task.error || "Canceled by subagent_cancel"; // never overwrite a real error
 			task.endedAt = Date.now();
 			aborted += 1;
-			if (task.branch) removeByBranch(task.cwd, task.branch); // dir only; branch keeps partial work
+			// The branch is recorded here so the leader can still merge partial work;
+			// runChild's finally commits + drops the dir (it owns the live worktree).
+			const wt = this.liveWorktrees.get(`${runId}:${task.id}`);
+			if (wt) task.branch = wt.branch;
 		}
 		run.status = "aborted";
 		run.endedAt = Date.now();
@@ -1219,7 +1277,7 @@ export class SubagentManager {
 			// IN the await result, no steering queue, no turn boundary needed.
 			this.parked.set(runId, { msgs, wake: () => resolve(cloneRun(run)) });
 		});
-		if (timeoutMs) {
+		if (timeoutMs !== undefined && timeoutMs > 0) {
 			return Promise.race([
 				settled.then((r) => {
 					finish();

package/src/worktree.ts

@@ -2,9 +2,12 @@
  *  Worktrees live inside `<repo>/.git/subagents/<runId>/<taskId>` so the child's
  *  ancestor walk still finds the project AGENTS.md chain. node_modules is
  *  symlinked from the main tree. The extension commits the child's changes on
- *  completion; the leader reviews and merges the branch manually; merged
- *  branches are cleaned automatically, crash leftovers are swept at session
- *  start (dir removed, branch kept — the work survives). */
+ *  completion; the leader reviews and merges the branch manually.
+ *
+ *  Cleanup, in order of trust: `reapDeadWorktrees` (session start — nothing can
+ *  be live yet, so every registered worktree is a crash leftover: commit its
+ *  work, drop the dir, keep the branch), `cleanupMerged` (merged branches, never
+ *  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";
@@ -17,6 +20,8 @@ export interface Worktree {
 	base: string; // SHA the branch was created from
 }
 
+const BRANCH_PREFIX = "subagents/";
+
 function git(root: string, args: string[]): string {
 	return execFileSync("git", ["-C", root, ...args], { encoding: "utf8" }).trim();
 }
@@ -35,6 +40,25 @@ 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;
+	try {
+		return realpathSync(a) === realpathSync(b);
+	} catch {
+		return false;
+	}
+}
+
 /** Repo root for cwd, or undefined when not a git repo (or cwd doesn't exist). */
 export function repoRoot(cwd: string): string | undefined {
 	if (!existsSync(cwd)) return undefined;
@@ -50,7 +74,7 @@ export function createWorktree(cwd: string, runId: string, taskId: string): Work
 	const root = repoRoot(cwd);
 	if (!root) return undefined;
 	const path = join(root, ".git", "subagents", runId, taskId);
-	const branch = `subagents/${runId}/${taskId}`;
+	const branch = `${BRANCH_PREFIX}${runId}/${taskId}`;
 	let base: string;
 	try {
 		base = git(root, ["rev-parse", "HEAD"]); // SHA — detached HEAD stays correct
@@ -71,11 +95,19 @@ export function createWorktree(cwd: string, runId: string, taskId: string): Work
 	return { root, path, branch, base };
 }
 
-/** Commit all child changes. No-op when the worktree is already clean. */
+/**
+ * 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.
+ */
 export function commitWorktree(wt: Worktree, message: string): void {
-	if (gitIn(wt.path, ["status", "--porcelain"]).length === 0) return;
-	gitIn(wt.path, ["add", "-A", "--", ".", ":(exclude)node_modules"]); // never stage the dep symlink
-	gitIn(wt.path, ["commit", "-m", message, "--no-verify"]);
+	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
+	gitIn(dir, ["commit", "-m", message, "--no-verify"]);
 }
 
 /** Diffstat + changed files of the branch vs its base SHA. */
@@ -89,23 +121,22 @@ export function branchDiff(wt: Worktree): { stat: string; files: string[] } {
 
 /** Remove the worktree dir. The branch is KEPT (the work survives for merging). */
 export function removeWorktree(wt: Worktree): void {
-	try {
-		git(wt.root, ["worktree", "remove", "--force", wt.path]);
-	} catch {
-		/* already gone */
-	}
-	prune(wt.root);
+	dropDir(wt.root, wt.path);
 }
 
 /** Remove a worktree dir by branch name (cancel paths that didn't keep a Worktree). */
 export function removeByBranch(cwd: string, branch: string): void {
+	if (!branch.startsWith(BRANCH_PREFIX)) return;
 	const root = repoRoot(cwd);
 	if (!root) return;
-	const path = join(root, ".git", "subagents", branch.slice("subagents/".length));
+	dropDir(root, join(root, ".git", "subagents", branch.slice(BRANCH_PREFIX.length)));
+}
+
+function dropDir(root: string, path: string): void {
 	try {
 		git(root, ["worktree", "remove", "--force", path]);
 	} catch {
-		/* already gone */
+		if (existsSync(path)) rmSync(path, { recursive: true, force: true });
 	}
 	prune(root);
 }
@@ -119,21 +150,30 @@ function prune(root: string): void {
 	}
 }
 
+/** Is this branch checked out in some worktree right now? Checked fresh, per call. */
+function isCheckedOut(root: string, branch: string): boolean {
+	return worktreeBranches(root).includes(branch);
+}
+
 /**
  * Delete branch + worktree for branches already merged into `target`.
- * SAFETY: branches checked out in a LIVE worktree (concurrent run) are skipped —
- * their tip equals the base until the child commits, so they look "merged".
+ * SAFETY: a branch checked out in a LIVE worktree is skipped — a fresh branch's
+ * tip equals its base until the child commits, so it looks "merged". The
+ * registration is re-checked immediately before each removal (a concurrent run
+ * may have created its worktree after the first listing).
  */
-export function cleanupMerged(root: string, target = "HEAD"): number {
+export function cleanupMerged(root: string, opts: { skipBranches?: Set<string>; target?: string } = {}): number {
 	root = realpathSync(root);
+	const target = opts.target ?? "HEAD";
 	const merged = git(root, ["branch", "--merged", target])
 		.split("\n")
 		.map((b) => b.trim().replace(/^[+*]\s*/, ""));
-	const live = new Set(worktreeBranches(root));
 	let cleaned = 0;
 	for (const branch of merged) {
-		if (!branch.startsWith("subagents/") || live.has(branch)) continue;
-		const path = join(root, ".git", "subagents", branch.slice("subagents/".length));
+		if (!branch.startsWith(BRANCH_PREFIX)) 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));
 		if (existsSync(path)) {
 			try {
 				git(root, ["worktree", "remove", "--force", path]);
@@ -160,25 +200,43 @@ function worktreeBranches(root: string): string[] {
 }
 
 /**
- * Remove worktree dirs that are NOT registered worktrees (crash leftovers).
- * A crash leaves the dir AND the branch, so branch-existence can't identify
- * leftovers — worktree registration can. Branches always survive.
+ * Session-start recovery: every registered subagent worktree is a crash leftover
+ * (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 {
+	root = realpathSync(root);
+	const sub = join(root, ".git", "subagents");
+	if (!existsSync(sub)) return 0;
+	let reaped = 0;
+	for (const path of worktreePaths(root)) {
+		if (!isInside(path, sub)) continue; // not ours
+		try {
+			commitIn(path, "subagent (recovered after interrupted session)");
+		} catch {
+			/* nothing committable */
+		}
+		dropDir(root, path);
+		reaped += 1;
+	}
+	return reaped;
+}
+
+/**
+ * Remove worktree dirs that git no longer knows about (partial-crash leftovers).
+ * Registered worktrees are never touched here — `reapDeadWorktrees` owns those,
+ * and a live child's checkout must survive.
  */
 export function sweepStale(root: string): void {
 	root = realpathSync(root);
 	const sub = join(root, ".git", "subagents");
 	if (!existsSync(sub)) return;
-	const registered = new Set(worktreePaths(root));
+	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.has(dir)) continue; // live worktree
-			try {
-				git(root, ["worktree", "remove", "--force", dir]);
-			} catch {
-				// dir not a registered worktree (partial crash) — plain rm
-				rmSync(dir, { recursive: true, force: true });
-			}
+			if (registered.some((p) => samePath(p, dir))) continue; // live/registered worktree
+			rmSync(dir, { recursive: true, force: true });
 		}
 	}
 	prune(root);
@@ -189,12 +247,26 @@ function worktreePaths(root: string): string[] {
 		return git(root, ["worktree", "list", "--porcelain"])
 			.split("\n")
 			.filter((l) => l.startsWith("worktree "))
-			.map((l) => l.slice("worktree ".length).trim());
+			.map((l) => unquote(l.slice("worktree ".length).trim()));
 	} catch {
 		return [];
 	}
 }
 
+function isInside(path: string, dir: string): boolean {
+	const p = safeReal(path);
+	const d = safeReal(dir);
+	return p === d || p.startsWith(`${d}/`);
+}
+
+function safeReal(path: string): string {
+	try {
+		return realpathSync(path);
+	} catch {
+		return path;
+	}
+}
+
 function readDirs(dir: string): string[] {
 	try {
 		return readdirSync(dir, { withFileTypes: true })