@ngockhoale/ukit 2.4.2 → 2.4.3
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/manifests/platform.full.yaml +19 -111
- package/package.json +2 -1
- package/scripts/index/refresh-index.mjs +47 -22
- package/src/core/compact/threshold.js +36 -6
- package/src/diagnostics/classifyHang.js +246 -0
- package/src/index/buildIndex.js +1033 -62
- package/templates/.claude/hooks/auto-allow-bash.sh +82 -93
- package/templates/.claude/hooks/block-dangerous.sh +31 -5
- package/templates/.claude/hooks/completion-gate.sh +51 -10
- package/templates/.claude/hooks/compress-output.sh +38 -6
- package/templates/.claude/hooks/context-hardcap-gate.sh +35 -6
- package/templates/.claude/hooks/context-window-guard.sh +128 -18
- package/templates/.claude/hooks/handoff-model-guard.sh +31 -5
- package/templates/.claude/hooks/handoff-resume.sh +31 -5
- package/templates/.claude/hooks/post-edit-verify.sh +31 -5
- package/templates/.claude/hooks/pre-edit-backup.sh +31 -5
- package/templates/.claude/hooks/protect-files.sh +31 -5
- package/templates/.claude/hooks/record-execution.sh +31 -5
- package/templates/.claude/hooks/sensitive-data-guard.sh +44 -8
- package/templates/.claude/hooks/skill-router.sh +31 -5
- package/templates/.claude/hooks/stale-spec-guard.sh +31 -5
- package/templates/.claude/hooks/task-watchdog.sh +108 -123
- package/templates/.claude/hooks/verification-guard.sh +107 -112
- package/templates/.claude/hooks/vision-router.sh +49 -13
- package/templates/.claude/settings.json +0 -5
- package/templates/.claude/ukit/index/lib/index-core.mjs +960 -63
- package/templates/.claude/ukit/index/refresh-index.mjs +47 -22
- package/templates/.claude/ukit/index/route-task.mjs +610 -4
- package/templates/.claude/ukit/runtime/async-lock.mjs +340 -0
- package/templates/.claude/ukit/runtime/compact-threshold.mjs +73 -24
- package/templates/.claude/ukit/runtime/context-capacity.mjs +144 -0
- package/templates/.claude/ukit/runtime/execution-ledger.mjs +664 -170
- package/templates/.claude/ukit/runtime/hook-chain-budget.mjs +92 -0
- package/templates/.claude/ukit/runtime/hook-chain-runner.mjs +84 -29
- package/templates/.claude/ukit/runtime/hook-input.sh +85 -5
- package/templates/.claude/ukit/runtime/hook-payload-store.mjs +160 -0
- package/templates/.claude/ukit/runtime/hook-process.mjs +250 -0
- package/templates/.claude/ukit/runtime/hook-telemetry.mjs +255 -0
- package/templates/.claude/ukit/runtime/hook-telemetry.sh +60 -0
- package/templates/.claude/ukit/runtime/stop-coordinator.mjs +509 -0
- package/templates/.claude/ukit/runtime/task-watchdog.mjs +180 -6
- package/templates/.claude/ukit/runtime/transcript-tail.mjs +1 -1
- package/templates/.omp/hooks/pre/ukit-bridge.js +171 -57
|
@@ -3,7 +3,93 @@ import path from 'node:path';
|
|
|
3
3
|
import crypto from 'node:crypto';
|
|
4
4
|
import { spawnSync } from 'node:child_process';
|
|
5
5
|
|
|
6
|
-
|
|
6
|
+
// One wall-clock ceiling for the WHOLE discovery phase: both `git ls-files`
|
|
7
|
+
// enumerations AND the non-git fallback walk share this single budget. Every
|
|
8
|
+
// stage receives only the remaining budget and expiry throws a typed
|
|
9
|
+
// IndexDiscoveryTimeoutError instead of degrading silently. Callers may tighten
|
|
10
|
+
// it with `deadlineMs` or cancel early with `signal` on
|
|
11
|
+
// `discoverProjectFiles(root, { signal, deadlineMs })`.
|
|
12
|
+
export const DEFAULT_DISCOVERY_DEADLINE_MS = 30_000;
|
|
13
|
+
|
|
14
|
+
// Typed liveness failure for project discovery: deadline expired or the
|
|
15
|
+
// caller's AbortSignal fired — never a silent partial result.
|
|
16
|
+
export class IndexDiscoveryTimeoutError extends Error {
|
|
17
|
+
constructor({ deadlineMs, phase, elapsedMs, aborted = false }) {
|
|
18
|
+
super(aborted
|
|
19
|
+
? `Index discovery was aborted ${elapsedMs}ms into its ${deadlineMs}ms deadline (during ${phase}).`
|
|
20
|
+
: `Index discovery exceeded its ${deadlineMs}ms wall-clock deadline (elapsed ${elapsedMs}ms, during ${phase}).`);
|
|
21
|
+
this.name = 'IndexDiscoveryTimeoutError';
|
|
22
|
+
this.code = 'INDEX_DISCOVERY_TIMEOUT';
|
|
23
|
+
this.deadlineMs = deadlineMs;
|
|
24
|
+
this.phase = phase;
|
|
25
|
+
this.elapsedMs = elapsedMs;
|
|
26
|
+
this.aborted = aborted;
|
|
27
|
+
}
|
|
28
|
+
}
|
|
29
|
+
|
|
30
|
+
// Upper bound on in-flight filesystem stat calls during discovery: git
|
|
31
|
+
// candidate stats and the non-git fallback walk share this fixed pool.
|
|
32
|
+
// The previous `Promise.all(candidates.map(stat))` produced one unbounded
|
|
33
|
+
// promise per candidate, so a 10k-file listing held 10k concurrent stats
|
|
34
|
+
// (descriptors + scheduler pressure) at once. With the pool, peak in-flight
|
|
35
|
+
// I/O stays flat no matter the repository size.
|
|
36
|
+
export const INDEX_IO_CONCURRENCY = 64;
|
|
37
|
+
|
|
38
|
+
// Typed cancellation failure for build phases beyond discovery (stat pools,
|
|
39
|
+
// file reads/parse, artifact publish) when the caller's AbortSignal fires.
|
|
40
|
+
// Discovery keeps its own deadline/abort classification in
|
|
41
|
+
// IndexDiscoveryTimeoutError; this error reports explicit caller cancellation
|
|
42
|
+
// of the rest of the pipeline.
|
|
43
|
+
export class IndexCancelledError extends Error {
|
|
44
|
+
constructor({ phase } = {}) {
|
|
45
|
+
super(`Index work was cancelled by its AbortSignal (during ${phase}).`);
|
|
46
|
+
this.name = 'IndexCancelledError';
|
|
47
|
+
this.code = 'INDEX_CANCELLED';
|
|
48
|
+
this.phase = phase;
|
|
49
|
+
}
|
|
50
|
+
}
|
|
51
|
+
|
|
52
|
+
// Fixed-concurrency worker pool over `items`. At most `limit` workers run at
|
|
53
|
+
// once, so in-flight I/O never scales with the item count. Each worker
|
|
54
|
+
// re-checks `signal` BEFORE claiming its next item, so an abort leaves queued
|
|
55
|
+
// work unstarted and the promise rejects with IndexCancelledError. A worker
|
|
56
|
+
// rejection maps that one item to `null` while the remaining items still
|
|
57
|
+
// process — a single failing stat/read can never hang or kill the pool.
|
|
58
|
+
// Returns results in `items` order (null for failed items).
|
|
59
|
+
export async function mapLimit(items, limit, worker, signal = null, phase = 'map') {
|
|
60
|
+
const concurrency = Math.max(1, Math.floor(Number(limit)) || 1);
|
|
61
|
+
const results = new Array(items.length);
|
|
62
|
+
let nextIndex = 0;
|
|
63
|
+
|
|
64
|
+
const runWorker = async () => {
|
|
65
|
+
while (nextIndex < items.length) {
|
|
66
|
+
if (signal?.aborted) {
|
|
67
|
+
throw new IndexCancelledError({ phase });
|
|
68
|
+
}
|
|
69
|
+
const index = nextIndex;
|
|
70
|
+
nextIndex += 1;
|
|
71
|
+
try {
|
|
72
|
+
results[index] = await worker(items[index], index);
|
|
73
|
+
} catch (error) {
|
|
74
|
+
if (error instanceof IndexDiscoveryTimeoutError || error instanceof IndexCancelledError) {
|
|
75
|
+
throw error;
|
|
76
|
+
}
|
|
77
|
+
results[index] = null;
|
|
78
|
+
}
|
|
79
|
+
}
|
|
80
|
+
};
|
|
81
|
+
|
|
82
|
+
await Promise.all(Array.from({ length: Math.min(concurrency, items.length) }, () => runWorker()));
|
|
83
|
+
return results;
|
|
84
|
+
}
|
|
85
|
+
|
|
86
|
+
// Gate for phases between discovery and publish: abort means nothing is written.
|
|
87
|
+
function assertNotAborted(signal, phase) {
|
|
88
|
+
if (signal?.aborted) {
|
|
89
|
+
throw new IndexCancelledError({ phase });
|
|
90
|
+
}
|
|
91
|
+
}
|
|
92
|
+
|
|
7
93
|
const EXCLUDED_DIR_NAMES = new Set([
|
|
8
94
|
'node_modules',
|
|
9
95
|
'.git',
|
|
@@ -30,7 +116,9 @@ const TRACKED_EXTENSIONS = new Set(['.js', '.mjs', '.cjs', '.ts', '.tsx', '.jsx'
|
|
|
30
116
|
const DISCOVERED_EXTENSIONS = new Set([...TRACKED_EXTENSIONS, ...STYLE_EXTENSIONS]);
|
|
31
117
|
export const INDEX_SCHEMA_VERSION = 8;
|
|
32
118
|
export const DEFAULT_INDEX_CACHE_MAX_AGE_MS = 24 * 60 * 60 * 1000;
|
|
33
|
-
|
|
119
|
+
// Parse-phase read/extract concurrency — deliberately tighter than the
|
|
120
|
+
// discovery stat pool because parsing holds file content in memory.
|
|
121
|
+
export const INDEX_PARSE_BATCH_SIZE = 8;
|
|
34
122
|
const MAX_IMPORTER_HOPS = 2;
|
|
35
123
|
const CALL_IGNORE_WORDS = new Set([
|
|
36
124
|
'if', 'for', 'while', 'switch', 'catch', 'function', 'return', 'typeof',
|
|
@@ -122,10 +210,26 @@ const VIETNAMESE_TOKEN_ALIASES = new Map([
|
|
|
122
210
|
// snapshots from a different version instead of trusting unknown fields.
|
|
123
211
|
export const DISCOVERY_SNAPSHOT_VERSION = 1;
|
|
124
212
|
|
|
125
|
-
export async function buildCodeIndex({ rootDir = process.cwd(), discoverySnapshot = null } = {}) {
|
|
213
|
+
export async function buildCodeIndex({ rootDir = process.cwd(), changedFiles = null, discoverySnapshot = null, signal = null, deadlineMs = DEFAULT_DISCOVERY_DEADLINE_MS } = {}) {
|
|
126
214
|
const absoluteRoot = path.resolve(rootDir);
|
|
127
215
|
const indexDir = getIndexDir(absoluteRoot);
|
|
128
216
|
|
|
217
|
+
// Incremental fast path (TASK-023): validated changed-file hints let an
|
|
218
|
+
// existing, schema-current index absorb the change set without a full
|
|
219
|
+
// repository discovery. Every unsafe hint, and every previous artifact that
|
|
220
|
+
// cannot be merged, falls back to the bounded full path below — correctness
|
|
221
|
+
// wins over the optimization, and incremental output must stay equivalent to
|
|
222
|
+
// a clean full rebuild.
|
|
223
|
+
const changedHintPlan = normalizeChangedFileHints(changedFiles);
|
|
224
|
+
if (changedHintPlan) {
|
|
225
|
+
const incremental = changedHintPlan.ok
|
|
226
|
+
? await tryIncrementalIndexUpdate({ absoluteRoot, indexDir, changedPaths: changedHintPlan.paths, signal })
|
|
227
|
+
: null;
|
|
228
|
+
if (incremental) {
|
|
229
|
+
return incremental;
|
|
230
|
+
}
|
|
231
|
+
}
|
|
232
|
+
|
|
129
233
|
const previousFilesArtifact = await readArtifactIfExists(absoluteRoot, INDEX_ARTIFACTS.files);
|
|
130
234
|
const canReusePrevious = previousFilesArtifact?.schemaVersion === INDEX_SCHEMA_VERSION;
|
|
131
235
|
const previousCodeFileRecords = canReusePrevious
|
|
@@ -142,7 +246,9 @@ export async function buildCodeIndex({ rootDir = process.cwd(), discoverySnapsho
|
|
|
142
246
|
// once here. Keeps a stale-check + rebuild refresh at a single enumeration.
|
|
143
247
|
const discoveredFiles = isDiscoverySnapshotUsable(discoverySnapshot, absoluteRoot)
|
|
144
248
|
? discoverySnapshot.files
|
|
145
|
-
|
|
249
|
+
// Discovery precedes every artifact write, so a deadline expiry here aborts
|
|
250
|
+
// the build before any prior index artifact can be replaced.
|
|
251
|
+
: await discoverProjectFiles(absoluteRoot, { signal, deadlineMs });
|
|
146
252
|
const sourceFingerprint = createSourceFingerprint(absoluteRoot, discoveredFiles);
|
|
147
253
|
const styleFilePaths = discoveredFiles
|
|
148
254
|
.map((entry) => {
|
|
@@ -221,32 +327,34 @@ export async function buildCodeIndex({ rootDir = process.cwd(), discoverySnapsho
|
|
|
221
327
|
}
|
|
222
328
|
const canReuseAllParsedArtifacts = canReuseParsedArtifacts && filesToParse.length === 0 && reusableCodeFiles.length === codeFiles.length;
|
|
223
329
|
|
|
224
|
-
|
|
225
|
-
|
|
226
|
-
|
|
227
|
-
|
|
228
|
-
|
|
229
|
-
|
|
230
|
-
|
|
231
|
-
|
|
232
|
-
|
|
233
|
-
|
|
234
|
-
|
|
235
|
-
|
|
236
|
-
|
|
237
|
-
|
|
238
|
-
|
|
239
|
-
|
|
240
|
-
|
|
241
|
-
|
|
242
|
-
|
|
243
|
-
|
|
244
|
-
|
|
245
|
-
for (const parsedFile of parsedBatch) {
|
|
246
|
-
symbols.push(...parsedFile.symbols);
|
|
247
|
-
imports.push(...parsedFile.imports);
|
|
248
|
-
calls.push(...parsedFile.calls);
|
|
330
|
+
// Parse through the same fixed pool as discovery stats: reads stay bounded
|
|
331
|
+
// by INDEX_PARSE_BATCH_SIZE and the caller's signal is re-checked before
|
|
332
|
+
// every new file, so an abort stops the phase with queued files unread.
|
|
333
|
+
// Results come back in filesToParse order, matching the previous batch loop.
|
|
334
|
+
const parsedFiles = await mapLimit(filesToParse, INDEX_PARSE_BATCH_SIZE, async (filePath) => {
|
|
335
|
+
try {
|
|
336
|
+
const absolutePath = path.join(absoluteRoot, filePath);
|
|
337
|
+
const content = await fs.readFile(absolutePath, 'utf8');
|
|
338
|
+
const scriptContent = extractScriptContent(filePath, content);
|
|
339
|
+
return {
|
|
340
|
+
filePath,
|
|
341
|
+
symbols: extractSymbols(filePath, scriptContent),
|
|
342
|
+
imports: [
|
|
343
|
+
...extractImports(filePath, scriptContent),
|
|
344
|
+
...extractSupplementalImports(filePath, content),
|
|
345
|
+
],
|
|
346
|
+
calls: extractFunctionCalls(filePath, scriptContent),
|
|
347
|
+
};
|
|
348
|
+
} catch {
|
|
349
|
+
return null;
|
|
249
350
|
}
|
|
351
|
+
}, signal, 'parse');
|
|
352
|
+
|
|
353
|
+
for (const parsedFile of parsedFiles) {
|
|
354
|
+
if (!parsedFile) continue;
|
|
355
|
+
symbols.push(...parsedFile.symbols);
|
|
356
|
+
imports.push(...parsedFile.imports);
|
|
357
|
+
calls.push(...parsedFile.calls);
|
|
250
358
|
}
|
|
251
359
|
|
|
252
360
|
const canReuseTestsMapCandidate = canReusePrevious
|
|
@@ -344,6 +452,10 @@ export async function buildCodeIndex({ rootDir = process.cwd(), discoverySnapsho
|
|
|
344
452
|
|
|
345
453
|
const generatedAt = new Date().toISOString();
|
|
346
454
|
|
|
455
|
+
// No partial artifact publish: an abort landing after parsing (during the
|
|
456
|
+
// artifact-graph phases above) must not replace any artifact either, so the
|
|
457
|
+
// whole write block sits behind this gate.
|
|
458
|
+
assertNotAborted(signal, 'publish');
|
|
347
459
|
await fs.mkdir(indexDir, { recursive: true });
|
|
348
460
|
if (!canReuseFilesArtifact) {
|
|
349
461
|
await writeArtifact(absoluteRoot, INDEX_ARTIFACTS.files, { schemaVersion: INDEX_SCHEMA_VERSION, generatedAt, items: fileRecords });
|
|
@@ -368,7 +480,20 @@ export async function buildCodeIndex({ rootDir = process.cwd(), discoverySnapsho
|
|
|
368
480
|
if (!canReuseAnalogs) {
|
|
369
481
|
await writeArtifact(absoluteRoot, INDEX_ARTIFACTS.analogs, { schemaVersion: INDEX_SCHEMA_VERSION, generatedAt, sourceSnapshot: styleSnapshot, items: analogs });
|
|
370
482
|
}
|
|
371
|
-
await writeArtifact(absoluteRoot, INDEX_ARTIFACTS.meta, {
|
|
483
|
+
await writeArtifact(absoluteRoot, INDEX_ARTIFACTS.meta, {
|
|
484
|
+
schemaVersion: INDEX_SCHEMA_VERSION,
|
|
485
|
+
generatedAt,
|
|
486
|
+
sourceFingerprint,
|
|
487
|
+
// Item counts let a later incremental update prove the artifacts it reuses
|
|
488
|
+
// were not truncated between builds (TASK-023 review, important #2).
|
|
489
|
+
artifactItemCounts: {
|
|
490
|
+
[INDEX_ARTIFACTS.files]: fileRecords.length,
|
|
491
|
+
[INDEX_ARTIFACTS.symbols]: symbols.length,
|
|
492
|
+
[INDEX_ARTIFACTS.imports]: imports.length,
|
|
493
|
+
[INDEX_ARTIFACTS.calls]: calls.length,
|
|
494
|
+
[INDEX_ARTIFACTS.relations]: relations.length,
|
|
495
|
+
},
|
|
496
|
+
});
|
|
372
497
|
const preservedRuntimeCaches = canReuseFilesArtifact
|
|
373
498
|
&& canReuseAllParsedArtifacts
|
|
374
499
|
&& canReuseTestsMap
|
|
@@ -400,6 +525,582 @@ export async function buildCodeIndex({ rootDir = process.cwd(), discoverySnapsho
|
|
|
400
525
|
reusedRelations: canReuseRelations,
|
|
401
526
|
reusedAnalogs: canReuseAnalogs,
|
|
402
527
|
preservedRuntimeCaches,
|
|
528
|
+
mode: 'full',
|
|
529
|
+
parsedFiles: filesToParse.slice(),
|
|
530
|
+
removedFiles: [],
|
|
531
|
+
};
|
|
532
|
+
}
|
|
533
|
+
|
|
534
|
+
// ── Incremental refresh (TASK-023) ──
|
|
535
|
+
//
|
|
536
|
+
// `changedFiles` hints let refresh skip repository discovery entirely: the
|
|
537
|
+
// caller (a watcher or git-status driven hook) asserts which paths changed and
|
|
538
|
+
// the update merges exactly those paths into the existing artifacts. The path
|
|
539
|
+
// is taken only when EVERY gate holds:
|
|
540
|
+
// - every hint is a safe project-relative path (no absolute form, no `..`
|
|
541
|
+
// segment, no NUL byte), deduplicated and normalized;
|
|
542
|
+
// - the previous meta/files/symbols/imports/calls/relations artifacts all
|
|
543
|
+
// carry the current schema version and a source fingerprint to refresh;
|
|
544
|
+
// - every stat and read resolves unambiguously (ENOENT means deletion,
|
|
545
|
+
// anything else means "unknown" and falls back to full discovery);
|
|
546
|
+
// - every existing hint still resolves (through `realpath`) to a target
|
|
547
|
+
// inside the resolved root, so a symlink cannot pull external content into
|
|
548
|
+
// the index that a clean discovery would have excluded;
|
|
549
|
+
// - every reused artifact is a well-formed item array whose entries stay
|
|
550
|
+
// contained in the files artifact, so a truncated artifact cannot make the
|
|
551
|
+
// merge publish a partial artifact.
|
|
552
|
+
// Style files need one extra gate: discovery lists them but files.json never
|
|
553
|
+
// tracks them, so the previous style list only survives inside the relations
|
|
554
|
+
// artifact's source snapshot — without it the refreshed fingerprint could not
|
|
555
|
+
// match a clean rebuild.
|
|
556
|
+
|
|
557
|
+
function normalizeChangedFileHints(changedFiles) {
|
|
558
|
+
if (!Array.isArray(changedFiles) || changedFiles.length === 0) {
|
|
559
|
+
return null;
|
|
560
|
+
}
|
|
561
|
+
|
|
562
|
+
const paths = [];
|
|
563
|
+
const seen = new Set();
|
|
564
|
+
for (const raw of changedFiles) {
|
|
565
|
+
if (typeof raw !== 'string' || raw.includes('\0')) {
|
|
566
|
+
return { ok: false, paths: [] };
|
|
567
|
+
}
|
|
568
|
+
const trimmed = raw.trim();
|
|
569
|
+
if (!trimmed) {
|
|
570
|
+
continue;
|
|
571
|
+
}
|
|
572
|
+
// An absolute hint (POSIX or Windows drive form) can never describe a
|
|
573
|
+
// project-relative path, and a `..` segment cannot be trusted to stay
|
|
574
|
+
// contained — either rejects the whole hint set so the safe full
|
|
575
|
+
// discovery path runs instead.
|
|
576
|
+
if (path.isAbsolute(trimmed) || /^[A-Za-z]:[\\/]/.test(trimmed)) {
|
|
577
|
+
return { ok: false, paths: [] };
|
|
578
|
+
}
|
|
579
|
+
const segments = trimmed.split(/[\\/]+/);
|
|
580
|
+
if (segments.some((segment) => segment === '..')) {
|
|
581
|
+
return { ok: false, paths: [] };
|
|
582
|
+
}
|
|
583
|
+
const normalized = segments.filter((segment) => segment && segment !== '.').join('/');
|
|
584
|
+
if (!normalized || seen.has(normalized)) {
|
|
585
|
+
continue;
|
|
586
|
+
}
|
|
587
|
+
seen.add(normalized);
|
|
588
|
+
paths.push(normalized);
|
|
589
|
+
}
|
|
590
|
+
|
|
591
|
+
if (paths.length === 0) {
|
|
592
|
+
// Hints were requested but none survived validation: treat the request as
|
|
593
|
+
// ambiguous instead of silently doing nothing.
|
|
594
|
+
return { ok: false, paths: [] };
|
|
595
|
+
}
|
|
596
|
+
return { ok: true, paths };
|
|
597
|
+
}
|
|
598
|
+
|
|
599
|
+
function isSafeContainedRelative(rootDir, relativePath) {
|
|
600
|
+
if (!relativePath || path.isAbsolute(relativePath)) {
|
|
601
|
+
return false;
|
|
602
|
+
}
|
|
603
|
+
// Round-trip containment check: joining and re-relativizing must return the
|
|
604
|
+
// exact hint, which rules out separator or prefix surprises.
|
|
605
|
+
return path.relative(rootDir, path.join(rootDir, relativePath)) === relativePath;
|
|
606
|
+
}
|
|
607
|
+
|
|
608
|
+
function isSafeArtifactItem(item) {
|
|
609
|
+
return Boolean(item)
|
|
610
|
+
&& typeof item === 'object'
|
|
611
|
+
&& !Array.isArray(item)
|
|
612
|
+
&& isSafeArtifactItemPath(item.filePath);
|
|
613
|
+
}
|
|
614
|
+
|
|
615
|
+
function isSafeArtifactItemPath(relativePath) {
|
|
616
|
+
return typeof relativePath === 'string'
|
|
617
|
+
&& relativePath !== ''
|
|
618
|
+
&& !path.isAbsolute(relativePath)
|
|
619
|
+
&& !relativePath.split(/[\\/]+/).some((segment) => segment === '..');
|
|
620
|
+
}
|
|
621
|
+
|
|
622
|
+
function isValidFileRecord(item) {
|
|
623
|
+
return isSafeArtifactItem(item)
|
|
624
|
+
&& typeof item.domain === 'string'
|
|
625
|
+
&& typeof item.ext === 'string'
|
|
626
|
+
&& Number.isFinite(Number(item.mtimeMs))
|
|
627
|
+
&& Number.isFinite(Number(item.size));
|
|
628
|
+
}
|
|
629
|
+
|
|
630
|
+
// The merge reuses every item of these artifacts verbatim and groups them by
|
|
631
|
+
// the covered file path. They are only safe to reuse when the array is present
|
|
632
|
+
// and every entry is a well-formed item whose covered path the files artifact
|
|
633
|
+
// still tracks — a malformed or foreign entry falls back to the bounded full
|
|
634
|
+
// path instead of publishing a partial artifact.
|
|
635
|
+
function isValidCoveredItems(items, fileRecords, keyFn) {
|
|
636
|
+
if (!Array.isArray(items)) {
|
|
637
|
+
return false;
|
|
638
|
+
}
|
|
639
|
+
const trackedPaths = new Set(fileRecords.map((record) => record.filePath));
|
|
640
|
+
return items.every((item) => {
|
|
641
|
+
if (!item || typeof item !== 'object' || Array.isArray(item)) {
|
|
642
|
+
return false;
|
|
643
|
+
}
|
|
644
|
+
const key = keyFn(item);
|
|
645
|
+
return isSafeArtifactItemPath(key) && trackedPaths.has(key);
|
|
646
|
+
});
|
|
647
|
+
}
|
|
648
|
+
|
|
649
|
+
// Item shapes alone cannot detect a schema-current artifact whose `items` array
|
|
650
|
+
// lost entries: every surviving entry still looks valid, yet the merge would
|
|
651
|
+
// reuse it verbatim and publish an artifact missing the unchanged files. Each
|
|
652
|
+
// required artifact therefore records its item count in meta, and the count must
|
|
653
|
+
// match the array actually loaded — a mismatch (or a pre-count meta) falls back
|
|
654
|
+
// to full discovery.
|
|
655
|
+
function artifactItemCountsMatch(metaArtifact, counts) {
|
|
656
|
+
const recorded = metaArtifact?.artifactItemCounts;
|
|
657
|
+
if (!recorded || typeof recorded !== 'object' || Array.isArray(recorded)) {
|
|
658
|
+
return false;
|
|
659
|
+
}
|
|
660
|
+
return Object.entries(counts).every(([name, count]) => Number(recorded[name]) === count);
|
|
661
|
+
}
|
|
662
|
+
|
|
663
|
+
// Stat every hinted path and classify it as an upsert, a removal, or — for
|
|
664
|
+
// extensions discovery never indexes — a no-op. Returns null when a hint
|
|
665
|
+
// cannot be resolved unambiguously so the caller falls back to full discovery.
|
|
666
|
+
//
|
|
667
|
+
// Containment is re-checked against the RESOLVED root for every hint that still
|
|
668
|
+
// exists: `fs.stat` and `fs.readFile` both follow symlinks, so the lexical check
|
|
669
|
+
// alone would let `src/escape.js -> /outside/secret.js` pull external content
|
|
670
|
+
// into an incremental update that a clean discovery excludes.
|
|
671
|
+
async function resolveChangedHintUpdates(rootDir, changedPaths, previousRecordsByPath) {
|
|
672
|
+
const upserts = new Map();
|
|
673
|
+
const removals = new Set();
|
|
674
|
+
const styleUpserts = new Set();
|
|
675
|
+
const styleRemovals = new Set();
|
|
676
|
+
|
|
677
|
+
let rootRealPath = null;
|
|
678
|
+
try {
|
|
679
|
+
rootRealPath = await fs.realpath(rootDir);
|
|
680
|
+
} catch {
|
|
681
|
+
// The root itself cannot be resolved: no hint is trustworthy.
|
|
682
|
+
return null;
|
|
683
|
+
}
|
|
684
|
+
|
|
685
|
+
for (const relativePath of changedPaths) {
|
|
686
|
+
if (!isSafeContainedRelative(rootDir, relativePath)) {
|
|
687
|
+
return null;
|
|
688
|
+
}
|
|
689
|
+
|
|
690
|
+
const absolutePath = path.join(rootDir, relativePath);
|
|
691
|
+
let stat = null;
|
|
692
|
+
try {
|
|
693
|
+
stat = await fs.stat(absolutePath);
|
|
694
|
+
} catch (error) {
|
|
695
|
+
if (error?.code !== 'ENOENT') {
|
|
696
|
+
return null;
|
|
697
|
+
}
|
|
698
|
+
}
|
|
699
|
+
|
|
700
|
+
if (stat && !stat.isFile()) {
|
|
701
|
+
return null;
|
|
702
|
+
}
|
|
703
|
+
|
|
704
|
+
if (stat) {
|
|
705
|
+
let realPath = null;
|
|
706
|
+
try {
|
|
707
|
+
realPath = await fs.realpath(absolutePath);
|
|
708
|
+
} catch {
|
|
709
|
+
// Vanished or unresolvable between stat and realpath: unknown state.
|
|
710
|
+
return null;
|
|
711
|
+
}
|
|
712
|
+
if (!isDiscoveryPathContained(rootRealPath, realPath)) {
|
|
713
|
+
return null;
|
|
714
|
+
}
|
|
715
|
+
}
|
|
716
|
+
|
|
717
|
+
const ext = path.extname(relativePath).toLowerCase();
|
|
718
|
+
if (!stat) {
|
|
719
|
+
// Missing file: a removal when the index knows it, a no-op otherwise.
|
|
720
|
+
if (previousRecordsByPath.has(relativePath)) {
|
|
721
|
+
removals.add(relativePath);
|
|
722
|
+
} else if (STYLE_EXTENSIONS.has(ext)) {
|
|
723
|
+
styleRemovals.add(relativePath);
|
|
724
|
+
}
|
|
725
|
+
continue;
|
|
726
|
+
}
|
|
727
|
+
|
|
728
|
+
if (TRACKED_EXTENSIONS.has(ext)) {
|
|
729
|
+
upserts.set(relativePath, {
|
|
730
|
+
filePath: relativePath,
|
|
731
|
+
domain: relativePath.split('/')[0] ?? 'other',
|
|
732
|
+
ext,
|
|
733
|
+
mtimeMs: Math.floor(stat.mtimeMs),
|
|
734
|
+
size: stat.size,
|
|
735
|
+
});
|
|
736
|
+
} else if (STYLE_EXTENSIONS.has(ext)) {
|
|
737
|
+
styleUpserts.add(relativePath);
|
|
738
|
+
}
|
|
739
|
+
// Any other extension is never discovered, so a full rebuild would ignore
|
|
740
|
+
// it too — the hint is a safe no-op.
|
|
741
|
+
}
|
|
742
|
+
|
|
743
|
+
return { upserts, removals, styleUpserts, styleRemovals };
|
|
744
|
+
}
|
|
745
|
+
|
|
746
|
+
async function tryIncrementalIndexUpdate({ absoluteRoot, indexDir, changedPaths, signal }) {
|
|
747
|
+
assertNotAborted(signal, 'incremental-stat');
|
|
748
|
+
|
|
749
|
+
const [
|
|
750
|
+
metaArtifact,
|
|
751
|
+
filesArtifact,
|
|
752
|
+
symbolsArtifact,
|
|
753
|
+
importsArtifact,
|
|
754
|
+
callsArtifact,
|
|
755
|
+
testsMapArtifact,
|
|
756
|
+
hotspotsArtifact,
|
|
757
|
+
archetypesArtifact,
|
|
758
|
+
relationsArtifact,
|
|
759
|
+
analogsArtifact,
|
|
760
|
+
] = await Promise.all([
|
|
761
|
+
readArtifactIfExists(absoluteRoot, INDEX_ARTIFACTS.meta),
|
|
762
|
+
readArtifactIfExists(absoluteRoot, INDEX_ARTIFACTS.files),
|
|
763
|
+
readArtifactIfExists(absoluteRoot, INDEX_ARTIFACTS.symbols),
|
|
764
|
+
readArtifactIfExists(absoluteRoot, INDEX_ARTIFACTS.imports),
|
|
765
|
+
readArtifactIfExists(absoluteRoot, INDEX_ARTIFACTS.calls),
|
|
766
|
+
readArtifactIfExists(absoluteRoot, INDEX_ARTIFACTS.testsMap),
|
|
767
|
+
readArtifactIfExists(absoluteRoot, INDEX_ARTIFACTS.hotspots),
|
|
768
|
+
readArtifactIfExists(absoluteRoot, INDEX_ARTIFACTS.archetypes),
|
|
769
|
+
readArtifactIfExists(absoluteRoot, INDEX_ARTIFACTS.relations),
|
|
770
|
+
readArtifactIfExists(absoluteRoot, INDEX_ARTIFACTS.analogs),
|
|
771
|
+
]);
|
|
772
|
+
|
|
773
|
+
// A schema version matching is not enough: the merge reuses these items
|
|
774
|
+
// verbatim, so each artifact must still be a well-formed item array whose
|
|
775
|
+
// entries belong to the files artifact, at the item count the meta artifact
|
|
776
|
+
// recorded when it was published. A schema-current but truncated
|
|
777
|
+
// symbols.json would otherwise cause the update to publish only the changed
|
|
778
|
+
// file's symbols and silently drop every unchanged file's.
|
|
779
|
+
const artifactItemCounts = {
|
|
780
|
+
[INDEX_ARTIFACTS.files]: filesArtifact?.items?.length ?? -1,
|
|
781
|
+
[INDEX_ARTIFACTS.symbols]: symbolsArtifact?.items?.length ?? -1,
|
|
782
|
+
[INDEX_ARTIFACTS.imports]: importsArtifact?.items?.length ?? -1,
|
|
783
|
+
[INDEX_ARTIFACTS.calls]: callsArtifact?.items?.length ?? -1,
|
|
784
|
+
[INDEX_ARTIFACTS.relations]: relationsArtifact?.items?.length ?? -1,
|
|
785
|
+
};
|
|
786
|
+
const previousArtifactsUsable = [
|
|
787
|
+
metaArtifact,
|
|
788
|
+
filesArtifact,
|
|
789
|
+
symbolsArtifact,
|
|
790
|
+
importsArtifact,
|
|
791
|
+
callsArtifact,
|
|
792
|
+
relationsArtifact,
|
|
793
|
+
].every((artifact) => artifact?.schemaVersion === INDEX_SCHEMA_VERSION)
|
|
794
|
+
&& Boolean(metaArtifact?.sourceFingerprint)
|
|
795
|
+
&& Array.isArray(filesArtifact.items)
|
|
796
|
+
&& filesArtifact.items.every((item) => isValidFileRecord(item))
|
|
797
|
+
&& isValidCoveredItems(symbolsArtifact.items, filesArtifact.items, (item) => item.filePath)
|
|
798
|
+
&& isValidCoveredItems(importsArtifact.items, filesArtifact.items, (item) => item.from)
|
|
799
|
+
&& isValidCoveredItems(callsArtifact.items, filesArtifact.items, (item) => item.filePath)
|
|
800
|
+
&& isValidCoveredItems(relationsArtifact.items, filesArtifact.items, (item) => item.filePath)
|
|
801
|
+
&& Array.isArray(relationsArtifact?.sourceSnapshot?.styleFiles)
|
|
802
|
+
&& artifactItemCountsMatch(metaArtifact, artifactItemCounts);
|
|
803
|
+
if (!previousArtifactsUsable) {
|
|
804
|
+
return null;
|
|
805
|
+
}
|
|
806
|
+
|
|
807
|
+
const previousRecordsByPath = new Map(filesArtifact.items.map((item) => [item.filePath, item]));
|
|
808
|
+
const hintUpdates = await resolveChangedHintUpdates(absoluteRoot, changedPaths, previousRecordsByPath);
|
|
809
|
+
if (!hintUpdates) {
|
|
810
|
+
return null;
|
|
811
|
+
}
|
|
812
|
+
|
|
813
|
+
// Merge the hint outcome into one sorted record set — the same shape and
|
|
814
|
+
// ordering the full discovery path produces.
|
|
815
|
+
const mergedRecordsByPath = new Map(previousRecordsByPath);
|
|
816
|
+
for (const removedPath of hintUpdates.removals) {
|
|
817
|
+
mergedRecordsByPath.delete(removedPath);
|
|
818
|
+
}
|
|
819
|
+
for (const [upsertedPath, record] of hintUpdates.upserts) {
|
|
820
|
+
mergedRecordsByPath.set(upsertedPath, record);
|
|
821
|
+
}
|
|
822
|
+
const mergedFileRecords = [...mergedRecordsByPath.values()]
|
|
823
|
+
.sort((a, b) => a.filePath.localeCompare(b.filePath));
|
|
824
|
+
|
|
825
|
+
const recordsChanged = !areFileRecordSnapshotsEqual(filesArtifact.items, mergedFileRecords);
|
|
826
|
+
|
|
827
|
+
// Only hinted code files whose on-disk identity actually moved are
|
|
828
|
+
// re-parsed; an unchanged hint must not rewrite anything.
|
|
829
|
+
const parsedFiles = [];
|
|
830
|
+
for (const [filePath, record] of hintUpdates.upserts) {
|
|
831
|
+
if (!CODE_EXTENSIONS.has(record.ext)) {
|
|
832
|
+
continue;
|
|
833
|
+
}
|
|
834
|
+
const previous = previousRecordsByPath.get(filePath);
|
|
835
|
+
const unchanged = previous
|
|
836
|
+
&& Number(previous.mtimeMs ?? -1) === Number(record.mtimeMs)
|
|
837
|
+
&& Number(previous.size ?? -1) === Number(record.size);
|
|
838
|
+
if (!unchanged) {
|
|
839
|
+
parsedFiles.push(filePath);
|
|
840
|
+
}
|
|
841
|
+
}
|
|
842
|
+
parsedFiles.sort((a, b) => a.localeCompare(b));
|
|
843
|
+
|
|
844
|
+
// Re-apply style hints to the persisted style list; a listed style file that
|
|
845
|
+
// vanished without a hint is real drift the update heals the same way a full
|
|
846
|
+
// rebuild would (by dropping it).
|
|
847
|
+
const styleSet = new Set(relationsArtifact.sourceSnapshot.styleFiles);
|
|
848
|
+
let styleListChanged = false;
|
|
849
|
+
for (const stylePath of hintUpdates.styleUpserts) {
|
|
850
|
+
if (!styleSet.has(stylePath)) {
|
|
851
|
+
styleSet.add(stylePath);
|
|
852
|
+
styleListChanged = true;
|
|
853
|
+
}
|
|
854
|
+
}
|
|
855
|
+
for (const stylePath of hintUpdates.styleRemovals) {
|
|
856
|
+
if (styleSet.has(stylePath)) {
|
|
857
|
+
styleSet.delete(stylePath);
|
|
858
|
+
styleListChanged = true;
|
|
859
|
+
}
|
|
860
|
+
}
|
|
861
|
+
const styleEntries = (await mapLimit([...styleSet].sort((a, b) => a.localeCompare(b)), INDEX_IO_CONCURRENCY, async (stylePath) => {
|
|
862
|
+
try {
|
|
863
|
+
const stat = await fs.stat(path.join(absoluteRoot, stylePath));
|
|
864
|
+
return {
|
|
865
|
+
absolutePath: path.join(absoluteRoot, stylePath),
|
|
866
|
+
mtimeMs: Math.floor(stat.mtimeMs),
|
|
867
|
+
size: stat.size,
|
|
868
|
+
};
|
|
869
|
+
} catch (error) {
|
|
870
|
+
if (error?.code === 'ENOENT') {
|
|
871
|
+
styleListChanged = true;
|
|
872
|
+
return null;
|
|
873
|
+
}
|
|
874
|
+
// Unknown stat failure: cannot reconstruct the fingerprint input.
|
|
875
|
+
return { statError: true };
|
|
876
|
+
}
|
|
877
|
+
}, signal, 'stat-style'));
|
|
878
|
+
if (styleEntries.some((entry) => entry?.statError)) {
|
|
879
|
+
return null;
|
|
880
|
+
}
|
|
881
|
+
const styleFilePaths = styleEntries.filter(Boolean).map((entry) => normalizeRelative(absoluteRoot, entry.absolutePath));
|
|
882
|
+
|
|
883
|
+
const previousSymbolsByPath = groupBy(symbolsArtifact.items ?? [], (item) => item.filePath);
|
|
884
|
+
const previousImportsByPath = groupBy(importsArtifact.items ?? [], (item) => item.from);
|
|
885
|
+
const previousCallsByPath = groupBy(callsArtifact.items ?? [], (item) => item.filePath);
|
|
886
|
+
|
|
887
|
+
const parsedSet = new Set(parsedFiles);
|
|
888
|
+
const codeFiles = mergedFileRecords.filter((record) => CODE_EXTENSIONS.has(record.ext));
|
|
889
|
+
// A removal (or an add) changes code-file membership even when nothing is
|
|
890
|
+
// re-parsed — the merged artifacts must drop (or carry) that file's items.
|
|
891
|
+
const codeIdentitiesChanged = !haveStableTrackedFileIdentities(
|
|
892
|
+
filesArtifact.items.filter((item) => CODE_EXTENSIONS.has(item?.ext)),
|
|
893
|
+
codeFiles,
|
|
894
|
+
);
|
|
895
|
+
const parsedArtifactsChanged = parsedFiles.length > 0 || codeIdentitiesChanged;
|
|
896
|
+
const symbols = [];
|
|
897
|
+
const imports = [];
|
|
898
|
+
const calls = [];
|
|
899
|
+
let reusedCodeFileCount = 0;
|
|
900
|
+
|
|
901
|
+
// Full-path ordering: reused files first (sorted), then newly parsed files
|
|
902
|
+
// (sorted), so a merged artifact carries the same item order a rebuild
|
|
903
|
+
// produces for the same state.
|
|
904
|
+
for (const file of codeFiles) {
|
|
905
|
+
if (parsedSet.has(file.filePath)) {
|
|
906
|
+
continue;
|
|
907
|
+
}
|
|
908
|
+
symbols.push(...(previousSymbolsByPath.get(file.filePath) ?? []));
|
|
909
|
+
imports.push(...(previousImportsByPath.get(file.filePath) ?? []));
|
|
910
|
+
calls.push(...(previousCallsByPath.get(file.filePath) ?? []));
|
|
911
|
+
reusedCodeFileCount += 1;
|
|
912
|
+
}
|
|
913
|
+
|
|
914
|
+
if (parsedFiles.length > 0) {
|
|
915
|
+
const parsed = await mapLimit(parsedFiles, INDEX_PARSE_BATCH_SIZE, async (filePath) => {
|
|
916
|
+
try {
|
|
917
|
+
const absolutePath = path.join(absoluteRoot, filePath);
|
|
918
|
+
const content = await fs.readFile(absolutePath, 'utf8');
|
|
919
|
+
const scriptContent = extractScriptContent(filePath, content);
|
|
920
|
+
return {
|
|
921
|
+
filePath,
|
|
922
|
+
symbols: extractSymbols(filePath, scriptContent),
|
|
923
|
+
imports: [
|
|
924
|
+
...extractImports(filePath, scriptContent),
|
|
925
|
+
...extractSupplementalImports(filePath, content),
|
|
926
|
+
],
|
|
927
|
+
calls: extractFunctionCalls(filePath, scriptContent),
|
|
928
|
+
};
|
|
929
|
+
} catch {
|
|
930
|
+
return null;
|
|
931
|
+
}
|
|
932
|
+
}, signal, 'parse');
|
|
933
|
+
|
|
934
|
+
for (const parsedFile of parsed) {
|
|
935
|
+
if (!parsedFile) {
|
|
936
|
+
continue;
|
|
937
|
+
}
|
|
938
|
+
symbols.push(...parsedFile.symbols);
|
|
939
|
+
imports.push(...parsedFile.imports);
|
|
940
|
+
calls.push(...parsedFile.calls);
|
|
941
|
+
}
|
|
942
|
+
}
|
|
943
|
+
|
|
944
|
+
// Derived artifacts are recomputed in memory from the merged inputs — the
|
|
945
|
+
// expensive work being avoided is discovery and re-parsing, not these pure
|
|
946
|
+
// transformations, so their output stays identical to a clean rebuild.
|
|
947
|
+
const canReuseTestsMap = haveStableTrackedFileIdentities(filesArtifact.items, mergedFileRecords)
|
|
948
|
+
&& testsMapArtifact?.schemaVersion === INDEX_SCHEMA_VERSION;
|
|
949
|
+
const testsMap = canReuseTestsMap ? testsMapArtifact.items : buildTestsMap(mergedFileRecords);
|
|
950
|
+
|
|
951
|
+
const bugIndexSnapshot = await readBugIndexSnapshot(absoluteRoot);
|
|
952
|
+
const canReuseHotspots = hotspotsArtifact?.schemaVersion === INDEX_SCHEMA_VERSION
|
|
953
|
+
&& areBugIndexSnapshotsEqual(hotspotsArtifact?.sourceSnapshot, bugIndexSnapshot);
|
|
954
|
+
const hotspots = canReuseHotspots ? hotspotsArtifact.items : await buildHotspots(absoluteRoot, bugIndexSnapshot);
|
|
955
|
+
|
|
956
|
+
const archetypes = classifyArchetypes(mergedFileRecords, symbols);
|
|
957
|
+
const previousCodePaths = new Set(
|
|
958
|
+
filesArtifact.items.filter((item) => CODE_EXTENSIONS.has(item?.ext)).map((item) => item.filePath),
|
|
959
|
+
);
|
|
960
|
+
const currentCodePaths = new Set(codeFiles.map((file) => file.filePath));
|
|
961
|
+
const canReuseArchetypes = parsedFiles.length === 0
|
|
962
|
+
&& !recordsChanged
|
|
963
|
+
&& archetypesArtifact?.schemaVersion === INDEX_SCHEMA_VERSION
|
|
964
|
+
&& codeFiles.every((file) => previousCodePaths.has(file.filePath))
|
|
965
|
+
&& archetypesArtifact?.items?.length === archetypes.length
|
|
966
|
+
&& (archetypesArtifact.items ?? []).every((item) => currentCodePaths.has(item?.filePath));
|
|
967
|
+
const archetypeItems = canReuseArchetypes ? archetypesArtifact.items : archetypes;
|
|
968
|
+
|
|
969
|
+
const needsAliasContext = importsNeedAliasContext(imports);
|
|
970
|
+
const importAliasState = needsAliasContext
|
|
971
|
+
? await loadImportAliasContextState({ rootDir: absoluteRoot })
|
|
972
|
+
: null;
|
|
973
|
+
const importAliasContext = importAliasState?.context ?? null;
|
|
974
|
+
const styleSnapshot = {
|
|
975
|
+
styleFiles: styleFilePaths,
|
|
976
|
+
importAliasSnapshot: importAliasState?.snapshot ?? [],
|
|
977
|
+
};
|
|
978
|
+
|
|
979
|
+
const graphInputsChanged = parsedArtifactsChanged || styleListChanged;
|
|
980
|
+
const canReuseRelations = !graphInputsChanged
|
|
981
|
+
&& areStringArraysEqual(relationsArtifact?.sourceSnapshot?.styleFiles, styleFilePaths)
|
|
982
|
+
&& arePathSnapshotsEqual(relationsArtifact?.sourceSnapshot?.importAliasSnapshot, styleSnapshot.importAliasSnapshot);
|
|
983
|
+
const canReuseAnalogs = canReuseRelations
|
|
984
|
+
&& analogsArtifact?.schemaVersion === INDEX_SCHEMA_VERSION;
|
|
985
|
+
|
|
986
|
+
let relations = relationsArtifact.items;
|
|
987
|
+
let analogs = analogsArtifact?.items ?? null;
|
|
988
|
+
if (!canReuseRelations || !canReuseAnalogs) {
|
|
989
|
+
const indexedFileSet = new Set(codeFiles.map((file) => file.filePath));
|
|
990
|
+
const resolvedImportsBySource = buildResolvedImportMap(absoluteRoot, imports, indexedFileSet, importAliasContext);
|
|
991
|
+
const resolvedStyleImportsBySource = buildResolvedImportMap(absoluteRoot, imports, new Set(styleFilePaths), importAliasContext);
|
|
992
|
+
if (!canReuseRelations) {
|
|
993
|
+
relations = buildRelations(mergedFileRecords, archetypeItems, testsMap, resolvedImportsBySource, resolvedStyleImportsBySource);
|
|
994
|
+
}
|
|
995
|
+
if (!canReuseAnalogs) {
|
|
996
|
+
analogs = buildAnalogs(mergedFileRecords, archetypeItems, relations, resolvedImportsBySource);
|
|
997
|
+
}
|
|
998
|
+
}
|
|
999
|
+
|
|
1000
|
+
// The fingerprint covers the merged discovery view (tracked records +
|
|
1001
|
+
// persisted style list), so recomputing it detects a style file whose
|
|
1002
|
+
// content/mtime moved without changing list membership — something
|
|
1003
|
+
// `styleListChanged` alone cannot see, and which a clean full rebuild would
|
|
1004
|
+
// fold into `meta.sourceFingerprint`.
|
|
1005
|
+
const mergedDiscoveredEntries = [
|
|
1006
|
+
...mergedFileRecords.map((record) => ({
|
|
1007
|
+
absolutePath: path.join(absoluteRoot, record.filePath),
|
|
1008
|
+
mtimeMs: record.mtimeMs,
|
|
1009
|
+
size: record.size,
|
|
1010
|
+
})),
|
|
1011
|
+
...styleEntries.filter(Boolean),
|
|
1012
|
+
];
|
|
1013
|
+
const refreshedFingerprint = createSourceFingerprint(absoluteRoot, mergedDiscoveredEntries);
|
|
1014
|
+
const fingerprintChanged = !areSourceFingerprintsEqual(metaArtifact.sourceFingerprint, refreshedFingerprint);
|
|
1015
|
+
|
|
1016
|
+
const stateChanged = recordsChanged
|
|
1017
|
+
|| parsedFiles.length > 0
|
|
1018
|
+
|| styleListChanged
|
|
1019
|
+
|| fingerprintChanged;
|
|
1020
|
+
const generatedAt = new Date().toISOString();
|
|
1021
|
+
|
|
1022
|
+
// Same publish gate as the full path: an abort landing after parsing must
|
|
1023
|
+
// not replace any artifact.
|
|
1024
|
+
assertNotAborted(signal, 'publish');
|
|
1025
|
+
await fs.mkdir(indexDir, { recursive: true });
|
|
1026
|
+
if (recordsChanged) {
|
|
1027
|
+
await writeArtifact(absoluteRoot, INDEX_ARTIFACTS.files, { schemaVersion: INDEX_SCHEMA_VERSION, generatedAt, items: mergedFileRecords });
|
|
1028
|
+
}
|
|
1029
|
+
if (parsedArtifactsChanged) {
|
|
1030
|
+
await writeArtifact(absoluteRoot, INDEX_ARTIFACTS.symbols, { schemaVersion: INDEX_SCHEMA_VERSION, generatedAt, items: symbols });
|
|
1031
|
+
await writeArtifact(absoluteRoot, INDEX_ARTIFACTS.imports, { schemaVersion: INDEX_SCHEMA_VERSION, generatedAt, items: imports });
|
|
1032
|
+
await writeArtifact(absoluteRoot, INDEX_ARTIFACTS.calls, { schemaVersion: INDEX_SCHEMA_VERSION, generatedAt, items: calls });
|
|
1033
|
+
}
|
|
1034
|
+
if (!canReuseTestsMap) {
|
|
1035
|
+
await writeArtifact(absoluteRoot, INDEX_ARTIFACTS.testsMap, { schemaVersion: INDEX_SCHEMA_VERSION, generatedAt, items: testsMap });
|
|
1036
|
+
}
|
|
1037
|
+
if (!canReuseHotspots) {
|
|
1038
|
+
await writeArtifact(absoluteRoot, INDEX_ARTIFACTS.hotspots, { schemaVersion: INDEX_SCHEMA_VERSION, generatedAt, sourceSnapshot: bugIndexSnapshot, items: hotspots });
|
|
1039
|
+
}
|
|
1040
|
+
if (!canReuseArchetypes) {
|
|
1041
|
+
await writeArtifact(absoluteRoot, INDEX_ARTIFACTS.archetypes, { schemaVersion: INDEX_SCHEMA_VERSION, generatedAt, items: archetypeItems });
|
|
1042
|
+
}
|
|
1043
|
+
if (!canReuseRelations) {
|
|
1044
|
+
await writeArtifact(absoluteRoot, INDEX_ARTIFACTS.relations, { schemaVersion: INDEX_SCHEMA_VERSION, generatedAt, sourceSnapshot: styleSnapshot, items: relations });
|
|
1045
|
+
}
|
|
1046
|
+
if (!canReuseAnalogs) {
|
|
1047
|
+
await writeArtifact(absoluteRoot, INDEX_ARTIFACTS.analogs, { schemaVersion: INDEX_SCHEMA_VERSION, generatedAt, sourceSnapshot: styleSnapshot, items: analogs });
|
|
1048
|
+
}
|
|
1049
|
+
if (stateChanged) {
|
|
1050
|
+
// Refresh the fingerprint over the merged discovery view (tracked records
|
|
1051
|
+
// + persisted style list), so staleness detection agrees with a clean
|
|
1052
|
+
// rebuild of the same disk state.
|
|
1053
|
+
await writeArtifact(absoluteRoot, INDEX_ARTIFACTS.meta, {
|
|
1054
|
+
schemaVersion: INDEX_SCHEMA_VERSION,
|
|
1055
|
+
generatedAt,
|
|
1056
|
+
sourceFingerprint: refreshedFingerprint,
|
|
1057
|
+
// Keep the truncation guard armed for the next incremental update: these
|
|
1058
|
+
// are the counts of the artifacts that now hold those items.
|
|
1059
|
+
artifactItemCounts: {
|
|
1060
|
+
[INDEX_ARTIFACTS.files]: mergedFileRecords.length,
|
|
1061
|
+
[INDEX_ARTIFACTS.symbols]: symbols.length,
|
|
1062
|
+
[INDEX_ARTIFACTS.imports]: imports.length,
|
|
1063
|
+
[INDEX_ARTIFACTS.calls]: calls.length,
|
|
1064
|
+
[INDEX_ARTIFACTS.relations]: relations.length,
|
|
1065
|
+
},
|
|
1066
|
+
});
|
|
1067
|
+
}
|
|
1068
|
+
|
|
1069
|
+
const preservedRuntimeCaches = !recordsChanged
|
|
1070
|
+
&& !parsedArtifactsChanged
|
|
1071
|
+
&& !styleListChanged
|
|
1072
|
+
&& canReuseTestsMap
|
|
1073
|
+
&& canReuseHotspots
|
|
1074
|
+
&& canReuseArchetypes
|
|
1075
|
+
&& canReuseRelations
|
|
1076
|
+
&& canReuseAnalogs;
|
|
1077
|
+
if (!preservedRuntimeCaches) {
|
|
1078
|
+
clearIndexArtifactCache(absoluteRoot);
|
|
1079
|
+
RELATED_TEST_LOOKUP_CACHE.delete(absoluteRoot);
|
|
1080
|
+
}
|
|
1081
|
+
|
|
1082
|
+
return {
|
|
1083
|
+
indexDir,
|
|
1084
|
+
generatedAt,
|
|
1085
|
+
generatedAtMs: Date.parse(generatedAt),
|
|
1086
|
+
fileCount: mergedFileRecords.length,
|
|
1087
|
+
symbolCount: symbols.length,
|
|
1088
|
+
importCount: imports.length,
|
|
1089
|
+
testsMapCount: testsMap.length,
|
|
1090
|
+
hotspotCount: hotspots.length,
|
|
1091
|
+
archetypeCount: archetypeItems.length,
|
|
1092
|
+
relationCount: relations.length,
|
|
1093
|
+
analogCount: analogs?.length ?? 0,
|
|
1094
|
+
parsedCodeFileCount: parsedFiles.length,
|
|
1095
|
+
reusedCodeFileCount,
|
|
1096
|
+
reusedTestsMap: canReuseTestsMap,
|
|
1097
|
+
reusedHotspots: canReuseHotspots,
|
|
1098
|
+
reusedRelations: canReuseRelations,
|
|
1099
|
+
reusedAnalogs: canReuseAnalogs,
|
|
1100
|
+
preservedRuntimeCaches,
|
|
1101
|
+
mode: 'incremental',
|
|
1102
|
+
parsedFiles,
|
|
1103
|
+
removedFiles: [...hintUpdates.removals].sort((a, b) => a.localeCompare(b)),
|
|
403
1104
|
};
|
|
404
1105
|
}
|
|
405
1106
|
|
|
@@ -1575,17 +2276,37 @@ function buildResolvedImportMap(rootDir, imports, indexedFileSet, importAliasCon
|
|
|
1575
2276
|
|
|
1576
2277
|
// ── Helpers ──
|
|
1577
2278
|
|
|
1578
|
-
async function collectFiles(scanRoots) {
|
|
2279
|
+
async function collectFiles(scanRoots, discovery = null) {
|
|
2280
|
+
const budget = discovery ?? createDiscoveryBudget({});
|
|
1579
2281
|
const result = [];
|
|
1580
2282
|
const stack = [...scanRoots];
|
|
2283
|
+
const rootPath = path.resolve(scanRoots[0]);
|
|
2284
|
+
let rootRealPath = null;
|
|
2285
|
+
try {
|
|
2286
|
+
rootRealPath = await resolveDiscoveryPath(rootPath, budget, 'realpath-root');
|
|
2287
|
+
} catch (error) {
|
|
2288
|
+
if (error instanceof IndexDiscoveryTimeoutError) {
|
|
2289
|
+
throw error;
|
|
2290
|
+
}
|
|
2291
|
+
}
|
|
2292
|
+
|
|
2293
|
+
const isInsideRoot = (realPath) => isDiscoveryPathContained(rootRealPath, realPath);
|
|
1581
2294
|
const visitedDirectoryRealPaths = new Set();
|
|
1582
2295
|
|
|
1583
2296
|
while (stack.length > 0) {
|
|
2297
|
+
assertDiscoveryAlive(budget, 'walk');
|
|
1584
2298
|
const current = stack.pop();
|
|
1585
2299
|
let currentRealPath;
|
|
1586
2300
|
try {
|
|
1587
|
-
currentRealPath = await
|
|
1588
|
-
} catch {
|
|
2301
|
+
currentRealPath = await resolveDiscoveryPath(current, budget, 'realpath');
|
|
2302
|
+
} catch (error) {
|
|
2303
|
+
if (error instanceof IndexDiscoveryTimeoutError) {
|
|
2304
|
+
throw error;
|
|
2305
|
+
}
|
|
2306
|
+
continue;
|
|
2307
|
+
}
|
|
2308
|
+
|
|
2309
|
+
if (!isInsideRoot(currentRealPath)) {
|
|
1589
2310
|
continue;
|
|
1590
2311
|
}
|
|
1591
2312
|
|
|
@@ -1596,8 +2317,15 @@ async function collectFiles(scanRoots) {
|
|
|
1596
2317
|
|
|
1597
2318
|
let entries;
|
|
1598
2319
|
try {
|
|
1599
|
-
entries = await
|
|
1600
|
-
|
|
2320
|
+
entries = await resolveDiscoveryOperation(
|
|
2321
|
+
() => fs.readdir(current, { withFileTypes: true }),
|
|
2322
|
+
budget,
|
|
2323
|
+
'readdir',
|
|
2324
|
+
);
|
|
2325
|
+
} catch (error) {
|
|
2326
|
+
if (error instanceof IndexDiscoveryTimeoutError) {
|
|
2327
|
+
throw error;
|
|
2328
|
+
}
|
|
1601
2329
|
continue;
|
|
1602
2330
|
}
|
|
1603
2331
|
|
|
@@ -1611,15 +2339,33 @@ async function collectFiles(scanRoots) {
|
|
|
1611
2339
|
directDirectories.push(fullPath);
|
|
1612
2340
|
} else if (entry.isSymbolicLink()) {
|
|
1613
2341
|
try {
|
|
1614
|
-
const stat = await
|
|
2342
|
+
const stat = await resolveDiscoveryOperation(
|
|
2343
|
+
() => fs.stat(fullPath),
|
|
2344
|
+
budget,
|
|
2345
|
+
'stat-symlink',
|
|
2346
|
+
);
|
|
1615
2347
|
if (stat.isDirectory() && !shouldSkipDirectory(entry.name)) {
|
|
2348
|
+
// Containment is re-checked from the link's realpath when the
|
|
2349
|
+
// directory is popped, so an escaping symlinked subtree is pruned.
|
|
1616
2350
|
symlinkDirectories.push(fullPath);
|
|
1617
2351
|
continue;
|
|
1618
2352
|
}
|
|
1619
2353
|
|
|
1620
2354
|
if (stat.isFile()) {
|
|
1621
2355
|
const ext = path.extname(entry.name).toLowerCase();
|
|
1622
|
-
if (DISCOVERED_EXTENSIONS.has(ext))
|
|
2356
|
+
if (!DISCOVERED_EXTENSIONS.has(ext)) continue;
|
|
2357
|
+
// A symlinked FILE is indexed only when its real path stays inside
|
|
2358
|
+
// the root; the link itself may point anywhere on disk.
|
|
2359
|
+
let realFilePath = null;
|
|
2360
|
+
try {
|
|
2361
|
+
realFilePath = await resolveDiscoveryPath(fullPath, budget, 'realpath-symlink');
|
|
2362
|
+
} catch (error) {
|
|
2363
|
+
if (error instanceof IndexDiscoveryTimeoutError) {
|
|
2364
|
+
throw error;
|
|
2365
|
+
}
|
|
2366
|
+
realFilePath = null;
|
|
2367
|
+
}
|
|
2368
|
+
if (isInsideRoot(realFilePath)) {
|
|
1623
2369
|
symlinkFiles.push({
|
|
1624
2370
|
absolutePath: fullPath,
|
|
1625
2371
|
mtimeMs: Math.floor(stat.mtimeMs),
|
|
@@ -1627,7 +2373,13 @@ async function collectFiles(scanRoots) {
|
|
|
1627
2373
|
});
|
|
1628
2374
|
}
|
|
1629
2375
|
}
|
|
1630
|
-
} catch {
|
|
2376
|
+
} catch (error) {
|
|
2377
|
+
// A stalled stat/realpath past the shared deadline must surface as the
|
|
2378
|
+
// typed timeout, not be misclassified as a broken symlink and resolve
|
|
2379
|
+
// to silently truncated discovery.
|
|
2380
|
+
if (error instanceof IndexDiscoveryTimeoutError) {
|
|
2381
|
+
throw error;
|
|
2382
|
+
}
|
|
1631
2383
|
// broken symlink, skip
|
|
1632
2384
|
}
|
|
1633
2385
|
}
|
|
@@ -1639,15 +2391,22 @@ async function collectFiles(scanRoots) {
|
|
|
1639
2391
|
const ext = path.extname(entry.name).toLowerCase();
|
|
1640
2392
|
return DISCOVERED_EXTENSIONS.has(ext);
|
|
1641
2393
|
});
|
|
1642
|
-
const fileStats = (await
|
|
2394
|
+
const fileStats = (await mapLimit(fileEntries, INDEX_IO_CONCURRENCY, async (entry) => {
|
|
1643
2395
|
try {
|
|
1644
2396
|
const fullPath = path.join(current, entry.name);
|
|
1645
|
-
const stat = await
|
|
2397
|
+
const stat = await resolveDiscoveryOperation(
|
|
2398
|
+
() => fs.stat(fullPath),
|
|
2399
|
+
budget,
|
|
2400
|
+
'stat-walk',
|
|
2401
|
+
);
|
|
1646
2402
|
return { absolutePath: fullPath, mtimeMs: Math.floor(stat.mtimeMs), size: stat.size };
|
|
1647
|
-
} catch {
|
|
2403
|
+
} catch (error) {
|
|
2404
|
+
if (error instanceof IndexDiscoveryTimeoutError) {
|
|
2405
|
+
throw error;
|
|
2406
|
+
}
|
|
1648
2407
|
return null;
|
|
1649
2408
|
}
|
|
1650
|
-
}))
|
|
2409
|
+
}, budget.signal, 'stat-walk')).filter(Boolean);
|
|
1651
2410
|
result.push(...fileStats, ...symlinkFiles);
|
|
1652
2411
|
}
|
|
1653
2412
|
|
|
@@ -2944,50 +3703,156 @@ function normalizeTestStem(testBase) {
|
|
|
2944
3703
|
* is self-tuning per layout and needs no configuration. Non-git projects fall
|
|
2945
3704
|
* back to walking the root with the same exclusion rules.
|
|
2946
3705
|
*/
|
|
2947
|
-
|
|
2948
|
-
|
|
3706
|
+
// Discover project files under ONE top-level wall-clock deadline. `deadlineMs`
|
|
3707
|
+
// is a single ceiling divided across both git enumerations and the non-git
|
|
3708
|
+
// fallback walk — not a per-call timeout. `signal` cancels early and surfaces
|
|
3709
|
+
// through the same typed error. Throws IndexDiscoveryTimeoutError on expiry.
|
|
3710
|
+
export async function discoverProjectFiles(rootDir, { signal = null, deadlineMs = DEFAULT_DISCOVERY_DEADLINE_MS } = {}) {
|
|
3711
|
+
const budget = createDiscoveryBudget({ signal, deadlineMs });
|
|
3712
|
+
assertDiscoveryAlive(budget, 'start');
|
|
3713
|
+
const gitFiles = await collectGitTrackedFiles(rootDir, budget);
|
|
2949
3714
|
if (gitFiles) return gitFiles;
|
|
2950
|
-
return collectFiles([rootDir]);
|
|
3715
|
+
return collectFiles([rootDir], budget);
|
|
3716
|
+
}
|
|
3717
|
+
|
|
3718
|
+
function createDiscoveryBudget({ signal = null, deadlineMs = DEFAULT_DISCOVERY_DEADLINE_MS } = {}) {
|
|
3719
|
+
const effectiveDeadlineMs = Number.isFinite(deadlineMs) && deadlineMs > 0 ? deadlineMs : 0;
|
|
3720
|
+
const startedAt = Date.now();
|
|
3721
|
+
return {
|
|
3722
|
+
signal,
|
|
3723
|
+
deadlineMs: effectiveDeadlineMs,
|
|
3724
|
+
startedAt,
|
|
3725
|
+
deadline: startedAt + effectiveDeadlineMs,
|
|
3726
|
+
};
|
|
3727
|
+
}
|
|
3728
|
+
|
|
3729
|
+
function assertDiscoveryAlive({ signal, deadlineMs, deadline, startedAt }, phase) {
|
|
3730
|
+
const elapsedMs = Date.now() - startedAt;
|
|
3731
|
+
if (signal?.aborted) {
|
|
3732
|
+
throw new IndexDiscoveryTimeoutError({ deadlineMs, phase, elapsedMs, aborted: true });
|
|
3733
|
+
}
|
|
3734
|
+
if (Date.now() >= deadline) {
|
|
3735
|
+
throw new IndexDiscoveryTimeoutError({ deadlineMs, phase, elapsedMs });
|
|
3736
|
+
}
|
|
3737
|
+
}
|
|
3738
|
+
|
|
3739
|
+
function discoveryOperationTimeout(budget, phase) {
|
|
3740
|
+
assertDiscoveryAlive(budget, phase);
|
|
3741
|
+
return Math.max(1, Math.ceil(remainingDiscoveryMs(budget)));
|
|
3742
|
+
}
|
|
3743
|
+
|
|
3744
|
+
async function resolveDiscoveryOperation(operation, budget, phase) {
|
|
3745
|
+
const timeoutMs = discoveryOperationTimeout(budget, phase);
|
|
3746
|
+
let timer;
|
|
3747
|
+
const timeout = new Promise((_, reject) => {
|
|
3748
|
+
timer = setTimeout(() => {
|
|
3749
|
+
reject(new IndexDiscoveryTimeoutError({
|
|
3750
|
+
deadlineMs: budget.deadlineMs,
|
|
3751
|
+
phase,
|
|
3752
|
+
elapsedMs: Date.now() - budget.startedAt,
|
|
3753
|
+
}));
|
|
3754
|
+
}, timeoutMs);
|
|
3755
|
+
});
|
|
3756
|
+
try {
|
|
3757
|
+
return await Promise.race([Promise.resolve().then(operation), timeout]);
|
|
3758
|
+
} finally {
|
|
3759
|
+
clearTimeout(timer);
|
|
3760
|
+
}
|
|
3761
|
+
}
|
|
3762
|
+
|
|
3763
|
+
async function resolveDiscoveryPath(filePath, budget, phase) {
|
|
3764
|
+
return resolveDiscoveryOperation(() => fs.realpath(filePath), budget, phase);
|
|
3765
|
+
}
|
|
3766
|
+
|
|
3767
|
+
function remainingDiscoveryMs({ deadline }) {
|
|
3768
|
+
return deadline - Date.now();
|
|
3769
|
+
}
|
|
3770
|
+
|
|
3771
|
+
function isDiscoveryPathContained(rootRealPath, realPath) {
|
|
3772
|
+
if (!rootRealPath || !realPath) {
|
|
3773
|
+
return false;
|
|
3774
|
+
}
|
|
3775
|
+
if (realPath === rootRealPath) {
|
|
3776
|
+
return true;
|
|
3777
|
+
}
|
|
3778
|
+
const relative = path.relative(rootRealPath, realPath);
|
|
3779
|
+
return relative !== '' && !relative.startsWith('..') && !path.isAbsolute(relative);
|
|
2951
3780
|
}
|
|
2952
3781
|
|
|
2953
3782
|
function createDiscoverySnapshot(rootDir, files, fingerprint) {
|
|
2954
|
-
//
|
|
2955
|
-
//
|
|
3783
|
+
// Deep-frozen, and each file entry is cloned into a fresh frozen record: the
|
|
3784
|
+
// snapshot must stay byte-stable between the staleness inspection and the
|
|
3785
|
+
// build that consumes it, and a caller that keeps a reference must not be able
|
|
3786
|
+
// to redirect the build by mutating `entry.absolutePath` after inspection.
|
|
2956
3787
|
return Object.freeze({
|
|
2957
3788
|
snapshotVersion: DISCOVERY_SNAPSHOT_VERSION,
|
|
2958
3789
|
schemaVersion: INDEX_SCHEMA_VERSION,
|
|
2959
3790
|
rootDir,
|
|
2960
|
-
fingerprint,
|
|
2961
|
-
files: Object.freeze(files.
|
|
3791
|
+
fingerprint: Object.freeze({ ...fingerprint }),
|
|
3792
|
+
files: Object.freeze(files.map((entry) => Object.freeze({
|
|
3793
|
+
absolutePath: entry?.absolutePath,
|
|
3794
|
+
mtimeMs: entry?.mtimeMs,
|
|
3795
|
+
size: entry?.size,
|
|
3796
|
+
}))),
|
|
2962
3797
|
});
|
|
2963
3798
|
}
|
|
2964
3799
|
|
|
3800
|
+
function isSnapshotPathContained(rootDir, absolutePath) {
|
|
3801
|
+
if (typeof absolutePath !== 'string' || absolutePath === '' || !path.isAbsolute(absolutePath)) {
|
|
3802
|
+
return false;
|
|
3803
|
+
}
|
|
3804
|
+
const resolved = path.resolve(absolutePath);
|
|
3805
|
+
// A file entry can never be the root directory itself.
|
|
3806
|
+
if (resolved === rootDir) {
|
|
3807
|
+
return false;
|
|
3808
|
+
}
|
|
3809
|
+
const relative = path.relative(rootDir, resolved);
|
|
3810
|
+
return relative !== '' && !relative.startsWith('..') && !path.isAbsolute(relative);
|
|
3811
|
+
}
|
|
3812
|
+
|
|
2965
3813
|
function isDiscoverySnapshotUsable(discoverySnapshot, absoluteRoot) {
|
|
2966
3814
|
return Boolean(discoverySnapshot)
|
|
2967
3815
|
&& discoverySnapshot.snapshotVersion === DISCOVERY_SNAPSHOT_VERSION
|
|
2968
3816
|
&& discoverySnapshot.schemaVersion === INDEX_SCHEMA_VERSION
|
|
2969
3817
|
&& typeof discoverySnapshot.rootDir === 'string'
|
|
2970
3818
|
&& path.resolve(discoverySnapshot.rootDir) === absoluteRoot
|
|
2971
|
-
&& Array.isArray(discoverySnapshot.files)
|
|
3819
|
+
&& Array.isArray(discoverySnapshot.files)
|
|
3820
|
+
// Fail closed: a snapshot is reusable only when EVERY entry is an absolute
|
|
3821
|
+
// path contained by the snapshotted root. A mutated, relative, or foreign
|
|
3822
|
+
// entry makes the whole snapshot untrusted — the repository is rediscovered
|
|
3823
|
+
// once instead of letting caller-supplied paths steer the build.
|
|
3824
|
+
&& discoverySnapshot.files.every((entry) => Boolean(entry)
|
|
3825
|
+
&& typeof entry === 'object'
|
|
3826
|
+
&& isSnapshotPathContained(absoluteRoot, entry.absolutePath));
|
|
2972
3827
|
}
|
|
2973
3828
|
|
|
2974
|
-
function runGit(rootDir, args) {
|
|
3829
|
+
function runGit(rootDir, args, budgetMs) {
|
|
3830
|
+
// spawnSync treats `timeout: 0` as "no timeout", so an exhausted budget must
|
|
3831
|
+
// short-circuit here instead of ever reaching an unbounded git call.
|
|
3832
|
+
if (!(budgetMs > 0)) return null;
|
|
3833
|
+
|
|
2975
3834
|
const result = spawnSync('git', args, {
|
|
2976
3835
|
cwd: rootDir,
|
|
2977
3836
|
encoding: 'utf8',
|
|
2978
3837
|
maxBuffer: 256 * 1024 * 1024,
|
|
2979
|
-
timeout:
|
|
3838
|
+
timeout: Math.ceil(budgetMs),
|
|
2980
3839
|
windowsHide: true,
|
|
2981
3840
|
});
|
|
2982
3841
|
if (result.error || result.status !== 0 || typeof result.stdout !== 'string') return null;
|
|
2983
3842
|
return result.stdout;
|
|
2984
3843
|
}
|
|
2985
3844
|
|
|
2986
|
-
async function collectGitTrackedFiles(rootDir) {
|
|
3845
|
+
async function collectGitTrackedFiles(rootDir, discovery = null) {
|
|
3846
|
+
const budget = discovery ?? createDiscoveryBudget({});
|
|
3847
|
+
assertDiscoveryAlive(budget, 'git');
|
|
2987
3848
|
// `-z` also disables git path quoting, which would mangle non-ASCII filenames.
|
|
2988
|
-
const tracked = runGit(rootDir, ['ls-files', '-z']);
|
|
3849
|
+
const tracked = runGit(rootDir, ['ls-files', '-z'], remainingDiscoveryMs(budget));
|
|
2989
3850
|
if (tracked === null) return null;
|
|
2990
|
-
|
|
3851
|
+
// Both enumerations draw from the same budget: a first call that hung pays
|
|
3852
|
+
// the whole ceiling, and the second call (and the fallback walk after it)
|
|
3853
|
+
// only ever gets the remaining time.
|
|
3854
|
+
assertDiscoveryAlive(budget, 'git');
|
|
3855
|
+
const untracked = runGit(rootDir, ['ls-files', '-z', '--others', '--exclude-standard'], remainingDiscoveryMs(budget)) ?? '';
|
|
2991
3856
|
|
|
2992
3857
|
const relativePaths = new Set();
|
|
2993
3858
|
for (const entry of `${tracked}${untracked}`.split('\0')) {
|
|
@@ -3002,16 +3867,44 @@ async function collectGitTrackedFiles(rootDir) {
|
|
|
3002
3867
|
return !relativePath.split('/').some((segment) => EXCLUDED_DIR_NAMES.has(segment));
|
|
3003
3868
|
});
|
|
3004
3869
|
|
|
3005
|
-
|
|
3870
|
+
// Fixed pool instead of one unbounded promise per candidate: peak in-flight
|
|
3871
|
+
// stats stay at INDEX_IO_CONCURRENCY for a listing of any size, and an
|
|
3872
|
+
// aborted signal stops the fan-out with queued candidates unstat'ed.
|
|
3873
|
+
let rootRealPath;
|
|
3874
|
+
try {
|
|
3875
|
+
rootRealPath = await resolveDiscoveryPath(path.resolve(rootDir), budget, 'realpath-root');
|
|
3876
|
+
} catch (error) {
|
|
3877
|
+
if (error instanceof IndexDiscoveryTimeoutError) {
|
|
3878
|
+
throw error;
|
|
3879
|
+
}
|
|
3880
|
+
return null;
|
|
3881
|
+
}
|
|
3882
|
+
|
|
3883
|
+
const entries = await mapLimit(candidates, INDEX_IO_CONCURRENCY, async (relativePath) => {
|
|
3006
3884
|
const absolutePath = path.join(rootDir, relativePath);
|
|
3007
3885
|
try {
|
|
3008
|
-
const stat = await
|
|
3886
|
+
const stat = await resolveDiscoveryOperation(
|
|
3887
|
+
() => fs.stat(absolutePath),
|
|
3888
|
+
budget,
|
|
3889
|
+
'stat-git',
|
|
3890
|
+
);
|
|
3009
3891
|
if (!stat.isFile()) return null;
|
|
3892
|
+
// fs.stat follows symlinks. Resolve the final target before accepting a
|
|
3893
|
+
// tracked path so Git cannot make an external file enter the index merely
|
|
3894
|
+
// by tracking an escaping symlink.
|
|
3895
|
+
const realPath = await resolveDiscoveryPath(absolutePath, budget, 'realpath-git');
|
|
3896
|
+
if (!isDiscoveryPathContained(rootRealPath, realPath)) {
|
|
3897
|
+
return null;
|
|
3898
|
+
}
|
|
3010
3899
|
return { absolutePath, mtimeMs: Math.floor(stat.mtimeMs), size: stat.size };
|
|
3011
|
-
} catch {
|
|
3900
|
+
} catch (error) {
|
|
3901
|
+
if (error instanceof IndexDiscoveryTimeoutError) {
|
|
3902
|
+
throw error;
|
|
3903
|
+
}
|
|
3904
|
+
// staged deletion, broken symlink, or an unreadable path
|
|
3012
3905
|
return null;
|
|
3013
3906
|
}
|
|
3014
|
-
})
|
|
3907
|
+
}, budget.signal, 'stat-git');
|
|
3015
3908
|
|
|
3016
3909
|
return entries.filter(Boolean);
|
|
3017
3910
|
}
|
|
@@ -3437,7 +4330,7 @@ export async function getIndexArtifactGeneratedAt({
|
|
|
3437
4330
|
return parseArtifactGeneratedAt(artifact);
|
|
3438
4331
|
}
|
|
3439
4332
|
|
|
3440
|
-
async function evaluateIndexStaleness({ rootDir, maxAgeMs, now, generatedAtMs, enumerateForSnapshot }) {
|
|
4333
|
+
async function evaluateIndexStaleness({ rootDir, maxAgeMs, now, generatedAtMs, enumerateForSnapshot, signal = null, deadlineMs = DEFAULT_DISCOVERY_DEADLINE_MS }) {
|
|
3441
4334
|
const absoluteRoot = path.resolve(rootDir);
|
|
3442
4335
|
const metaArtifact = await readArtifactIfExists(absoluteRoot, INDEX_ARTIFACTS.meta);
|
|
3443
4336
|
const effectiveGeneratedAtMs = Number.isFinite(generatedAtMs)
|
|
@@ -3460,7 +4353,7 @@ async function evaluateIndexStaleness({ rootDir, maxAgeMs, now, generatedAtMs, e
|
|
|
3460
4353
|
// Exactly one enumeration per inspection. In the refresh flow this snapshot is
|
|
3461
4354
|
// handed to buildCodeIndex, so stale-check + rebuild enumerates the repository
|
|
3462
4355
|
// once instead of twice.
|
|
3463
|
-
const discoveredFiles = await discoverProjectFiles(absoluteRoot);
|
|
4356
|
+
const discoveredFiles = await discoverProjectFiles(absoluteRoot, { signal, deadlineMs });
|
|
3464
4357
|
const fingerprint = createSourceFingerprint(absoluteRoot, discoveredFiles);
|
|
3465
4358
|
|
|
3466
4359
|
return {
|
|
@@ -3476,8 +4369,10 @@ export async function inspectIndexStaleness({
|
|
|
3476
4369
|
maxAgeMs = DEFAULT_INDEX_CACHE_MAX_AGE_MS,
|
|
3477
4370
|
now = Date.now(),
|
|
3478
4371
|
generatedAtMs = null,
|
|
4372
|
+
signal = null,
|
|
4373
|
+
deadlineMs = DEFAULT_DISCOVERY_DEADLINE_MS,
|
|
3479
4374
|
} = {}) {
|
|
3480
|
-
return evaluateIndexStaleness({ rootDir, maxAgeMs, now, generatedAtMs, enumerateForSnapshot: true });
|
|
4375
|
+
return evaluateIndexStaleness({ rootDir, maxAgeMs, now, generatedAtMs, enumerateForSnapshot: true, signal, deadlineMs });
|
|
3481
4376
|
}
|
|
3482
4377
|
|
|
3483
4378
|
export async function isIndexStale({
|
|
@@ -3485,7 +4380,9 @@ export async function isIndexStale({
|
|
|
3485
4380
|
maxAgeMs = DEFAULT_INDEX_CACHE_MAX_AGE_MS,
|
|
3486
4381
|
now = Date.now(),
|
|
3487
4382
|
generatedAtMs = null,
|
|
4383
|
+
signal = null,
|
|
4384
|
+
deadlineMs = DEFAULT_DISCOVERY_DEADLINE_MS,
|
|
3488
4385
|
} = {}) {
|
|
3489
|
-
const { stale } = await evaluateIndexStaleness({ rootDir, maxAgeMs, now, generatedAtMs, enumerateForSnapshot: false });
|
|
4386
|
+
const { stale } = await evaluateIndexStaleness({ rootDir, maxAgeMs, now, generatedAtMs, enumerateForSnapshot: false, signal, deadlineMs });
|
|
3490
4387
|
return stale;
|
|
3491
4388
|
}
|