@gavin-hu/my-pi-agent 0.12.0 → 0.13.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (153) hide show
  1. package/README.md +28 -21
  2. package/README.zh-CN.md +30 -22
  3. package/extensions/bytefly/README.md +139 -67
  4. package/extensions/bytefly/chrome.ts +7 -2
  5. package/extensions/bytefly/doctor/register.ts +6 -0
  6. package/extensions/bytefly/doctor/runtime.ts +36 -6
  7. package/extensions/bytefly/footer/collect/snapshot.ts +12 -2
  8. package/extensions/bytefly/footer/config.ts +7 -1
  9. package/extensions/bytefly/footer/delegation.ts +100 -0
  10. package/extensions/bytefly/footer/register.ts +13 -2
  11. package/extensions/bytefly/footer/render/lines.ts +2 -0
  12. package/extensions/bytefly/footer/render/segments.ts +27 -0
  13. package/extensions/bytefly/footer/types.ts +11 -0
  14. package/extensions/bytefly/index.ts +4 -3
  15. package/extensions/bytefly/notify/config.ts +2 -0
  16. package/extensions/bytefly/notify/format.ts +19 -1
  17. package/extensions/bytefly/notify/register.ts +37 -13
  18. package/extensions/bytefly/notify/types.ts +2 -0
  19. package/extensions/bytefly/panel/commands.ts +34 -13
  20. package/extensions/bytefly/panel/settings.ts +98 -16
  21. package/extensions/bytefly/update/cache.ts +73 -0
  22. package/extensions/bytefly/update/format.ts +65 -10
  23. package/extensions/bytefly/update/register.ts +99 -13
  24. package/extensions/bytefly/update/state.ts +13 -1
  25. package/extensions/document/README.md +22 -12
  26. package/extensions/document/config.ts +4 -0
  27. package/extensions/document/deadline.ts +61 -0
  28. package/extensions/document/tool.ts +13 -2
  29. package/extensions/file-browse/README.md +12 -8
  30. package/extensions/file-browse/client.ts +6 -7
  31. package/extensions/file-browse/files.ts +0 -4
  32. package/extensions/file-browse/git.ts +9 -10
  33. package/extensions/file-browse/html.ts +17 -7
  34. package/extensions/file-browse/router.ts +24 -15
  35. package/extensions/goal/README.md +7 -1
  36. package/extensions/goal/schema.ts +6 -2
  37. package/extensions/goal/types.ts +6 -2
  38. package/extensions/job/README.md +16 -12
  39. package/extensions/job/runtime.ts +23 -6
  40. package/extensions/job/status.ts +4 -2
  41. package/extensions/memory/README.md +55 -13
  42. package/extensions/memory/commands.ts +20 -2
  43. package/extensions/memory/doctor.ts +137 -0
  44. package/extensions/memory/format.ts +24 -7
  45. package/extensions/memory/runtime.ts +49 -0
  46. package/extensions/memory/store.ts +31 -0
  47. package/extensions/memory/tui.ts +136 -12
  48. package/extensions/plan/README.md +60 -29
  49. package/extensions/plan/commands.ts +67 -8
  50. package/extensions/plan/index.ts +35 -11
  51. package/extensions/plan/plan-index.ts +234 -0
  52. package/extensions/plan/plan-text.ts +95 -0
  53. package/extensions/plan/plans.ts +90 -1
  54. package/extensions/plan/review.ts +12 -9
  55. package/extensions/plan/runtime.ts +72 -27
  56. package/extensions/plan/steps.ts +35 -25
  57. package/extensions/plan/todos.ts +24 -2
  58. package/extensions/plan/tools.ts +124 -24
  59. package/extensions/plan/tui.ts +7 -28
  60. package/extensions/plan/types.ts +9 -0
  61. package/extensions/plan/web-page.ts +2 -2
  62. package/extensions/plan/web.ts +12 -11
  63. package/extensions/rewind/README.md +28 -8
  64. package/extensions/rewind/commands.ts +1 -1
  65. package/extensions/rewind/format.ts +4 -4
  66. package/extensions/rewind/git.ts +56 -18
  67. package/extensions/rewind/index.ts +3 -5
  68. package/extensions/rewind/path-picker.ts +198 -0
  69. package/extensions/rewind/paths.ts +128 -0
  70. package/extensions/rewind/restore.ts +69 -28
  71. package/extensions/rewind/rewind.ts +80 -3
  72. package/extensions/rewind/runtime.ts +13 -5
  73. package/extensions/rewind/snapshot-run.ts +5 -2
  74. package/extensions/rewind/types.ts +2 -0
  75. package/extensions/subagent/README.md +7 -1
  76. package/extensions/subagent/accounting.ts +34 -0
  77. package/extensions/subagent/index.ts +23 -1
  78. package/extensions/subagent/stream.ts +2 -0
  79. package/extensions/subagent/types.ts +2 -0
  80. package/extensions/watch/README.md +8 -3
  81. package/extensions/watch/index.ts +19 -0
  82. package/extensions/web-access/README.md +43 -13
  83. package/extensions/web-access/charset.ts +58 -0
  84. package/extensions/web-access/fetch/find.ts +11 -7
  85. package/extensions/web-access/fetch/format.ts +64 -18
  86. package/extensions/web-access/fetch/page.ts +5 -1
  87. package/extensions/web-access/fetch/tool.ts +19 -18
  88. package/extensions/web-access/http.ts +49 -8
  89. package/extensions/web-access/search/facts.ts +38 -0
  90. package/extensions/web-access/search/tool.ts +13 -0
  91. package/extensions/web-access/text.ts +50 -11
  92. package/extensions/worktree/README.md +74 -15
  93. package/extensions/worktree/config.ts +11 -0
  94. package/extensions/worktree/exit.ts +35 -3
  95. package/extensions/worktree/gaps.ts +74 -0
  96. package/extensions/worktree/guard.ts +50 -4
  97. package/extensions/worktree/handoff.ts +55 -0
  98. package/extensions/worktree/index.ts +12 -2
  99. package/extensions/worktree/runtime.ts +20 -0
  100. package/extensions/worktree/status.ts +15 -2
  101. package/extensions/worktree/tools.ts +3 -2
  102. package/lib/README.md +10 -8
  103. package/lib/events.ts +86 -1
  104. package/lib/goal-state.ts +95 -0
  105. package/lib/requirements.ts +282 -62
  106. package/lib/tui/README.md +91 -0
  107. package/lib/tui/glyphs.ts +1 -0
  108. package/lib/web/layout.ts +12 -2
  109. package/package.json +5 -3
  110. package/themes/bytefly-dark.json +1 -1
  111. package/assets/chrome.png +0 -0
  112. package/assets/file-browse.png +0 -0
  113. package/assets/job.png +0 -0
  114. package/assets/plan-review.png +0 -0
  115. package/assets/question.png +0 -0
  116. package/assets/rewind.png +0 -0
  117. package/assets/todo.png +0 -0
  118. package/assets/tool-renderings.png +0 -0
  119. package/assets/tools/ask_user_question-cancelled.png +0 -0
  120. package/assets/tools/ask_user_question.png +0 -0
  121. package/assets/tools/bash.png +0 -0
  122. package/assets/tools/edit.png +0 -0
  123. package/assets/tools/enter_plan_mode.png +0 -0
  124. package/assets/tools/enter_worktree.png +0 -0
  125. package/assets/tools/exit_plan_mode.png +0 -0
  126. package/assets/tools/exit_worktree.png +0 -0
  127. package/assets/tools/find.png +0 -0
  128. package/assets/tools/goal.png +0 -0
  129. package/assets/tools/grep.png +0 -0
  130. package/assets/tools/job.png +0 -0
  131. package/assets/tools/list_worktrees.png +0 -0
  132. package/assets/tools/ls.png +0 -0
  133. package/assets/tools/memory-long-subject.png +0 -0
  134. package/assets/tools/memory.png +0 -0
  135. package/assets/tools/notice.png +0 -0
  136. package/assets/tools/open_path.png +0 -0
  137. package/assets/tools/prune_worktrees.png +0 -0
  138. package/assets/tools/read.png +0 -0
  139. package/assets/tools/read_doc-running.png +0 -0
  140. package/assets/tools/read_doc.png +0 -0
  141. package/assets/tools/run_doctor.png +0 -0
  142. package/assets/tools/send_wechat.png +0 -0
  143. package/assets/tools/set_plan_steps.png +0 -0
  144. package/assets/tools/subagent.png +0 -0
  145. package/assets/tools/todo.png +0 -0
  146. package/assets/tools/turn-rule.png +0 -0
  147. package/assets/tools/user-message.png +0 -0
  148. package/assets/tools/watch.png +0 -0
  149. package/assets/tools/web_fetch-rejected.png +0 -0
  150. package/assets/tools/web_fetch.png +0 -0
  151. package/assets/tools/web_search.png +0 -0
  152. package/assets/tools/write.png +0 -0
  153. package/assets/tools/write_plan.png +0 -0
@@ -27,7 +27,7 @@ import { charLength } from "../text.ts";
27
27
  import { oneLine } from "../transcript.ts";
28
28
  import { createPageCache, type PageCache } from "./cache.ts";
29
29
  import type { WebFetchConfig } from "./config.ts";
30
- import { findPassages } from "./find.ts";
30
+ import { findPassages, type Passage } from "./find.ts";
31
31
  import { formatMatches, formatPage } from "./format.ts";
32
32
  import { runFetch, type FetchDeps, type PageResult } from "./page.ts";
33
33
  import { resolveRequest, WebFetchOutput, WebFetchParams, type FetchResponse, type WebFetchArgs } from "./schema.ts";
@@ -85,29 +85,33 @@ async function loadPage(
85
85
  function pageResult(loaded: LoadedPage, request: FetchRequest): { response: FetchResponse; text: string } {
86
86
  const { page, cached, fetchedAt } = loaded;
87
87
  const finding = request.find.length > 0;
88
- const matches = finding
89
- ? findPassages(page.text, request.find, {
90
- mode: request.mode,
91
- contextChars: request.contextChars,
92
- maxMatches: request.maxMatches + 1,
93
- })
94
- : [];
95
- // Request one extra so a full list proves more hits exist; only the cap is shown.
96
- const matchesTruncated = finding && matches.length > request.maxMatches;
97
- const visibleMatches = matchesTruncated ? matches.slice(0, request.maxMatches) : matches;
98
88
 
99
89
  let formatted: { body: string; text: string; truncated: boolean };
100
90
  let nextIndex: number;
91
+ let matches: Passage[] = [];
92
+ let matchesTruncated = false;
93
+
101
94
  if (finding) {
102
- formatted = formatMatches({
95
+ // Request one extra so a full list proves more hits exist; only the cap is shown.
96
+ const found = findPassages(page.text, request.find, {
97
+ mode: request.mode,
98
+ contextChars: request.contextChars,
99
+ maxMatches: request.maxMatches + 1,
100
+ });
101
+ const cappedByMaxMatches = found.length > request.maxMatches;
102
+ const rendered = formatMatches({
103
103
  finalUrl: page.finalUrl,
104
104
  title: page.title,
105
105
  status: page.status,
106
106
  contentType: page.contentType,
107
- matches: visibleMatches,
107
+ matches: cappedByMaxMatches ? found.slice(0, request.maxMatches) : found,
108
108
  find: request.find,
109
- truncated: matchesTruncated,
109
+ maxChars: request.maxChars,
110
+ cappedByMaxMatches,
110
111
  });
112
+ formatted = rendered;
113
+ matches = rendered.included;
114
+ matchesTruncated = rendered.truncated;
111
115
  // Find results are not paged, so the caller keeps reading from the same offset.
112
116
  nextIndex = request.startIndex;
113
117
  } else {
@@ -136,7 +140,7 @@ function pageResult(loaded: LoadedPage, request: FetchRequest): { response: Fetc
136
140
  nextIndex,
137
141
  truncated: formatted.truncated,
138
142
  cached,
139
- matches: visibleMatches,
143
+ matches,
140
144
  matchesTruncated,
141
145
  fetchedAt,
142
146
  error: "",
@@ -192,9 +196,6 @@ export function registerFetchTool(pi: ExtensionAPI, deps: FetchToolDeps = {}): v
192
196
  parameters: WebFetchParams,
193
197
  outputSchema: WebFetchOutput,
194
198
  annotations: { readOnlyHint: true, openWorldHint: true },
195
- // The page cache is shared in-memory state created per registration, so
196
- // sibling calls must not run concurrently (matches web_search's throttle).
197
- executionMode: "sequential",
198
199
 
199
200
  async execute(_toolCallId, params, signal, onUpdate, ctx) {
200
201
  const config = loadFetchConfig(ctx.cwd);
@@ -3,10 +3,12 @@
3
3
  *
4
4
  * No `curl`, no dependencies: one `fetch` call per request (with optional
5
5
  * manual redirect handling) with a combined caller-signal and timeout, optional
6
- * request body, and a size cap. `HttpRunner` is the seam tests use to avoid the
7
- * network.
6
+ * request body, and a size cap enforced while the body is read. `HttpRunner` is
7
+ * the seam tests use to avoid the network.
8
8
  */
9
9
 
10
+ import { bomCharset, charsetOf, decodeBytes } from "./charset.ts";
11
+
10
12
  /** A request to send. */
11
13
  interface HttpRequest {
12
14
  url: string;
@@ -15,7 +17,7 @@ interface HttpRequest {
15
17
  /** Raw request body (for example a JSON-RPC payload). */
16
18
  body?: string;
17
19
  timeoutMs: number;
18
- /** Maximum response size in bytes; larger responses raise `HttpTooLargeError`. */
20
+ /** Maximum response size in bytes, enforced while the body is read. */
19
21
  maxBytes?: number;
20
22
  /** Redirect handling: `"follow"` (default) or `"manual"` to inspect each hop. */
21
23
  redirect?: "follow" | "manual";
@@ -63,6 +65,48 @@ export class HttpUnavailableError extends Error {
63
65
  }
64
66
  }
65
67
 
68
+ /**
69
+ * Concatenate chunks into one buffer. `total` is the caller's running byte count,
70
+ * so the allocation happens once rather than growing an array per chunk.
71
+ */
72
+ function concatChunks(chunks: Uint8Array[], total: number): Uint8Array {
73
+ const merged = new Uint8Array(total);
74
+ let offset = 0;
75
+ for (const chunk of chunks) {
76
+ merged.set(chunk, offset);
77
+ offset += chunk.byteLength;
78
+ }
79
+ return merged;
80
+ }
81
+
82
+ /**
83
+ * Read a response body to completion, aborting the read as soon as it exceeds
84
+ * `maxBytes`. `response.arrayBuffer()` would buffer a whole chunked or
85
+ * compressed response first, which is the case the cap exists to prevent.
86
+ */
87
+ async function readCappedBody(response: Response, maxBytes?: number): Promise<Uint8Array> {
88
+ if (!response.body) {
89
+ const bytes = new Uint8Array(await response.arrayBuffer());
90
+ if (maxBytes !== undefined && bytes.byteLength > maxBytes) throw new HttpTooLargeError(maxBytes);
91
+ return bytes;
92
+ }
93
+
94
+ const reader = response.body.getReader();
95
+ const chunks: Uint8Array[] = [];
96
+ let total = 0;
97
+ for (;;) {
98
+ const { done, value } = await reader.read();
99
+ if (done) break;
100
+ total += value.byteLength;
101
+ if (maxBytes !== undefined && total > maxBytes) {
102
+ await reader.cancel().catch(() => undefined);
103
+ throw new HttpTooLargeError(maxBytes);
104
+ }
105
+ chunks.push(value);
106
+ }
107
+ return concatChunks(chunks, total);
108
+ }
109
+
66
110
  /**
67
111
  * Create a runner backed by `fetch`.
68
112
  *
@@ -103,11 +147,8 @@ export function createFetchRunner(fetchImpl: typeof fetch = globalThis.fetch): H
103
147
  throw new HttpTooLargeError(request.maxBytes);
104
148
  }
105
149
 
106
- const bytes = new Uint8Array(await response.arrayBuffer());
107
- if (request.maxBytes !== undefined && bytes.byteLength > request.maxBytes) {
108
- throw new HttpTooLargeError(request.maxBytes);
109
- }
110
- const body = new TextDecoder().decode(bytes);
150
+ const bytes = await readCappedBody(response, request.maxBytes);
151
+ const body = decodeBytes(bytes, bomCharset(bytes) ?? charsetOf(contentType));
111
152
  const location = response.headers.get("location") ?? undefined;
112
153
 
113
154
  return {
@@ -0,0 +1,38 @@
1
+ /**
2
+ * What `web_search` resolved to, published for `doctor`.
3
+ *
4
+ * The provider registry, each provider's `isConfigured`, and the `auto`
5
+ * resolution rule all live in this extension, so this module is where the facts
6
+ * for a diagnostic come from. It publishes a summary only — the key itself never
7
+ * leaves the extension, just the variable name and whether a literal key exists.
8
+ *
9
+ * `endpoint` is deliberately null for the providers without a configurable
10
+ * endpoint: probing an API host would report a 401 as a failure for a provider
11
+ * that works, and there is nothing for the user to fix.
12
+ */
13
+
14
+ import type { WebSearchProviderEvent } from "../../../lib/events.ts";
15
+ import type { WebSearchConfig } from "./config.ts";
16
+ import { DEFAULT_PROVIDER_ID, resolveProvider } from "./registry.ts";
17
+
18
+ /**
19
+ * The facts for one session's config. `reason` distinguishes an explicit choice
20
+ * from the auto rule, so a report can tell "you asked for this and it is not
21
+ * configured" from "this is the keyless fallback".
22
+ */
23
+ export function providerFacts(config: WebSearchConfig): WebSearchProviderEvent {
24
+ const provider = resolveProvider(config);
25
+ // `auto` only leaves the keyless default when another provider is configured,
26
+ // so the resolved id tells the two apart.
27
+ const reason: WebSearchProviderEvent["reason"] =
28
+ config.provider !== "auto" ? "explicit" : provider.id === DEFAULT_PROVIDER_ID ? "default" : "configured";
29
+ return {
30
+ provider: provider.id,
31
+ reason,
32
+ configured: provider.isConfigured(config),
33
+ // Only SearXNG's base URL is a setting a user can get wrong.
34
+ endpoint: provider.id === "searxng" && config.endpoint !== "" ? config.endpoint : null,
35
+ apiKeyEnv: config.apiKeyEnv || null,
36
+ keyConfigured: config.apiKey.trim() !== "",
37
+ };
38
+ }
@@ -8,6 +8,7 @@
8
8
  */
9
9
 
10
10
  import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
11
+ import { WEB_SEARCH_PROVIDER_CHANNEL } from "../../../lib/events.ts";
11
12
  import { loadSearchConfig } from "../config.ts";
12
13
  import { oneLine } from "../transcript.ts";
13
14
  import { reuseText } from "../../../lib/tui/chrome.ts";
@@ -21,6 +22,7 @@ import {
21
22
  } from "../../../lib/tui/tool.ts";
22
23
  import { paint } from "../../../lib/tui/tokens.ts";
23
24
  import { expandHint, SEPARATORS } from "../../../lib/tui/glyphs.ts";
25
+ import { providerFacts } from "./facts.ts";
24
26
  import { formatResults } from "./format.ts";
25
27
  import { createThrottle, runSearch, type SearchDeps } from "./search.ts";
26
28
  import { resolveRequest, WebSearchOutput, WebSearchParams, type SearchResponse, type WebSearchArgs } from "./schema.ts";
@@ -32,6 +34,17 @@ export function registerSearchTool(pi: ExtensionAPI, deps: SearchDeps = {}): voi
32
34
  // without any module-level state.
33
35
  const runDeps: SearchDeps = { http: deps.http, throttle: deps.throttle ?? createThrottle() };
34
36
 
37
+ // Publish what this session resolved to, so `doctor` can report the provider,
38
+ // its key, and its endpoint without importing this extension. Accounting only:
39
+ // a bus listener that throws must not break the tool.
40
+ pi.on("session_start", (_event, ctx) => {
41
+ try {
42
+ pi.events.emit(WEB_SEARCH_PROVIDER_CHANNEL, providerFacts(loadSearchConfig(ctx.cwd)));
43
+ } catch {
44
+ // Diagnostics are best-effort.
45
+ }
46
+ });
47
+
35
48
  pi.registerTool({
36
49
  name: TOOL_NAME,
37
50
  label: "Web search",
@@ -3,24 +3,63 @@
3
3
  *
4
4
  * Character budgets are counted in code points, not UTF-16 units, so a slice
5
5
  * never splits a CJK character or an emoji, and a declared limit is honoured for
6
- * the text the model sees. `web_search` and `web_fetch` measure and slice here
7
- * rather than re-deriving `Array.from` at each site.
6
+ * the text the model sees. Everything here walks the string in place: a
7
+ * multi-megabyte page is never materialised as an array of one-character
8
+ * strings, which is what `Array.from(text)` would cost on every call.
8
9
  */
9
10
 
10
- /** Split text into code points (an array of one-code-point strings). */
11
- export function toChars(text: string): string[] {
12
- return Array.from(text);
13
- }
11
+ const SURROGATE = /[\uD800-\uDFFF]/;
14
12
 
15
13
  /** Length of text in code points. */
16
14
  export function charLength(text: string): number {
17
- return Array.from(text).length;
15
+ // A native scan with no allocation covers the common all-BMP page.
16
+ if (!SURROGATE.test(text)) return text.length;
17
+
18
+ let count = 0;
19
+ for (let i = 0; i < text.length; i++) {
20
+ const code = text.charCodeAt(i);
21
+ if (code >= 0xd800 && code <= 0xdbff) {
22
+ const next = text.charCodeAt(i + 1);
23
+ if (next >= 0xdc00 && next <= 0xdfff) i++;
24
+ }
25
+ count++;
26
+ }
27
+ return count;
28
+ }
29
+
30
+ /**
31
+ * Slice text by code-point offsets, like `Array.from(text).slice(from, to).join("")`
32
+ * without building the array. Out-of-range offsets clamp, and `to` defaults to
33
+ * the end of the string.
34
+ */
35
+ export function sliceChars(text: string, from: number, to: number = Number.POSITIVE_INFINITY): string {
36
+ const start = Math.max(0, Math.floor(from));
37
+ const end = Math.max(start, Math.floor(to));
38
+ if (start === 0 && end >= text.length) return text;
39
+
40
+ let startIndex = text.length;
41
+ let endIndex = text.length;
42
+ let codePoint = 0;
43
+ for (let i = 0; i < text.length; i++) {
44
+ if (codePoint === start && startIndex === text.length) startIndex = i;
45
+ if (codePoint === end) {
46
+ endIndex = i;
47
+ break;
48
+ }
49
+ const code = text.charCodeAt(i);
50
+ if (code >= 0xd800 && code <= 0xdbff) {
51
+ const next = text.charCodeAt(i + 1);
52
+ if (next >= 0xdc00 && next <= 0xdfff) i++;
53
+ }
54
+ codePoint++;
55
+ }
56
+ return text.slice(startIndex, endIndex);
18
57
  }
19
58
 
20
59
  /** Truncate to `maxChars` code points, appending `marker` when anything was cut. */
21
60
  export function truncateChars(text: string, maxChars: number, marker = "…"): string {
22
- const points = toChars(text);
23
- if (points.length <= maxChars) return text;
24
- if (maxChars <= marker.length) return marker;
25
- return `${points.slice(0, maxChars - marker.length).join("")}${marker}`;
61
+ if (charLength(text) <= maxChars) return text;
62
+ const markerLength = charLength(marker);
63
+ if (maxChars <= markerLength) return marker;
64
+ return `${sliceChars(text, 0, maxChars - markerLength)}${marker}`;
26
65
  }
@@ -23,11 +23,16 @@ How the `list_worktrees` call and its result look in the transcript — the call
23
23
  ## What it does
24
24
 
25
25
  - `enter_worktree` creates `<repo>/.pi/worktrees/<name>` on branch
26
- `worktree-<name>` (or enters an existing worktree by `path`) and rebinds the
27
- session's working root to it.
26
+ `worktree-<name>` cut from `origin/main` (which can lag the local `main`), or
27
+ enters an existing worktree by `path`, and rebinds the session's working root
28
+ to it.
28
29
  - `exit_worktree` returns to the main checkout. A clean worktree is removed
29
30
  automatically; one with uncommitted changes or new commits prompts (or is
30
- kept in non-interactive mode). A branch that still holds work (uncommitted
31
+ kept in non-interactive mode). Its result ends with a **handoff**: the branch,
32
+ how many commits it holds beyond the merge target, whether that target already
33
+ contains them, and the `git merge --no-ff` and `gh pr create` commands that
34
+ finish the task. The tool never commits, merges, pushes, or opens the PR. A
35
+ branch that still holds work (uncommitted
31
36
  changes, or commits not reachable from the recorded base) is never
32
37
  force-deleted, even with `remove: true`.
33
38
  - While isolated, `read`/`write`/`edit`/`bash`/`grep`/`find`/`ls` all resolve
@@ -56,14 +61,32 @@ changes; a branch that still holds work is kept. Declining a confirmation
56
61
  (entering an outside path, exiting a worktree with work) is reported as normal
57
62
  feedback, not an error.
58
63
 
64
+ `/worktree status` (and `list_worktrees`) reports the active worktree, any
65
+ override that is not re-rooted, what isolation does not cover, and the managed
66
+ worktrees:
67
+
68
+ ```text
69
+ Not isolated:
70
+ ! shell runs in the main checkout; Pi has no mutable session cwd
71
+ powershell allowed by worktree.guard.blockPowershell: false; not re-rooted or guarded
72
+ unverifiable $(...), backticks, eval, and sh -c pass through (blockUnparsableCommands is off); 2 ran this session
73
+ ```
74
+
75
+ `powershell` appears only on Windows, with that tool active, and only when
76
+ `guard.blockPowershell` lets it through — by default the guard refuses the call
77
+ instead, so there is no gap to report. `unverifiable` appears only while
78
+ `guard.blockUnparsableCommands` is off; its trailing count is this session's,
79
+ reset at session start. `doctor` reports the `powershell` switch as `degraded`
80
+ when it is off and notes the two structural gaps, which no setting closes.
81
+
59
82
  The model surface mirrors this with four tools:
60
83
 
61
84
  | Tool | Usage |
62
85
  |---|---|
63
86
  | `enter_worktree` | `name` to create a worktree, or `path` to enter an existing one. Errors if already isolated. |
64
- | `exit_worktree` | optional `remove` and `keepBranch`. Returns the clean/keep/remove decision. |
87
+ | `exit_worktree` | optional `remove` and `keepBranch`. Returns the clean/keep/remove decision and a `handoff` (`branch`, `base`, `ahead`, `merged`). |
65
88
  | `prune_worktrees` | no parameters. Removes eligible managed worktrees and reports the rest. |
66
- | `list_worktrees` | no parameters. Read-only status: active root, inactive overrides, managed worktrees. |
89
+ | `list_worktrees` | no parameters. Read-only status: active root, inactive overrides, isolation gaps, managed worktrees. |
67
90
 
68
91
  ## Configuration
69
92
 
@@ -87,7 +110,8 @@ The `worktree` section of the shared `bytefly.json` (global document, then
87
110
  "blockReadEscapes": false,
88
111
  "blockGitRedirects": true,
89
112
  "blockSymlinkEscapes": true,
90
- "blockUnparsableCommands": false
113
+ "blockUnparsableCommands": false,
114
+ "blockPowershell": true
91
115
  }
92
116
  }
93
117
  ```
@@ -118,7 +142,8 @@ variable, `~user`, a backtick) is treated as escaping. File-descriptor
118
142
  duplications (`2>&1`, `>&-`), heredocs, and here-strings are not treated as
119
143
  paths, and options are only recognized in real argument position, so
120
144
  `git log --grep='--git-dir=/tmp/x'` and `git log -C` are not false positives.
121
- `GIT_DIR=`/`GIT_WORK_TREE=` are refused both as a command prefix and when
145
+ The null device is exempt: `>/dev/null` and `< /dev/null` write and read nothing,
146
+ so they are allowed. `GIT_DIR=`/`GIT_WORK_TREE=` are refused both as a command prefix and when
122
147
  exported (`export GIT_DIR=/main`), and a `sh -c`/`bash -c` sub-command is refused
123
148
  when `blockUnparsableCommands` is enabled.
124
149
 
@@ -126,6 +151,15 @@ This guard covers the built-in file tools and these common shell forms. It
126
151
  cannot contain a program that computes and writes an absolute path itself (for
127
152
  example a script or an inlined interpreter).
128
153
 
154
+ Windows `powershell` is refused outright while the session is isolated rather
155
+ than parsed. It is not one of the tools this extension re-roots, and its language
156
+ is a second grammar — backtick escapes, `Set-Location`, `$env:GIT_DIR`, `--%` —
157
+ so the bash checks would be guessing about a shell that starts in the main
158
+ checkout. Refusing it is the same answer plan mode gives, and `bash` (re-rooted
159
+ and guarded) stays available. `guard.blockPowershell: false` allows the tool
160
+ again; that is a deliberate trade, so it is reported: `/worktree` lists the gap
161
+ and `doctor` reports the switch as `degraded`.
162
+
129
163
  ## Gitignored files
130
164
 
131
165
  A worktree is a fresh checkout, so untracked, gitignored files such as `.env`
@@ -187,13 +221,22 @@ use, falling back to directory mtime for a worktree with no record.
187
221
  writes are guarded (relative paths are not isolated), and `list_worktrees`
188
222
  reports it. `skipOverrides` is read at load time, from the process working
189
223
  directory.
224
+ - `git branch --merged main` calls a worktree branch merged whenever it matches
225
+ its base, so a worktree that never committed looks landed while its
226
+ uncommitted work is still there. Read `git -C <path> status` before removing
227
+ one; `list_worktrees` reports the same clean/merged flags.
190
228
  - The guard is best-effort at the shell level: command substitution, `eval`, and
191
229
  `sh -c` are only covered when `blockUnparsableCommands` is enabled, and real
192
- isolation of arbitrary shell writes needs an OS sandbox.
230
+ isolation of arbitrary shell writes needs an OS sandbox. `/worktree` names the
231
+ gap and counts the commands that passed through it this session.
193
232
  - Only the model's `bash` tool is re-rooted and guarded. A shell command the
194
233
  user runs directly (`!` in the TUI / the `user_bash` event) is not re-rooted
195
234
  and still runs in the main checkout; Pi has no mutable session cwd to change
196
- that. `/worktree` and `list_worktrees` report the active root.
235
+ that. `/worktree` and `list_worktrees` report the active root, and list the
236
+ shell under `Not isolated:` on every status read.
237
+ - Windows `powershell` is refused while isolated, or — with
238
+ `guard.blockPowershell: false` — allowed and unguarded. There is no third
239
+ state: a parsed PowerShell command that keeps the tool.
197
240
 
198
241
  ## Non-goals
199
242
 
@@ -202,19 +245,19 @@ use, falling back to directory mtime for a worktree with no record.
202
245
  knows about. It does not rename, rebase, merge, or push branches.
203
246
  - **No user-shell re-rooting.** A shell the user runs directly (`!` /
204
247
  `user_bash`) is not isolated.
205
- - **No Windows `powershell` surface.** The overrides guard the POSIX `bash`
206
- tool only.
248
+ - **No PowerShell guard.** `powershell` is refused while isolated, not parsed;
249
+ teaching the guard a second grammar belongs with an OS sandbox, not here.
207
250
 
208
251
  ## Pi integration
209
252
 
210
253
  | Contract | Detail |
211
254
  |---|---|
212
255
  | Registration | The default export calls `registerWorktree`: root-tool overrides, the `--worktree` flag, lifecycle listeners, model tools, and the `/worktree` command. Nothing long-lived starts at load. |
213
- | Tools | `enter_worktree`, `exit_worktree`, `prune_worktrees`, `list_worktrees` via `registerTool`. All `executionMode: "sequential"`. Annotations: `enter_worktree` destructive + open-world; `exit_worktree`/`prune_worktrees` destructive; `list_worktrees` read-only. No `outputSchema`; structured state is returned in tool-result `details`. |
214
- | Built-in overrides | `read`/`write`/`edit`/`bash`/`grep`/`find`/`ls` are re-registered under the same names with `defaultActive: false`; each spreads the built-in definition and replaces only `execute`. `skipOverrides` is applied at load time. |
256
+ | Tools | `enter_worktree`, `exit_worktree`, `prune_worktrees`, `list_worktrees` via `registerTool`. All `executionMode: "sequential"`. Annotations: `enter_worktree` destructive + open-world; `exit_worktree`/`prune_worktrees` destructive; `list_worktrees` read-only. No `outputSchema`; structured state is returned in tool-result `details` — `exit_worktree` carries `{ state, removed, handoff }`, where `handoff` is `{ branch, base, ahead, merged }`. |
257
+ | Built-in overrides | `read`/`write`/`edit`/`bash`/`grep`/`find`/`ls` are re-registered under the same names with `defaultActive: false`; each spreads the built-in definition and replaces only `execute`. `skipOverrides` is applied at load time. `powershell` is deliberately not overridden: it cannot be re-rooted without a guard, so the `tool_call` handler refuses it (see `guard.blockPowershell`). |
215
258
  | Commands | `/worktree` via `registerCommand`, with argument completions for the subcommands and exit flags. |
216
259
  | Flags | `--worktree` via `registerFlag`. |
217
- | Events | `session_start` (bind inherited env, restore or refuse recorded state, read the flag), `session_shutdown` (release the lock, clear status), `before_agent_start` (set the prompt cwd and worktree section), `tool_call` (guard). |
260
+ | Events | `session_start` (bind inherited env, restore or refuse recorded state, read the flag), `session_shutdown` (release the lock, clear status), `before_agent_start` (set the prompt cwd and worktree section), `tool_call` (guard file tools and `bash`, refuse `powershell`, and count a command it lets through unverified). |
218
261
  | State | The binding is a custom session entry (`worktree`) written with `pi.appendEntry` and rebuilt from `ctx.sessionManager.getBranch()` on `session_start`. `.pi/worktrees/index.json` is a disposable managed-record cache, reconciled against `git worktree list`. Child processes inherit `PI_WORKTREE_*` env. |
219
262
  | UI/events | Status chip via `ctx.ui.setStatus`; `worktree:changed` published on `pi.events`. |
220
263
 
@@ -229,6 +272,14 @@ use, falling back to directory mtime for a worktree with no record.
229
272
  - **Two layers: re-root and guard.** Re-rooting relative paths is not enough on
230
273
  its own — an absolute path, `cd`, or `git -C` still escapes. The guard runs
231
274
  centrally from the `tool_call` handler, so nested tool calls are covered too.
275
+ - **A gap the user cannot see is a gap they cannot weigh.** Re-rooting is not
276
+ isolating: Pi has no mutable session cwd, so the user's own `!` shell stays in
277
+ the main checkout, an unverifiable command passes by default, and `powershell`
278
+ would run unguarded if it were allowed. Those rules live in `gaps.ts` as data,
279
+ the status text renders them, and the `tool_call` handler counts the one gap the
280
+ transcript cannot show: a command the guard let through without verifying it.
281
+ The two gaps a setting can close are closed — `powershell` is refused, and an
282
+ opt-out is reported by the status and by `doctor`.
232
283
  - **Core model: Checkout / Managed record / Binding.** A *Checkout* is a real
233
284
  `git worktree` and git is authoritative about it. A *Managed record* is
234
285
  provenance and usage in a disposable index. A *Binding* is this session's
@@ -245,6 +296,12 @@ use, falling back to directory mtime for a worktree with no record.
245
296
  - **Never force out work.** `exit_worktree` never force-deletes a branch that
246
297
  still holds uncommitted changes or commits not reachable from the recorded
247
298
  base, even with `remove: true`.
299
+ - **A handoff is read, not performed.** The exit reports the branch's state and
300
+ the command that finishes it, because a merge touches the main checkout and can
301
+ conflict — the user should be the one in that state. The merge target is
302
+ `origin/HEAD`'s branch, falling back to the main checkout's own branch when the
303
+ repository has no remote; when neither resolves, the block names the branch and
304
+ prints no command rather than guessing `main`.
248
305
  - **Lock, then prune.** Entering a managed worktree takes a `git worktree lock`
249
306
  so a concurrent sweep cannot remove it; prune releases stale locks first and
250
307
  keeps any worktree it cannot safely remove, with a reason.
@@ -262,10 +319,12 @@ use, falling back to directory mtime for a worktree with no record.
262
319
  | `lifecycle.ts` | Barrel: `enterWorktree` / `exitWorktree` / `pruneWorktrees` / status |
263
320
  | `errors.ts` | Cancellation error that distinguishes a user decline from a failure |
264
321
  | `enter.ts` / `exit.ts` / `prune.ts` | The lifecycle phases |
322
+ | `handoff.ts` | The `Handoff:` block an exit leaves: the branch's lead, whether the base contains it, and the merge or PR command. |
265
323
  | `tools.ts` / `commands.ts` | Model tools / slash commands |
266
324
  | `root-tools.ts` | Load-time built-in overrides bound to the active root |
267
325
  | `git.ts` | Git plumbing (shared runner + read helpers) |
268
- | `guard.ts` | Path, shell, and git-redirect isolation checks |
326
+ | `guard.ts` | Path, shell, and git-redirect isolation checks, plus the `powershell` refusal |
327
+ | `gaps.ts` | The isolation gaps a session carries, and how they read (pure) |
269
328
  | `include.ts` | `.worktreeinclude` matcher and copy |
270
329
  | `config.ts` / `state.ts` / `status.ts` | Config, session entry shape, status text |
271
330
  | `registry.ts` | Disposable managed-worktree index + derived status |
@@ -20,6 +20,15 @@ export interface WorktreeGuardConfig {
20
20
  blockSymlinkEscapes: boolean;
21
21
  /** Block commands whose git behavior cannot be verified. Default: false */
22
22
  blockUnparsableCommands: boolean;
23
+ /**
24
+ * Refuse the `powershell` tool while isolated. Default: true
25
+ *
26
+ * Pi's `powershell` tool is neither re-rooted nor guarded — a second grammar
27
+ * (backtick escapes, `Set-Location`, `$env:GIT_DIR`) would need a second parser —
28
+ * so inside a worktree its relative paths resolve at the main checkout. Refusing
29
+ * it is the only honest isolation; `false` allows it and is reported as a gap.
30
+ */
31
+ blockPowershell: boolean;
23
32
  }
24
33
 
25
34
  export interface WorktreeConfig {
@@ -57,6 +66,7 @@ export const DEFAULT_CONFIG: WorktreeConfig = {
57
66
  blockGitRedirects: true,
58
67
  blockSymlinkEscapes: true,
59
68
  blockUnparsableCommands: false,
69
+ blockPowershell: true,
60
70
  },
61
71
  };
62
72
 
@@ -81,6 +91,7 @@ function normalizeGuard(raw: unknown, base: WorktreeGuardConfig): WorktreeGuardC
81
91
  blockGitRedirects: booleanOr(value.blockGitRedirects, base.blockGitRedirects),
82
92
  blockSymlinkEscapes: booleanOr(value.blockSymlinkEscapes, base.blockSymlinkEscapes),
83
93
  blockUnparsableCommands: booleanOr(value.blockUnparsableCommands, base.blockUnparsableCommands),
94
+ blockPowershell: booleanOr(value.blockPowershell, base.blockPowershell),
84
95
  };
85
96
  }
86
97
 
@@ -6,7 +6,17 @@
6
6
  import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
7
7
  import { loadConfig } from "./config.ts";
8
8
  import { WorktreeCancelledError } from "./errors.ts";
9
- import { deleteBranch, inspectWork, lockWorktree, unlockWorktree, worktreeRemove } from "./git.ts";
9
+ import {
10
+ currentBranch,
11
+ defaultBranch,
12
+ deleteBranch,
13
+ inspectWork,
14
+ isBranchMerged,
15
+ lockWorktree,
16
+ unlockWorktree,
17
+ worktreeRemove,
18
+ } from "./git.ts";
19
+ import { handoffLines, type Handoff } from "./handoff.ts";
10
20
  import { removeRecord, touchRecord } from "./registry.ts";
11
21
  import { applyWorktreeEnv, clearConfigCache, getActive, publishWorktree, setActive, setStatus } from "./runtime.ts";
12
22
  import { persistState, type WorktreeState } from "./state.ts";
@@ -17,11 +27,30 @@ export interface ExitOptions {
17
27
  keepBranch?: boolean;
18
28
  }
19
29
 
30
+ /**
31
+ * What the branch holds and how to finish it, computed after the outcome is known.
32
+ * A failed lookup degrades to the facts already in hand: an exit that succeeded
33
+ * must not fail because the default branch could not be read.
34
+ */
35
+ async function describeHandoff(pi: ExtensionAPI, state: WorktreeState, ahead: number): Promise<Handoff> {
36
+ if (!state.branch) return { branch: null, base: null, ahead, merged: false };
37
+ try {
38
+ // `origin/HEAD` names the merge target when the repository has a remote; a
39
+ // local-only repository has none, and the main checkout's own branch is the
40
+ // branch a merge would actually land on.
41
+ const base = (await defaultBranch(pi, state.repoRoot)) ?? (await currentBranch(pi, state.repoRoot)) ?? null;
42
+ const merged = base ? await isBranchMerged(pi, state.repoRoot, state.branch, base) : false;
43
+ return { branch: state.branch, base, ahead, merged };
44
+ } catch {
45
+ return { branch: state.branch, base: null, ahead, merged: false };
46
+ }
47
+ }
48
+
20
49
  export async function exitWorktree(
21
50
  pi: ExtensionAPI,
22
51
  ctx: ExtensionContext,
23
52
  options: ExitOptions,
24
- ): Promise<{ state: WorktreeState; removed: boolean; output: string }> {
53
+ ): Promise<{ state: WorktreeState; removed: boolean; handoff: Handoff; output: string }> {
25
54
  const state = getActive();
26
55
  if (!state) throw new Error("Not currently in a worktree.");
27
56
  if (state.borrowed) {
@@ -118,5 +147,8 @@ export async function exitWorktree(
118
147
  touchRecord(state.repoRoot, config, state.path);
119
148
  }
120
149
 
121
- return { state, removed, output: lines.join("\n") };
150
+ const handoff = await describeHandoff(pi, state, ahead);
151
+ lines.push(...handoffLines(handoff));
152
+
153
+ return { state, removed, handoff, output: lines.join("\n") };
122
154
  }
@@ -0,0 +1,74 @@
1
+ /**
2
+ * The isolation gaps a worktree session carries, as data (pure).
3
+ *
4
+ * Re-rooting is not the same as isolating: Pi has no mutable session cwd, so a
5
+ * shell the user runs directly still runs in the main checkout; a command whose
6
+ * git behavior cannot be verified passes the guard unless
7
+ * `blockUnparsableCommands` is set; and the Windows `powershell` tool has no
8
+ * worktree surface, so the guard refuses it unless `blockPowershell` is off.
9
+ * The README documents all three, which is no help mid-task, so the session
10
+ * names them where the user already looks: `/worktree` status, and `doctor`.
11
+ *
12
+ * Only the unverifiable-command gap carries a count. A `!` shell or a
13
+ * `powershell` call is visible in the transcript; a command the guard let
14
+ * through unverified is otherwise silent, so `index.ts` counts those per
15
+ * session and this module says how many ran.
16
+ */
17
+
18
+ import type { WorktreeConfig } from "./config.ts";
19
+
20
+ /** One surface isolation does not cover. */
21
+ export interface IsolationGap {
22
+ /** Short label for the surface, e.g. `! shell`. */
23
+ surface: string;
24
+ /** What is not isolated, and why. */
25
+ detail: string;
26
+ }
27
+
28
+ /** The session facts the gap rules read. */
29
+ export interface GapInput {
30
+ /** `process.platform`; `powershell` only exists on Windows. */
31
+ platform: string;
32
+ /** Tool names active in this session, to tell whether `powershell` is offered. */
33
+ activeTools: readonly string[];
34
+ config: WorktreeConfig;
35
+ /** Commands this session let through although their git behavior could not be verified. */
36
+ unverifiableCommands: number;
37
+ }
38
+
39
+ /** The gaps a session carries, in the order a reader should meet them. */
40
+ export function isolationGaps(input: GapInput): IsolationGap[] {
41
+ const gaps: IsolationGap[] = [
42
+ {
43
+ surface: "! shell",
44
+ detail: "runs in the main checkout; Pi has no mutable session cwd",
45
+ },
46
+ ];
47
+ // Only while the guard lets the tool through: refusing it closes the gap, and a
48
+ // switch that reopens it is exactly what this row is for.
49
+ if (input.platform === "win32" && input.activeTools.includes("powershell") && !input.config.guard.blockPowershell) {
50
+ gaps.push({
51
+ surface: "powershell",
52
+ detail: "allowed by worktree.guard.blockPowershell: false; not re-rooted or guarded",
53
+ });
54
+ }
55
+ if (!input.config.guard.blockUnparsableCommands) {
56
+ const ran = input.unverifiableCommands > 0 ? `; ${input.unverifiableCommands} ran this session` : "";
57
+ gaps.push({
58
+ surface: "unverifiable",
59
+ detail: `$(...), backticks, eval, and sh -c pass through (blockUnparsableCommands is off)${ran}`,
60
+ });
61
+ }
62
+ return gaps;
63
+ }
64
+
65
+ /**
66
+ * The rendered `Not isolated:` block: one indented line per gap, labels padded
67
+ * to the longest one. No gaps, no lines — a section that says nothing should
68
+ * not be printed.
69
+ */
70
+ export function gapLines(gaps: readonly IsolationGap[]): string[] {
71
+ if (gaps.length === 0) return [];
72
+ const column = Math.max(...gaps.map((gap) => gap.surface.length));
73
+ return gaps.map((gap) => ` ${gap.surface.padEnd(column)} ${gap.detail}`);
74
+ }