@earendil-works/pi-coding-agent 0.77.0 → 0.78.1

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 (147) hide show
  1. package/CHANGELOG.md +63 -1
  2. package/README.md +10 -2
  3. package/dist/cli/args.d.ts +1 -0
  4. package/dist/cli/args.d.ts.map +1 -1
  5. package/dist/cli/args.js +15 -0
  6. package/dist/cli/args.js.map +1 -1
  7. package/dist/config.d.ts.map +1 -1
  8. package/dist/config.js +9 -1
  9. package/dist/config.js.map +1 -1
  10. package/dist/core/agent-session.d.ts +3 -1
  11. package/dist/core/agent-session.d.ts.map +1 -1
  12. package/dist/core/agent-session.js +7 -1
  13. package/dist/core/agent-session.js.map +1 -1
  14. package/dist/core/auth-storage.d.ts.map +1 -1
  15. package/dist/core/auth-storage.js +4 -3
  16. package/dist/core/auth-storage.js.map +1 -1
  17. package/dist/core/compaction/branch-summarization.d.ts +3 -1
  18. package/dist/core/compaction/branch-summarization.d.ts.map +1 -1
  19. package/dist/core/compaction/branch-summarization.js +9 -3
  20. package/dist/core/compaction/branch-summarization.js.map +1 -1
  21. package/dist/core/export-html/template.js +19 -6
  22. package/dist/core/extensions/index.d.ts +1 -1
  23. package/dist/core/extensions/index.d.ts.map +1 -1
  24. package/dist/core/extensions/index.js.map +1 -1
  25. package/dist/core/extensions/runner.d.ts +4 -2
  26. package/dist/core/extensions/runner.d.ts.map +1 -1
  27. package/dist/core/extensions/runner.js +13 -1
  28. package/dist/core/extensions/runner.js.map +1 -1
  29. package/dist/core/extensions/types.d.ts +7 -1
  30. package/dist/core/extensions/types.d.ts.map +1 -1
  31. package/dist/core/extensions/types.js.map +1 -1
  32. package/dist/core/footer-data-provider.d.ts +2 -0
  33. package/dist/core/footer-data-provider.d.ts.map +1 -1
  34. package/dist/core/footer-data-provider.js +29 -1
  35. package/dist/core/footer-data-provider.js.map +1 -1
  36. package/dist/core/model-resolver.d.ts.map +1 -1
  37. package/dist/core/model-resolver.js +3 -0
  38. package/dist/core/model-resolver.js.map +1 -1
  39. package/dist/core/package-manager.d.ts +2 -0
  40. package/dist/core/package-manager.d.ts.map +1 -1
  41. package/dist/core/package-manager.js +22 -6
  42. package/dist/core/package-manager.js.map +1 -1
  43. package/dist/core/provider-attribution.d.ts +4 -0
  44. package/dist/core/provider-attribution.d.ts.map +1 -0
  45. package/dist/core/provider-attribution.js +72 -0
  46. package/dist/core/provider-attribution.js.map +1 -0
  47. package/dist/core/provider-display-names.d.ts.map +1 -1
  48. package/dist/core/provider-display-names.js +3 -0
  49. package/dist/core/provider-display-names.js.map +1 -1
  50. package/dist/core/sdk.d.ts.map +1 -1
  51. package/dist/core/sdk.js +7 -33
  52. package/dist/core/sdk.js.map +1 -1
  53. package/dist/core/session-manager.d.ts.map +1 -1
  54. package/dist/core/session-manager.js +92 -68
  55. package/dist/core/session-manager.js.map +1 -1
  56. package/dist/core/tools/edit.d.ts.map +1 -1
  57. package/dist/core/tools/edit.js +7 -10
  58. package/dist/core/tools/edit.js.map +1 -1
  59. package/dist/core/tools/find.d.ts.map +1 -1
  60. package/dist/core/tools/find.js.map +1 -1
  61. package/dist/core/tools/grep.d.ts.map +1 -1
  62. package/dist/core/tools/grep.js.map +1 -1
  63. package/dist/core/tools/ls.d.ts.map +1 -1
  64. package/dist/core/tools/ls.js +5 -7
  65. package/dist/core/tools/ls.js.map +1 -1
  66. package/dist/core/tools/read.d.ts.map +1 -1
  67. package/dist/core/tools/read.js +6 -7
  68. package/dist/core/tools/read.js.map +1 -1
  69. package/dist/core/tools/render-utils.d.ts +5 -2
  70. package/dist/core/tools/render-utils.d.ts.map +1 -1
  71. package/dist/core/tools/render-utils.js +17 -1
  72. package/dist/core/tools/render-utils.js.map +1 -1
  73. package/dist/core/tools/write.d.ts.map +1 -1
  74. package/dist/core/tools/write.js +5 -6
  75. package/dist/core/tools/write.js.map +1 -1
  76. package/dist/index.d.ts +2 -0
  77. package/dist/index.d.ts.map +1 -1
  78. package/dist/index.js +2 -0
  79. package/dist/index.js.map +1 -1
  80. package/dist/main.d.ts.map +1 -1
  81. package/dist/main.js +8 -0
  82. package/dist/main.js.map +1 -1
  83. package/dist/modes/interactive/components/login-dialog.d.ts +0 -1
  84. package/dist/modes/interactive/components/login-dialog.d.ts.map +1 -1
  85. package/dist/modes/interactive/components/login-dialog.js +3 -12
  86. package/dist/modes/interactive/components/login-dialog.js.map +1 -1
  87. package/dist/modes/interactive/components/tool-execution.d.ts.map +1 -1
  88. package/dist/modes/interactive/components/tool-execution.js +22 -0
  89. package/dist/modes/interactive/components/tool-execution.js.map +1 -1
  90. package/dist/modes/interactive/interactive-mode.d.ts +3 -0
  91. package/dist/modes/interactive/interactive-mode.d.ts.map +1 -1
  92. package/dist/modes/interactive/interactive-mode.js +36 -0
  93. package/dist/modes/interactive/interactive-mode.js.map +1 -1
  94. package/dist/modes/print-mode.d.ts.map +1 -1
  95. package/dist/modes/print-mode.js +1 -0
  96. package/dist/modes/print-mode.js.map +1 -1
  97. package/dist/modes/rpc/rpc-mode.d.ts.map +1 -1
  98. package/dist/modes/rpc/rpc-mode.js +1 -0
  99. package/dist/modes/rpc/rpc-mode.js.map +1 -1
  100. package/dist/utils/git.d.ts.map +1 -1
  101. package/dist/utils/git.js +54 -22
  102. package/dist/utils/git.js.map +1 -1
  103. package/dist/utils/open-browser.d.ts +9 -0
  104. package/dist/utils/open-browser.d.ts.map +1 -0
  105. package/dist/utils/open-browser.js +22 -0
  106. package/dist/utils/open-browser.js.map +1 -0
  107. package/docs/containerization.md +111 -0
  108. package/docs/docs.json +4 -0
  109. package/docs/extensions.md +36 -11
  110. package/docs/index.md +1 -0
  111. package/docs/providers.md +5 -0
  112. package/docs/quickstart.md +1 -0
  113. package/docs/rpc.md +3 -2
  114. package/docs/session-format.md +1 -1
  115. package/docs/sessions.md +8 -0
  116. package/docs/settings.md +1 -1
  117. package/docs/terminal-setup.md +2 -0
  118. package/docs/tui.md +12 -3
  119. package/docs/usage.md +6 -1
  120. package/examples/extensions/README.md +1 -0
  121. package/examples/extensions/custom-header.ts +1 -1
  122. package/examples/extensions/custom-provider-anthropic/package-lock.json +2 -2
  123. package/examples/extensions/custom-provider-anthropic/package.json +1 -1
  124. package/examples/extensions/custom-provider-gitlab-duo/index.ts +53 -2
  125. package/examples/extensions/custom-provider-gitlab-duo/package.json +1 -1
  126. package/examples/extensions/doom-overlay/index.ts +1 -1
  127. package/examples/extensions/gondolin/index.ts +531 -0
  128. package/examples/extensions/gondolin/package-lock.json +185 -0
  129. package/examples/extensions/gondolin/package.json +19 -0
  130. package/examples/extensions/handoff.ts +1 -1
  131. package/examples/extensions/interactive-shell.ts +1 -1
  132. package/examples/extensions/overlay-qa-tests.ts +152 -81
  133. package/examples/extensions/qna.ts +1 -1
  134. package/examples/extensions/question.ts +1 -1
  135. package/examples/extensions/questionnaire.ts +1 -1
  136. package/examples/extensions/sandbox/package-lock.json +2 -2
  137. package/examples/extensions/sandbox/package.json +1 -1
  138. package/examples/extensions/snake.ts +1 -1
  139. package/examples/extensions/space-invaders.ts +1 -1
  140. package/examples/extensions/summarize.ts +1 -1
  141. package/examples/extensions/tic-tac-toe.ts +1 -1
  142. package/examples/extensions/todo.ts +1 -1
  143. package/examples/extensions/tools.ts +5 -0
  144. package/examples/extensions/with-deps/package-lock.json +2 -2
  145. package/examples/extensions/with-deps/package.json +1 -1
  146. package/npm-shrinkwrap.json +12 -12
  147. package/package.json +5 -4
@@ -20,6 +20,7 @@ import {
20
20
  type SimpleStreamOptions,
21
21
  streamSimpleAnthropic,
22
22
  streamSimpleOpenAIResponses,
23
+ type ThinkingLevelMap,
23
24
  } from "@earendil-works/pi-ai";
24
25
  import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
25
26
 
@@ -49,6 +50,7 @@ interface GitLabModel {
49
50
  backend: Backend;
50
51
  baseUrl: string;
51
52
  reasoning: boolean;
53
+ thinkingLevelMap?: ThinkingLevelMap;
52
54
  input: ("text" | "image")[];
53
55
  cost: { input: number; output: number; cacheRead: number; cacheWrite: number };
54
56
  contextWindow: number;
@@ -57,12 +59,37 @@ interface GitLabModel {
57
59
 
58
60
  export const MODELS: GitLabModel[] = [
59
61
  // Anthropic
62
+ {
63
+ id: "claude-opus-4-8",
64
+ name: "Claude Opus 4.8",
65
+ backend: "anthropic",
66
+ baseUrl: ANTHROPIC_PROXY_URL,
67
+ reasoning: true,
68
+ thinkingLevelMap: { xhigh: "max" },
69
+ input: ["text", "image"],
70
+ cost: { input: 5, output: 25, cacheRead: 0.5, cacheWrite: 6.25 },
71
+ contextWindow: 1000000,
72
+ maxTokens: 128000,
73
+ },
74
+ {
75
+ id: "claude-sonnet-4-6",
76
+ name: "Claude Sonnet 4.6",
77
+ backend: "anthropic",
78
+ baseUrl: ANTHROPIC_PROXY_URL,
79
+ reasoning: true,
80
+ thinkingLevelMap: { xhigh: "max" },
81
+ input: ["text", "image"],
82
+ cost: { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
83
+ contextWindow: 1000000,
84
+ maxTokens: 64000,
85
+ },
60
86
  {
61
87
  id: "claude-opus-4-5-20251101",
62
88
  name: "Claude Opus 4.5",
63
89
  backend: "anthropic",
64
90
  baseUrl: ANTHROPIC_PROXY_URL,
65
91
  reasoning: true,
92
+ thinkingLevelMap: { xhigh: "max" },
66
93
  input: ["text", "image"],
67
94
  cost: { input: 15, output: 75, cacheRead: 1.5, cacheWrite: 18.75 },
68
95
  contextWindow: 200000,
@@ -74,6 +101,7 @@ export const MODELS: GitLabModel[] = [
74
101
  backend: "anthropic",
75
102
  baseUrl: ANTHROPIC_PROXY_URL,
76
103
  reasoning: true,
104
+ thinkingLevelMap: { xhigh: "max" },
77
105
  input: ["text", "image"],
78
106
  cost: { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
79
107
  contextWindow: 200000,
@@ -85,12 +113,24 @@ export const MODELS: GitLabModel[] = [
85
113
  backend: "anthropic",
86
114
  baseUrl: ANTHROPIC_PROXY_URL,
87
115
  reasoning: true,
116
+ thinkingLevelMap: { xhigh: "max" },
88
117
  input: ["text", "image"],
89
118
  cost: { input: 1, output: 5, cacheRead: 0.1, cacheWrite: 1.25 },
90
119
  contextWindow: 200000,
91
120
  maxTokens: 8192,
92
121
  },
93
122
  // OpenAI (all use Responses API)
123
+ {
124
+ id: "gpt-5.5-2026-04-23",
125
+ name: "GPT-5.5",
126
+ backend: "openai",
127
+ baseUrl: OPENAI_PROXY_URL,
128
+ reasoning: true,
129
+ input: ["text", "image"],
130
+ cost: { input: 5, output: 30, cacheRead: 0.5, cacheWrite: 0 },
131
+ contextWindow: 272000,
132
+ maxTokens: 128000,
133
+ },
94
134
  {
95
135
  id: "gpt-5.1-2025-11-13",
96
136
  name: "GPT-5.1",
@@ -285,7 +325,17 @@ export function streamGitLabDuo(
285
325
 
286
326
  const innerStream =
287
327
  cfg.backend === "anthropic"
288
- ? streamSimpleAnthropic(modelWithBaseUrl as Model<"anthropic-messages">, context, streamOptions)
328
+ ? streamSimpleAnthropic(
329
+ {
330
+ ...(modelWithBaseUrl as Model<"anthropic-messages">),
331
+ compat: {
332
+ ...(modelWithBaseUrl as Model<"anthropic-messages">).compat,
333
+ forceAdaptiveThinking: true,
334
+ },
335
+ },
336
+ context,
337
+ streamOptions,
338
+ )
289
339
  : streamSimpleOpenAIResponses(modelWithBaseUrl as Model<"openai-responses">, context, streamOptions);
290
340
 
291
341
  for await (const event of innerStream) stream.push(event);
@@ -329,10 +379,11 @@ export default function (pi: ExtensionAPI) {
329
379
  baseUrl: AI_GATEWAY_URL,
330
380
  apiKey: "$GITLAB_TOKEN",
331
381
  api: "gitlab-duo-api",
332
- models: MODELS.map(({ id, name, reasoning, input, cost, contextWindow, maxTokens }) => ({
382
+ models: MODELS.map(({ id, name, reasoning, thinkingLevelMap, input, cost, contextWindow, maxTokens }) => ({
333
383
  id,
334
384
  name,
335
385
  reasoning,
386
+ thinkingLevelMap,
336
387
  input,
337
388
  cost,
338
389
  contextWindow,
@@ -1,7 +1,7 @@
1
1
  {
2
2
  "name": "pi-extension-custom-provider-gitlab-duo",
3
3
  "private": true,
4
- "version": "0.77.0",
4
+ "version": "0.78.1",
5
5
  "type": "module",
6
6
  "scripts": {
7
7
  "clean": "echo 'nothing to clean'",
@@ -23,7 +23,7 @@ export default function (pi: ExtensionAPI) {
23
23
  description: "Play DOOM as an overlay. Q to pause and exit.",
24
24
 
25
25
  handler: async (args, ctx) => {
26
- if (!ctx.hasUI) {
26
+ if (ctx.mode !== "tui") {
27
27
  ctx.ui.notify("DOOM requires interactive mode", "error");
28
28
  return;
29
29
  }
@@ -0,0 +1,531 @@
1
+ /**
2
+ * Gondolin Tool Routing Example
3
+ *
4
+ * Runs pi's built-in tools inside a local Gondolin micro-VM. The host working
5
+ * directory is mounted at /workspace in the guest. File changes under
6
+ * /workspace write through to the host; other guest filesystem changes are
7
+ * isolated to the VM.
8
+ *
9
+ * Setup:
10
+ * cd packages/coding-agent/examples/extensions/gondolin
11
+ * npm install --ignore-scripts
12
+ *
13
+ * Usage:
14
+ * cd /path/to/project
15
+ * pi -e /path/to/pi/packages/coding-agent/examples/extensions/gondolin
16
+ *
17
+ * Requirements:
18
+ * - Node.js >= 23.6.0 for @earendil-works/gondolin
19
+ * - QEMU installed (for example, `brew install qemu` on macOS)
20
+ */
21
+
22
+ import path from "node:path";
23
+ import { RealFSProvider, VM } from "@earendil-works/gondolin";
24
+ import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
25
+ import {
26
+ type BashOperations,
27
+ createBashTool,
28
+ createEditTool,
29
+ createFindTool,
30
+ createGrepTool,
31
+ createLsTool,
32
+ createReadTool,
33
+ createWriteTool,
34
+ DEFAULT_MAX_BYTES,
35
+ type EditOperations,
36
+ type FindOperations,
37
+ formatSize,
38
+ type GrepToolDetails,
39
+ type GrepToolInput,
40
+ type LsOperations,
41
+ type ReadOperations,
42
+ truncateHead,
43
+ truncateLine,
44
+ type WriteOperations,
45
+ } from "@earendil-works/pi-coding-agent";
46
+
47
+ const GUEST_WORKSPACE = "/workspace";
48
+ const DEFAULT_GREP_LIMIT = 100;
49
+
50
+ type TextToolResult<TDetails> = {
51
+ content: Array<{ type: "text"; text: string }>;
52
+ details: TDetails | undefined;
53
+ };
54
+
55
+ function stripAtPrefix(value: string): string {
56
+ return value.startsWith("@") ? value.slice(1) : value;
57
+ }
58
+
59
+ function toPosix(value: string): string {
60
+ return value.split(path.sep).join(path.posix.sep);
61
+ }
62
+
63
+ function isInsideHostPath(root: string, value: string): boolean {
64
+ const relativePath = path.relative(root, value);
65
+ return relativePath === "" || (!relativePath.startsWith("..") && !path.isAbsolute(relativePath));
66
+ }
67
+
68
+ function hostPathToGuest(localCwd: string, hostPath: string): string {
69
+ const relativePath = path.relative(localCwd, hostPath);
70
+ if (!isInsideHostPath(localCwd, hostPath)) return toPosix(hostPath);
71
+ return relativePath ? path.posix.join(GUEST_WORKSPACE, toPosix(relativePath)) : GUEST_WORKSPACE;
72
+ }
73
+
74
+ function toGuestPath(localCwd: string, inputPath: string): string {
75
+ const trimmed = stripAtPrefix(inputPath.trim());
76
+ if (!trimmed) return GUEST_WORKSPACE;
77
+ if (path.isAbsolute(trimmed)) {
78
+ if (isInsideHostPath(localCwd, trimmed)) return hostPathToGuest(localCwd, trimmed);
79
+ return path.posix.resolve("/", toPosix(trimmed));
80
+ }
81
+ return path.posix.resolve(GUEST_WORKSPACE, toPosix(trimmed));
82
+ }
83
+
84
+ function createGondolinReadOps(vm: VM, localCwd: string): ReadOperations {
85
+ return {
86
+ readFile: async (filePath) => vm.fs.readFile(toGuestPath(localCwd, filePath)),
87
+ access: async (filePath) => {
88
+ await vm.fs.access(toGuestPath(localCwd, filePath));
89
+ },
90
+ detectImageMimeType: async (filePath) => {
91
+ const ext = path.posix.extname(toGuestPath(localCwd, filePath)).toLowerCase();
92
+ if (ext === ".png") return "image/png";
93
+ if (ext === ".jpg" || ext === ".jpeg") return "image/jpeg";
94
+ if (ext === ".gif") return "image/gif";
95
+ if (ext === ".webp") return "image/webp";
96
+ return null;
97
+ },
98
+ };
99
+ }
100
+
101
+ function createGondolinWriteOps(vm: VM, localCwd: string): WriteOperations {
102
+ return {
103
+ writeFile: async (filePath, content) => {
104
+ await vm.fs.writeFile(toGuestPath(localCwd, filePath), content, { encoding: "utf8" });
105
+ },
106
+ mkdir: async (dirPath) => {
107
+ await vm.fs.mkdir(toGuestPath(localCwd, dirPath), { recursive: true });
108
+ },
109
+ };
110
+ }
111
+
112
+ function createGondolinEditOps(vm: VM, localCwd: string): EditOperations {
113
+ const readOps = createGondolinReadOps(vm, localCwd);
114
+ const writeOps = createGondolinWriteOps(vm, localCwd);
115
+ return {
116
+ readFile: readOps.readFile,
117
+ writeFile: writeOps.writeFile,
118
+ access: readOps.access,
119
+ };
120
+ }
121
+
122
+ function createGondolinLsOps(vm: VM, localCwd: string): LsOperations {
123
+ return {
124
+ exists: async (filePath) => {
125
+ try {
126
+ await vm.fs.access(toGuestPath(localCwd, filePath));
127
+ return true;
128
+ } catch {
129
+ return false;
130
+ }
131
+ },
132
+ stat: async (filePath) => vm.fs.stat(toGuestPath(localCwd, filePath)),
133
+ readdir: async (dirPath) => vm.fs.listDir(toGuestPath(localCwd, dirPath)),
134
+ };
135
+ }
136
+
137
+ async function walkGuestFiles(
138
+ vm: VM,
139
+ root: string,
140
+ visit: (guestPath: string, relativePath: string) => Promise<boolean>,
141
+ signal?: AbortSignal,
142
+ ): Promise<boolean> {
143
+ if (signal?.aborted) throw new Error("Operation aborted");
144
+ const stat = await vm.fs.stat(root, { signal });
145
+ if (!stat.isDirectory()) return visit(root, path.posix.basename(root));
146
+
147
+ const walkDirectory = async (dir: string, relativeDir: string): Promise<boolean> => {
148
+ if (signal?.aborted) throw new Error("Operation aborted");
149
+ const entries = await vm.fs.listDir(dir, { signal });
150
+ for (const entry of entries) {
151
+ if (entry === ".git" || entry === "node_modules") continue;
152
+ const guestPath = path.posix.join(dir, entry);
153
+ const relativePath = relativeDir ? path.posix.join(relativeDir, entry) : entry;
154
+ let entryStat: Awaited<ReturnType<VM["fs"]["stat"]>>;
155
+ try {
156
+ entryStat = await vm.fs.stat(guestPath, { signal });
157
+ } catch {
158
+ continue;
159
+ }
160
+ if (entryStat.isDirectory()) {
161
+ if (!(await walkDirectory(guestPath, relativePath))) return false;
162
+ } else if (!(await visit(guestPath, relativePath))) {
163
+ return false;
164
+ }
165
+ }
166
+ return true;
167
+ };
168
+
169
+ return walkDirectory(root, "");
170
+ }
171
+
172
+ function matchesToolGlob(relativePath: string, pattern: string): boolean {
173
+ const normalizedPattern = toPosix(pattern);
174
+ if (normalizedPattern.includes("/")) {
175
+ return (
176
+ path.posix.matchesGlob(relativePath, normalizedPattern) ||
177
+ path.posix.matchesGlob(relativePath, `**/${normalizedPattern}`)
178
+ );
179
+ }
180
+ return path.posix.matchesGlob(path.posix.basename(relativePath), normalizedPattern);
181
+ }
182
+
183
+ function createGondolinFindOps(vm: VM, localCwd: string): FindOperations {
184
+ return {
185
+ exists: async (filePath) => {
186
+ try {
187
+ await vm.fs.access(toGuestPath(localCwd, filePath));
188
+ return true;
189
+ } catch {
190
+ return false;
191
+ }
192
+ },
193
+ glob: async (pattern, cwd, options) => {
194
+ const root = toGuestPath(localCwd, cwd);
195
+ const results: string[] = [];
196
+ await walkGuestFiles(vm, root, async (guestPath, relativePath) => {
197
+ if (results.length >= options.limit) return false;
198
+ if (matchesToolGlob(relativePath, pattern)) results.push(guestPath);
199
+ return results.length < options.limit;
200
+ });
201
+ return results;
202
+ },
203
+ };
204
+ }
205
+
206
+ function createLineMatcher(pattern: string, literal: boolean | undefined, ignoreCase: boolean | undefined) {
207
+ if (literal) {
208
+ const needle = ignoreCase ? pattern.toLowerCase() : pattern;
209
+ return (line: string) => (ignoreCase ? line.toLowerCase() : line).includes(needle);
210
+ }
211
+ const regex = new RegExp(pattern, ignoreCase ? "i" : undefined);
212
+ return (line: string) => regex.test(line);
213
+ }
214
+
215
+ function appendGrepBlock(params: {
216
+ outputLines: string[];
217
+ lines: string[];
218
+ relativePath: string;
219
+ lineIndex: number;
220
+ contextLines: number;
221
+ }): boolean {
222
+ let linesTruncated = false;
223
+ const start = params.contextLines > 0 ? Math.max(0, params.lineIndex - params.contextLines) : params.lineIndex;
224
+ const end =
225
+ params.contextLines > 0
226
+ ? Math.min(params.lines.length - 1, params.lineIndex + params.contextLines)
227
+ : params.lineIndex;
228
+
229
+ for (let index = start; index <= end; index++) {
230
+ const rawLine = params.lines[index] ?? "";
231
+ const { text, wasTruncated } = truncateLine(rawLine.replace(/\r/g, ""));
232
+ if (wasTruncated) linesTruncated = true;
233
+ const separator = index === params.lineIndex ? ":" : "-";
234
+ params.outputLines.push(`${params.relativePath}${separator}${index + 1}${separator} ${text}`);
235
+ }
236
+ return linesTruncated;
237
+ }
238
+
239
+ async function executeGondolinGrep(
240
+ vm: VM,
241
+ localCwd: string,
242
+ params: GrepToolInput,
243
+ signal?: AbortSignal,
244
+ ): Promise<TextToolResult<GrepToolDetails>> {
245
+ const root = toGuestPath(localCwd, params.path ?? ".");
246
+ const rootStat = await vm.fs.stat(root, { signal });
247
+ const rootIsDirectory = rootStat.isDirectory();
248
+ const matcher = createLineMatcher(params.pattern, params.literal, params.ignoreCase);
249
+ const contextLines = params.context && params.context > 0 ? params.context : 0;
250
+ const effectiveLimit = Math.max(1, params.limit ?? DEFAULT_GREP_LIMIT);
251
+ const outputLines: string[] = [];
252
+ const details: GrepToolDetails = {};
253
+ let matchCount = 0;
254
+ let matchLimitReached = false;
255
+ let linesTruncated = false;
256
+
257
+ await walkGuestFiles(
258
+ vm,
259
+ root,
260
+ async (guestPath, relativePath) => {
261
+ if (matchCount >= effectiveLimit) return false;
262
+ if (params.glob && !matchesToolGlob(relativePath, params.glob)) return true;
263
+ let content: string;
264
+ try {
265
+ content = await vm.fs.readFile(guestPath, { encoding: "utf8", signal });
266
+ } catch {
267
+ return true;
268
+ }
269
+ const lines = content.replace(/\r\n/g, "\n").replace(/\r/g, "\n").split("\n");
270
+ const displayPath = rootIsDirectory ? relativePath : path.posix.basename(guestPath);
271
+ for (let index = 0; index < lines.length; index++) {
272
+ if (signal?.aborted) throw new Error("Operation aborted");
273
+ if (!matcher(lines[index] ?? "")) continue;
274
+ matchCount++;
275
+ if (appendGrepBlock({ outputLines, lines, relativePath: displayPath, lineIndex: index, contextLines })) {
276
+ linesTruncated = true;
277
+ }
278
+ if (matchCount >= effectiveLimit) {
279
+ matchLimitReached = true;
280
+ return false;
281
+ }
282
+ }
283
+ return true;
284
+ },
285
+ signal,
286
+ );
287
+
288
+ if (matchCount === 0) return { content: [{ type: "text", text: "No matches found" }], details: undefined };
289
+
290
+ const rawOutput = outputLines.join("\n");
291
+ const truncation = truncateHead(rawOutput, { maxLines: Number.MAX_SAFE_INTEGER });
292
+ const notices: string[] = [];
293
+ let output = truncation.content;
294
+
295
+ if (matchLimitReached) {
296
+ details.matchLimitReached = effectiveLimit;
297
+ notices.push(`${effectiveLimit} matches limit reached`);
298
+ }
299
+ if (linesTruncated) {
300
+ details.linesTruncated = true;
301
+ notices.push("long lines truncated");
302
+ }
303
+ if (truncation.truncated) {
304
+ details.truncation = truncation;
305
+ notices.push(`${formatSize(DEFAULT_MAX_BYTES)} limit reached`);
306
+ }
307
+ if (notices.length > 0) output += `\n\n[${notices.join(". ")}]`;
308
+
309
+ return {
310
+ content: [{ type: "text", text: output }],
311
+ details: Object.keys(details).length > 0 ? details : undefined,
312
+ };
313
+ }
314
+
315
+ function sanitizeEnv(env: NodeJS.ProcessEnv | undefined): Record<string, string> | undefined {
316
+ if (!env) return undefined;
317
+ const result: Record<string, string> = {};
318
+ for (const [key, value] of Object.entries(env)) {
319
+ if (typeof value === "string") result[key] = value;
320
+ }
321
+ return result;
322
+ }
323
+
324
+ function createGondolinBashOps(vm: VM, localCwd: string, shellPath: string): BashOperations {
325
+ return {
326
+ exec: async (command, cwd, { onData, signal, timeout, env }) => {
327
+ if (signal?.aborted) throw new Error("aborted");
328
+ const guestCwd = toGuestPath(localCwd, cwd);
329
+ const controller = new AbortController();
330
+ const onAbort = () => controller.abort();
331
+ signal?.addEventListener("abort", onAbort, { once: true });
332
+
333
+ let timedOut = false;
334
+ const timer =
335
+ timeout && timeout > 0
336
+ ? setTimeout(() => {
337
+ timedOut = true;
338
+ controller.abort();
339
+ }, timeout * 1000)
340
+ : undefined;
341
+
342
+ try {
343
+ const proc = vm.exec([shellPath, "-lc", command], {
344
+ cwd: guestCwd,
345
+ env: sanitizeEnv(env),
346
+ signal: controller.signal,
347
+ stdout: "pipe",
348
+ stderr: "pipe",
349
+ });
350
+ for await (const chunk of proc.output()) onData(chunk.data);
351
+ const result = await proc;
352
+ return { exitCode: result.exitCode };
353
+ } catch (error) {
354
+ if (signal?.aborted) throw new Error("aborted");
355
+ if (timedOut) throw new Error(`timeout:${timeout}`);
356
+ throw error;
357
+ } finally {
358
+ if (timer) clearTimeout(timer);
359
+ signal?.removeEventListener("abort", onAbort);
360
+ }
361
+ },
362
+ };
363
+ }
364
+
365
+ export default function (pi: ExtensionAPI) {
366
+ const localCwd = process.cwd();
367
+ const localRead = createReadTool(localCwd);
368
+ const localWrite = createWriteTool(localCwd);
369
+ const localEdit = createEditTool(localCwd);
370
+ const localBash = createBashTool(localCwd);
371
+ const localGrep = createGrepTool(localCwd);
372
+ const localFind = createFindTool(localCwd);
373
+ const localLs = createLsTool(localCwd);
374
+
375
+ let vm: VM | undefined;
376
+ let vmStarting: Promise<VM> | undefined;
377
+ let shellPath = "/bin/sh";
378
+
379
+ async function startVm(ctx?: ExtensionContext): Promise<VM> {
380
+ ctx?.ui.setStatus("gondolin", ctx.ui.theme.fg("accent", `Gondolin: starting ${GUEST_WORKSPACE}`));
381
+ const created = await VM.create({
382
+ sessionLabel: `pi ${path.basename(localCwd)}`,
383
+ vfs: {
384
+ mounts: {
385
+ [GUEST_WORKSPACE]: new RealFSProvider(localCwd),
386
+ },
387
+ },
388
+ });
389
+ const bashProbe = await created.exec(["/bin/sh", "-lc", "command -v bash || true"]);
390
+ shellPath = bashProbe.stdout.trim() || "/bin/sh";
391
+ vm = created;
392
+ ctx?.ui.setStatus(
393
+ "gondolin",
394
+ ctx.ui.theme.fg("accent", `Gondolin: ${created.id.slice(0, 8)} (${GUEST_WORKSPACE})`),
395
+ );
396
+ ctx?.ui.notify(`Gondolin VM ready. ${localCwd} is mounted at ${GUEST_WORKSPACE}.`, "info");
397
+ return created;
398
+ }
399
+
400
+ async function ensureVm(ctx?: ExtensionContext): Promise<VM> {
401
+ if (vm) return vm;
402
+ if (!vmStarting) {
403
+ vmStarting = startVm(ctx).finally(() => {
404
+ vmStarting = undefined;
405
+ });
406
+ }
407
+ return vmStarting;
408
+ }
409
+
410
+ pi.on("session_start", async (_event, ctx) => {
411
+ await ensureVm(ctx);
412
+ });
413
+
414
+ pi.on("session_shutdown", async (_event, ctx) => {
415
+ const activeVm = vm;
416
+ vm = undefined;
417
+ vmStarting = undefined;
418
+ if (!activeVm) return;
419
+ ctx.ui.setStatus("gondolin", ctx.ui.theme.fg("muted", "Gondolin: stopping"));
420
+ try {
421
+ await activeVm.close();
422
+ } finally {
423
+ ctx.ui.setStatus("gondolin", undefined);
424
+ }
425
+ });
426
+
427
+ pi.registerCommand("gondolin", {
428
+ description: "Show Gondolin VM status",
429
+ handler: async (_args, ctx) => {
430
+ const activeVm = await ensureVm(ctx);
431
+ ctx.ui.notify(
432
+ [
433
+ `Gondolin VM: ${activeVm.id}`,
434
+ `Host workspace: ${localCwd}`,
435
+ `Guest workspace: ${GUEST_WORKSPACE}`,
436
+ `Shell: ${shellPath}`,
437
+ ].join("\n"),
438
+ "info",
439
+ );
440
+ },
441
+ });
442
+
443
+ pi.registerTool({
444
+ ...localRead,
445
+ async execute(id, params, signal, onUpdate, ctx) {
446
+ const activeVm = await ensureVm(ctx);
447
+ const tool = createReadTool(GUEST_WORKSPACE, {
448
+ operations: createGondolinReadOps(activeVm, localCwd),
449
+ });
450
+ return tool.execute(id, params, signal, onUpdate);
451
+ },
452
+ });
453
+
454
+ pi.registerTool({
455
+ ...localWrite,
456
+ async execute(id, params, signal, onUpdate, ctx) {
457
+ const activeVm = await ensureVm(ctx);
458
+ const tool = createWriteTool(GUEST_WORKSPACE, {
459
+ operations: createGondolinWriteOps(activeVm, localCwd),
460
+ });
461
+ return tool.execute(id, params, signal, onUpdate);
462
+ },
463
+ });
464
+
465
+ pi.registerTool({
466
+ ...localEdit,
467
+ async execute(id, params, signal, onUpdate, ctx) {
468
+ const activeVm = await ensureVm(ctx);
469
+ const tool = createEditTool(GUEST_WORKSPACE, {
470
+ operations: createGondolinEditOps(activeVm, localCwd),
471
+ });
472
+ return tool.execute(id, params, signal, onUpdate);
473
+ },
474
+ });
475
+
476
+ pi.registerTool({
477
+ ...localBash,
478
+ async execute(id, params, signal, onUpdate, ctx) {
479
+ const activeVm = await ensureVm(ctx);
480
+ const tool = createBashTool(GUEST_WORKSPACE, {
481
+ operations: createGondolinBashOps(activeVm, localCwd, shellPath),
482
+ });
483
+ return tool.execute(id, params, signal, onUpdate);
484
+ },
485
+ });
486
+
487
+ pi.registerTool({
488
+ ...localLs,
489
+ async execute(id, params, signal, onUpdate, ctx) {
490
+ const activeVm = await ensureVm(ctx);
491
+ const tool = createLsTool(GUEST_WORKSPACE, {
492
+ operations: createGondolinLsOps(activeVm, localCwd),
493
+ });
494
+ return tool.execute(id, params, signal, onUpdate);
495
+ },
496
+ });
497
+
498
+ pi.registerTool({
499
+ ...localFind,
500
+ async execute(id, params, signal, onUpdate, ctx) {
501
+ const activeVm = await ensureVm(ctx);
502
+ const tool = createFindTool(GUEST_WORKSPACE, {
503
+ operations: createGondolinFindOps(activeVm, localCwd),
504
+ });
505
+ return tool.execute(id, params, signal, onUpdate);
506
+ },
507
+ });
508
+
509
+ pi.registerTool({
510
+ ...localGrep,
511
+ async execute(_id, params, signal, _onUpdate, ctx) {
512
+ const activeVm = await ensureVm(ctx);
513
+ return executeGondolinGrep(activeVm, localCwd, params, signal);
514
+ },
515
+ });
516
+
517
+ pi.on("user_bash", async (_event, ctx) => {
518
+ const activeVm = await ensureVm(ctx);
519
+ return { operations: createGondolinBashOps(activeVm, localCwd, shellPath) };
520
+ });
521
+
522
+ pi.on("before_agent_start", async (event, ctx) => {
523
+ await ensureVm(ctx);
524
+ const localLine = `Current working directory: ${localCwd}`;
525
+ const guestLine = `Current working directory: ${GUEST_WORKSPACE} (Gondolin VM; host workspace mounted from ${localCwd})`;
526
+ const systemPrompt = event.systemPrompt.includes(localLine)
527
+ ? event.systemPrompt.replace(localLine, guestLine)
528
+ : `${event.systemPrompt}\n\n${guestLine}`;
529
+ return { systemPrompt };
530
+ });
531
+ }