@pugi/cli 0.1.0-alpha.9 → 0.1.0-beta.2

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 (68) hide show
  1. package/README.md +33 -0
  2. package/assets/pugi-mascot.ansi +41 -0
  3. package/dist/commands/deploy.js +439 -0
  4. package/dist/core/agents/loader.js +104 -0
  5. package/dist/core/agents/registry.js +1 -1
  6. package/dist/core/consensus/anvil-fanout.js +276 -0
  7. package/dist/core/consensus/diff-capture.js +382 -0
  8. package/dist/core/consensus/rubric.js +233 -0
  9. package/dist/core/context/index.js +21 -0
  10. package/dist/core/context/pugiignore.js +316 -0
  11. package/dist/core/context/repo-skeleton.js +533 -0
  12. package/dist/core/context/watcher.js +342 -0
  13. package/dist/core/context/working-set.js +165 -0
  14. package/dist/core/edits/dispatch.js +185 -0
  15. package/dist/core/edits/index.js +15 -0
  16. package/dist/core/edits/layer-a-apply.js +217 -0
  17. package/dist/core/edits/layer-b-apply.js +211 -0
  18. package/dist/core/edits/layer-c-apply.js +160 -0
  19. package/dist/core/edits/layer-d-ast.js +29 -0
  20. package/dist/core/edits/marker-parser.js +401 -0
  21. package/dist/core/edits/security-gate.js +223 -0
  22. package/dist/core/edits/worktree.js +229 -0
  23. package/dist/core/engine/native-pugi.js +6 -1
  24. package/dist/core/engine/prompts.js +4 -1
  25. package/dist/core/engine/tool-bridge.js +33 -1
  26. package/dist/core/lsp/client.js +631 -0
  27. package/dist/core/repl/ask.js +512 -0
  28. package/dist/core/repl/cancellation.js +98 -0
  29. package/dist/core/repl/dispatch-fsm.js +220 -0
  30. package/dist/core/repl/privacy-banner.js +71 -0
  31. package/dist/core/repl/session.js +1896 -13
  32. package/dist/core/repl/slash-commands.js +59 -32
  33. package/dist/core/repl/store/index.js +12 -0
  34. package/dist/core/repl/store/jsonl-log.js +321 -0
  35. package/dist/core/repl/store/lockfile.js +155 -0
  36. package/dist/core/repl/store/session-store.js +792 -0
  37. package/dist/core/repl/store/types.js +44 -0
  38. package/dist/core/repl/store/uuid-v7.js +68 -0
  39. package/dist/core/repl/workspace-context.js +72 -1
  40. package/dist/core/skills/loader.js +454 -0
  41. package/dist/core/skills/sources.js +480 -0
  42. package/dist/core/skills/trust.js +172 -0
  43. package/dist/runtime/cli.js +767 -10
  44. package/dist/runtime/commands/agents.js +385 -0
  45. package/dist/runtime/commands/config.js +338 -8
  46. package/dist/runtime/commands/lsp.js +184 -0
  47. package/dist/runtime/commands/patch.js +111 -0
  48. package/dist/runtime/commands/review-consensus.js +399 -0
  49. package/dist/runtime/commands/skills.js +401 -0
  50. package/dist/runtime/commands/worktree.js +133 -0
  51. package/dist/tools/apply-patch.js +314 -0
  52. package/dist/tools/file-tools.js +90 -0
  53. package/dist/tools/lsp-tools.js +189 -0
  54. package/dist/tools/registry.js +18 -0
  55. package/dist/tools/web-fetch.js +1 -1
  56. package/dist/tui/agent-tree-pane.js +9 -0
  57. package/dist/tui/ask-cli.js +52 -0
  58. package/dist/tui/ask-modal.js +211 -0
  59. package/dist/tui/conversation-pane.js +48 -3
  60. package/dist/tui/input-box.js +48 -5
  61. package/dist/tui/markdown-render.js +266 -0
  62. package/dist/tui/repl-render.js +185 -0
  63. package/dist/tui/repl-splash-mascot.js +130 -0
  64. package/dist/tui/repl-splash.js +7 -1
  65. package/dist/tui/repl.js +82 -11
  66. package/dist/tui/status-bar.js +63 -3
  67. package/dist/tui/tool-stream-pane.js +91 -0
  68. package/package.json +11 -5
@@ -0,0 +1,401 @@
1
+ export class MarkerParseError extends Error {
2
+ code = 'MARKER_PARSE_ERROR';
3
+ modelHint;
4
+ atLine;
5
+ constructor(message, modelHint, atLine) {
6
+ super(message);
7
+ this.name = 'MarkerParseError';
8
+ this.modelHint = modelHint;
9
+ this.atLine = atLine;
10
+ }
11
+ }
12
+ /**
13
+ * Convert a raw LLM response into parsed edits. Marker family routes
14
+ * by tag; `auto` triggers detection.
15
+ *
16
+ * Layer C/D markers ALWAYS use the universal `@@@` envelope (vendor-
17
+ * neutral) because their wire format is heavy enough that re-emitting
18
+ * per-vendor would invite drift. Layer A/B follow the per-vendor
19
+ * SEARCH/REPLACE convention.
20
+ */
21
+ export function parseMarkers(raw, family) {
22
+ if (typeof raw !== 'string') {
23
+ throw new MarkerParseError('parseMarkers requires a string input', family);
24
+ }
25
+ const effective = family === 'auto' ? detectFamily(raw) : family;
26
+ // Layer C and Layer D envelopes are vendor-neutral and trimmed out
27
+ // before per-vendor Layer A/B parsing so the per-vendor pass cannot
28
+ // accidentally tokenise the rewrite/AST body.
29
+ const universalEdits = [];
30
+ const stripped = extractUniversalBlocks(raw, universalEdits, family);
31
+ const familyEdits = parseFamilyMarkers(stripped, effective);
32
+ return [...universalEdits, ...familyEdits];
33
+ }
34
+ /**
35
+ * Heuristic auto-detection — looks for the first distinctive marker
36
+ * and routes accordingly. Cheap to compute (single linear scan via
37
+ * `indexOf`) so we run it whenever the dispatcher gets `modelTag:
38
+ * undefined`.
39
+ */
40
+ export function detectFamily(raw) {
41
+ // Order matters: Gemini's `<<<<<<< SEARCH` is distinctive enough to
42
+ // win even when the same payload contains a unified diff in a
43
+ // comment block.
44
+ if (raw.includes('<<<<<<< SEARCH'))
45
+ return 'gemini';
46
+ if (raw.includes('+++ NEW') && raw.includes('--- OLD'))
47
+ return 'anthropic';
48
+ // Unified-diff signature: `--- a/` immediately followed by `+++ b/`
49
+ // somewhere in the payload. Looser than Anthropic's `+++ NEW`.
50
+ if (/^--- a\//m.test(raw) && /^\+\+\+ b\//m.test(raw))
51
+ return 'openai';
52
+ // Default to Anthropic — covers the Claude family which is the
53
+ // primary alpha target.
54
+ return 'anthropic';
55
+ }
56
+ // ---------------------------------------------------------------------------
57
+ // Universal Layer C / Layer D envelope extractor.
58
+ // ---------------------------------------------------------------------------
59
+ function extractUniversalBlocks(raw, out, family) {
60
+ const lines = raw.split('\n');
61
+ const kept = [];
62
+ let i = 0;
63
+ while (i < lines.length) {
64
+ const line = lines[i] ?? '';
65
+ const rewrite = line.match(/^@@@ REWRITE\s+(\S+)\s+sha256=([a-f0-9]+)\s*$/);
66
+ if (rewrite) {
67
+ const file = rewrite[1] ?? '';
68
+ const baseSha256 = (rewrite[2] ?? '').toLowerCase();
69
+ const bodyStart = i + 1;
70
+ let bodyEnd = bodyStart;
71
+ while (bodyEnd < lines.length && lines[bodyEnd] !== '@@@ END')
72
+ bodyEnd += 1;
73
+ if (bodyEnd >= lines.length) {
74
+ throw new MarkerParseError(`Layer C REWRITE block for ${file} missing '@@@ END' terminator`, family, i + 1);
75
+ }
76
+ const newContents = lines.slice(bodyStart, bodyEnd).join('\n');
77
+ out.push({
78
+ kind: 'layer-c',
79
+ edit: { file, baseSha256, newContents },
80
+ });
81
+ i = bodyEnd + 1;
82
+ continue;
83
+ }
84
+ const ast = line.match(/^@@@ AST\s+(\S+)\s+op=(\S+)\s*$/);
85
+ if (ast) {
86
+ const file = ast[1] ?? '';
87
+ const operation = (ast[2] ?? '');
88
+ const bodyStart = i + 1;
89
+ let bodyEnd = bodyStart;
90
+ while (bodyEnd < lines.length && lines[bodyEnd] !== '@@@ END')
91
+ bodyEnd += 1;
92
+ if (bodyEnd >= lines.length) {
93
+ throw new MarkerParseError(`Layer D AST block for ${file} missing '@@@ END' terminator`, family, i + 1);
94
+ }
95
+ const paramsRaw = lines.slice(bodyStart, bodyEnd).join('\n').trim();
96
+ let params = {};
97
+ if (paramsRaw.length > 0) {
98
+ try {
99
+ const parsed = JSON.parse(paramsRaw);
100
+ if (parsed && typeof parsed === 'object' && !Array.isArray(parsed)) {
101
+ params = parsed;
102
+ }
103
+ }
104
+ catch (error) {
105
+ throw new MarkerParseError(`Layer D AST params JSON parse failed for ${file}: ${error instanceof Error ? error.message : String(error)}`, family, bodyStart + 1);
106
+ }
107
+ }
108
+ out.push({
109
+ kind: 'layer-d',
110
+ edit: { file, operation, params },
111
+ });
112
+ i = bodyEnd + 1;
113
+ continue;
114
+ }
115
+ kept.push(line);
116
+ i += 1;
117
+ }
118
+ return kept.join('\n');
119
+ }
120
+ // ---------------------------------------------------------------------------
121
+ // Per-family Layer A / Layer B parsers.
122
+ // ---------------------------------------------------------------------------
123
+ function parseFamilyMarkers(raw, family) {
124
+ switch (family) {
125
+ case 'anthropic':
126
+ return parseAnthropic(raw);
127
+ case 'gemini':
128
+ return parseGemini(raw);
129
+ case 'openai':
130
+ return parseOpenAI(raw);
131
+ case 'auto':
132
+ // Unreachable — `parseMarkers` resolves `auto` to a concrete
133
+ // family before calling this. We keep the case so the switch is
134
+ // exhaustive under `noFallthroughCasesInSwitch` / `noImplicitReturns`.
135
+ return [];
136
+ }
137
+ }
138
+ /**
139
+ * Anthropic / Claude — Cline three-segment envelope:
140
+ * +++ NEW <file>
141
+ * <new>
142
+ * --- OLD <file>
143
+ * <old>
144
+ * ===
145
+ *
146
+ * Multiple blocks targeting the same `<file>` collapse into a single
147
+ * Layer B batch (preserving order). Single blocks return as Layer A.
148
+ */
149
+ // Shared between the body-terminator scan and the header-match check
150
+ // below so the two cannot drift. Triple-review 2026-05-25 P1 (Claude):
151
+ // the previous implementation used `lines[i].startsWith('--- OLD')`
152
+ // for the terminator and the regex below only for the header, so a
153
+ // model-emitted line like `// --- OLD usage was foo` inside the NEW
154
+ // body would falsely terminate the body.
155
+ const ANTHROPIC_OLD_HEADER = /^--- OLD\s+(\S+)\s*$/;
156
+ function parseAnthropic(raw) {
157
+ const blocks = [];
158
+ const lines = raw.split('\n');
159
+ let i = 0;
160
+ while (i < lines.length) {
161
+ const line = lines[i] ?? '';
162
+ const newHeader = line.match(/^\+\+\+ NEW\s+(\S+)\s*$/);
163
+ if (!newHeader) {
164
+ i += 1;
165
+ continue;
166
+ }
167
+ const file = newHeader[1] ?? '';
168
+ const startedAt = i + 1;
169
+ const newBody = [];
170
+ i += 1;
171
+ // Body terminator MUST match the same exact regex as `oldHeader`
172
+ // below (`/^--- OLD\s+\S+$/`), not a loose `startsWith('--- OLD')`.
173
+ // Triple-review 2026-05-25 P1 (Claude): a NEW body that contains a
174
+ // comment line beginning with `--- OLD foo` (e.g. a git-diff
175
+ // excerpt in a TypeScript comment, a markdown code fence showing
176
+ // what an old patch looked like) would falsely terminate the body
177
+ // and rip the rest of the NEW segment into the OLD segment, then
178
+ // throw a misleading "file mismatch" or write half-edits to disk.
179
+ while (i < lines.length && !ANTHROPIC_OLD_HEADER.test(lines[i] ?? '')) {
180
+ newBody.push(lines[i] ?? '');
181
+ i += 1;
182
+ }
183
+ if (i >= lines.length) {
184
+ throw new MarkerParseError(`Anthropic block for ${file} missing '--- OLD' segment`, 'anthropic', startedAt);
185
+ }
186
+ const oldHeader = (lines[i] ?? '').match(ANTHROPIC_OLD_HEADER);
187
+ if (!oldHeader || oldHeader[1] !== file) {
188
+ throw new MarkerParseError(`Anthropic block file mismatch: NEW ${file} vs OLD ${oldHeader?.[1] ?? '?'}`, 'anthropic', i + 1);
189
+ }
190
+ i += 1;
191
+ const oldBody = [];
192
+ while (i < lines.length && (lines[i] ?? '') !== '===') {
193
+ oldBody.push(lines[i] ?? '');
194
+ i += 1;
195
+ }
196
+ if (i >= lines.length) {
197
+ throw new MarkerParseError(`Anthropic block for ${file} missing '===' terminator`, 'anthropic', startedAt);
198
+ }
199
+ // Anthropic emits the bodies WITHOUT trailing newlines on the
200
+ // last line; our `.join('\n')` produces the same shape so a
201
+ // round-trip is byte-for-byte stable.
202
+ blocks.push({
203
+ file,
204
+ newString: newBody.join('\n'),
205
+ oldString: oldBody.join('\n'),
206
+ atLine: startedAt,
207
+ });
208
+ i += 1; // consume '==='
209
+ }
210
+ return collapseByFile(blocks);
211
+ }
212
+ /**
213
+ * Gemini / xAI — GitHub-conflict-marker envelope:
214
+ * <<<<<<< SEARCH [file]
215
+ * <old>
216
+ * =======
217
+ * <new>
218
+ * >>>>>>> REPLACE [file]
219
+ *
220
+ * `[file]` may appear on the SEARCH line, the REPLACE line, or BOTH.
221
+ * If both are present they must match.
222
+ */
223
+ function parseGemini(raw) {
224
+ const blocks = [];
225
+ const lines = raw.split('\n');
226
+ let i = 0;
227
+ while (i < lines.length) {
228
+ const line = lines[i] ?? '';
229
+ const search = line.match(/^<<<<<<< SEARCH(?:\s+(\S+))?\s*$/);
230
+ if (!search) {
231
+ i += 1;
232
+ continue;
233
+ }
234
+ const startedAt = i + 1;
235
+ const searchFile = search[1] ?? '';
236
+ i += 1;
237
+ const oldBody = [];
238
+ while (i < lines.length && (lines[i] ?? '') !== '=======') {
239
+ oldBody.push(lines[i] ?? '');
240
+ i += 1;
241
+ }
242
+ if (i >= lines.length) {
243
+ throw new MarkerParseError(`Gemini SEARCH block missing '=======' divider`, 'gemini', startedAt);
244
+ }
245
+ i += 1; // consume '======='
246
+ const newBody = [];
247
+ while (i < lines.length && !(lines[i] ?? '').startsWith('>>>>>>> REPLACE')) {
248
+ newBody.push(lines[i] ?? '');
249
+ i += 1;
250
+ }
251
+ if (i >= lines.length) {
252
+ throw new MarkerParseError(`Gemini SEARCH block missing '>>>>>>> REPLACE' terminator`, 'gemini', startedAt);
253
+ }
254
+ const replace = (lines[i] ?? '').match(/^>>>>>>> REPLACE(?:\s+(\S+))?\s*$/);
255
+ const replaceFile = replace?.[1] ?? '';
256
+ const file = searchFile || replaceFile;
257
+ if (!file) {
258
+ throw new MarkerParseError(`Gemini block missing file path on both SEARCH and REPLACE lines`, 'gemini', startedAt);
259
+ }
260
+ if (searchFile && replaceFile && searchFile !== replaceFile) {
261
+ throw new MarkerParseError(`Gemini SEARCH/REPLACE file mismatch: ${searchFile} vs ${replaceFile}`, 'gemini', i + 1);
262
+ }
263
+ i += 1; // consume REPLACE line
264
+ blocks.push({
265
+ file,
266
+ oldString: oldBody.join('\n'),
267
+ newString: newBody.join('\n'),
268
+ atLine: startedAt,
269
+ });
270
+ }
271
+ return collapseByFile(blocks);
272
+ }
273
+ /**
274
+ * OpenAI / generic — unified diff. We parse a single contiguous hunk
275
+ * per file into a Layer A block by collapsing all `-` lines into
276
+ * oldString and all `+` lines into newString, preserving leading
277
+ * context. Multi-hunk files become a Layer B batch (one sub-edit per
278
+ * hunk). The grammar is intentionally minimal — Pugi does not need
279
+ * full GNU diff compatibility because the model owns the format
280
+ * choice.
281
+ */
282
+ function parseOpenAI(raw) {
283
+ const blocks = [];
284
+ const lines = raw.split('\n');
285
+ let i = 0;
286
+ let currentFile = null;
287
+ while (i < lines.length) {
288
+ const line = lines[i] ?? '';
289
+ const aHeader = line.match(/^--- a\/(.+)$/);
290
+ if (aHeader) {
291
+ // Expect the matching `+++ b/<file>` on the next line.
292
+ const next = lines[i + 1] ?? '';
293
+ const bHeader = next.match(/^\+\+\+ b\/(.+)$/);
294
+ if (!bHeader) {
295
+ throw new MarkerParseError(`OpenAI unified diff: '--- a/${aHeader[1]}' not followed by '+++ b/'`, 'openai', i + 1);
296
+ }
297
+ if (aHeader[1] !== bHeader[1]) {
298
+ throw new MarkerParseError(`OpenAI unified diff file mismatch: a/${aHeader[1]} vs b/${bHeader[1]}`, 'openai', i + 2);
299
+ }
300
+ currentFile = bHeader[1] ?? '';
301
+ i += 2;
302
+ continue;
303
+ }
304
+ if (line.startsWith('@@') && currentFile) {
305
+ const startedAt = i + 1;
306
+ i += 1;
307
+ const oldLines = [];
308
+ const newLines = [];
309
+ while (i < lines.length &&
310
+ !(lines[i] ?? '').startsWith('@@') &&
311
+ !(lines[i] ?? '').startsWith('--- a/')) {
312
+ const hl = lines[i] ?? '';
313
+ if (hl.startsWith('+'))
314
+ newLines.push(hl.slice(1));
315
+ else if (hl.startsWith('-'))
316
+ oldLines.push(hl.slice(1));
317
+ else if (hl.startsWith(' ')) {
318
+ // Context line — preserved in both sides to keep the
319
+ // oldString unique. Strip the leading space.
320
+ const ctx = hl.slice(1);
321
+ oldLines.push(ctx);
322
+ newLines.push(ctx);
323
+ }
324
+ else if (hl === '') {
325
+ // Bare empty lines in a unified diff are ambiguous — some
326
+ // emitters use them as a terminator, others as a literal
327
+ // empty context line. We treat them as context (preserved
328
+ // on both sides) which is the conservative choice; if the
329
+ // model meant terminator the next `@@` or `--- a/` will
330
+ // catch it.
331
+ oldLines.push('');
332
+ newLines.push('');
333
+ }
334
+ else {
335
+ // `\` lines ('') and any other
336
+ // metadata are ignored.
337
+ }
338
+ i += 1;
339
+ }
340
+ blocks.push({
341
+ file: currentFile,
342
+ oldString: oldLines.join('\n'),
343
+ newString: newLines.join('\n'),
344
+ atLine: startedAt,
345
+ });
346
+ continue;
347
+ }
348
+ i += 1;
349
+ }
350
+ if (blocks.length === 0 && raw.includes('--- a/')) {
351
+ throw new MarkerParseError('OpenAI unified diff parsed zero hunks — payload may be malformed', 'openai');
352
+ }
353
+ return collapseByFile(blocks);
354
+ }
355
+ /**
356
+ * Collapse N blocks-per-file into either a single Layer A (N=1) or a
357
+ * single Layer B batch (N>1). The dispatcher routes Layer A and
358
+ * Layer B differently; we make that choice at parse time so the
359
+ * dispatcher stays format-agnostic.
360
+ */
361
+ function collapseByFile(blocks) {
362
+ const byFile = new Map();
363
+ // Preserve first-seen order across files.
364
+ const order = [];
365
+ for (const block of blocks) {
366
+ const list = byFile.get(block.file);
367
+ if (list) {
368
+ list.push({ oldString: block.oldString, newString: block.newString });
369
+ }
370
+ else {
371
+ byFile.set(block.file, [{ oldString: block.oldString, newString: block.newString }]);
372
+ order.push(block.file);
373
+ }
374
+ }
375
+ const out = [];
376
+ for (const file of order) {
377
+ const subs = byFile.get(file) ?? [];
378
+ if (subs.length === 1) {
379
+ const only = subs[0];
380
+ out.push({
381
+ kind: 'layer-a',
382
+ edit: {
383
+ file,
384
+ oldString: only.oldString,
385
+ newString: only.newString,
386
+ },
387
+ });
388
+ }
389
+ else {
390
+ out.push({
391
+ kind: 'layer-b',
392
+ edit: {
393
+ file,
394
+ edits: subs,
395
+ },
396
+ });
397
+ }
398
+ }
399
+ return out;
400
+ }
401
+ //# sourceMappingURL=marker-parser.js.map
@@ -0,0 +1,223 @@
1
+ /**
2
+ * Shared security gate for the α6.6 diff escalation Layer A/B/C
3
+ * applicators.
4
+ *
5
+ * Before α6.6.1 (triple-review remediation 2026-05-25) every layer
6
+ * built its own path resolver via `isAbsolute(file) ? file : resolve(cwd, file)`.
7
+ * That bypassed the workspace-scoped resolver + protected-basename
8
+ * gate + symlink-escape check used by the `edit`/`write` tools in
9
+ * `apps/pugi-cli/src/tools/file-tools.ts`. A model that emitted
10
+ * `+++ NEW ../../../../etc/passwd` (or `+++ NEW .env`) would have
11
+ * happily landed an atomic write outside the workspace or onto a
12
+ * protected basename.
13
+ *
14
+ * This module is the single chokepoint every layer goes through.
15
+ * Centralised here so:
16
+ *
17
+ * 1. Future layers (D = AST, planned β-tier) inherit the gate for
18
+ * free instead of re-implementing it.
19
+ * 2. New protected-file rules added to `decidePermission` propagate
20
+ * to dispatcher writes uniformly.
21
+ * 3. A single spec surface covers every layer's path-handling
22
+ * contract (`apps/pugi-cli/test/edits-security-gate.spec.ts`).
23
+ *
24
+ * Defense layers, in order (fail-fast — first reject wins):
25
+ *
26
+ * a. `resolveWorkspacePath` — workspace-scoped resolver with URL-decode
27
+ * and realpath gate at the target. Throws on traversal; we
28
+ * translate the throw into `path_outside_workspace`.
29
+ * b. `decidePermission` (kind: 'edit') — protected basename / suffix /
30
+ * credential-path check (`.env`, `*.pem`, `id_rsa`, `~/.ssh/**`,
31
+ * etc.). Mirrors what `writeTool` / `editTool` already enforce.
32
+ * c. Explicit `realpathSync.native` re-check on the target (or
33
+ * target's parent when the target does not yet exist) to defend
34
+ * against TOCTOU symlink swap between `resolveWorkspacePath` and
35
+ * the apply-time write.
36
+ * d. Re-run `decidePermission` on the REAL target's workspace-relative
37
+ * form. R2 triple-review (2026-05-25, Codex P1) caught the
38
+ * symlink → protected-file bypass: a workspace-local symlink
39
+ * `alias -> .env` passes step (b) because `alias` is not a
40
+ * protected basename, and passes step (c) because `.env`
41
+ * legitimately lives inside the workspace realpath. Without step
42
+ * (d) the gate would allow the atomic write to land on `.env`
43
+ * via the symlink's resolved target. Re-running the protected-file
44
+ * check against the resolved name closes that loop.
45
+ */
46
+ import { realpathSync } from 'node:fs';
47
+ import { basename, resolve, sep, relative } from 'node:path';
48
+ import { decidePermission } from '../permission.js';
49
+ import { resolveWorkspacePath } from '../path-security.js';
50
+ import { loadSettings } from '../settings.js';
51
+ /**
52
+ * Apply the full path-traversal + protected-file + symlink-escape gate
53
+ * for an inbound edit request. Returns the resolved absolute path on
54
+ * success; never throws on the security-relevant rejections. Filesystem
55
+ * errors that are NOT security-relevant (an unexpected EACCES on a
56
+ * realpath that doesn't involve a symlink, etc.) are re-thrown so the
57
+ * layer's existing error path records them as `write_error`.
58
+ */
59
+ export function applySecurityGate(inputPath, opts) {
60
+ // (a) Workspace-scoped path resolution. `resolveWorkspacePath`
61
+ // throws on URL-encoded traversal, parent-chain escapes via `..`,
62
+ // and target-symlinks that point outside the workspace root. We
63
+ // funnel the throw into a structured rejection so the dispatcher
64
+ // can render a deterministic operator-facing message.
65
+ let resolved;
66
+ try {
67
+ resolved = resolveWorkspacePath(opts.cwd, inputPath);
68
+ }
69
+ catch (error) {
70
+ const message = error instanceof Error ? error.message : String(error);
71
+ return {
72
+ ok: false,
73
+ reason: 'path_outside_workspace',
74
+ detail: `${message}`,
75
+ };
76
+ }
77
+ // (b) Protected-file gate. The same `decidePermission` call the
78
+ // `edit` / `write` tools make. We pass the workspace-relative form
79
+ // because `protectedTargetReason` uses `basename(resolve(root, target))`
80
+ // internally — passing an absolute path that already lives under
81
+ // `root` works because Node's `resolve` is idempotent, but the
82
+ // workspace-relative form keeps the audit log compact.
83
+ const settings = opts.settings ?? loadSettings(opts.cwd);
84
+ const targetForPermission = isInsideRoot(resolved, opts.cwd)
85
+ ? relative(opts.cwd, resolved) || inputPath
86
+ : inputPath;
87
+ const decision = decidePermission({ tool: opts.toolName, kind: 'edit', target: targetForPermission }, settings, opts.cwd);
88
+ if (decision.decision !== 'allow') {
89
+ return {
90
+ ok: false,
91
+ reason: 'protected_file',
92
+ detail: `${decision.decision} for ${targetForPermission}: ${decision.reason}`,
93
+ };
94
+ }
95
+ // (c) Symlink-escape re-check. Closes the TOCTOU window between
96
+ // `resolveWorkspacePath` and the apply-time write. If the target
97
+ // file does not yet exist (legitimate for any future "create"
98
+ // path), we walk up to the parent — realpathSync throws ENOENT on
99
+ // a missing leaf — and require the parent realpath to live inside
100
+ // the workspace realpath. The target's basename is then anchored to
101
+ // the parent realpath so a final-segment symlink swap cannot escape.
102
+ let realRoot;
103
+ try {
104
+ realRoot = realpathSync.native(opts.cwd);
105
+ }
106
+ catch (error) {
107
+ const message = error instanceof Error ? error.message : String(error);
108
+ return {
109
+ ok: false,
110
+ reason: 'symlink_escape',
111
+ detail: `cannot realpath workspace root: ${message}`,
112
+ };
113
+ }
114
+ let realTarget;
115
+ try {
116
+ realTarget = realpathSync.native(resolved);
117
+ }
118
+ catch (error) {
119
+ const code = error.code;
120
+ if (code !== 'ENOENT' && code !== 'ENOTDIR') {
121
+ // Unexpected filesystem error — re-throw so the layer records
122
+ // it as `write_error` rather than silently rejecting as
123
+ // `symlink_escape` (which would mis-attribute the cause to the
124
+ // operator).
125
+ throw error;
126
+ }
127
+ // Target does not exist — anchor against the parent's realpath.
128
+ let realParent;
129
+ try {
130
+ realParent = realpathSync.native(resolve(resolved, '..'));
131
+ }
132
+ catch (parentError) {
133
+ const parentCode = parentError.code;
134
+ if (parentCode !== 'ENOENT' && parentCode !== 'ENOTDIR')
135
+ throw parentError;
136
+ return {
137
+ ok: false,
138
+ reason: 'symlink_escape',
139
+ detail: `parent directory does not exist for ${inputPath}`,
140
+ };
141
+ }
142
+ if (!isInsideRoot(realParent, realRoot)) {
143
+ return {
144
+ ok: false,
145
+ reason: 'symlink_escape',
146
+ detail: `parent realpath ${realParent} escapes workspace ${realRoot}`,
147
+ };
148
+ }
149
+ // (d) Re-run the protected-file check against the parent's
150
+ // realpath-anchored basename. The basename of `resolved` is the
151
+ // final path segment the write will land on; pairing it with the
152
+ // resolved parent gives the workspace-relative form the
153
+ // protected-basename matcher needs. Closes the "create-new-file
154
+ // symlink-parent" variant: a dir symlink whose realpath stays
155
+ // inside the workspace must not let an attacker create a new
156
+ // `.env` (or any other protected basename) via a non-protected
157
+ // relative path.
158
+ const realCreateTarget = resolve(realParent, basename(resolved));
159
+ const recheck = recheckProtectedOnReal(realCreateTarget, realRoot, opts);
160
+ if (recheck)
161
+ return recheck;
162
+ return { ok: true, absPath: resolved };
163
+ }
164
+ if (!isInsideRoot(realTarget, realRoot)) {
165
+ return {
166
+ ok: false,
167
+ reason: 'symlink_escape',
168
+ detail: `target realpath ${realTarget} escapes workspace ${realRoot}`,
169
+ };
170
+ }
171
+ // (d) Re-run the protected-file check on the resolved target. The
172
+ // pre-fix gate only checked the operator-supplied path; a symlink
173
+ // `alias -> .env` would slip through because `alias` is not
174
+ // protected. Now we re-key the permission decision on the realpath's
175
+ // workspace-relative form so the basename matcher sees `.env` (or
176
+ // any other protected suffix) the symlink points at. The settings
177
+ // pull is free — it is the same `settings` we loaded at step (b).
178
+ const recheck = recheckProtectedOnReal(realTarget, realRoot, opts);
179
+ if (recheck)
180
+ return recheck;
181
+ return { ok: true, absPath: resolved };
182
+ }
183
+ /**
184
+ * Helper for step (d) of the gate. Re-runs `decidePermission` on the
185
+ * real (symlink-resolved) target's workspace-relative form and returns
186
+ * a `protected_file` rejection when the basename matches a protected
187
+ * pattern. Returns null when the recheck passes; the caller continues
188
+ * to `ok: true`.
189
+ *
190
+ * Lifted into a helper because both the existing-target branch and the
191
+ * missing-target (create) branch in `applySecurityGate` need the same
192
+ * check, and inlining it twice would invite drift the next time the
193
+ * protected-file rules grow.
194
+ */
195
+ function recheckProtectedOnReal(realTarget, realRoot, opts) {
196
+ if (!isInsideRoot(realTarget, realRoot))
197
+ return null;
198
+ const realRelative = relative(realRoot, realTarget);
199
+ if (realRelative === '')
200
+ return null;
201
+ const settings = opts.settings ?? loadSettings(opts.cwd);
202
+ const realDecision = decidePermission({ tool: opts.toolName, kind: 'edit', target: realRelative }, settings, realRoot);
203
+ if (realDecision.decision !== 'allow') {
204
+ return {
205
+ ok: false,
206
+ reason: 'protected_file',
207
+ detail: `${realDecision.decision} for symlink target ${realRelative}: ${realDecision.reason}`,
208
+ };
209
+ }
210
+ return null;
211
+ }
212
+ /**
213
+ * True iff `child` is `root` or strictly nested inside it. Uses the
214
+ * platform separator + a trailing `sep` guard to avoid the
215
+ * `/workspace` vs `/workspace-evil` false-positive that a naive
216
+ * `startsWith` check would produce.
217
+ */
218
+ function isInsideRoot(child, root) {
219
+ if (child === root)
220
+ return true;
221
+ return child.startsWith(root + sep);
222
+ }
223
+ //# sourceMappingURL=security-gate.js.map