@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.
- package/README.md +33 -0
- package/assets/pugi-mascot.ansi +41 -0
- package/dist/commands/deploy.js +439 -0
- package/dist/core/agents/loader.js +104 -0
- package/dist/core/agents/registry.js +1 -1
- package/dist/core/consensus/anvil-fanout.js +276 -0
- package/dist/core/consensus/diff-capture.js +382 -0
- package/dist/core/consensus/rubric.js +233 -0
- package/dist/core/context/index.js +21 -0
- package/dist/core/context/pugiignore.js +316 -0
- package/dist/core/context/repo-skeleton.js +533 -0
- package/dist/core/context/watcher.js +342 -0
- package/dist/core/context/working-set.js +165 -0
- package/dist/core/edits/dispatch.js +185 -0
- package/dist/core/edits/index.js +15 -0
- package/dist/core/edits/layer-a-apply.js +217 -0
- package/dist/core/edits/layer-b-apply.js +211 -0
- package/dist/core/edits/layer-c-apply.js +160 -0
- package/dist/core/edits/layer-d-ast.js +29 -0
- package/dist/core/edits/marker-parser.js +401 -0
- package/dist/core/edits/security-gate.js +223 -0
- package/dist/core/edits/worktree.js +229 -0
- package/dist/core/engine/native-pugi.js +6 -1
- package/dist/core/engine/prompts.js +4 -1
- package/dist/core/engine/tool-bridge.js +33 -1
- package/dist/core/lsp/client.js +631 -0
- package/dist/core/repl/ask.js +512 -0
- package/dist/core/repl/cancellation.js +98 -0
- package/dist/core/repl/dispatch-fsm.js +220 -0
- package/dist/core/repl/privacy-banner.js +71 -0
- package/dist/core/repl/session.js +1896 -13
- package/dist/core/repl/slash-commands.js +59 -32
- package/dist/core/repl/store/index.js +12 -0
- package/dist/core/repl/store/jsonl-log.js +321 -0
- package/dist/core/repl/store/lockfile.js +155 -0
- package/dist/core/repl/store/session-store.js +792 -0
- package/dist/core/repl/store/types.js +44 -0
- package/dist/core/repl/store/uuid-v7.js +68 -0
- package/dist/core/repl/workspace-context.js +72 -1
- package/dist/core/skills/loader.js +454 -0
- package/dist/core/skills/sources.js +480 -0
- package/dist/core/skills/trust.js +172 -0
- package/dist/runtime/cli.js +767 -10
- package/dist/runtime/commands/agents.js +385 -0
- package/dist/runtime/commands/config.js +338 -8
- package/dist/runtime/commands/lsp.js +184 -0
- package/dist/runtime/commands/patch.js +111 -0
- package/dist/runtime/commands/review-consensus.js +399 -0
- package/dist/runtime/commands/skills.js +401 -0
- package/dist/runtime/commands/worktree.js +133 -0
- package/dist/tools/apply-patch.js +314 -0
- package/dist/tools/file-tools.js +90 -0
- package/dist/tools/lsp-tools.js +189 -0
- package/dist/tools/registry.js +18 -0
- package/dist/tools/web-fetch.js +1 -1
- package/dist/tui/agent-tree-pane.js +9 -0
- package/dist/tui/ask-cli.js +52 -0
- package/dist/tui/ask-modal.js +211 -0
- package/dist/tui/conversation-pane.js +48 -3
- package/dist/tui/input-box.js +48 -5
- package/dist/tui/markdown-render.js +266 -0
- package/dist/tui/repl-render.js +185 -0
- package/dist/tui/repl-splash-mascot.js +130 -0
- package/dist/tui/repl-splash.js +7 -1
- package/dist/tui/repl.js +82 -11
- package/dist/tui/status-bar.js +63 -3
- package/dist/tui/tool-stream-pane.js +91 -0
- package/package.json +11 -5
|
@@ -0,0 +1,342 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* chokidar-based filewatch - α6.5 Phase 1 (three-tier context).
|
|
3
|
+
*
|
|
4
|
+
* The REPL needs a live signal when files change so the operator's
|
|
5
|
+
* agent can re-read a stale file, and so the status bar can surface a
|
|
6
|
+
* "file changed" badge. chokidar wraps fsevents / inotify / kqueue so
|
|
7
|
+
* the cross-platform surface stays sane.
|
|
8
|
+
*
|
|
9
|
+
* Design choices:
|
|
10
|
+
*
|
|
11
|
+
* 1. **Ignore-aware**: every watched path is filtered through the
|
|
12
|
+
* same `PugiIgnore` matcher the skeleton walker uses. We pass the
|
|
13
|
+
* matcher into chokidar's `ignored` option so `node_modules/`,
|
|
14
|
+
* `dist/`, `.git/`, and secret files are never watched at all -
|
|
15
|
+
* this matters because chokidar's resource bound is the number
|
|
16
|
+
* of watched paths, not the number of total files in the repo.
|
|
17
|
+
*
|
|
18
|
+
* 2. **Throttle**: chokidar can fire dozens of `change` events when
|
|
19
|
+
* a code formatter rewrites a directory. We batch events in a
|
|
20
|
+
* `THROTTLE_WINDOW_MS` window and emit a single aggregated event
|
|
21
|
+
* so the REPL system line / status bar do not flicker.
|
|
22
|
+
*
|
|
23
|
+
* 3. **Watch cap**: per spec, the watcher caps watched paths at
|
|
24
|
+
* `MAX_WATCHED_PATHS`. Once the count crosses the cap we close
|
|
25
|
+
* the watcher and fall back to "no live updates this session" -
|
|
26
|
+
* better to silently lose the badge than to consume thousands of
|
|
27
|
+
* file descriptors. The fallback path emits one warning event so
|
|
28
|
+
* the operator knows live updates are off.
|
|
29
|
+
*
|
|
30
|
+
* 4. **Lifecycle**: `start()` is async (chokidar's `ready` event)
|
|
31
|
+
* so the caller can `await` the initial scan. `close()` is also
|
|
32
|
+
* async (chokidar's close is async). The watcher is reusable
|
|
33
|
+
* across multiple `start` / `close` pairs, but Phase 1 expects
|
|
34
|
+
* one-watcher-per-REPL-session.
|
|
35
|
+
*
|
|
36
|
+
* 5. **Testability**: the chokidar handle is injectable via the
|
|
37
|
+
* `watchFactory` option. Production wires `chokidar.watch`;
|
|
38
|
+
* tests pass a fake emitter that lets the spec drive `add` /
|
|
39
|
+
* `change` / `unlink` events synchronously.
|
|
40
|
+
*/
|
|
41
|
+
import { EventEmitter } from 'node:events';
|
|
42
|
+
import chokidar from 'chokidar';
|
|
43
|
+
/** Per-spec watch-path cap. */
|
|
44
|
+
export const MAX_WATCHED_PATHS = 10_000;
|
|
45
|
+
/** Batching window for the change throttle. */
|
|
46
|
+
export const THROTTLE_WINDOW_MS = 250;
|
|
47
|
+
/**
|
|
48
|
+
* Number of `add` events between cap-check re-evaluations. The ready
|
|
49
|
+
* handler triggers the FIRST cap check, but a `git checkout` of a
|
|
50
|
+
* giant subtree post-ready can blow past the cap without ever
|
|
51
|
+
* re-evaluating. Sample every N adds so the cap fires within a
|
|
52
|
+
* bounded window even on post-ready growth. Cheap: one Map
|
|
53
|
+
* iteration per N adds. triple-review P2 (PR #380).
|
|
54
|
+
*/
|
|
55
|
+
export const CAP_CHECK_ADD_INTERVAL = 1_000;
|
|
56
|
+
/**
|
|
57
|
+
* Concrete watcher. Construct, then `await start()`. Subscribe via
|
|
58
|
+
* `.on('batch', cb)`. `await close()` to tear down.
|
|
59
|
+
*/
|
|
60
|
+
export class PugiWatcher extends EventEmitter {
|
|
61
|
+
cwd;
|
|
62
|
+
ignore;
|
|
63
|
+
maxWatchedPaths;
|
|
64
|
+
throttleWindowMs;
|
|
65
|
+
now;
|
|
66
|
+
watchFactory;
|
|
67
|
+
setTimeoutImpl;
|
|
68
|
+
clearTimeoutImpl;
|
|
69
|
+
handle = null;
|
|
70
|
+
buffer = [];
|
|
71
|
+
windowStart = null;
|
|
72
|
+
timer = null;
|
|
73
|
+
closed = false;
|
|
74
|
+
capExceededEmitted = false;
|
|
75
|
+
capCheckTimer = null;
|
|
76
|
+
/**
|
|
77
|
+
* Counter of `add` events received since the last cap-check sample.
|
|
78
|
+
* When it crosses CAP_CHECK_ADD_INTERVAL we schedule another check.
|
|
79
|
+
* Resets on every cap-check run. triple-review P2 (PR #380).
|
|
80
|
+
*/
|
|
81
|
+
addsSinceLastCapCheck = 0;
|
|
82
|
+
constructor(options) {
|
|
83
|
+
super();
|
|
84
|
+
this.cwd = options.cwd;
|
|
85
|
+
this.ignore = options.ignore;
|
|
86
|
+
this.maxWatchedPaths = options.maxWatchedPaths ?? MAX_WATCHED_PATHS;
|
|
87
|
+
this.throttleWindowMs = options.throttleWindowMs ?? THROTTLE_WINDOW_MS;
|
|
88
|
+
this.now = options.now ?? (() => Date.now());
|
|
89
|
+
this.watchFactory = options.watchFactory ?? defaultWatchFactory;
|
|
90
|
+
this.setTimeoutImpl = options.setTimeoutImpl ?? ((fn, ms) => setTimeout(fn, ms));
|
|
91
|
+
this.clearTimeoutImpl = options.clearTimeoutImpl ?? ((h) => clearTimeout(h));
|
|
92
|
+
// Default no-op error listener - Node's EventEmitter.emit('error', ...)
|
|
93
|
+
// THROWS synchronously when no listener is registered. Production
|
|
94
|
+
// bootstrap awaits start() before constructing ReplSession, so there
|
|
95
|
+
// is no listener on the watcher at the moment chokidar may surface
|
|
96
|
+
// an error during the initial scan. The default listener swallows
|
|
97
|
+
// the event so emit() does not crash the CLI; any real consumer
|
|
98
|
+
// that subscribes later receives the events as normal because
|
|
99
|
+
// EventEmitter delivers to every registered listener, not just one.
|
|
100
|
+
// Codex P1 (PR #380).
|
|
101
|
+
this.on('error', () => {
|
|
102
|
+
/* defensive no-op so emit('error') never throws */
|
|
103
|
+
});
|
|
104
|
+
}
|
|
105
|
+
/**
|
|
106
|
+
* Open the chokidar handle and wait for the `ready` event. Resolves
|
|
107
|
+
* once the initial scan completes. Rejects when the underlying
|
|
108
|
+
* watcher errors out before `ready`.
|
|
109
|
+
*/
|
|
110
|
+
async start() {
|
|
111
|
+
if (this.handle)
|
|
112
|
+
return;
|
|
113
|
+
if (this.closed) {
|
|
114
|
+
throw new Error('PugiWatcher: cannot start a closed watcher');
|
|
115
|
+
}
|
|
116
|
+
const opts = {
|
|
117
|
+
// chokidar does not tell the predicate whether the path is a
|
|
118
|
+
// file or a directory, so we ask the matcher both ways. If
|
|
119
|
+
// either form matches, the path is excluded. This catches
|
|
120
|
+
// gitignore-style dir patterns (`node_modules/`) that would
|
|
121
|
+
// otherwise only match descendants, leaving the dir itself
|
|
122
|
+
// visible to chokidar's recursion.
|
|
123
|
+
ignored: (path) => this.ignore.isIgnored(path) || this.ignore.isIgnored(path, true),
|
|
124
|
+
ignoreInitial: true,
|
|
125
|
+
persistent: true,
|
|
126
|
+
// Disable polling - we want fsevents / inotify / kqueue speed.
|
|
127
|
+
usePolling: false,
|
|
128
|
+
// Aggregate events so we get one `change` per save, not three.
|
|
129
|
+
awaitWriteFinish: {
|
|
130
|
+
stabilityThreshold: 50,
|
|
131
|
+
pollInterval: 25,
|
|
132
|
+
},
|
|
133
|
+
// Keep the depth shallow at the chokidar layer too - even after
|
|
134
|
+
// ignore filtering, a multi-million file dir could exhaust file
|
|
135
|
+
// descriptors. The skeleton walker has its own depth cap; this
|
|
136
|
+
// one is a safety net.
|
|
137
|
+
depth: 12,
|
|
138
|
+
};
|
|
139
|
+
const watcher = this.watchFactory(this.cwd, opts);
|
|
140
|
+
this.handle = watcher;
|
|
141
|
+
return new Promise((resolveReady, rejectReady) => {
|
|
142
|
+
let settled = false;
|
|
143
|
+
watcher.on('ready', () => {
|
|
144
|
+
if (settled)
|
|
145
|
+
return;
|
|
146
|
+
settled = true;
|
|
147
|
+
this.emit('ready');
|
|
148
|
+
this.scheduleCapCheck();
|
|
149
|
+
resolveReady();
|
|
150
|
+
});
|
|
151
|
+
watcher.on('error', (err) => {
|
|
152
|
+
const error = err instanceof Error ? err : new Error(String(err));
|
|
153
|
+
this.emit('error', error);
|
|
154
|
+
if (!settled) {
|
|
155
|
+
settled = true;
|
|
156
|
+
rejectReady(error);
|
|
157
|
+
}
|
|
158
|
+
});
|
|
159
|
+
watcher.on('add', (path) => {
|
|
160
|
+
this.queue('add', path);
|
|
161
|
+
// Re-evaluate the cap every CAP_CHECK_ADD_INTERVAL adds so a
|
|
162
|
+
// post-ready burst (e.g. `git checkout` of a giant subtree)
|
|
163
|
+
// does not silently blow past the watched-paths cap. The
|
|
164
|
+
// ready-time check is preserved; this is the missing follow-up
|
|
165
|
+
// sampler the comment promised. triple-review P2 (PR #380).
|
|
166
|
+
this.addsSinceLastCapCheck += 1;
|
|
167
|
+
if (this.addsSinceLastCapCheck >= CAP_CHECK_ADD_INTERVAL) {
|
|
168
|
+
this.addsSinceLastCapCheck = 0;
|
|
169
|
+
this.scheduleCapCheck();
|
|
170
|
+
}
|
|
171
|
+
});
|
|
172
|
+
watcher.on('change', (path) => this.queue('change', path));
|
|
173
|
+
watcher.on('unlink', (path) => this.queue('unlink', path));
|
|
174
|
+
});
|
|
175
|
+
}
|
|
176
|
+
/** Close the underlying watcher and clear the throttle timer. */
|
|
177
|
+
async close() {
|
|
178
|
+
this.closed = true;
|
|
179
|
+
if (this.timer !== null) {
|
|
180
|
+
this.clearTimeoutImpl(this.timer);
|
|
181
|
+
this.timer = null;
|
|
182
|
+
}
|
|
183
|
+
if (this.capCheckTimer !== null) {
|
|
184
|
+
this.clearTimeoutImpl(this.capCheckTimer);
|
|
185
|
+
this.capCheckTimer = null;
|
|
186
|
+
}
|
|
187
|
+
if (this.handle) {
|
|
188
|
+
try {
|
|
189
|
+
await this.handle.close();
|
|
190
|
+
}
|
|
191
|
+
catch {
|
|
192
|
+
/* idempotent - chokidar may already be closed */
|
|
193
|
+
}
|
|
194
|
+
this.handle = null;
|
|
195
|
+
}
|
|
196
|
+
this.buffer = [];
|
|
197
|
+
this.windowStart = null;
|
|
198
|
+
}
|
|
199
|
+
/**
|
|
200
|
+
* Force-flush any pending events. Exposed primarily for tests that
|
|
201
|
+
* want to assert the batched payload without waiting for the
|
|
202
|
+
* timeout. Production code can also use this on REPL teardown so
|
|
203
|
+
* pending events are surfaced before close.
|
|
204
|
+
*/
|
|
205
|
+
flush() {
|
|
206
|
+
if (this.buffer.length === 0)
|
|
207
|
+
return;
|
|
208
|
+
if (this.timer !== null) {
|
|
209
|
+
this.clearTimeoutImpl(this.timer);
|
|
210
|
+
this.timer = null;
|
|
211
|
+
}
|
|
212
|
+
const events = this.buffer;
|
|
213
|
+
this.buffer = [];
|
|
214
|
+
const start = this.windowStart ?? this.now();
|
|
215
|
+
const end = this.now();
|
|
216
|
+
this.windowStart = null;
|
|
217
|
+
this.emit('batch', {
|
|
218
|
+
events: dedupeBatch(events),
|
|
219
|
+
windowStartEpochMs: start,
|
|
220
|
+
windowEndEpochMs: end,
|
|
221
|
+
});
|
|
222
|
+
}
|
|
223
|
+
/** Test surface - returns the current pending buffer. */
|
|
224
|
+
pendingCount() {
|
|
225
|
+
return this.buffer.length;
|
|
226
|
+
}
|
|
227
|
+
/* ------------------------------------------------------------ */
|
|
228
|
+
/* Internals */
|
|
229
|
+
/* ------------------------------------------------------------ */
|
|
230
|
+
queue(kind, path) {
|
|
231
|
+
if (this.closed)
|
|
232
|
+
return;
|
|
233
|
+
// Defensive double-check: even if chokidar's `ignored` predicate
|
|
234
|
+
// accepted the path, the matcher gets the last word. Same dir-aware
|
|
235
|
+
// double-call as the predicate above.
|
|
236
|
+
if (this.ignore.isIgnored(path) || this.ignore.isIgnored(path, true))
|
|
237
|
+
return;
|
|
238
|
+
const atEpochMs = this.now();
|
|
239
|
+
const event = {
|
|
240
|
+
path: relPathPosix(this.cwd, path),
|
|
241
|
+
absPath: path,
|
|
242
|
+
kind,
|
|
243
|
+
atEpochMs,
|
|
244
|
+
};
|
|
245
|
+
if (this.windowStart === null)
|
|
246
|
+
this.windowStart = atEpochMs;
|
|
247
|
+
this.buffer.push(event);
|
|
248
|
+
if (this.timer === null) {
|
|
249
|
+
this.timer = this.setTimeoutImpl(() => this.flush(), this.throttleWindowMs);
|
|
250
|
+
}
|
|
251
|
+
}
|
|
252
|
+
/**
|
|
253
|
+
* Lazy cap check - chokidar reports its watched paths via
|
|
254
|
+
* `getWatched()`. We sample once on ready, and again every
|
|
255
|
+
* CAP_CHECK_ADD_INTERVAL `add` events thereafter (e.g. a `git
|
|
256
|
+
* checkout` of a giant subtree post-ready). The cost is one Map
|
|
257
|
+
* iteration; the payoff is silent fallback when an operator targets
|
|
258
|
+
* a giant repo. Guarded against double-scheduling so the add-handler
|
|
259
|
+
* cannot stack multiple timers.
|
|
260
|
+
*/
|
|
261
|
+
scheduleCapCheck() {
|
|
262
|
+
if (this.closed || this.capExceededEmitted)
|
|
263
|
+
return;
|
|
264
|
+
if (this.capCheckTimer !== null)
|
|
265
|
+
return;
|
|
266
|
+
this.capCheckTimer = this.setTimeoutImpl(() => this.runCapCheck(), 100);
|
|
267
|
+
}
|
|
268
|
+
runCapCheck() {
|
|
269
|
+
this.capCheckTimer = null;
|
|
270
|
+
if (this.closed || this.capExceededEmitted || !this.handle)
|
|
271
|
+
return;
|
|
272
|
+
try {
|
|
273
|
+
const watched = this.handle.getWatched();
|
|
274
|
+
let count = 0;
|
|
275
|
+
for (const entries of Object.values(watched)) {
|
|
276
|
+
count += entries.length;
|
|
277
|
+
}
|
|
278
|
+
if (count > this.maxWatchedPaths) {
|
|
279
|
+
this.capExceededEmitted = true;
|
|
280
|
+
this.emit('capExceeded', {
|
|
281
|
+
watchedCount: count,
|
|
282
|
+
cap: this.maxWatchedPaths,
|
|
283
|
+
});
|
|
284
|
+
// Stop watching so we do not consume more file descriptors.
|
|
285
|
+
void this.close();
|
|
286
|
+
}
|
|
287
|
+
}
|
|
288
|
+
catch {
|
|
289
|
+
/* chokidar threw on getWatched - safe to ignore, cap check is best-effort */
|
|
290
|
+
}
|
|
291
|
+
}
|
|
292
|
+
}
|
|
293
|
+
/* ------------------------------------------------------------------ */
|
|
294
|
+
/* Helpers */
|
|
295
|
+
/* ------------------------------------------------------------------ */
|
|
296
|
+
/**
|
|
297
|
+
* Convert an absolute path to a repo-relative POSIX path. chokidar
|
|
298
|
+
* already uses forward slashes on Windows because it normalises
|
|
299
|
+
* internally, but we run the same fix unconditionally so the contract
|
|
300
|
+
* is platform-agnostic.
|
|
301
|
+
*/
|
|
302
|
+
function relPathPosix(cwd, abs) {
|
|
303
|
+
// chokidar may report paths already relative to cwd. Normalise both
|
|
304
|
+
// sides to forward slashes for the prefix compare.
|
|
305
|
+
const cwdPosix = cwd.replace(/\\/g, '/');
|
|
306
|
+
const absPosix = abs.replace(/\\/g, '/');
|
|
307
|
+
if (absPosix.startsWith(`${cwdPosix}/`)) {
|
|
308
|
+
return absPosix.slice(cwdPosix.length + 1);
|
|
309
|
+
}
|
|
310
|
+
if (absPosix === cwdPosix)
|
|
311
|
+
return '';
|
|
312
|
+
return absPosix;
|
|
313
|
+
}
|
|
314
|
+
/**
|
|
315
|
+
* Collapse consecutive (path, kind) duplicates inside a batch. We
|
|
316
|
+
* keep the FIRST occurrence so the order is "first-touch first".
|
|
317
|
+
* Different kinds for the same path are preserved (`add` then
|
|
318
|
+
* `change`) - the model wants both signals.
|
|
319
|
+
*/
|
|
320
|
+
function dedupeBatch(events) {
|
|
321
|
+
const seen = new Set();
|
|
322
|
+
const out = [];
|
|
323
|
+
for (const event of events) {
|
|
324
|
+
const key = `${event.kind}:${event.path}`;
|
|
325
|
+
if (seen.has(key))
|
|
326
|
+
continue;
|
|
327
|
+
seen.add(key);
|
|
328
|
+
out.push(event);
|
|
329
|
+
}
|
|
330
|
+
return out;
|
|
331
|
+
}
|
|
332
|
+
/**
|
|
333
|
+
* Production chokidar factory. Returns the real `FSWatcher` cast to
|
|
334
|
+
* the minimal interface so the watcher class never directly imports
|
|
335
|
+
* chokidar types beyond `WatchOptions`. The cast is safe - chokidar's
|
|
336
|
+
* `FSWatcher` is a superset of `ChokidarLike`.
|
|
337
|
+
*/
|
|
338
|
+
function defaultWatchFactory(cwd, opts) {
|
|
339
|
+
const watcher = chokidar.watch(cwd, opts);
|
|
340
|
+
return watcher;
|
|
341
|
+
}
|
|
342
|
+
//# sourceMappingURL=watcher.js.map
|
|
@@ -0,0 +1,165 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Tier 1 working-set tracker - α6.5 Phase 1 (three-tier context).
|
|
3
|
+
*
|
|
4
|
+
* The model context window is finite. Tier 0 (repo skeleton, ~5KB) is
|
|
5
|
+
* always loaded; Tier 2 (RAG, Anvil-side, deferred to α6.5b) answers
|
|
6
|
+
* queries on demand. Tier 1 - the working set - lives between them: the
|
|
7
|
+
* subset of files the agent has touched THIS session, bounded so a
|
|
8
|
+
* long REPL session does not unbounded-grow the system prompt.
|
|
9
|
+
*
|
|
10
|
+
* Design choices:
|
|
11
|
+
*
|
|
12
|
+
* 1. **LRU semantics**: the most-recently-touched file stays in the
|
|
13
|
+
* window; the oldest is evicted when the cap is hit. We use a
|
|
14
|
+
* `Map` so insertion order doubles as recency order (delete +
|
|
15
|
+
* re-set bubbles the entry to the tail).
|
|
16
|
+
*
|
|
17
|
+
* 2. **Source tracking**: every `track()` call records the source -
|
|
18
|
+
* `read` (agent fetched the file), `edit` (agent modified it),
|
|
19
|
+
* `write` (agent created it). The source seeds future heuristics
|
|
20
|
+
* (e.g. always pin edited files near the cap, never evict files
|
|
21
|
+
* written this session) but Phase 1 keeps the policy simple:
|
|
22
|
+
* pure LRU.
|
|
23
|
+
*
|
|
24
|
+
* 3. **No persistence**: working set is in-memory only. A REPL
|
|
25
|
+
* restart rebuilds from scratch (and from `/resume` SessionStore
|
|
26
|
+
* replay if/when we wire that). Per α6.5 spec: "node:sqlite NOT
|
|
27
|
+
* needed for this sprint."
|
|
28
|
+
*
|
|
29
|
+
* 4. **Path normalisation**: we resolve every path to an absolute
|
|
30
|
+
* form on insert so two `track()` calls with `./foo.ts` and
|
|
31
|
+
* `/abs/cwd/foo.ts` collapse to one entry.
|
|
32
|
+
*
|
|
33
|
+
* 5. **Defensive cap**: `MAX_WORKING_SET = 50` per spec. We hold the
|
|
34
|
+
* cap as a configurable option so tests can dial it down to
|
|
35
|
+
* exercise eviction without 51 fixture files.
|
|
36
|
+
*/
|
|
37
|
+
import { resolve } from 'node:path';
|
|
38
|
+
/** Spec-mandated default cap. Mirrors the doc string in the α6.5 sprint plan. */
|
|
39
|
+
export const DEFAULT_WORKING_SET_CAPACITY = 50;
|
|
40
|
+
export class WorkingSet {
|
|
41
|
+
entries = new Map();
|
|
42
|
+
capacity;
|
|
43
|
+
now;
|
|
44
|
+
constructor(options = {}) {
|
|
45
|
+
this.capacity = options.capacity ?? DEFAULT_WORKING_SET_CAPACITY;
|
|
46
|
+
if (!Number.isInteger(this.capacity) || this.capacity <= 0) {
|
|
47
|
+
throw new Error(`WorkingSet capacity must be a positive integer, got ${options.capacity}`);
|
|
48
|
+
}
|
|
49
|
+
this.now = options.now ?? (() => Date.now());
|
|
50
|
+
}
|
|
51
|
+
/**
|
|
52
|
+
* Record that the agent touched `path`. The path is normalised to an
|
|
53
|
+
* absolute form so duplicate calls with different relative paths
|
|
54
|
+
* collapse. Returns the entry as stored.
|
|
55
|
+
*
|
|
56
|
+
* LRU bump: an existing entry is deleted and re-inserted so insertion
|
|
57
|
+
* order tracks recency. The oldest entry is evicted when the
|
|
58
|
+
* insertion would breach `capacity`.
|
|
59
|
+
*/
|
|
60
|
+
track(path, source, sizeBytes = 0) {
|
|
61
|
+
const absPath = resolve(path);
|
|
62
|
+
const previous = this.entries.get(absPath);
|
|
63
|
+
const touchCount = previous ? previous.touchCount + 1 : 1;
|
|
64
|
+
// Prefer the new size when the caller supplies one; otherwise keep
|
|
65
|
+
// the previous reading so we do not "forget" the file's footprint
|
|
66
|
+
// just because the second track() call did not stat the file.
|
|
67
|
+
const nextSize = sizeBytes > 0
|
|
68
|
+
? sizeBytes
|
|
69
|
+
: (previous?.sizeBytes ?? 0);
|
|
70
|
+
const entry = Object.freeze({
|
|
71
|
+
absPath,
|
|
72
|
+
source,
|
|
73
|
+
touchedAtEpochMs: this.now(),
|
|
74
|
+
touchCount,
|
|
75
|
+
sizeBytes: nextSize,
|
|
76
|
+
});
|
|
77
|
+
// Delete first so re-insert lands at the tail (most-recent).
|
|
78
|
+
if (previous)
|
|
79
|
+
this.entries.delete(absPath);
|
|
80
|
+
this.entries.set(absPath, entry);
|
|
81
|
+
this.evictExcess();
|
|
82
|
+
return entry;
|
|
83
|
+
}
|
|
84
|
+
/**
|
|
85
|
+
* Drop a single path from the set. Returns `true` when the entry
|
|
86
|
+
* existed. Used by the file watcher's `unlink` event - a removed
|
|
87
|
+
* file is no longer in the working set.
|
|
88
|
+
*/
|
|
89
|
+
forget(path) {
|
|
90
|
+
const absPath = resolve(path);
|
|
91
|
+
return this.entries.delete(absPath);
|
|
92
|
+
}
|
|
93
|
+
/** Clear the entire set - used by `/clear` slash + REPL teardown. */
|
|
94
|
+
clear() {
|
|
95
|
+
this.entries.clear();
|
|
96
|
+
}
|
|
97
|
+
/** True when the path is currently retained. */
|
|
98
|
+
has(path) {
|
|
99
|
+
return this.entries.has(resolve(path));
|
|
100
|
+
}
|
|
101
|
+
/** Live count - O(1). */
|
|
102
|
+
size() {
|
|
103
|
+
return this.entries.size;
|
|
104
|
+
}
|
|
105
|
+
/** Configured cap - surfaced so `/context` can render `12/50`. */
|
|
106
|
+
capacityLimit() {
|
|
107
|
+
return this.capacity;
|
|
108
|
+
}
|
|
109
|
+
/**
|
|
110
|
+
* Snapshot of the set ordered MOST RECENT first. Suitable for
|
|
111
|
+
* direct injection into the agent's system prompt; the oldest
|
|
112
|
+
* entries are the least relevant and trail the list so the model's
|
|
113
|
+
* recency bias works in our favour.
|
|
114
|
+
*/
|
|
115
|
+
forSerialization() {
|
|
116
|
+
// `Map` iterates in insertion order = LRU order with oldest first.
|
|
117
|
+
// We want recency-first so reverse.
|
|
118
|
+
const all = Array.from(this.entries.values());
|
|
119
|
+
all.reverse();
|
|
120
|
+
return all;
|
|
121
|
+
}
|
|
122
|
+
/** Aggregate summary used by `/context` slash + status bar. */
|
|
123
|
+
summary() {
|
|
124
|
+
let totalSize = 0;
|
|
125
|
+
let oldest = null;
|
|
126
|
+
for (const entry of this.entries.values()) {
|
|
127
|
+
totalSize += entry.sizeBytes;
|
|
128
|
+
if (oldest === null || entry.touchedAtEpochMs < oldest) {
|
|
129
|
+
oldest = entry.touchedAtEpochMs;
|
|
130
|
+
}
|
|
131
|
+
}
|
|
132
|
+
return {
|
|
133
|
+
count: this.entries.size,
|
|
134
|
+
capacity: this.capacity,
|
|
135
|
+
totalSizeBytes: totalSize,
|
|
136
|
+
oldestTouchedAtEpochMs: oldest,
|
|
137
|
+
};
|
|
138
|
+
}
|
|
139
|
+
/**
|
|
140
|
+
* Public LRU eviction trigger - called automatically on every
|
|
141
|
+
* `track()` but exposed so callers (tests, future memory-pressure
|
|
142
|
+
* hook) can force a sweep without inserting a sentinel entry.
|
|
143
|
+
* Returns the count of evicted entries.
|
|
144
|
+
*/
|
|
145
|
+
evict() {
|
|
146
|
+
return this.evictExcess();
|
|
147
|
+
}
|
|
148
|
+
/**
|
|
149
|
+
* Walk the head of the insertion-ordered map (oldest first) and
|
|
150
|
+
* delete entries until size <= capacity. Returns the number of
|
|
151
|
+
* entries dropped.
|
|
152
|
+
*/
|
|
153
|
+
evictExcess() {
|
|
154
|
+
let removed = 0;
|
|
155
|
+
while (this.entries.size > this.capacity) {
|
|
156
|
+
const oldestKey = this.entries.keys().next().value;
|
|
157
|
+
if (oldestKey === undefined)
|
|
158
|
+
break;
|
|
159
|
+
this.entries.delete(oldestKey);
|
|
160
|
+
removed += 1;
|
|
161
|
+
}
|
|
162
|
+
return removed;
|
|
163
|
+
}
|
|
164
|
+
}
|
|
165
|
+
//# sourceMappingURL=working-set.js.map
|
|
@@ -0,0 +1,185 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Diff dispatch — α6.6 escalation Phase 1.
|
|
3
|
+
*
|
|
4
|
+
* Reads a raw model response containing one or more SEARCH/REPLACE
|
|
5
|
+
* envelopes, normalises them through `marker-parser`, and routes each
|
|
6
|
+
* parsed edit to the correct applicator:
|
|
7
|
+
*
|
|
8
|
+
* - `layer-a` → applyLayerA
|
|
9
|
+
* - `layer-b` → applyLayerB
|
|
10
|
+
* - `layer-c` → applyLayerC
|
|
11
|
+
* - `layer-d` → throws LayerDDeferredError, surfaced as a clean
|
|
12
|
+
* dispatch failure (Layer D ships in α6.6b)
|
|
13
|
+
*
|
|
14
|
+
* Per-edit results are aggregated into `DispatchResult[]` so callers
|
|
15
|
+
* can render the full apply transcript even when some edits failed.
|
|
16
|
+
* Order is preserved across the response.
|
|
17
|
+
*
|
|
18
|
+
* Crash recovery hook: when a SessionStore-style appendEvent callback
|
|
19
|
+
* is supplied, the dispatcher records the INTENT (parsed edit) BEFORE
|
|
20
|
+
* calling the applicator. The matching `applied` event lands AFTER
|
|
21
|
+
* the writeFile. A crash between the two leaves a recoverable trail —
|
|
22
|
+
* the operator (or `pugi resume`) sees the intent and can re-attempt.
|
|
23
|
+
*
|
|
24
|
+
* The dispatcher is intentionally side-effect-light: no logging, no
|
|
25
|
+
* stdout writes, no exit-code mutation. The CLI integration layer in
|
|
26
|
+
* `cli.ts` owns operator-facing rendering; the dispatcher returns
|
|
27
|
+
* structured data and lets the caller decide UX.
|
|
28
|
+
*/
|
|
29
|
+
import { LayerDDeferredError, applyLayerD } from './layer-d-ast.js';
|
|
30
|
+
import { applyLayerA } from './layer-a-apply.js';
|
|
31
|
+
import { applyLayerB } from './layer-b-apply.js';
|
|
32
|
+
import { applyLayerC } from './layer-c-apply.js';
|
|
33
|
+
import { MarkerParseError, parseMarkers, } from './marker-parser.js';
|
|
34
|
+
/**
|
|
35
|
+
* Parse `raw` into edits and apply each in order. Aggregate results,
|
|
36
|
+
* preserving order. Never throws — parse failures surface as a single
|
|
37
|
+
* synthetic DispatchResult with `ok: false, layer: 'layer-a', reason:
|
|
38
|
+
* 'marker_parse_error'`. Applicator failures are recorded per-edit.
|
|
39
|
+
*/
|
|
40
|
+
export async function dispatchEdit(raw, opts) {
|
|
41
|
+
const family = resolveFamily(opts.modelTag);
|
|
42
|
+
let parsed;
|
|
43
|
+
try {
|
|
44
|
+
parsed = parseMarkers(raw, family);
|
|
45
|
+
}
|
|
46
|
+
catch (error) {
|
|
47
|
+
if (error instanceof MarkerParseError) {
|
|
48
|
+
const result = {
|
|
49
|
+
layer: 'layer-a',
|
|
50
|
+
file: '',
|
|
51
|
+
ok: false,
|
|
52
|
+
bytesWritten: 0,
|
|
53
|
+
reason: 'marker_parse_error',
|
|
54
|
+
detail: `${error.message}${error.atLine ? ` (line ${error.atLine})` : ''} — modelHint=${error.modelHint}`,
|
|
55
|
+
};
|
|
56
|
+
opts.onResult?.(result);
|
|
57
|
+
return [result];
|
|
58
|
+
}
|
|
59
|
+
throw error;
|
|
60
|
+
}
|
|
61
|
+
if (parsed.length === 0) {
|
|
62
|
+
// Empty parse but no error == no markers in the payload. This is
|
|
63
|
+
// not necessarily a failure (the model may have answered with
|
|
64
|
+
// prose only); surface a single neutral result so the caller can
|
|
65
|
+
// render "no edits proposed".
|
|
66
|
+
return [];
|
|
67
|
+
}
|
|
68
|
+
const out = [];
|
|
69
|
+
for (const edit of parsed) {
|
|
70
|
+
const intent = makeIntent(edit);
|
|
71
|
+
opts.onIntent?.(intent);
|
|
72
|
+
const result = await applyOne(edit, opts);
|
|
73
|
+
out.push(result);
|
|
74
|
+
opts.onResult?.(result);
|
|
75
|
+
}
|
|
76
|
+
return out;
|
|
77
|
+
}
|
|
78
|
+
/**
|
|
79
|
+
* Public helper exposed for the marker parser tests + CLI surface that
|
|
80
|
+
* may want to know the resolved family without re-running the auto
|
|
81
|
+
* detector.
|
|
82
|
+
*/
|
|
83
|
+
export function resolveFamily(modelTag) {
|
|
84
|
+
if (!modelTag)
|
|
85
|
+
return 'auto';
|
|
86
|
+
const tag = modelTag.toLowerCase();
|
|
87
|
+
if (tag.startsWith('claude') || tag.startsWith('anthropic/'))
|
|
88
|
+
return 'anthropic';
|
|
89
|
+
if (tag.startsWith('gemini') || tag.startsWith('xai/') || tag.startsWith('grok'))
|
|
90
|
+
return 'gemini';
|
|
91
|
+
if (tag.startsWith('gpt') || tag.startsWith('o1') || tag.startsWith('openai/'))
|
|
92
|
+
return 'openai';
|
|
93
|
+
return 'auto';
|
|
94
|
+
}
|
|
95
|
+
function makeIntent(edit) {
|
|
96
|
+
switch (edit.kind) {
|
|
97
|
+
case 'layer-a':
|
|
98
|
+
return {
|
|
99
|
+
layer: 'layer-a',
|
|
100
|
+
file: edit.edit.file,
|
|
101
|
+
intentSummary: `Layer A: ${edit.edit.file} (oldString ${edit.edit.oldString.length} bytes)`,
|
|
102
|
+
};
|
|
103
|
+
case 'layer-b':
|
|
104
|
+
return {
|
|
105
|
+
layer: 'layer-b',
|
|
106
|
+
file: edit.edit.file,
|
|
107
|
+
intentSummary: `Layer B: ${edit.edit.file} (${edit.edit.edits.length} sub-edits)`,
|
|
108
|
+
};
|
|
109
|
+
case 'layer-c':
|
|
110
|
+
return {
|
|
111
|
+
layer: 'layer-c',
|
|
112
|
+
file: edit.edit.file,
|
|
113
|
+
intentSummary: `Layer C: ${edit.edit.file} (rewrite, ${edit.edit.newContents.length} bytes, baseSha ${edit.edit.baseSha256.slice(0, 12)})`,
|
|
114
|
+
};
|
|
115
|
+
case 'layer-d':
|
|
116
|
+
return {
|
|
117
|
+
layer: 'layer-d',
|
|
118
|
+
file: edit.edit.file,
|
|
119
|
+
intentSummary: `Layer D: ${edit.edit.file} op=${edit.edit.operation}`,
|
|
120
|
+
};
|
|
121
|
+
}
|
|
122
|
+
}
|
|
123
|
+
async function applyOne(edit, opts) {
|
|
124
|
+
const applyOpts = { cwd: opts.cwd, dryRun: opts.dryRun };
|
|
125
|
+
switch (edit.kind) {
|
|
126
|
+
case 'layer-a': {
|
|
127
|
+
const r = await applyLayerA(edit.edit, applyOpts);
|
|
128
|
+
return toResult('layer-a', edit.edit.file, r);
|
|
129
|
+
}
|
|
130
|
+
case 'layer-b': {
|
|
131
|
+
const r = await applyLayerB(edit.edit, applyOpts);
|
|
132
|
+
return {
|
|
133
|
+
...toResult('layer-b', edit.edit.file, r),
|
|
134
|
+
appliedCount: r.appliedCount,
|
|
135
|
+
};
|
|
136
|
+
}
|
|
137
|
+
case 'layer-c': {
|
|
138
|
+
const r = await applyLayerC(edit.edit, applyOpts);
|
|
139
|
+
return {
|
|
140
|
+
...toResult('layer-c', edit.edit.file, r),
|
|
141
|
+
expectedSha256: r.expectedSha256,
|
|
142
|
+
actualSha256: r.actualSha256,
|
|
143
|
+
};
|
|
144
|
+
}
|
|
145
|
+
case 'layer-d': {
|
|
146
|
+
try {
|
|
147
|
+
const r = await applyLayerD(edit.edit, applyOpts);
|
|
148
|
+
return toResult('layer-d', edit.edit.file, r);
|
|
149
|
+
}
|
|
150
|
+
catch (error) {
|
|
151
|
+
if (error instanceof LayerDDeferredError) {
|
|
152
|
+
return {
|
|
153
|
+
layer: 'layer-d',
|
|
154
|
+
file: edit.edit.file,
|
|
155
|
+
ok: false,
|
|
156
|
+
bytesWritten: 0,
|
|
157
|
+
reason: 'layer_d_deferred',
|
|
158
|
+
detail: error.message,
|
|
159
|
+
};
|
|
160
|
+
}
|
|
161
|
+
return {
|
|
162
|
+
layer: 'layer-d',
|
|
163
|
+
file: edit.edit.file,
|
|
164
|
+
ok: false,
|
|
165
|
+
bytesWritten: 0,
|
|
166
|
+
reason: 'apply_error',
|
|
167
|
+
detail: error instanceof Error ? error.message : String(error),
|
|
168
|
+
};
|
|
169
|
+
}
|
|
170
|
+
}
|
|
171
|
+
}
|
|
172
|
+
}
|
|
173
|
+
function toResult(layer, file, r) {
|
|
174
|
+
return {
|
|
175
|
+
layer,
|
|
176
|
+
file,
|
|
177
|
+
ok: r.ok,
|
|
178
|
+
bytesWritten: r.bytesWritten,
|
|
179
|
+
reason: r.reason,
|
|
180
|
+
detail: r.detail,
|
|
181
|
+
matchCount: r.matchCount,
|
|
182
|
+
absPath: r.absPath,
|
|
183
|
+
};
|
|
184
|
+
}
|
|
185
|
+
//# sourceMappingURL=dispatch.js.map
|