@codeyam/codeyam-cli 0.1.22 → 0.1.23

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (52) hide show
  1. package/analyzer-template/.build-info.json +7 -7
  2. package/analyzer-template/log.txt +3 -3
  3. package/analyzer-template/packages/ai/src/lib/astScopes/methodSemantics.ts +135 -0
  4. package/analyzer-template/packages/ai/src/lib/astScopes/nodeToSource.ts +19 -0
  5. package/analyzer-template/packages/ai/src/lib/astScopes/paths.ts +11 -4
  6. package/analyzer-template/packages/analyze/src/lib/files/scenarios/generateDataStructure.ts +5 -1
  7. package/analyzer-template/packages/analyze/src/lib/files/scenarios/mergeInDependentDataStructure.ts +1632 -1554
  8. package/codeyam-cli/src/commands/__tests__/editor.analyzeImportsArgs.test.js +47 -0
  9. package/codeyam-cli/src/commands/__tests__/editor.analyzeImportsArgs.test.js.map +1 -0
  10. package/codeyam-cli/src/commands/__tests__/editor.auditNoAutoAnalysis.test.js +17 -9
  11. package/codeyam-cli/src/commands/__tests__/editor.auditNoAutoAnalysis.test.js.map +1 -1
  12. package/codeyam-cli/src/commands/editor.js +135 -18
  13. package/codeyam-cli/src/commands/editor.js.map +1 -1
  14. package/codeyam-cli/src/commands/editorAnalyzeImportsArgs.js +23 -0
  15. package/codeyam-cli/src/commands/editorAnalyzeImportsArgs.js.map +1 -0
  16. package/codeyam-cli/src/utils/__tests__/editorAudit.test.js +103 -1
  17. package/codeyam-cli/src/utils/__tests__/editorAudit.test.js.map +1 -1
  18. package/codeyam-cli/src/utils/__tests__/editorScenarios.test.js +140 -1
  19. package/codeyam-cli/src/utils/__tests__/editorScenarios.test.js.map +1 -1
  20. package/codeyam-cli/src/utils/__tests__/editorSeedAdapter.test.js +50 -1
  21. package/codeyam-cli/src/utils/__tests__/editorSeedAdapter.test.js.map +1 -1
  22. package/codeyam-cli/src/utils/editorAudit.js +38 -2
  23. package/codeyam-cli/src/utils/editorAudit.js.map +1 -1
  24. package/codeyam-cli/src/utils/editorScenarios.js +60 -0
  25. package/codeyam-cli/src/utils/editorScenarios.js.map +1 -1
  26. package/codeyam-cli/src/utils/editorSeedAdapter.js +42 -2
  27. package/codeyam-cli/src/utils/editorSeedAdapter.js.map +1 -1
  28. package/codeyam-cli/src/webserver/__tests__/editorProxy.test.js +30 -11
  29. package/codeyam-cli/src/webserver/__tests__/editorProxy.test.js.map +1 -1
  30. package/codeyam-cli/src/webserver/build/client/assets/{editor.entity.(_sha)-aIHKLB-m.js → editor.entity.(_sha)-DMv5ESGo.js} +18 -18
  31. package/codeyam-cli/src/webserver/build/client/assets/{manifest-bcbb3d49.js → manifest-1a45e154.js} +1 -1
  32. package/codeyam-cli/src/webserver/build/server/assets/{analysisRunner-DjF-soOH.js → analysisRunner-By5slFjw.js} +1 -1
  33. package/codeyam-cli/src/webserver/build/server/assets/{index-nAvHGWbz.js → index-DXaOwBnm.js} +1 -1
  34. package/codeyam-cli/src/webserver/build/server/assets/{init-XhpIt-OT.js → init-CLG1LjQM.js} +1 -1
  35. package/codeyam-cli/src/webserver/build/server/assets/{server-build-DVwiibFu.js → server-build-NZmUqQv6.js} +155 -111
  36. package/codeyam-cli/src/webserver/build/server/index.js +1 -1
  37. package/codeyam-cli/src/webserver/build-info.json +5 -5
  38. package/codeyam-cli/src/webserver/editorProxy.js +55 -3
  39. package/codeyam-cli/src/webserver/editorProxy.js.map +1 -1
  40. package/codeyam-cli/templates/codeyam-editor-reference.md +8 -6
  41. package/codeyam-cli/templates/nextjs-prisma-sqlite/seed-adapter.ts +42 -34
  42. package/package.json +1 -1
  43. package/packages/ai/src/lib/astScopes/methodSemantics.js +99 -0
  44. package/packages/ai/src/lib/astScopes/methodSemantics.js.map +1 -1
  45. package/packages/ai/src/lib/astScopes/nodeToSource.js +16 -0
  46. package/packages/ai/src/lib/astScopes/nodeToSource.js.map +1 -1
  47. package/packages/ai/src/lib/astScopes/paths.js +12 -3
  48. package/packages/ai/src/lib/astScopes/paths.js.map +1 -1
  49. package/packages/analyze/src/lib/files/scenarios/generateDataStructure.js +5 -1
  50. package/packages/analyze/src/lib/files/scenarios/generateDataStructure.js.map +1 -1
  51. package/packages/analyze/src/lib/files/scenarios/mergeInDependentDataStructure.js +1330 -1270
  52. package/packages/analyze/src/lib/files/scenarios/mergeInDependentDataStructure.js.map +1 -1
@@ -54,6 +54,23 @@ function isCollectionMethodPath(path: string): boolean {
54
54
  return COLLECTION_METHOD_PATTERN.test(path);
55
55
  }
56
56
 
57
+ /**
58
+ * Check if a path contains an inline object literal inside a function call argument.
59
+ * e.g., setUndoEntry({ label: '...', undo: () => {} }) has '{' inside '(' ')'.
60
+ * These paths are call-site snapshots where the source code text was captured
61
+ * as the path. They don't represent schema structure and are very expensive to
62
+ * parse (avg 319 chars). They account for ~55% of equivalencies in complex entities.
63
+ */
64
+ const INLINE_OBJECT_ARG_PATTERN = /\([^)]*\{[^}]*:/;
65
+ function hasInlineObjectArg(path: string): boolean {
66
+ // Match function calls containing object literals with key-value pairs.
67
+ // Pattern: open paren, then { with a : inside before closing }.
68
+ // e.g., setUndoEntry({ label: '...' }) matches
69
+ // e.g., find(item) does NOT match
70
+ // e.g., fn({a:1, b:2}) matches
71
+ return INLINE_OBJECT_ARG_PATTERN.test(path);
72
+ }
73
+
57
74
  // Primitive types that should not have child paths
58
75
  const PRIMITIVE_TYPES = new Set([
59
76
  'number',
@@ -159,6 +176,11 @@ function bestValueFromOptions(options: Array<string | undefined>) {
159
176
  * All successful merges complete in <300ms. Anything exceeding 2s is pathological. */
160
177
  const MERGE_TIMEOUT_MS = 2_000;
161
178
 
179
+ /** Cap for schema size during postfix application and dep copy.
180
+ * Successful merges produce <3K ret keys. Beyond 5K, further entries
181
+ * are cross-products of unrelated equivalencies — noise, not signal. */
182
+ const SCHEMA_KEY_CAP = 5_000;
183
+
162
184
  export class DataStructureTimeoutError extends Error {
163
185
  constructor(entityName: string, elapsedMs: number) {
164
186
  super(
@@ -225,749 +247,768 @@ export default function mergeInDependentDataStructure({
225
247
  environmentVariables: [...(dataStructure.environmentVariables || [])],
226
248
  };
227
249
 
228
- // Build a set of functions that have multiple DIFFERENT type parameters.
229
- // For these functions, we must NOT normalize paths to avoid merging different schemas.
230
- // e.g., if we have both useFetcher<{ data: UserData }>() and useFetcher<{ data: ConfigData }>(),
231
- // they must stay separate and not both become 'returnValue'.
232
- const functionsWithMultipleTypeParams = new Set<string>();
233
- const typeParamsByFunction: Record<string, Set<string>> = {};
234
-
235
- // Helper to scan a schema for type parameters
236
- const scanSchemaForTypeParams = (schema: { [key: string]: string }) => {
237
- for (const schemaPath of Object.keys(schema ?? {})) {
238
- const parts = splitOutsideParenthesesAndArrays(schemaPath);
239
- if (parts.length > 0) {
240
- const firstPart = parts[0];
241
- const typeParam = getTypeParameter(firstPart);
242
- if (typeParam) {
243
- const baseName = cleanFunctionName(firstPart);
244
- typeParamsByFunction[baseName] ||= new Set();
245
- typeParamsByFunction[baseName].add(typeParam);
246
- if (typeParamsByFunction[baseName].size > 1) {
247
- functionsWithMultipleTypeParams.add(baseName);
250
+ try {
251
+ // Build a set of functions that have multiple DIFFERENT type parameters.
252
+ // For these functions, we must NOT normalize paths to avoid merging different schemas.
253
+ // e.g., if we have both useFetcher<{ data: UserData }>() and useFetcher<{ data: ConfigData }>(),
254
+ // they must stay separate and not both become 'returnValue'.
255
+ const functionsWithMultipleTypeParams = new Set<string>();
256
+ const typeParamsByFunction: Record<string, Set<string>> = {};
257
+
258
+ // Helper to scan a schema for type parameters
259
+ const scanSchemaForTypeParams = (schema: { [key: string]: string }) => {
260
+ for (const schemaPath of Object.keys(schema ?? {})) {
261
+ const parts = splitOutsideParenthesesAndArrays(schemaPath);
262
+ if (parts.length > 0) {
263
+ const firstPart = parts[0];
264
+ const typeParam = getTypeParameter(firstPart);
265
+ if (typeParam) {
266
+ const baseName = cleanFunctionName(firstPart);
267
+ typeParamsByFunction[baseName] ||= new Set();
268
+ typeParamsByFunction[baseName].add(typeParam);
269
+ if (typeParamsByFunction[baseName].size > 1) {
270
+ functionsWithMultipleTypeParams.add(baseName);
271
+ }
248
272
  }
249
273
  }
250
274
  }
251
- }
252
- };
275
+ };
253
276
 
254
- // Scan the root entity's schema
255
- scanSchemaForTypeParams(dataStructure.returnValueSchema);
277
+ // Scan the root entity's schema
278
+ scanSchemaForTypeParams(dataStructure.returnValueSchema);
256
279
 
257
- // Also scan all dependency schemas for type parameters
258
- for (const filePath of Object.keys(dependencySchemas ?? {})) {
259
- for (const name of Object.keys(dependencySchemas[filePath] ?? {})) {
260
- scanSchemaForTypeParams(
261
- dependencySchemas[filePath][name]?.returnValueSchema,
262
- );
280
+ // Also scan all dependency schemas for type parameters
281
+ for (const filePath of Object.keys(dependencySchemas ?? {})) {
282
+ for (const name of Object.keys(dependencySchemas[filePath] ?? {})) {
283
+ scanSchemaForTypeParams(
284
+ dependencySchemas[filePath][name]?.returnValueSchema,
285
+ );
286
+ }
263
287
  }
264
- }
265
288
 
266
- let equivalentSchemaPaths: {
267
- equivalentRoots: {
268
- schemaRootPath: string;
269
- function?: { filePath?: string; name: string };
270
- postfix?: string;
271
- }[];
272
- equivalentPostfixes: Record<string, string>;
273
- }[] = [];
274
-
275
- // O(1) index for findOrCreateEquivalentSchemaPathsEntry.
276
- // Maps "(rootPath)::(normalizedFuncName)" → the entry containing that root.
277
- // This replaces the O(E) linear search that was causing O(E²) gather performance.
278
- const espIndex = new Map<string, (typeof equivalentSchemaPaths)[0]>();
279
- const espIndexKey = (path: string, functionName: string | undefined) => {
280
- const normalized = cleanFunctionName(functionName);
281
- const funcKey =
282
- normalized === rootScopeName ? '__self__' : normalized || '__self__';
283
- return `${path}::${funcKey}`;
284
- };
285
- const updateEspIndex = (entry: (typeof equivalentSchemaPaths)[0]) => {
286
- for (const root of entry.equivalentRoots) {
287
- const funcName = root.function?.name ?? rootScopeName;
288
- espIndex.set(espIndexKey(root.schemaRootPath, funcName), entry);
289
- }
290
- };
289
+ let equivalentSchemaPaths: {
290
+ equivalentRoots: {
291
+ schemaRootPath: string;
292
+ function?: { filePath?: string; name: string };
293
+ postfix?: string;
294
+ }[];
295
+ equivalentPostfixes: Record<string, string>;
296
+ }[] = [];
297
+
298
+ // O(1) index for findOrCreateEquivalentSchemaPathsEntry.
299
+ // Maps "(rootPath)::(normalizedFuncName)" → the entry containing that root.
300
+ // This replaces the O(E) linear search that was causing O(E²) gather performance.
301
+ const espIndex = new Map<string, (typeof equivalentSchemaPaths)[0]>();
302
+ const espIndexKey = (path: string, functionName: string | undefined) => {
303
+ const normalized = cleanFunctionName(functionName);
304
+ const funcKey =
305
+ normalized === rootScopeName ? '__self__' : normalized || '__self__';
306
+ return `${path}::${funcKey}`;
307
+ };
308
+ const updateEspIndex = (entry: (typeof equivalentSchemaPaths)[0]) => {
309
+ for (const root of entry.equivalentRoots) {
310
+ const funcName = root.function?.name ?? rootScopeName;
311
+ espIndex.set(espIndexKey(root.schemaRootPath, funcName), entry);
312
+ }
313
+ };
291
314
 
292
- // Pre-build a lookup map from cleaned function name to dependency for O(1) lookups.
293
- // This avoids O(n) linear search in findRelevantDependency which was causing O(n²) performance.
294
- const dependencyByCleanedName = new Map<
295
- string,
296
- (typeof importedExports)[0]
297
- >();
298
- for (const dep of importedExports) {
299
- const cleanedName = cleanFunctionName(dep.name);
300
- if (!dependencyByCleanedName.has(cleanedName)) {
301
- dependencyByCleanedName.set(cleanedName, dep);
315
+ // Pre-build a lookup map from cleaned function name to dependency for O(1) lookups.
316
+ // This avoids O(n) linear search in findRelevantDependency which was causing O(n²) performance.
317
+ const dependencyByCleanedName = new Map<
318
+ string,
319
+ (typeof importedExports)[0]
320
+ >();
321
+ for (const dep of importedExports) {
322
+ const cleanedName = cleanFunctionName(dep.name);
323
+ if (!dependencyByCleanedName.has(cleanedName)) {
324
+ dependencyByCleanedName.set(cleanedName, dep);
325
+ }
302
326
  }
303
- }
304
-
305
- const findRelevantDependency = (functionName: any) => {
306
- return dependencyByCleanedName.get(cleanFunctionName(functionName));
307
- };
308
327
 
309
- const findRelevantDependentDataStructure = (functionName: any) => {
310
- const dependency = findRelevantDependency(functionName);
311
- if (!dependency) return;
328
+ const findRelevantDependency = (functionName: any) => {
329
+ return dependencyByCleanedName.get(cleanFunctionName(functionName));
330
+ };
312
331
 
313
- return dependencySchemas?.[dependency.filePath]?.[dependency.name];
314
- };
332
+ const findRelevantDependentDataStructure = (functionName: any) => {
333
+ const dependency = findRelevantDependency(functionName);
334
+ if (!dependency) return;
315
335
 
316
- const findRelevantDependentAnalysisDataStructure = (functionName: any) => {
317
- const dependency = findRelevantDependency(functionName);
318
- if (!dependency) return;
336
+ return dependencySchemas?.[dependency.filePath]?.[dependency.name];
337
+ };
319
338
 
320
- const dependentAnalysis =
321
- dependentAnalyses[dependency.filePath]?.[dependency.name];
322
- if (!dependentAnalysis?.metadata?.mergedDataStructure) {
323
- return;
324
- }
339
+ const findRelevantDependentAnalysisDataStructure = (functionName: any) => {
340
+ const dependency = findRelevantDependency(functionName);
341
+ if (!dependency) return;
325
342
 
326
- return dependentAnalysis.metadata.mergedDataStructure;
327
- };
343
+ const dependentAnalysis =
344
+ dependentAnalyses[dependency.filePath]?.[dependency.name];
345
+ if (!dependentAnalysis?.metadata?.mergedDataStructure) {
346
+ return;
347
+ }
328
348
 
329
- const findOrCreateDependentSchemas = (dependency: {
330
- filePath?: string;
331
- name: string;
332
- }) => {
333
- const { filePath, name } = dependency;
334
- mergedDataStructure.dependencySchemas[filePath] ||= {};
335
- mergedDataStructure.dependencySchemas[filePath][name] ||= {
336
- signatureSchema: {},
337
- returnValueSchema: {},
349
+ return dependentAnalysis.metadata.mergedDataStructure;
338
350
  };
339
351
 
340
- return mergedDataStructure.dependencySchemas[filePath][name];
341
- };
342
-
343
- const cleanSchema = (
344
- schema: { [key: string]: string },
345
- context?: Record<string, any>,
346
- ) => {
347
- transformationTracer.traceSchemaTransform(
348
- rootScopeName,
349
- 'cleanKnownObjectFunctionsFromMapping',
350
- schema,
351
- cleanKnownObjectFunctionsFromMapping,
352
- context,
353
- );
354
- };
352
+ const findOrCreateDependentSchemas = (dependency: {
353
+ filePath?: string;
354
+ name: string;
355
+ }) => {
356
+ const { filePath, name } = dependency;
357
+ mergedDataStructure.dependencySchemas[filePath] ||= {};
358
+ mergedDataStructure.dependencySchemas[filePath][name] ||= {
359
+ signatureSchema: {},
360
+ returnValueSchema: {},
361
+ };
355
362
 
356
- // Cache translatePath results — the same path is often translated multiple times
357
- // (once per equivalency entry that references it). Avoids redundant
358
- // splitOutsideParenthesesAndArrays calls on long paths.
359
- const translatePathCache = new Map<string, string>();
360
-
361
- const translatePath = (path: string, dependencyName: string) => {
362
- const cacheKey = `${dependencyName}\0${path}`;
363
- const cached = translatePathCache.get(cacheKey);
364
- if (cached !== undefined) return cached;
365
-
366
- let result = path;
367
- if (path.startsWith(dependencyName)) {
368
- const pathParts = splitOutsideParenthesesAndArrays(path);
369
- if (pathParts.length > 1) {
370
- if (pathParts[1].startsWith('functionCallReturnValue')) {
371
- // Check if this function has multiple DIFFERENT type parameters.
372
- // If so, DON'T normalize to returnValue - keep the full path to avoid
373
- // merging different type-parameterized variants together.
374
- const baseName = cleanFunctionName(pathParts[0]);
375
- if (!functionsWithMultipleTypeParams.has(baseName)) {
376
- // functionCallReturnValue immediately follows - normalize to returnValue
377
- result = joinParenthesesAndArrays([
378
- 'returnValue',
379
- ...pathParts.slice(2),
380
- ]);
381
- }
382
- } else if (
383
- pathParts[0].endsWith(')') &&
384
- pathParts[1].startsWith('signature[')
385
- ) {
386
- // Hook-style with signature access
387
- result = joinParenthesesAndArrays(pathParts.slice(1));
388
- }
389
- }
390
- }
363
+ return mergedDataStructure.dependencySchemas[filePath][name];
364
+ };
391
365
 
392
- translatePathCache.set(cacheKey, result);
393
- return result;
394
- };
366
+ const cleanSchema = (
367
+ schema: { [key: string]: string },
368
+ context?: Record<string, any>,
369
+ ) => {
370
+ transformationTracer.traceSchemaTransform(
371
+ rootScopeName,
372
+ 'cleanKnownObjectFunctionsFromMapping',
373
+ schema,
374
+ cleanKnownObjectFunctionsFromMapping,
375
+ context,
376
+ );
377
+ };
395
378
 
396
- const gatherAllEquivalentSchemaPaths = (
397
- functionName: string,
398
- sourceAndUsageEquivalencies: Pick<
399
- DataStructure,
400
- 'sourceEquivalencies' | 'usageEquivalencies'
401
- >,
402
- dataStructure?: Pick<
403
- DataStructure,
404
- 'signatureSchema' | 'returnValueSchema'
405
- >,
406
- ) => {
407
- checkDeadline();
408
- if (!sourceAndUsageEquivalencies) return;
409
-
410
- // Pre-computed normalized schema index cache.
411
- // Avoids repeated splitOutsideParenthesesAndArrays calls and function-name
412
- // normalization for the same schema paths across multiple equivalency iterations.
413
- // The normalization depends on `functionName` (constant per gatherAllEquivalentSchemaPaths call),
414
- // so this cache is scoped to this call.
415
- type NormalizedEntry = { path: string; parts: string[] };
416
- const normalizedSchemaCache = new Map<
417
- object,
418
- {
419
- byFirstPart: Map<string, NormalizedEntry[]>;
420
- }
421
- >();
422
- const getSchemaIndex = (
423
- schema: Record<string, string> | undefined,
424
- ): { byFirstPart: Map<string, NormalizedEntry[]> } => {
425
- if (!schema) return { byFirstPart: new Map() };
426
- const cached = normalizedSchemaCache.get(schema);
427
- if (cached) return cached;
428
- const byFirstPart = new Map<string, NormalizedEntry[]>();
429
- for (const path in schema) {
430
- checkDeadline();
431
- let parts = splitOutsideParenthesesAndArrays(path);
432
- if (parts[0].startsWith(functionName)) {
433
- const baseName = cleanFunctionName(parts[0]);
434
- if (!functionsWithMultipleTypeParams.has(baseName)) {
435
- parts =
436
- parts[1] === 'functionCallReturnValue'
437
- ? ['returnValue', ...parts.slice(2)]
438
- : parts.slice(1);
379
+ // Cache translatePath results — the same path is often translated multiple times
380
+ // (once per equivalency entry that references it). Avoids redundant
381
+ // splitOutsideParenthesesAndArrays calls on long paths.
382
+ const translatePathCache = new Map<string, string>();
383
+
384
+ const translatePath = (path: string, dependencyName: string) => {
385
+ const cacheKey = `${dependencyName}\0${path}`;
386
+ const cached = translatePathCache.get(cacheKey);
387
+ if (cached !== undefined) return cached;
388
+
389
+ let result = path;
390
+ if (path.startsWith(dependencyName)) {
391
+ const pathParts = splitOutsideParenthesesAndArrays(path);
392
+ if (pathParts.length > 1) {
393
+ if (pathParts[1].startsWith('functionCallReturnValue')) {
394
+ // Check if this function has multiple DIFFERENT type parameters.
395
+ // If so, DON'T normalize to returnValue - keep the full path to avoid
396
+ // merging different type-parameterized variants together.
397
+ const baseName = cleanFunctionName(pathParts[0]);
398
+ if (!functionsWithMultipleTypeParams.has(baseName)) {
399
+ // functionCallReturnValue immediately follows - normalize to returnValue
400
+ result = joinParenthesesAndArrays([
401
+ 'returnValue',
402
+ ...pathParts.slice(2),
403
+ ]);
404
+ }
405
+ } else if (
406
+ pathParts[0].endsWith(')') &&
407
+ pathParts[1].startsWith('signature[')
408
+ ) {
409
+ // Hook-style with signature access
410
+ result = joinParenthesesAndArrays(pathParts.slice(1));
439
411
  }
440
412
  }
441
- const entry: NormalizedEntry = { path, parts };
442
- // Index by the base of the first part (before any function call args)
443
- const firstPart = parts[0] ?? '';
444
- const parenIdx = firstPart.indexOf('(');
445
- const firstPartBase =
446
- parenIdx >= 0 ? firstPart.slice(0, parenIdx) : firstPart;
447
- let bucket = byFirstPart.get(firstPartBase);
448
- if (!bucket) {
449
- bucket = [];
450
- byFirstPart.set(firstPartBase, bucket);
451
- }
452
- bucket.push(entry);
453
413
  }
454
- const result = { byFirstPart };
455
- normalizedSchemaCache.set(schema, result);
414
+
415
+ translatePathCache.set(cacheKey, result);
456
416
  return result;
457
417
  };
458
418
 
459
- const findOrCreateEquivalentSchemaPathsEntry = (
460
- allPaths: { path: string; functionName?: string }[],
419
+ const gatherAllEquivalentSchemaPaths = (
420
+ functionName: string,
421
+ sourceAndUsageEquivalencies: Pick<
422
+ DataStructure,
423
+ 'sourceEquivalencies' | 'usageEquivalencies'
424
+ >,
425
+ dataStructure?: Pick<
426
+ DataStructure,
427
+ 'signatureSchema' | 'returnValueSchema'
428
+ >,
461
429
  ) => {
462
- const equivalentRoots = allPaths
463
- .filter(
464
- (p) =>
465
- p.functionName === rootScopeName ||
466
- !!findRelevantDependency(p.functionName),
467
- )
468
- .map((p) => ({
469
- schemaRootPath: p.path,
470
- function:
471
- p.functionName === rootScopeName
472
- ? undefined
473
- : findRelevantDependency(p.functionName),
474
- }));
475
-
476
- let equivalentSchemaPathsEntry:
477
- | (typeof equivalentSchemaPaths)[0]
478
- | undefined;
479
-
480
- // Collect the signature indices from the new roots we want to add
481
- const newRootSignatureIndices = new Set<number>();
482
- for (const root of equivalentRoots) {
483
- const idx = extractSignatureIndex(root.schemaRootPath);
484
- if (idx !== undefined) {
485
- newRootSignatureIndices.add(idx);
430
+ checkDeadline();
431
+ if (!sourceAndUsageEquivalencies) return;
432
+
433
+ // Pre-computed normalized schema index cache.
434
+ // Avoids repeated splitOutsideParenthesesAndArrays calls and function-name
435
+ // normalization for the same schema paths across multiple equivalency iterations.
436
+ // The normalization depends on `functionName` (constant per gatherAllEquivalentSchemaPaths call),
437
+ // so this cache is scoped to this call.
438
+ type NormalizedEntry = { path: string; parts: string[] };
439
+ const normalizedSchemaCache = new Map<
440
+ object,
441
+ {
442
+ byFirstPart: Map<string, NormalizedEntry[]>;
486
443
  }
487
- }
444
+ >();
445
+ const getSchemaIndex = (
446
+ schema: Record<string, string> | undefined,
447
+ ): { byFirstPart: Map<string, NormalizedEntry[]> } => {
448
+ if (!schema) return { byFirstPart: new Map() };
449
+ const cached = normalizedSchemaCache.get(schema);
450
+ if (cached) return cached;
451
+ const byFirstPart = new Map<string, NormalizedEntry[]>();
452
+ for (const path in schema) {
453
+ checkDeadline();
454
+ let parts = splitOutsideParenthesesAndArrays(path);
455
+ if (parts[0].startsWith(functionName)) {
456
+ const baseName = cleanFunctionName(parts[0]);
457
+ if (!functionsWithMultipleTypeParams.has(baseName)) {
458
+ parts =
459
+ parts[1] === 'functionCallReturnValue'
460
+ ? ['returnValue', ...parts.slice(2)]
461
+ : parts.slice(1);
462
+ }
463
+ }
464
+ const entry: NormalizedEntry = { path, parts };
465
+ // Index by the base of the first part (before any function call args)
466
+ const firstPart = parts[0] ?? '';
467
+ const parenIdx = firstPart.indexOf('(');
468
+ const firstPartBase =
469
+ parenIdx >= 0 ? firstPart.slice(0, parenIdx) : firstPart;
470
+ let bucket = byFirstPart.get(firstPartBase);
471
+ if (!bucket) {
472
+ bucket = [];
473
+ byFirstPart.set(firstPartBase, bucket);
474
+ }
475
+ bucket.push(entry);
476
+ }
477
+ const result = { byFirstPart };
478
+ normalizedSchemaCache.set(schema, result);
479
+ return result;
480
+ };
488
481
 
489
- // Use espIndex Map for O(1) lookup instead of O(E) linear search.
490
- // Falls back to linear search only when Map hit has a signature index conflict.
491
- for (const pathInfo of allPaths) {
492
- checkDeadline();
493
- if (equivalentSchemaPathsEntry) break;
494
- const candidate = espIndex.get(
495
- espIndexKey(pathInfo.path, pathInfo.functionName),
496
- );
497
- if (!candidate) continue;
498
-
499
- // Verify no signature index conflict with the candidate entry
500
- if (newRootSignatureIndices.size > 0) {
501
- const existingIndicesByFunction = new Map<string, Set<number>>();
502
- for (const er of candidate.equivalentRoots) {
503
- const funcKey = er.function
504
- ? `${er.function.name}::${er.function.filePath}`
505
- : '__self__';
506
- const idx = extractSignatureIndex(er.schemaRootPath);
507
- if (idx !== undefined) {
508
- if (!existingIndicesByFunction.has(funcKey)) {
509
- existingIndicesByFunction.set(funcKey, new Set());
482
+ const findOrCreateEquivalentSchemaPathsEntry = (
483
+ allPaths: { path: string; functionName?: string }[],
484
+ ) => {
485
+ const equivalentRoots = allPaths
486
+ .filter(
487
+ (p) =>
488
+ p.functionName === rootScopeName ||
489
+ !!findRelevantDependency(p.functionName),
490
+ )
491
+ .map((p) => ({
492
+ schemaRootPath: p.path,
493
+ function:
494
+ p.functionName === rootScopeName
495
+ ? undefined
496
+ : findRelevantDependency(p.functionName),
497
+ }));
498
+
499
+ let equivalentSchemaPathsEntry:
500
+ | (typeof equivalentSchemaPaths)[0]
501
+ | undefined;
502
+
503
+ // Collect the signature indices from the new roots we want to add
504
+ const newRootSignatureIndices = new Set<number>();
505
+ for (const root of equivalentRoots) {
506
+ const idx = extractSignatureIndex(root.schemaRootPath);
507
+ if (idx !== undefined) {
508
+ newRootSignatureIndices.add(idx);
509
+ }
510
+ }
511
+
512
+ // Use espIndex Map for O(1) lookup instead of O(E) linear search.
513
+ // Falls back to linear search only when Map hit has a signature index conflict.
514
+ for (const pathInfo of allPaths) {
515
+ checkDeadline();
516
+ if (equivalentSchemaPathsEntry) break;
517
+ const candidate = espIndex.get(
518
+ espIndexKey(pathInfo.path, pathInfo.functionName),
519
+ );
520
+ if (!candidate) continue;
521
+
522
+ // Verify no signature index conflict with the candidate entry
523
+ if (newRootSignatureIndices.size > 0) {
524
+ const existingIndicesByFunction = new Map<string, Set<number>>();
525
+ for (const er of candidate.equivalentRoots) {
526
+ const funcKey = er.function
527
+ ? `${er.function.name}::${er.function.filePath}`
528
+ : '__self__';
529
+ const idx = extractSignatureIndex(er.schemaRootPath);
530
+ if (idx !== undefined) {
531
+ if (!existingIndicesByFunction.has(funcKey)) {
532
+ existingIndicesByFunction.set(funcKey, new Set());
533
+ }
534
+ existingIndicesByFunction.get(funcKey)!.add(idx);
510
535
  }
511
- existingIndicesByFunction.get(funcKey)!.add(idx);
512
536
  }
513
- }
514
537
 
515
- let hasConflict = false;
516
- for (const newRoot of equivalentRoots) {
517
- const funcKey = newRoot.function
518
- ? `${newRoot.function.name}::${newRoot.function.filePath}`
519
- : '__self__';
520
- const newIdx = extractSignatureIndex(newRoot.schemaRootPath);
521
- if (newIdx !== undefined) {
522
- const existingIndices = existingIndicesByFunction.get(funcKey);
523
- if (existingIndices && existingIndices.size > 0) {
524
- if (!existingIndices.has(newIdx)) {
525
- hasConflict = true;
526
- break;
538
+ let hasConflict = false;
539
+ for (const newRoot of equivalentRoots) {
540
+ const funcKey = newRoot.function
541
+ ? `${newRoot.function.name}::${newRoot.function.filePath}`
542
+ : '__self__';
543
+ const newIdx = extractSignatureIndex(newRoot.schemaRootPath);
544
+ if (newIdx !== undefined) {
545
+ const existingIndices = existingIndicesByFunction.get(funcKey);
546
+ if (existingIndices && existingIndices.size > 0) {
547
+ if (!existingIndices.has(newIdx)) {
548
+ hasConflict = true;
549
+ break;
550
+ }
527
551
  }
528
552
  }
529
553
  }
554
+
555
+ if (hasConflict) continue;
530
556
  }
531
557
 
532
- if (hasConflict) continue;
558
+ equivalentSchemaPathsEntry = candidate;
533
559
  }
534
560
 
535
- equivalentSchemaPathsEntry = candidate;
536
- }
537
-
538
- if (!equivalentSchemaPathsEntry) {
539
- // Before creating a new entry, filter out roots that have conflicting
540
- // signature indices from the same function. An entry should never contain
541
- // roots with different signature indices from the same function.
542
- // This prevents the bug where signature[1], signature[2], signature[4]
543
- // all get merged together due to incorrect sourceEquivalencies.
544
- let filteredRoots = equivalentRoots;
545
- if (newRootSignatureIndices.size > 1) {
546
- // There are multiple signature indices - we need to filter to keep only
547
- // one consistent set. We'll keep the roots that match the PRIMARY index
548
- // (the first signature index we encounter from self, or the lowest index).
549
-
550
- // First, determine the primary index - prefer the self root's index
551
- let primaryIndex: number | undefined;
552
- for (const root of equivalentRoots) {
553
- if (!root.function) {
554
- // This is a self root
555
- const idx = extractSignatureIndex(root.schemaRootPath);
556
- if (idx !== undefined) {
557
- primaryIndex = idx;
558
- break;
561
+ if (!equivalentSchemaPathsEntry) {
562
+ // Before creating a new entry, filter out roots that have conflicting
563
+ // signature indices from the same function. An entry should never contain
564
+ // roots with different signature indices from the same function.
565
+ // This prevents the bug where signature[1], signature[2], signature[4]
566
+ // all get merged together due to incorrect sourceEquivalencies.
567
+ let filteredRoots = equivalentRoots;
568
+ if (newRootSignatureIndices.size > 1) {
569
+ // There are multiple signature indices - we need to filter to keep only
570
+ // one consistent set. We'll keep the roots that match the PRIMARY index
571
+ // (the first signature index we encounter from self, or the lowest index).
572
+
573
+ // First, determine the primary index - prefer the self root's index
574
+ let primaryIndex: number | undefined;
575
+ for (const root of equivalentRoots) {
576
+ if (!root.function) {
577
+ // This is a self root
578
+ const idx = extractSignatureIndex(root.schemaRootPath);
579
+ if (idx !== undefined) {
580
+ primaryIndex = idx;
581
+ break;
582
+ }
559
583
  }
560
584
  }
561
- }
562
- // If no self root has a signature index, use the lowest index
563
- if (primaryIndex === undefined) {
564
- primaryIndex = Math.min(...newRootSignatureIndices);
585
+ // If no self root has a signature index, use the lowest index
586
+ if (primaryIndex === undefined) {
587
+ primaryIndex = Math.min(...newRootSignatureIndices);
588
+ }
589
+
590
+ // Filter roots: keep if no signature index OR signature index matches primary
591
+ filteredRoots = equivalentRoots.filter((root) => {
592
+ const idx = extractSignatureIndex(root.schemaRootPath);
593
+ return idx === undefined || idx === primaryIndex;
594
+ });
565
595
  }
566
596
 
567
- // Filter roots: keep if no signature index OR signature index matches primary
568
- filteredRoots = equivalentRoots.filter((root) => {
569
- const idx = extractSignatureIndex(root.schemaRootPath);
570
- return idx === undefined || idx === primaryIndex;
571
- });
597
+ equivalentSchemaPathsEntry = {
598
+ equivalentRoots: filteredRoots,
599
+ equivalentPostfixes: {},
600
+ };
601
+ equivalentSchemaPaths.push(equivalentSchemaPathsEntry);
602
+ } else {
603
+ equivalentSchemaPathsEntry.equivalentRoots.push(...equivalentRoots);
572
604
  }
573
605
 
574
- equivalentSchemaPathsEntry = {
575
- equivalentRoots: filteredRoots,
576
- equivalentPostfixes: {},
577
- };
578
- equivalentSchemaPaths.push(equivalentSchemaPathsEntry);
579
- } else {
580
- equivalentSchemaPathsEntry.equivalentRoots.push(...equivalentRoots);
581
- }
582
-
583
- // Deduplicate roots using a Set for O(n) instead of O(n²)
584
- const seenRoots = new Set<string>();
585
- equivalentSchemaPathsEntry.equivalentRoots =
586
- equivalentSchemaPathsEntry.equivalentRoots.filter((er) => {
587
- const key = er.schemaRootPath + '::' + (er.function?.name ?? '');
588
- if (seenRoots.has(key)) return false;
589
- seenRoots.add(key);
590
- return true;
591
- });
606
+ // Deduplicate roots using a Set for O(n) instead of O(n²)
607
+ const seenRoots = new Set<string>();
608
+ equivalentSchemaPathsEntry.equivalentRoots =
609
+ equivalentSchemaPathsEntry.equivalentRoots.filter((er) => {
610
+ const key = er.schemaRootPath + '::' + (er.function?.name ?? '');
611
+ if (seenRoots.has(key)) return false;
612
+ seenRoots.add(key);
613
+ return true;
614
+ });
592
615
 
593
- // Keep the espIndex in sync after adding/deduplicating roots
594
- updateEspIndex(equivalentSchemaPathsEntry);
616
+ // Keep the espIndex in sync after adding/deduplicating roots
617
+ updateEspIndex(equivalentSchemaPathsEntry);
595
618
 
596
- return equivalentSchemaPathsEntry;
597
- };
619
+ return equivalentSchemaPathsEntry;
620
+ };
598
621
 
599
- // Helper to extract function name from a path that starts with a function call.
600
- // e.g., 'ScenarioViewer().signature[0].scenario' -> 'ScenarioViewer'
601
- // Returns undefined if the path doesn't start with a function call or the function isn't a dependency.
602
- const extractFunctionNameFromPath = (path: string): string | undefined => {
603
- const parts = splitOutsideParenthesesAndArrays(path);
604
- if (parts.length > 0 && parts[0].endsWith(')')) {
605
- // Extract the function name without the () suffix and type params
606
- const funcCallPart = parts[0];
607
- const funcName = cleanFunctionName(funcCallPart.replace(/\(\)$/, ''));
608
- // Check if this function is a dependency
609
- if (findRelevantDependency(funcName)) {
610
- return funcName;
622
+ // Helper to extract function name from a path that starts with a function call.
623
+ // e.g., 'ScenarioViewer().signature[0].scenario' -> 'ScenarioViewer'
624
+ // Returns undefined if the path doesn't start with a function call or the function isn't a dependency.
625
+ const extractFunctionNameFromPath = (
626
+ path: string,
627
+ ): string | undefined => {
628
+ const parts = splitOutsideParenthesesAndArrays(path);
629
+ if (parts.length > 0 && parts[0].endsWith(')')) {
630
+ // Extract the function name without the () suffix and type params
631
+ const funcCallPart = parts[0];
632
+ const funcName = cleanFunctionName(funcCallPart.replace(/\(\)$/, ''));
633
+ // Check if this function is a dependency
634
+ if (findRelevantDependency(funcName)) {
635
+ return funcName;
636
+ }
611
637
  }
612
- }
613
- return undefined;
614
- };
638
+ return undefined;
639
+ };
615
640
 
616
- const allEquivalencies = [
617
- sourceAndUsageEquivalencies.usageEquivalencies,
618
- sourceAndUsageEquivalencies.sourceEquivalencies,
619
- ].filter(Boolean);
641
+ const allEquivalencies = [
642
+ sourceAndUsageEquivalencies.usageEquivalencies,
643
+ sourceAndUsageEquivalencies.sourceEquivalencies,
644
+ ].filter(Boolean);
620
645
 
621
- for (const equivalencies of allEquivalencies) {
622
- const schemaPathEntries = Object.entries(equivalencies);
623
- for (const [schemaPath, usages] of schemaPathEntries) {
624
- checkDeadline();
646
+ // Global dedup across ALL equivalency entries. The same (scope, targetPath)
647
+ // pair often appears in 30-50 different source entries (e.g., every variable
648
+ // that flows through loadView references the same 50 target paths).
649
+ // Processing these redundantly accounts for 96% of work in the gather phase.
650
+ const globalSeenTargets = new Set<string>();
625
651
 
626
- // Skip equivalency entries whose source path is a Set/Map membership operation.
627
- // Patterns like `.has(articleId)`, `.delete(articleId)`, `.add(articleId)` on
628
- // Sets/Maps represent membership checks, not meaningful data flow for schema generation.
629
- // In the Margo LibraryPage case, these account for 74% of all equivalency targets
630
- // (19,444 of 26,340) and cause a combinatorial explosion in the merge.
631
- if (isCollectionMethodPath(schemaPath)) continue;
632
-
633
- // First, check if the raw schemaPath starts with a function call to a dependency.
634
- // If so, use that dependency name for translation (so translatePath can strip the prefix).
635
- const extractedFuncName = extractFunctionNameFromPath(schemaPath);
636
- const effectiveFunctionName = extractedFuncName || functionName;
637
- const translatedPath = translatePath(schemaPath, effectiveFunctionName);
638
-
639
- const allPathsRaw: { path: string; functionName?: string }[] = [
640
- { path: translatedPath, functionName: effectiveFunctionName },
641
- ...usages
642
- .filter((u) => !isCollectionMethodPath(u.schemaPath))
643
- .map((u) => ({
644
- path: translatePath(u.schemaPath, u.scopeNodeName),
645
- functionName: u.scopeNodeName,
646
- })),
647
- ].filter((pathInfo) => !pathInfo.path.includes('.map('));
648
-
649
- // Deduplicate by translated path + function name.
650
- // Multiple call variants (e.g., loadView(viewKey(null,null)) vs loadView(viewKey(newTag,newCol)))
651
- // translate to the same path after stripping arguments. Processing duplicates
652
- // creates O(n²) work in the schema matching loops below.
653
- const seenPathKeys = new Set<string>();
654
- const allPaths = allPathsRaw.filter((p) => {
655
- const key = `${p.functionName ?? ''}::${p.path}`;
656
- if (seenPathKeys.has(key)) return false;
657
- seenPathKeys.add(key);
658
- return true;
659
- });
652
+ for (const equivalencies of allEquivalencies) {
653
+ const schemaPathEntries = Object.entries(equivalencies);
654
+ for (const [schemaPath, usages] of schemaPathEntries) {
655
+ checkDeadline();
660
656
 
661
- // Fix 38: Derive base paths from property access paths.
662
- // When we have equivalent paths like:
663
- // Parent: signature[0].scenarios[].name
664
- // Child: signature[0].selectedScenario.name
665
- // We want to derive the base paths by finding the common suffix:
666
- // Common suffix: .name
667
- // Parent base: signature[0].scenarios[]
668
- // Child base: signature[0].selectedScenario
669
- // This allows the merge to find nested child schema fields under the base prop.
670
-
671
- // Find child signature paths (paths from child components)
672
- const childPaths = allPaths.filter(
673
- (p) =>
674
- p.functionName &&
675
- p.functionName !== rootScopeName &&
676
- p.functionName !== effectiveFunctionName,
677
- );
678
- // Find parent paths (paths from this component)
679
- const parentPaths = allPaths.filter(
680
- (p) =>
681
- !p.functionName ||
682
- p.functionName === rootScopeName ||
683
- p.functionName === effectiveFunctionName,
684
- );
657
+ // Skip equivalency entries whose source path is a Set/Map membership operation.
658
+ // Patterns like `.has(articleId)`, `.delete(articleId)`, `.add(articleId)` on
659
+ // Sets/Maps represent membership checks, not meaningful data flow for schema generation.
660
+ // In the Margo LibraryPage case, these account for 74% of all equivalency targets
661
+ // (19,444 of 26,340) and cause a combinatorial explosion in the merge.
662
+ if (isCollectionMethodPath(schemaPath)) continue;
663
+
664
+ // Skip paths with inline object literals in function call arguments.
665
+ // These are call-site snapshots (e.g., setUndoEntry({ label: '...', undo: ... }))
666
+ // that embed source code text as path strings. They're expensive to parse
667
+ // and don't contribute useful schema information.
668
+ if (hasInlineObjectArg(schemaPath)) continue;
669
+
670
+ // First, check if the raw schemaPath starts with a function call to a dependency.
671
+ // If so, use that dependency name for translation (so translatePath can strip the prefix).
672
+ const extractedFuncName = extractFunctionNameFromPath(schemaPath);
673
+ const effectiveFunctionName = extractedFuncName || functionName;
674
+ const translatedPath = translatePath(
675
+ schemaPath,
676
+ effectiveFunctionName,
677
+ );
685
678
 
686
- const derivedBasePaths: { path: string; functionName?: string }[] = [];
687
- const allPathSet = new Set(allPaths.map((p) => p.path));
688
- const derivedBasePathSet = new Set<string>();
679
+ const allPathsRaw: { path: string; functionName?: string }[] = [
680
+ { path: translatedPath, functionName: effectiveFunctionName },
681
+ ...usages
682
+ .filter((u) => !isCollectionMethodPath(u.schemaPath))
683
+ .map((u) => ({
684
+ path: translatePath(u.schemaPath, u.scopeNodeName),
685
+ functionName: u.scopeNodeName,
686
+ })),
687
+ ].filter((pathInfo) => !pathInfo.path.includes('.map('));
688
+
689
+ // Deduplicate by translated path + function name, both within this entry
690
+ // AND across all entries. The same target path appears in 30-50 different
691
+ // source entries (every variable flowing through loadView references the same
692
+ // 50 target paths). Without global dedup, we process 5,533 targets instead of 217.
693
+ const allPaths = allPathsRaw.filter((p) => {
694
+ const key = `${p.functionName ?? ''}::${p.path}`;
695
+ if (globalSeenTargets.has(key)) return false;
696
+ globalSeenTargets.add(key);
697
+ return true;
698
+ });
689
699
 
690
- // For each child path, find its equivalent parent path and derive bases
691
- for (const childPathInfo of childPaths) {
692
- checkDeadline();
693
- const childParts = splitOutsideParenthesesAndArrays(
694
- childPathInfo.path,
700
+ // Fix 38: Derive base paths from property access paths.
701
+ // When we have equivalent paths like:
702
+ // Parent: signature[0].scenarios[].name
703
+ // Child: signature[0].selectedScenario.name
704
+ // We want to derive the base paths by finding the common suffix:
705
+ // Common suffix: .name
706
+ // Parent base: signature[0].scenarios[]
707
+ // Child base: signature[0].selectedScenario
708
+ // This allows the merge to find nested child schema fields under the base prop.
709
+
710
+ // Find child signature paths (paths from child components)
711
+ const childPaths = allPaths.filter(
712
+ (p) =>
713
+ p.functionName &&
714
+ p.functionName !== rootScopeName &&
715
+ p.functionName !== effectiveFunctionName,
716
+ );
717
+ // Find parent paths (paths from this component)
718
+ const parentPaths = allPaths.filter(
719
+ (p) =>
720
+ !p.functionName ||
721
+ p.functionName === rootScopeName ||
722
+ p.functionName === effectiveFunctionName,
695
723
  );
696
724
 
697
- // Look for a parent path that shares a common suffix with this child path
698
- for (const parentPathInfo of parentPaths) {
699
- const parentParts = splitOutsideParenthesesAndArrays(
700
- parentPathInfo.path,
701
- );
725
+ const derivedBasePaths: { path: string; functionName?: string }[] =
726
+ [];
727
+ const allPathSet = new Set(allPaths.map((p) => p.path));
728
+ const derivedBasePathSet = new Set<string>();
702
729
 
703
- // Find the common suffix (from the end)
704
- let commonSuffixLength = 0;
705
- const minLen = Math.min(childParts.length, parentParts.length);
706
- for (let i = 1; i <= minLen; i++) {
707
- if (
708
- childParts[childParts.length - i] ===
709
- parentParts[parentParts.length - i]
710
- ) {
711
- commonSuffixLength = i;
712
- } else {
713
- break;
714
- }
715
- }
730
+ // For each child path, find its equivalent parent path and derive bases
731
+ for (const childPathInfo of childPaths) {
732
+ checkDeadline();
733
+ const childParts = splitOutsideParenthesesAndArrays(
734
+ childPathInfo.path,
735
+ );
716
736
 
717
- // If there's a common suffix and both paths have more parts than the suffix
718
- if (
719
- commonSuffixLength > 0 &&
720
- childParts.length > commonSuffixLength &&
721
- parentParts.length > commonSuffixLength
722
- ) {
723
- const childBaseParts = childParts.slice(
724
- 0,
725
- childParts.length - commonSuffixLength,
737
+ // Look for a parent path that shares a common suffix with this child path
738
+ for (const parentPathInfo of parentPaths) {
739
+ const parentParts = splitOutsideParenthesesAndArrays(
740
+ parentPathInfo.path,
726
741
  );
727
- const parentBaseParts = parentParts.slice(
728
- 0,
729
- parentParts.length - commonSuffixLength,
730
- );
731
-
732
- // Only derive if BOTH paths look like signature paths.
733
- // This ensures we're handling JSX child-to-parent prop mappings,
734
- // not other complex equivalencies like function call returns.
735
- const isChildSignaturePath =
736
- childBaseParts[0]?.startsWith('signature[') ||
737
- (childBaseParts[0]?.endsWith(')') &&
738
- childBaseParts[1]?.startsWith('signature['));
739
- const isParentSignaturePath =
740
- parentBaseParts[0]?.startsWith('signature[');
741
-
742
- if (isChildSignaturePath && isParentSignaturePath) {
743
- const childBase = joinParenthesesAndArrays(childBaseParts);
744
- const parentBase = joinParenthesesAndArrays(parentBaseParts);
745
-
746
- // Only derive if:
747
- // 1. Parent has array iteration (e.g., scenarios[]) and child does NOT
748
- // 2. Bases are different
749
- // 3. Child base is NOT just "signature[N]" (too generic - every component has this)
750
- // We only want specific prop paths like "signature[0].selectedScenario"
751
- // This targets array-to-object mappings like scenarios[] -> selectedScenario
752
- const parentHasArrayIterator = parentBase.includes('[]');
753
- const childHasArrayIterator = childBase.includes('[]');
754
-
755
- // Skip if child base is just the generic signature marker (e.g., "signature[0]")
756
- const childBaseIsGenericSignature = /^signature\[\d+\]$/.test(
757
- childBase,
758
- );
759
742
 
743
+ // Find the common suffix (from the end)
744
+ let commonSuffixLength = 0;
745
+ const minLen = Math.min(childParts.length, parentParts.length);
746
+ for (let i = 1; i <= minLen; i++) {
760
747
  if (
761
- childBase !== parentBase &&
762
- parentHasArrayIterator &&
763
- !childHasArrayIterator &&
764
- !childBaseIsGenericSignature
748
+ childParts[childParts.length - i] ===
749
+ parentParts[parentParts.length - i]
765
750
  ) {
766
- // Add child base if not already present (O(1) Set lookup)
767
- if (
768
- !allPathSet.has(childBase) &&
769
- !derivedBasePathSet.has(childBase)
770
- ) {
771
- derivedBasePaths.push({
772
- path: childBase,
773
- functionName: childPathInfo.functionName,
774
- });
775
- derivedBasePathSet.add(childBase);
776
- }
751
+ commonSuffixLength = i;
752
+ } else {
753
+ break;
754
+ }
755
+ }
756
+
757
+ // If there's a common suffix and both paths have more parts than the suffix
758
+ if (
759
+ commonSuffixLength > 0 &&
760
+ childParts.length > commonSuffixLength &&
761
+ parentParts.length > commonSuffixLength
762
+ ) {
763
+ const childBaseParts = childParts.slice(
764
+ 0,
765
+ childParts.length - commonSuffixLength,
766
+ );
767
+ const parentBaseParts = parentParts.slice(
768
+ 0,
769
+ parentParts.length - commonSuffixLength,
770
+ );
771
+
772
+ // Only derive if BOTH paths look like signature paths.
773
+ // This ensures we're handling JSX child-to-parent prop mappings,
774
+ // not other complex equivalencies like function call returns.
775
+ const isChildSignaturePath =
776
+ childBaseParts[0]?.startsWith('signature[') ||
777
+ (childBaseParts[0]?.endsWith(')') &&
778
+ childBaseParts[1]?.startsWith('signature['));
779
+ const isParentSignaturePath =
780
+ parentBaseParts[0]?.startsWith('signature[');
781
+
782
+ if (isChildSignaturePath && isParentSignaturePath) {
783
+ const childBase = joinParenthesesAndArrays(childBaseParts);
784
+ const parentBase = joinParenthesesAndArrays(parentBaseParts);
785
+
786
+ // Only derive if:
787
+ // 1. Parent has array iteration (e.g., scenarios[]) and child does NOT
788
+ // 2. Bases are different
789
+ // 3. Child base is NOT just "signature[N]" (too generic - every component has this)
790
+ // We only want specific prop paths like "signature[0].selectedScenario"
791
+ // This targets array-to-object mappings like scenarios[] -> selectedScenario
792
+ const parentHasArrayIterator = parentBase.includes('[]');
793
+ const childHasArrayIterator = childBase.includes('[]');
794
+
795
+ // Skip if child base is just the generic signature marker (e.g., "signature[0]")
796
+ const childBaseIsGenericSignature = /^signature\[\d+\]$/.test(
797
+ childBase,
798
+ );
777
799
 
778
- // Add parent base if not already present (O(1) Set lookup)
779
800
  if (
780
- !allPathSet.has(parentBase) &&
781
- !derivedBasePathSet.has(parentBase)
801
+ childBase !== parentBase &&
802
+ parentHasArrayIterator &&
803
+ !childHasArrayIterator &&
804
+ !childBaseIsGenericSignature
782
805
  ) {
783
- derivedBasePaths.push({
784
- path: parentBase,
785
- functionName: parentPathInfo.functionName,
786
- });
787
- derivedBasePathSet.add(parentBase);
806
+ // Add child base if not already present (O(1) Set lookup)
807
+ if (
808
+ !allPathSet.has(childBase) &&
809
+ !derivedBasePathSet.has(childBase)
810
+ ) {
811
+ derivedBasePaths.push({
812
+ path: childBase,
813
+ functionName: childPathInfo.functionName,
814
+ });
815
+ derivedBasePathSet.add(childBase);
816
+ }
817
+
818
+ // Add parent base if not already present (O(1) Set lookup)
819
+ if (
820
+ !allPathSet.has(parentBase) &&
821
+ !derivedBasePathSet.has(parentBase)
822
+ ) {
823
+ derivedBasePaths.push({
824
+ path: parentBase,
825
+ functionName: parentPathInfo.functionName,
826
+ });
827
+ derivedBasePathSet.add(parentBase);
828
+ }
788
829
  }
789
830
  }
790
831
  }
791
832
  }
792
833
  }
793
- }
794
834
 
795
- allPaths.push(...derivedBasePaths);
835
+ allPaths.push(...derivedBasePaths);
796
836
 
797
- const entry = findOrCreateEquivalentSchemaPathsEntry(allPaths);
837
+ const entry = findOrCreateEquivalentSchemaPathsEntry(allPaths);
798
838
 
799
- // Trace equivalency gathering - helps debug why paths may not be connected
800
- if (allPaths.length > 1) {
801
- transformationTracer.operation(rootScopeName, {
802
- operation: 'gatherEquivalency',
803
- stage: 'gathering',
804
- path: translatedPath,
805
- context: {
806
- sourceFunction: functionName,
807
- equivalentPaths: allPaths.map((p) => ({
808
- path: p.path,
809
- function: p.functionName,
810
- })),
811
- equivalentRoots: entry.equivalentRoots.map((r) => ({
812
- path: r.schemaRootPath,
813
- function: r.function?.name,
814
- })),
815
- },
816
- });
817
- }
818
- for (const equivalentRoot of entry.equivalentRoots) {
819
- checkDeadline();
820
- const dataStructures =
821
- equivalentRoot.function &&
822
- equivalentRoot.function.name !== rootScopeName
823
- ? [
824
- findRelevantDependentDataStructure(
825
- equivalentRoot.function.name,
826
- ),
827
- findRelevantDependentAnalysisDataStructure(
828
- equivalentRoot.function.name,
829
- ),
830
- ]
831
- : [dataStructure];
832
-
833
- // Determine if this is a signature schema path.
834
- // The path might be 'signature[0]...' directly, or 'FuncName().signature[0]...' if it has a function prefix.
835
- const schemaRootParts = splitOutsideParenthesesAndArrays(
836
- equivalentRoot.schemaRootPath,
837
- );
838
- const isSignaturePath =
839
- equivalentRoot.schemaRootPath.startsWith('signature[') ||
840
- (schemaRootParts[0]?.endsWith(')') &&
841
- schemaRootParts[1]?.startsWith('signature['));
842
-
843
- const schemas = dataStructures.map((dataStructure) =>
844
- isSignaturePath
845
- ? dataStructure?.signatureSchema
846
- : dataStructure?.returnValueSchema,
847
- );
839
+ // Trace equivalency gathering - helps debug why paths may not be connected
840
+ if (allPaths.length > 1) {
841
+ transformationTracer.operation(rootScopeName, {
842
+ operation: 'gatherEquivalency',
843
+ stage: 'gathering',
844
+ path: translatedPath,
845
+ context: {
846
+ sourceFunction: functionName,
847
+ equivalentPaths: allPaths.map((p) => ({
848
+ path: p.path,
849
+ function: p.functionName,
850
+ })),
851
+ equivalentRoots: entry.equivalentRoots.map((r) => ({
852
+ path: r.schemaRootPath,
853
+ function: r.function?.name,
854
+ })),
855
+ },
856
+ });
857
+ }
858
+ for (const equivalentRoot of entry.equivalentRoots) {
859
+ checkDeadline();
860
+ const dataStructures =
861
+ equivalentRoot.function &&
862
+ equivalentRoot.function.name !== rootScopeName
863
+ ? [
864
+ findRelevantDependentDataStructure(
865
+ equivalentRoot.function.name,
866
+ ),
867
+ findRelevantDependentAnalysisDataStructure(
868
+ equivalentRoot.function.name,
869
+ ),
870
+ ]
871
+ : [dataStructure];
872
+
873
+ // Determine if this is a signature schema path.
874
+ // The path might be 'signature[0]...' directly, or 'FuncName().signature[0]...' if it has a function prefix.
875
+ const schemaRootParts = splitOutsideParenthesesAndArrays(
876
+ equivalentRoot.schemaRootPath,
877
+ );
878
+ const isSignaturePath =
879
+ equivalentRoot.schemaRootPath.startsWith('signature[') ||
880
+ (schemaRootParts[0]?.endsWith(')') &&
881
+ schemaRootParts[1]?.startsWith('signature['));
882
+
883
+ const schemas = dataStructures.map((dataStructure) =>
884
+ isSignaturePath
885
+ ? dataStructure?.signatureSchema
886
+ : dataStructure?.returnValueSchema,
887
+ );
848
888
 
849
- let pathParts = splitOutsideParenthesesAndArrays(
850
- equivalentRoot.schemaRootPath,
851
- );
889
+ let pathParts = splitOutsideParenthesesAndArrays(
890
+ equivalentRoot.schemaRootPath,
891
+ );
852
892
 
853
- // Fix: When processing a child component's schema, the schemaRootPath has the function
854
- // prefix (e.g., 'ScenarioViewer().signature[0].scenario'), but the child's schema paths
855
- // don't have that prefix (e.g., 'signature[0].scenario.metadata.screenshotPaths').
856
- // Strip the function prefix from pathParts so they can match.
857
- if (
858
- equivalentRoot.function &&
859
- pathParts[0].endsWith(')') &&
860
- pathParts[1]?.startsWith('signature[')
861
- ) {
862
- pathParts = pathParts.slice(1);
863
- }
893
+ // Fix: When processing a child component's schema, the schemaRootPath has the function
894
+ // prefix (e.g., 'ScenarioViewer().signature[0].scenario'), but the child's schema paths
895
+ // don't have that prefix (e.g., 'signature[0].scenario.metadata.screenshotPaths').
896
+ // Strip the function prefix from pathParts so they can match.
897
+ if (
898
+ equivalentRoot.function &&
899
+ pathParts[0].endsWith(')') &&
900
+ pathParts[1]?.startsWith('signature[')
901
+ ) {
902
+ pathParts = pathParts.slice(1);
903
+ }
864
904
 
865
- for (const schema of schemas) {
866
- // Use pre-computed index to only iterate schema entries whose
867
- // normalized first part matches pathParts[0], instead of all entries.
868
- const schemaIndex = getSchemaIndex(schema);
869
- const lookupPart = pathParts[0] ?? '';
870
- const lookupParenIdx = lookupPart.indexOf('(');
871
- const lookupBase =
872
- lookupParenIdx >= 0
873
- ? lookupPart.slice(0, lookupParenIdx)
874
- : lookupPart;
875
- const candidates = schemaIndex.byFirstPart.get(lookupBase) || [];
876
- for (const {
877
- path: schemaPath,
878
- parts: schemaPathParts,
879
- } of candidates) {
880
- checkDeadline();
881
- if (schemaPathParts.length < pathParts.length) continue;
882
-
883
- // Check if all path parts match (allowing function call variants)
884
- let allMatch = true;
885
- let matchedUpToIndex = pathParts.length;
886
- for (let i = 0; i < pathParts.length; i++) {
887
- if (!pathPartMatches(pathParts[i], schemaPathParts[i])) {
888
- allMatch = false;
889
- break;
890
- }
891
- // If the last pathPart matched a function call variant,
892
- // we need to include it in the postfix calculation
893
- if (
894
- i === pathParts.length - 1 &&
895
- schemaPathParts[i] !== pathParts[i] &&
896
- schemaPathParts[i].startsWith(pathParts[i] + '(')
897
- ) {
898
- // The schemaPathPart is a function call variant (e.g., 'isEntityBeingAnalyzed(entity.sha)')
899
- // We want to include this as part of the postfix
900
- matchedUpToIndex = i;
905
+ for (const schema of schemas) {
906
+ // Use pre-computed index to only iterate schema entries whose
907
+ // normalized first part matches pathParts[0], instead of all entries.
908
+ const schemaIndex = getSchemaIndex(schema);
909
+ const lookupPart = pathParts[0] ?? '';
910
+ const lookupParenIdx = lookupPart.indexOf('(');
911
+ const lookupBase =
912
+ lookupParenIdx >= 0
913
+ ? lookupPart.slice(0, lookupParenIdx)
914
+ : lookupPart;
915
+ const candidates = schemaIndex.byFirstPart.get(lookupBase) || [];
916
+ for (const {
917
+ path: schemaPath,
918
+ parts: schemaPathParts,
919
+ } of candidates) {
920
+ checkDeadline();
921
+ if (schemaPathParts.length < pathParts.length) continue;
922
+
923
+ // Check if all path parts match (allowing function call variants)
924
+ let allMatch = true;
925
+ let matchedUpToIndex = pathParts.length;
926
+ for (let i = 0; i < pathParts.length; i++) {
927
+ if (!pathPartMatches(pathParts[i], schemaPathParts[i])) {
928
+ allMatch = false;
929
+ break;
930
+ }
931
+ // If the last pathPart matched a function call variant,
932
+ // we need to include it in the postfix calculation
933
+ if (
934
+ i === pathParts.length - 1 &&
935
+ schemaPathParts[i] !== pathParts[i] &&
936
+ schemaPathParts[i].startsWith(pathParts[i] + '(')
937
+ ) {
938
+ // The schemaPathPart is a function call variant (e.g., 'isEntityBeingAnalyzed(entity.sha)')
939
+ // We want to include this as part of the postfix
940
+ matchedUpToIndex = i;
941
+ }
901
942
  }
902
- }
903
-
904
- if (allMatch) {
905
- // When we matched a function call variant at the end (e.g., 'foo' matched 'foo(args)'),
906
- // the base itself should be marked as a function, and the function call details
907
- // should be included as sub-paths
908
- if (matchedUpToIndex < pathParts.length) {
909
- // This is a function call variant match at the last position
910
- // Mark the base as a function (empty postfix = the base path itself)
911
- entry.equivalentPostfixes[''] = bestValueFromOptions([
912
- entry.equivalentPostfixes[''],
913
- 'function',
914
- ]);
915
-
916
- // Also capture the function call and any remaining parts
917
- // e.g., 'isEntityBeingAnalyzed(entity.sha)' or 'isEntityBeingAnalyzed(entity.sha).functionCallReturnValue'
918
- const funcCallPart = schemaPathParts[matchedUpToIndex];
919
- const baseName = pathParts[matchedUpToIndex]; // e.g., 'isEntityBeingAnalyzed'
920
- const argsMatch = funcCallPart.match(/\(.*\)$/);
921
-
922
- if (argsMatch) {
923
- // Create postfix using just the args portion: (entity.sha) instead of isEntityBeingAnalyzed(entity.sha)
924
- // This avoids duplicating the base name in the final path
925
- const argsPortion = argsMatch[0]; // e.g., '(entity.sha)'
926
- const remainingParts = schemaPathParts.slice(
927
- matchedUpToIndex + 1,
928
- );
929
943
 
930
- // Build the postfix as: (args).remaining.parts
931
- const funcPostfix = joinParenthesesAndArrays([
932
- argsPortion,
933
- ...remainingParts,
944
+ if (allMatch) {
945
+ // When we matched a function call variant at the end (e.g., 'foo' matched 'foo(args)'),
946
+ // the base itself should be marked as a function, and the function call details
947
+ // should be included as sub-paths
948
+ if (matchedUpToIndex < pathParts.length) {
949
+ // This is a function call variant match at the last position
950
+ // Mark the base as a function (empty postfix = the base path itself)
951
+ entry.equivalentPostfixes[''] = bestValueFromOptions([
952
+ entry.equivalentPostfixes[''],
953
+ 'function',
934
954
  ]);
935
- entry.equivalentPostfixes[funcPostfix] = entry
936
- .equivalentPostfixes[funcPostfix]
937
- ? bestValueFromOptions([
938
- entry.equivalentPostfixes[funcPostfix],
939
- schema[schemaPath],
940
- ])
941
- : schema[schemaPath];
942
- }
943
- } else {
944
- // Regular exact match - use the standard postfix logic
945
- const postfix = joinParenthesesAndArrays(
946
- schemaPathParts.slice(matchedUpToIndex),
947
- );
948
955
 
949
- const previousValue = entry.equivalentPostfixes[postfix];
950
- const newValue = schema[schemaPath];
951
- entry.equivalentPostfixes[postfix] = previousValue
952
- ? bestValueFromOptions([previousValue, newValue])
953
- : newValue;
954
-
955
- // Trace postfix gathering - shows where type info comes from
956
- if (entry.equivalentPostfixes[postfix] !== previousValue) {
957
- transformationTracer.operation(rootScopeName, {
958
- operation: 'gatherPostfix',
959
- stage: 'gathering',
960
- path: postfix || '(root)',
961
- before: previousValue,
962
- after: entry.equivalentPostfixes[postfix],
963
- context: {
964
- sourceSchemaPath: schemaPath,
965
- sourceFunction:
966
- equivalentRoot.function?.name || rootScopeName,
967
- equivalentRootPath: equivalentRoot.schemaRootPath,
968
- rawValue: newValue,
969
- },
970
- });
956
+ // Also capture the function call and any remaining parts
957
+ // e.g., 'isEntityBeingAnalyzed(entity.sha)' or 'isEntityBeingAnalyzed(entity.sha).functionCallReturnValue'
958
+ const funcCallPart = schemaPathParts[matchedUpToIndex];
959
+ const baseName = pathParts[matchedUpToIndex]; // e.g., 'isEntityBeingAnalyzed'
960
+ const argsMatch = funcCallPart.match(/\(.*\)$/);
961
+
962
+ if (argsMatch) {
963
+ // Create postfix using just the args portion: (entity.sha) instead of isEntityBeingAnalyzed(entity.sha)
964
+ // This avoids duplicating the base name in the final path
965
+ const argsPortion = argsMatch[0]; // e.g., '(entity.sha)'
966
+ const remainingParts = schemaPathParts.slice(
967
+ matchedUpToIndex + 1,
968
+ );
969
+
970
+ // Build the postfix as: (args).remaining.parts
971
+ const funcPostfix = joinParenthesesAndArrays([
972
+ argsPortion,
973
+ ...remainingParts,
974
+ ]);
975
+ entry.equivalentPostfixes[funcPostfix] = entry
976
+ .equivalentPostfixes[funcPostfix]
977
+ ? bestValueFromOptions([
978
+ entry.equivalentPostfixes[funcPostfix],
979
+ schema[schemaPath],
980
+ ])
981
+ : schema[schemaPath];
982
+ }
983
+ } else {
984
+ // Regular exact match - use the standard postfix logic
985
+ const postfix = joinParenthesesAndArrays(
986
+ schemaPathParts.slice(matchedUpToIndex),
987
+ );
988
+
989
+ const previousValue = entry.equivalentPostfixes[postfix];
990
+ const newValue = schema[schemaPath];
991
+ entry.equivalentPostfixes[postfix] = previousValue
992
+ ? bestValueFromOptions([previousValue, newValue])
993
+ : newValue;
994
+
995
+ // Trace postfix gathering - shows where type info comes from
996
+ if (entry.equivalentPostfixes[postfix] !== previousValue) {
997
+ transformationTracer.operation(rootScopeName, {
998
+ operation: 'gatherPostfix',
999
+ stage: 'gathering',
1000
+ path: postfix || '(root)',
1001
+ before: previousValue,
1002
+ after: entry.equivalentPostfixes[postfix],
1003
+ context: {
1004
+ sourceSchemaPath: schemaPath,
1005
+ sourceFunction:
1006
+ equivalentRoot.function?.name || rootScopeName,
1007
+ equivalentRootPath: equivalentRoot.schemaRootPath,
1008
+ rawValue: newValue,
1009
+ },
1010
+ });
1011
+ }
971
1012
  }
972
1013
  }
973
1014
  }
@@ -975,1059 +1016,1096 @@ export default function mergeInDependentDataStructure({
975
1016
  }
976
1017
  }
977
1018
  }
978
- }
979
1019
 
980
- if (Object.keys(dataStructure?.returnValueSchema ?? {}).length > 0) {
981
- // Find all paths that contain functionCallReturnValue and extract unique base paths
982
- // For each path containing functionCallReturnValue, find the FIRST occurrence and use
983
- // that as a base path. This handles nested cases like:
984
- // X().functionCallReturnValue.A.B.Y().functionCallReturnValue
985
- // where we want both X().functionCallReturnValue and Y().functionCallReturnValue as bases
986
- const allBasePaths = new Set<string>();
987
- for (const path of Object.keys(dataStructure.returnValueSchema)) {
988
- checkDeadline();
989
- const parts = splitOutsideParenthesesAndArrays(path);
990
- // Find all positions of functionCallReturnValue and create base paths for each
991
- for (let i = 0; i < parts.length; i++) {
992
- if (parts[i] === 'functionCallReturnValue') {
993
- const basePath = joinParenthesesAndArrays(parts.slice(0, i + 1));
994
- allBasePaths.add(basePath);
1020
+ if (Object.keys(dataStructure?.returnValueSchema ?? {}).length > 0) {
1021
+ // Find all paths that contain functionCallReturnValue and extract unique base paths
1022
+ // For each path containing functionCallReturnValue, find the FIRST occurrence and use
1023
+ // that as a base path. This handles nested cases like:
1024
+ // X().functionCallReturnValue.A.B.Y().functionCallReturnValue
1025
+ // where we want both X().functionCallReturnValue and Y().functionCallReturnValue as bases
1026
+ const allBasePaths = new Set<string>();
1027
+ for (const path of Object.keys(dataStructure.returnValueSchema)) {
1028
+ checkDeadline();
1029
+ const parts = splitOutsideParenthesesAndArrays(path);
1030
+ // Find all positions of functionCallReturnValue and create base paths for each
1031
+ for (let i = 0; i < parts.length; i++) {
1032
+ if (parts[i] === 'functionCallReturnValue') {
1033
+ const basePath = joinParenthesesAndArrays(parts.slice(0, i + 1));
1034
+ allBasePaths.add(basePath);
1035
+ }
995
1036
  }
996
1037
  }
997
- }
998
-
999
- // Sort by length so shorter paths are processed first
1000
- const sortedBasePaths = [...allBasePaths].sort(
1001
- (a, b) => a.length - b.length,
1002
- );
1003
1038
 
1004
- for (const basePath of sortedBasePaths) {
1005
- const translatedBasePath = translatePath(basePath, functionName);
1006
- const entry = findOrCreateEquivalentSchemaPathsEntry([
1007
- { path: translatedBasePath, functionName: functionName },
1008
- ]);
1009
- const newRoot = {
1010
- schemaRootPath: translatedBasePath,
1011
- function: findRelevantDependency(functionName),
1012
- };
1013
- entry.equivalentRoots.push(newRoot);
1014
- // Update index for the newly added root
1015
- const newRootFuncName = newRoot.function?.name ?? rootScopeName;
1016
- espIndex.set(
1017
- espIndexKey(newRoot.schemaRootPath, newRootFuncName),
1018
- entry,
1039
+ // Sort by length so shorter paths are processed first
1040
+ const sortedBasePaths = [...allBasePaths].sort(
1041
+ (a, b) => a.length - b.length,
1019
1042
  );
1020
1043
 
1021
- const basePathParts = splitOutsideParenthesesAndArrays(basePath);
1022
- for (const schemaPath in dataStructure.returnValueSchema) {
1023
- checkDeadline();
1024
- const schemaPathParts = splitOutsideParenthesesAndArrays(schemaPath);
1025
- if (schemaPathParts.length < basePathParts.length) continue;
1044
+ for (const basePath of sortedBasePaths) {
1045
+ const translatedBasePath = translatePath(basePath, functionName);
1046
+ const entry = findOrCreateEquivalentSchemaPathsEntry([
1047
+ { path: translatedBasePath, functionName: functionName },
1048
+ ]);
1049
+ const newRoot = {
1050
+ schemaRootPath: translatedBasePath,
1051
+ function: findRelevantDependency(functionName),
1052
+ };
1053
+ entry.equivalentRoots.push(newRoot);
1054
+ // Update index for the newly added root
1055
+ const newRootFuncName = newRoot.function?.name ?? rootScopeName;
1056
+ espIndex.set(
1057
+ espIndexKey(newRoot.schemaRootPath, newRootFuncName),
1058
+ entry,
1059
+ );
1026
1060
 
1027
- // Check if this schemaPath actually starts with this basePath
1028
- // (not just has the same length prefix)
1029
- const prefixParts = schemaPathParts.slice(0, basePathParts.length);
1030
- if (
1031
- joinParenthesesAndArrays(prefixParts) !==
1032
- joinParenthesesAndArrays(basePathParts)
1033
- ) {
1034
- continue;
1035
- }
1061
+ const basePathParts = splitOutsideParenthesesAndArrays(basePath);
1062
+ for (const schemaPath in dataStructure.returnValueSchema) {
1063
+ checkDeadline();
1064
+ const schemaPathParts =
1065
+ splitOutsideParenthesesAndArrays(schemaPath);
1066
+ if (schemaPathParts.length < basePathParts.length) continue;
1036
1067
 
1037
- const postfix = joinParenthesesAndArrays(
1038
- schemaPathParts.slice(basePathParts.length),
1039
- );
1040
- const newValue = entry.equivalentPostfixes[postfix]
1041
- ? bestValueFromOptions([
1042
- entry.equivalentPostfixes[postfix],
1043
- dataStructure.returnValueSchema[schemaPath],
1044
- ])
1045
- : dataStructure.returnValueSchema[schemaPath];
1046
-
1047
- entry.equivalentPostfixes[postfix] = newValue;
1068
+ // Check if this schemaPath actually starts with this basePath
1069
+ // (not just has the same length prefix)
1070
+ const prefixParts = schemaPathParts.slice(0, basePathParts.length);
1071
+ if (
1072
+ joinParenthesesAndArrays(prefixParts) !==
1073
+ joinParenthesesAndArrays(basePathParts)
1074
+ ) {
1075
+ continue;
1076
+ }
1077
+
1078
+ const postfix = joinParenthesesAndArrays(
1079
+ schemaPathParts.slice(basePathParts.length),
1080
+ );
1081
+ const newValue = entry.equivalentPostfixes[postfix]
1082
+ ? bestValueFromOptions([
1083
+ entry.equivalentPostfixes[postfix],
1084
+ dataStructure.returnValueSchema[schemaPath],
1085
+ ])
1086
+ : dataStructure.returnValueSchema[schemaPath];
1087
+
1088
+ entry.equivalentPostfixes[postfix] = newValue;
1089
+ }
1048
1090
  }
1049
1091
  }
1050
- }
1051
- };
1052
-
1053
- const mergeAllEquivalentSchemaPaths = () => {
1054
- const mergedEquivalentSchemaPaths: typeof equivalentSchemaPaths = [];
1092
+ };
1055
1093
 
1056
- // Pre-pass: Connect entries with array/array-element relationships.
1057
- // This handles cases like:
1058
- // - Entry A has root 'surveys' (array)
1059
- // - Entry B has root 'surveys[]' (array element)
1060
- // These need to be connected so Entry B's field postfixes flow to Entry A.
1061
- // We do this before the main merge to ensure the connection happens regardless
1062
- // of processing order.
1063
- for (const esp of equivalentSchemaPaths) {
1064
- checkDeadline();
1065
- for (const root of esp.equivalentRoots) {
1066
- if (root.schemaRootPath.endsWith('[]')) {
1067
- // Find a matching parent entry with the base array path
1068
- const baseArrayPath = root.schemaRootPath.slice(0, -2);
1069
- const parentEntry = equivalentSchemaPaths.find(
1070
- (other) =>
1071
- other !== esp &&
1072
- other.equivalentRoots.some(
1073
- (otherRoot) =>
1074
- otherRoot.schemaRootPath === baseArrayPath &&
1075
- otherRoot.function?.name === root.function?.name &&
1076
- otherRoot.function?.filePath === root.function?.filePath,
1077
- ),
1078
- );
1079
- if (parentEntry) {
1080
- // Add transformed postfixes from child (array element) to parent (array)
1081
- // so they can be applied with [] prefix to parent paths
1082
- for (const [postfixPath, postfixValue] of Object.entries(
1083
- esp.equivalentPostfixes,
1084
- )) {
1085
- checkDeadline();
1086
- const transformedPostfix = joinParenthesesAndArrays(
1087
- ['[]', postfixPath].filter(Boolean),
1088
- );
1089
- if (!(transformedPostfix in parentEntry.equivalentPostfixes)) {
1090
- parentEntry.equivalentPostfixes[transformedPostfix] =
1091
- postfixValue;
1094
+ const mergeAllEquivalentSchemaPaths = () => {
1095
+ const mergedEquivalentSchemaPaths: typeof equivalentSchemaPaths = [];
1096
+
1097
+ // Pre-pass: Connect entries with array/array-element relationships.
1098
+ // This handles cases like:
1099
+ // - Entry A has root 'surveys' (array)
1100
+ // - Entry B has root 'surveys[]' (array element)
1101
+ // These need to be connected so Entry B's field postfixes flow to Entry A.
1102
+ // We do this before the main merge to ensure the connection happens regardless
1103
+ // of processing order.
1104
+ for (const esp of equivalentSchemaPaths) {
1105
+ checkDeadline();
1106
+ for (const root of esp.equivalentRoots) {
1107
+ if (root.schemaRootPath.endsWith('[]')) {
1108
+ // Find a matching parent entry with the base array path
1109
+ const baseArrayPath = root.schemaRootPath.slice(0, -2);
1110
+ const parentEntry = equivalentSchemaPaths.find(
1111
+ (other) =>
1112
+ other !== esp &&
1113
+ other.equivalentRoots.some(
1114
+ (otherRoot) =>
1115
+ otherRoot.schemaRootPath === baseArrayPath &&
1116
+ otherRoot.function?.name === root.function?.name &&
1117
+ otherRoot.function?.filePath === root.function?.filePath,
1118
+ ),
1119
+ );
1120
+ if (parentEntry) {
1121
+ // Add transformed postfixes from child (array element) to parent (array)
1122
+ // so they can be applied with [] prefix to parent paths
1123
+ for (const [postfixPath, postfixValue] of Object.entries(
1124
+ esp.equivalentPostfixes,
1125
+ )) {
1126
+ checkDeadline();
1127
+ const transformedPostfix = joinParenthesesAndArrays(
1128
+ ['[]', postfixPath].filter(Boolean),
1129
+ );
1130
+ if (!(transformedPostfix in parentEntry.equivalentPostfixes)) {
1131
+ parentEntry.equivalentPostfixes[transformedPostfix] =
1132
+ postfixValue;
1133
+ }
1092
1134
  }
1093
1135
  }
1094
1136
  }
1095
1137
  }
1096
1138
  }
1097
- }
1098
1139
 
1099
- const findEquivalentSchemaPathEntry = (
1100
- schemaSubPath: string,
1101
- equivalentRootFunction: (typeof equivalentSchemaPaths)[0]['equivalentRoots'][0]['function'],
1102
- ) => {
1103
- let postfix: string | undefined;
1140
+ const findEquivalentSchemaPathEntry = (
1141
+ schemaSubPath: string,
1142
+ equivalentRootFunction: (typeof equivalentSchemaPaths)[0]['equivalentRoots'][0]['function'],
1143
+ ) => {
1144
+ let postfix: string | undefined;
1104
1145
 
1105
- // Get the signature index we're looking for (if any)
1106
- const lookingForSignatureIndex = extractSignatureIndex(schemaSubPath);
1146
+ // Get the signature index we're looking for (if any)
1147
+ const lookingForSignatureIndex = extractSignatureIndex(schemaSubPath);
1107
1148
 
1108
- const equivalentEntry = mergedEquivalentSchemaPaths.find((esp) =>
1109
- esp.equivalentRoots.some((er) => {
1110
- if (
1111
- (schemaSubPath.startsWith('returnValue') ||
1112
- schemaSubPath.startsWith('signature[')) &&
1113
- (er.function?.name !== equivalentRootFunction?.name ||
1114
- er.function?.filePath !== equivalentRootFunction?.filePath)
1115
- ) {
1116
- return false;
1117
- }
1149
+ const equivalentEntry = mergedEquivalentSchemaPaths.find((esp) =>
1150
+ esp.equivalentRoots.some((er) => {
1151
+ if (
1152
+ (schemaSubPath.startsWith('returnValue') ||
1153
+ schemaSubPath.startsWith('signature[')) &&
1154
+ (er.function?.name !== equivalentRootFunction?.name ||
1155
+ er.function?.filePath !== equivalentRootFunction?.filePath)
1156
+ ) {
1157
+ return false;
1158
+ }
1118
1159
 
1119
- if (schemaSubPath === er.schemaRootPath) {
1120
- // Additional check: if we're looking for a signature path, make sure
1121
- // the entry doesn't already have DIFFERENT signature indices.
1122
- // This prevents entries with signature[1], signature[2], signature[4]
1123
- // from all being merged together.
1124
- if (lookingForSignatureIndex !== undefined) {
1125
- const hasConflictingSignatureIndex = esp.equivalentRoots.some(
1126
- (otherRoot) => {
1127
- // Only check roots from the same function
1128
- if (
1129
- otherRoot.function?.name !== equivalentRootFunction?.name ||
1130
- otherRoot.function?.filePath !==
1131
- equivalentRootFunction?.filePath
1132
- ) {
1133
- return false;
1134
- }
1135
- const otherIndex = extractSignatureIndex(
1136
- otherRoot.schemaRootPath,
1137
- );
1138
- return (
1139
- otherIndex !== undefined &&
1140
- otherIndex !== lookingForSignatureIndex
1141
- );
1142
- },
1143
- );
1144
- if (hasConflictingSignatureIndex) {
1145
- return false;
1160
+ if (schemaSubPath === er.schemaRootPath) {
1161
+ // Additional check: if we're looking for a signature path, make sure
1162
+ // the entry doesn't already have DIFFERENT signature indices.
1163
+ // This prevents entries with signature[1], signature[2], signature[4]
1164
+ // from all being merged together.
1165
+ if (lookingForSignatureIndex !== undefined) {
1166
+ const hasConflictingSignatureIndex = esp.equivalentRoots.some(
1167
+ (otherRoot) => {
1168
+ // Only check roots from the same function
1169
+ if (
1170
+ otherRoot.function?.name !==
1171
+ equivalentRootFunction?.name ||
1172
+ otherRoot.function?.filePath !==
1173
+ equivalentRootFunction?.filePath
1174
+ ) {
1175
+ return false;
1176
+ }
1177
+ const otherIndex = extractSignatureIndex(
1178
+ otherRoot.schemaRootPath,
1179
+ );
1180
+ return (
1181
+ otherIndex !== undefined &&
1182
+ otherIndex !== lookingForSignatureIndex
1183
+ );
1184
+ },
1185
+ );
1186
+ if (hasConflictingSignatureIndex) {
1187
+ return false;
1188
+ }
1146
1189
  }
1147
- }
1148
1190
 
1149
- postfix = er.postfix;
1150
- return true;
1151
- }
1191
+ postfix = er.postfix;
1192
+ return true;
1193
+ }
1152
1194
 
1153
- return false;
1154
- }),
1155
- );
1195
+ return false;
1196
+ }),
1197
+ );
1156
1198
 
1157
- return { equivalentEntry, postfix };
1158
- };
1199
+ return { equivalentEntry, postfix };
1200
+ };
1159
1201
 
1160
- const sortedEquivalentSchemaPaths = equivalentSchemaPaths.sort(
1161
- (a, b) =>
1162
- Math.max(
1163
- ...a.equivalentRoots.map(
1164
- (er) => splitOutsideParenthesesAndArrays(er.schemaRootPath).length,
1165
- ),
1166
- ) -
1167
- Math.max(
1168
- ...b.equivalentRoots.map(
1169
- (er) => splitOutsideParenthesesAndArrays(er.schemaRootPath).length,
1202
+ const sortedEquivalentSchemaPaths = equivalentSchemaPaths.sort(
1203
+ (a, b) =>
1204
+ Math.max(
1205
+ ...a.equivalentRoots.map(
1206
+ (er) =>
1207
+ splitOutsideParenthesesAndArrays(er.schemaRootPath).length,
1208
+ ),
1209
+ ) -
1210
+ Math.max(
1211
+ ...b.equivalentRoots.map(
1212
+ (er) =>
1213
+ splitOutsideParenthesesAndArrays(er.schemaRootPath).length,
1214
+ ),
1170
1215
  ),
1171
- ),
1172
- );
1216
+ );
1173
1217
 
1174
- for (const esp of sortedEquivalentSchemaPaths) {
1175
- checkDeadline();
1176
- if (esp.equivalentRoots.length === 0) continue;
1177
- let bestCandidateLength: number | undefined;
1178
- let bestCandidate: (typeof equivalentSchemaPaths)[0] | undefined;
1179
- let postfix: string | undefined;
1180
- for (const equivalentRoot of esp.equivalentRoots) {
1181
- const rootSchemaPath = equivalentRoot.schemaRootPath;
1182
- const schemaPathParts =
1183
- splitOutsideParenthesesAndArrays(rootSchemaPath);
1184
-
1185
- for (let i = 0; i < schemaPathParts.length; i++) {
1186
- const subPath = joinParenthesesAndArrays(
1187
- schemaPathParts.slice(0, i + 1),
1188
- );
1189
-
1190
- const { equivalentEntry, postfix: equivalentEntryPostfix } =
1191
- findEquivalentSchemaPathEntry(subPath, equivalentRoot.function);
1192
- if (
1193
- equivalentEntry &&
1194
- (!bestCandidateLength || bestCandidateLength > i + 1)
1195
- ) {
1196
- bestCandidate = equivalentEntry;
1197
- bestCandidateLength = i + 1;
1198
- postfix = joinParenthesesAndArrays(
1199
- [equivalentEntryPostfix, ...schemaPathParts.slice(i + 1)].filter(
1200
- Boolean,
1201
- ),
1218
+ for (const esp of sortedEquivalentSchemaPaths) {
1219
+ checkDeadline();
1220
+ if (esp.equivalentRoots.length === 0) continue;
1221
+ let bestCandidateLength: number | undefined;
1222
+ let bestCandidate: (typeof equivalentSchemaPaths)[0] | undefined;
1223
+ let postfix: string | undefined;
1224
+ for (const equivalentRoot of esp.equivalentRoots) {
1225
+ const rootSchemaPath = equivalentRoot.schemaRootPath;
1226
+ const schemaPathParts =
1227
+ splitOutsideParenthesesAndArrays(rootSchemaPath);
1228
+
1229
+ for (let i = 0; i < schemaPathParts.length; i++) {
1230
+ const subPath = joinParenthesesAndArrays(
1231
+ schemaPathParts.slice(0, i + 1),
1202
1232
  );
1233
+
1234
+ const { equivalentEntry, postfix: equivalentEntryPostfix } =
1235
+ findEquivalentSchemaPathEntry(subPath, equivalentRoot.function);
1236
+ if (
1237
+ equivalentEntry &&
1238
+ (!bestCandidateLength || bestCandidateLength > i + 1)
1239
+ ) {
1240
+ bestCandidate = equivalentEntry;
1241
+ bestCandidateLength = i + 1;
1242
+ postfix = joinParenthesesAndArrays(
1243
+ [
1244
+ equivalentEntryPostfix,
1245
+ ...schemaPathParts.slice(i + 1),
1246
+ ].filter(Boolean),
1247
+ );
1248
+ }
1203
1249
  }
1204
1250
  }
1205
- }
1206
1251
 
1207
- if (bestCandidate) {
1208
- for (const root of esp.equivalentRoots) {
1209
- if (postfix.length > 0) {
1210
- root.postfix = postfix;
1252
+ if (bestCandidate) {
1253
+ for (const root of esp.equivalentRoots) {
1254
+ if (postfix.length > 0) {
1255
+ root.postfix = postfix;
1256
+ }
1257
+
1258
+ bestCandidate.equivalentRoots.push(root);
1211
1259
  }
1212
1260
 
1213
- bestCandidate.equivalentRoots.push(root);
1261
+ const postfixesToMerge =
1262
+ postfix.length > 0
1263
+ ? Object.keys(esp.equivalentPostfixes).reduce(
1264
+ (acc, postfixPath) => {
1265
+ const fullPath = joinParenthesesAndArrays([
1266
+ postfix,
1267
+ postfixPath,
1268
+ ]);
1269
+ acc[fullPath] = esp.equivalentPostfixes[postfixPath];
1270
+ return acc;
1271
+ },
1272
+ {} as Record<string, string>,
1273
+ )
1274
+ : esp.equivalentPostfixes;
1275
+
1276
+ bestCandidate.equivalentPostfixes = {
1277
+ ...bestCandidate.equivalentPostfixes,
1278
+ ...postfixesToMerge,
1279
+ };
1280
+ } else {
1281
+ mergedEquivalentSchemaPaths.push(esp);
1214
1282
  }
1283
+ }
1215
1284
 
1216
- const postfixesToMerge =
1217
- postfix.length > 0
1218
- ? Object.keys(esp.equivalentPostfixes).reduce(
1219
- (acc, postfixPath) => {
1220
- const fullPath = joinParenthesesAndArrays([
1221
- postfix,
1222
- postfixPath,
1223
- ]);
1224
- acc[fullPath] = esp.equivalentPostfixes[postfixPath];
1225
- return acc;
1226
- },
1227
- {} as Record<string, string>,
1228
- )
1229
- : esp.equivalentPostfixes;
1285
+ return mergedEquivalentSchemaPaths;
1286
+ };
1230
1287
 
1231
- bestCandidate.equivalentPostfixes = {
1232
- ...bestCandidate.equivalentPostfixes,
1233
- ...postfixesToMerge,
1234
- };
1235
- } else {
1236
- mergedEquivalentSchemaPaths.push(esp);
1288
+ // Build a lookup of mocked dependencies to skip their internal implementation
1289
+ const mockedDependencies = new Set<string>();
1290
+ for (const dep of importedExports) {
1291
+ if (dep.isMocked) {
1292
+ mockedDependencies.add(`${dep.filePath}::${dep.name}`);
1237
1293
  }
1238
1294
  }
1239
1295
 
1240
- return mergedEquivalentSchemaPaths;
1241
- };
1296
+ gatherAllEquivalentSchemaPaths(rootScopeName, dataStructure);
1242
1297
 
1243
- // Build a lookup of mocked dependencies to skip their internal implementation
1244
- const mockedDependencies = new Set<string>();
1245
- for (const dep of importedExports) {
1246
- if (dep.isMocked) {
1247
- mockedDependencies.add(`${dep.filePath}::${dep.name}`);
1298
+ // Process dependencySchemas for all dependencies (including mocked ones)
1299
+ // dependencySchemas contains usage information (how dependencies are called),
1300
+ // not internal implementation, so we want this for mocked dependencies too
1301
+ for (const dependency of importedExports) {
1302
+ checkDeadline();
1303
+ const dependentDataStructure =
1304
+ dependencySchemas?.[dependency.filePath]?.[dependency.name];
1305
+ if (!dependentDataStructure) continue;
1306
+ gatherAllEquivalentSchemaPaths(
1307
+ dependency.name,
1308
+ dependentDataStructure,
1309
+ dependentDataStructure,
1310
+ );
1248
1311
  }
1249
- }
1250
1312
 
1251
- gatherAllEquivalentSchemaPaths(rootScopeName, dataStructure);
1252
-
1253
- // Process dependencySchemas for all dependencies (including mocked ones)
1254
- // dependencySchemas contains usage information (how dependencies are called),
1255
- // not internal implementation, so we want this for mocked dependencies too
1256
- for (const dependency of importedExports) {
1257
- checkDeadline();
1258
- const dependentDataStructure =
1259
- dependencySchemas?.[dependency.filePath]?.[dependency.name];
1260
- if (!dependentDataStructure) continue;
1261
- gatherAllEquivalentSchemaPaths(
1262
- dependency.name,
1263
- dependentDataStructure,
1264
- dependentDataStructure,
1265
- );
1266
- }
1313
+ for (const filePath in dependentAnalyses) {
1314
+ for (const name in dependentAnalyses[filePath]) {
1315
+ // Skip mocked dependencies - we don't want to merge in their internal implementation
1316
+ if (mockedDependencies.has(`${filePath}::${name}`)) {
1317
+ continue;
1318
+ }
1267
1319
 
1268
- for (const filePath in dependentAnalyses) {
1269
- for (const name in dependentAnalyses[filePath]) {
1270
- // Skip mocked dependencies - we don't want to merge in their internal implementation
1271
- if (mockedDependencies.has(`${filePath}::${name}`)) {
1272
- continue;
1320
+ const childMergedDataStructure =
1321
+ dependentAnalyses[filePath][name].metadata?.mergedDataStructure || {};
1322
+
1323
+ gatherAllEquivalentSchemaPaths(name, childMergedDataStructure as any);
1273
1324
  }
1325
+ }
1274
1326
 
1275
- const childMergedDataStructure =
1276
- dependentAnalyses[filePath][name].metadata?.mergedDataStructure || {};
1327
+ const gatherElapsed = Date.now() - mergeStartTime;
1277
1328
 
1278
- gatherAllEquivalentSchemaPaths(name, childMergedDataStructure as any);
1279
- }
1280
- }
1329
+ equivalentSchemaPaths = mergeAllEquivalentSchemaPaths();
1281
1330
 
1282
- const gatherElapsed = Date.now() - mergeStartTime;
1283
-
1284
- equivalentSchemaPaths = mergeAllEquivalentSchemaPaths();
1285
-
1286
- const mergeEspElapsed = Date.now() - mergeStartTime;
1287
-
1288
- // Collect schemas that need cleaning — batch the calls for the end instead of
1289
- // calling cleanSchema inside the inner root loop (which was O(roots * schemaSize)).
1290
- const schemasToClean = new Set<{ [key: string]: string }>();
1291
-
1292
- for (const esp of equivalentSchemaPaths) {
1293
- checkDeadline();
1294
- // Pre-compute which postfixes have children to avoid O(n²) lookups in the inner loop.
1295
- // A postfix "has children" if there are other postfixes that extend it.
1296
- const postfixesWithChildren = new Set<string>();
1297
- const postfixKeys = Object.keys(esp.equivalentPostfixes);
1298
-
1299
- // Pre-parse ALL postfix paths once. These parsed parts are reused in:
1300
- // 1. The children detection loop below
1301
- // 2. The inner postfix application loop (lines that split postfixPath and equivalentRoot.postfix)
1302
- // This eliminates thousands of redundant splitOutsideParenthesesAndArrays calls.
1303
- const postfixPartsCache = new Map<string, string[]>();
1304
- for (const postfixPath of postfixKeys) {
1305
- if (!postfixPath) continue;
1306
- postfixPartsCache.set(
1307
- postfixPath,
1308
- splitOutsideParenthesesAndArrays(postfixPath),
1309
- );
1310
- }
1331
+ const mergeEspElapsed = Date.now() - mergeStartTime;
1311
1332
 
1312
- // Check for empty postfix having children (any other postfixes exist)
1313
- if (postfixKeys.length > 1 && '' in esp.equivalentPostfixes) {
1314
- postfixesWithChildren.add('');
1315
- }
1333
+ // Collect schemas that need cleaning batch the calls for the end instead of
1334
+ // calling cleanSchema inside the inner root loop (which was O(roots * schemaSize)).
1335
+ const schemasToClean = new Set<{ [key: string]: string }>();
1316
1336
 
1317
- // Check for array element postfixes having children using a prefix set.
1318
- // This avoids O() scans across large postfix lists.
1319
- // e.g., 'currentEntities[]' has children if a path like 'currentEntities[].sha' exists.
1320
- const postfixPrefixSet = new Set<string>();
1321
- for (const postfixPath of postfixKeys) {
1322
- if (!postfixPath) continue;
1323
- const parts = postfixPartsCache.get(postfixPath)!;
1324
- for (let i = 1; i < parts.length; i++) {
1325
- postfixPrefixSet.add(joinParenthesesAndArrays(parts.slice(0, i)));
1337
+ for (const esp of equivalentSchemaPaths) {
1338
+ checkDeadline();
1339
+ // Pre-compute which postfixes have children to avoid O(n²) lookups in the inner loop.
1340
+ // A postfix "has children" if there are other postfixes that extend it.
1341
+ const postfixesWithChildren = new Set<string>();
1342
+ const postfixKeys = Object.keys(esp.equivalentPostfixes);
1343
+
1344
+ // Pre-parse ALL postfix paths once. These parsed parts are reused in:
1345
+ // 1. The children detection loop below
1346
+ // 2. The inner postfix application loop (lines that split postfixPath and equivalentRoot.postfix)
1347
+ // This eliminates thousands of redundant splitOutsideParenthesesAndArrays calls.
1348
+ const postfixPartsCache = new Map<string, string[]>();
1349
+ for (const postfixPath of postfixKeys) {
1350
+ if (!postfixPath) continue;
1351
+ postfixPartsCache.set(
1352
+ postfixPath,
1353
+ splitOutsideParenthesesAndArrays(postfixPath),
1354
+ );
1326
1355
  }
1327
- }
1328
- for (const postfixPath of postfixKeys) {
1329
- if (postfixPath.endsWith('[]') && postfixPrefixSet.has(postfixPath)) {
1330
- postfixesWithChildren.add(postfixPath);
1356
+
1357
+ // Check for empty postfix having children (any other postfixes exist)
1358
+ if (postfixKeys.length > 1 && '' in esp.equivalentPostfixes) {
1359
+ postfixesWithChildren.add('');
1331
1360
  }
1332
- }
1333
1361
 
1334
- // Deduplicate equivalentRoots that would write to the same schema paths.
1335
- // Roots with the same (function, schemaRootPath, postfix) are redundant.
1336
- const seenRootKeys = new Set<string>();
1337
- const uniqueRoots = esp.equivalentRoots.filter((root) => {
1338
- const key = `${root.function?.filePath ?? ''}::${root.function?.name ?? ''}::${root.schemaRootPath}::${root.postfix ?? ''}`;
1339
- if (seenRootKeys.has(key)) return false;
1340
- seenRootKeys.add(key);
1341
- return true;
1342
- });
1362
+ // Check for array element postfixes having children using a prefix set.
1363
+ // This avoids O(n²) scans across large postfix lists.
1364
+ // e.g., 'currentEntities[]' has children if a path like 'currentEntities[].sha' exists.
1365
+ const postfixPrefixSet = new Set<string>();
1366
+ for (const postfixPath of postfixKeys) {
1367
+ if (!postfixPath) continue;
1368
+ const parts = postfixPartsCache.get(postfixPath)!;
1369
+ for (let i = 1; i < parts.length; i++) {
1370
+ postfixPrefixSet.add(joinParenthesesAndArrays(parts.slice(0, i)));
1371
+ }
1372
+ }
1373
+ for (const postfixPath of postfixKeys) {
1374
+ if (postfixPath.endsWith('[]') && postfixPrefixSet.has(postfixPath)) {
1375
+ postfixesWithChildren.add(postfixPath);
1376
+ }
1377
+ }
1343
1378
 
1344
- // Cap schema size to prevent combinatorial explosion.
1345
- // Successful merges produce <3K ret keys. Beyond 5K, further postfixes
1346
- // add noise but no useful data — they're cross-products of unrelated equivalencies.
1347
- const SCHEMA_KEY_CAP = 5_000;
1379
+ // Deduplicate equivalentRoots that would write to the same schema paths.
1380
+ // Roots with the same (function, schemaRootPath, postfix) are redundant.
1381
+ const seenRootKeys = new Set<string>();
1382
+ const uniqueRoots = esp.equivalentRoots.filter((root) => {
1383
+ const key = `${root.function?.filePath ?? ''}::${root.function?.name ?? ''}::${root.schemaRootPath}::${root.postfix ?? ''}`;
1384
+ if (seenRootKeys.has(key)) return false;
1385
+ seenRootKeys.add(key);
1386
+ return true;
1387
+ });
1348
1388
 
1349
- for (const equivalentRoot of uniqueRoots) {
1350
- checkDeadline();
1351
- let merged:
1352
- | {
1353
- signatureSchema: { [key: string]: string };
1354
- returnValueSchema: { [key: string]: string };
1355
- }
1356
- | undefined;
1389
+ for (const equivalentRoot of uniqueRoots) {
1390
+ checkDeadline();
1391
+ let merged:
1392
+ | {
1393
+ signatureSchema: { [key: string]: string };
1394
+ returnValueSchema: { [key: string]: string };
1395
+ }
1396
+ | undefined;
1357
1397
 
1358
- if (equivalentRoot.function) {
1359
- merged = findOrCreateDependentSchemas(equivalentRoot.function);
1360
- } else {
1361
- merged = mergedDataStructure;
1362
- }
1398
+ if (equivalentRoot.function) {
1399
+ merged = findOrCreateDependentSchemas(equivalentRoot.function);
1400
+ } else {
1401
+ merged = mergedDataStructure;
1402
+ }
1363
1403
 
1364
- if (!merged) continue;
1404
+ if (!merged) continue;
1365
1405
 
1366
- const schema = equivalentRoot.schemaRootPath.startsWith('signature[')
1367
- ? merged.signatureSchema
1368
- : merged.returnValueSchema;
1406
+ const schema = equivalentRoot.schemaRootPath.startsWith('signature[')
1407
+ ? merged.signatureSchema
1408
+ : merged.returnValueSchema;
1369
1409
 
1370
- // Skip if this schema has already grown past the cap
1371
- if (Object.keys(schema).length > SCHEMA_KEY_CAP) continue;
1410
+ // Skip if this schema has already grown past the cap
1411
+ if (Object.keys(schema).length > SCHEMA_KEY_CAP) continue;
1372
1412
 
1373
- for (const [postfixPath, postfixValue] of Object.entries(
1374
- esp.equivalentPostfixes,
1375
- )) {
1376
- checkDeadline();
1377
- let relevantPostfix = postfixPath;
1378
- if (equivalentRoot.postfix) {
1379
- // Check if postfixPath starts with equivalentRoot.postfix at a path boundary.
1380
- // Must ensure exact path part match - "entityCode" should NOT match "entity" prefix.
1381
- // Valid: "entity.foo" starts with "entity" (boundary at '.')
1382
- // Valid: "entity[0]" starts with "entity" (boundary at '[')
1383
- // Invalid: "entityCode" starts with "entity" (no boundary, different property)
1384
- if (!postfixPath.startsWith(equivalentRoot.postfix)) {
1385
- continue;
1386
- }
1387
- // Additional check: ensure the match is at a path boundary
1388
- const nextChar = postfixPath[equivalentRoot.postfix.length];
1389
- if (nextChar !== undefined && nextChar !== '.' && nextChar !== '[') {
1390
- // The postfixPath continues with more characters that aren't a path separator.
1391
- // This means "entity" matched "entityCode" which is wrong - they're different properties.
1392
- continue;
1393
- }
1413
+ for (const [postfixPath, postfixValue] of Object.entries(
1414
+ esp.equivalentPostfixes,
1415
+ )) {
1416
+ checkDeadline();
1417
+ let relevantPostfix = postfixPath;
1418
+ if (equivalentRoot.postfix) {
1419
+ // Check if postfixPath starts with equivalentRoot.postfix at a path boundary.
1420
+ // Must ensure exact path part match - "entityCode" should NOT match "entity" prefix.
1421
+ // Valid: "entity.foo" starts with "entity" (boundary at '.')
1422
+ // Valid: "entity[0]" starts with "entity" (boundary at '[')
1423
+ // Invalid: "entityCode" starts with "entity" (no boundary, different property)
1424
+ if (!postfixPath.startsWith(equivalentRoot.postfix)) {
1425
+ continue;
1426
+ }
1427
+ // Additional check: ensure the match is at a path boundary
1428
+ const nextChar = postfixPath[equivalentRoot.postfix.length];
1429
+ if (
1430
+ nextChar !== undefined &&
1431
+ nextChar !== '.' &&
1432
+ nextChar !== '['
1433
+ ) {
1434
+ // The postfixPath continues with more characters that aren't a path separator.
1435
+ // This means "entity" matched "entityCode" which is wrong - they're different properties.
1436
+ continue;
1437
+ }
1394
1438
 
1395
- const postFixPathParts =
1396
- postfixPartsCache.get(postfixPath) ??
1397
- splitOutsideParenthesesAndArrays(postfixPath);
1398
- // Cache equivalentRoot.postfix parts — same root reused across all postfixes
1399
- if (!postfixPartsCache.has(equivalentRoot.postfix)) {
1400
- postfixPartsCache.set(
1439
+ const postFixPathParts =
1440
+ postfixPartsCache.get(postfixPath) ??
1441
+ splitOutsideParenthesesAndArrays(postfixPath);
1442
+ // Cache equivalentRoot.postfix parts — same root reused across all postfixes
1443
+ if (!postfixPartsCache.has(equivalentRoot.postfix)) {
1444
+ postfixPartsCache.set(
1445
+ equivalentRoot.postfix,
1446
+ splitOutsideParenthesesAndArrays(equivalentRoot.postfix),
1447
+ );
1448
+ }
1449
+ const equivalentRootPostFixParts = postfixPartsCache.get(
1401
1450
  equivalentRoot.postfix,
1402
- splitOutsideParenthesesAndArrays(equivalentRoot.postfix),
1451
+ )!;
1452
+ relevantPostfix = joinParenthesesAndArrays(
1453
+ postFixPathParts.slice(equivalentRootPostFixParts.length),
1403
1454
  );
1404
1455
  }
1405
- const equivalentRootPostFixParts = postfixPartsCache.get(
1406
- equivalentRoot.postfix,
1407
- )!;
1408
- relevantPostfix = joinParenthesesAndArrays(
1409
- postFixPathParts.slice(equivalentRootPostFixParts.length),
1410
- );
1411
- }
1412
-
1413
- const newSchemaPath = joinParenthesesAndArrays([
1414
- equivalentRoot.schemaRootPath,
1415
- relevantPostfix,
1416
- ]);
1417
1456
 
1418
- // Skip paths that would go through a primitive type
1419
- // e.g., if schema has 'entities[].scenarioCount': 'number', skip 'entities[].scenarioCount.sha'
1420
- if (wouldGoThroughPrimitive(newSchemaPath, schema)) {
1421
- transformationTracer.operation(rootScopeName, {
1422
- operation: 'skipPrimitivePath',
1423
- stage: 'merged',
1424
- path: newSchemaPath,
1425
- context: {
1426
- reason: 'would go through primitive type',
1427
- postfixValue,
1428
- },
1429
- });
1430
- continue;
1431
- }
1432
-
1433
- // Skip setting primitive type when there are child postfixes that indicate structure.
1434
- // This prevents downgrading an object/array element to a primitive type.
1435
- // Uses pre-computed postfixesWithChildren Set for O(1) lookup instead of O(n) iteration.
1436
- const hasChildPostfixes =
1437
- (relevantPostfix === '' || relevantPostfix.endsWith('[]')) &&
1438
- postfixesWithChildren.has(postfixPath);
1439
- if (PRIMITIVE_TYPES.has(postfixValue) && hasChildPostfixes) {
1440
- continue;
1441
- }
1457
+ const newSchemaPath = joinParenthesesAndArrays([
1458
+ equivalentRoot.schemaRootPath,
1459
+ relevantPostfix,
1460
+ ]);
1442
1461
 
1443
- // Don't overwrite a more specific type with a less specific one
1444
- // This can happen when nested roots share entries with their parent roots
1445
- const existingType = schema[newSchemaPath];
1446
- if (existingType) {
1447
- // Don't overwrite a primitive type with 'object' or 'array'
1448
- // e.g., if schema has 'entities[].scenarioCount': 'number', don't overwrite with 'object'
1449
- if (
1450
- PRIMITIVE_TYPES.has(existingType) &&
1451
- (postfixValue === 'object' || postfixValue === 'array')
1452
- ) {
1462
+ // Skip paths that would go through a primitive type
1463
+ // e.g., if schema has 'entities[].scenarioCount': 'number', skip 'entities[].scenarioCount.sha'
1464
+ if (wouldGoThroughPrimitive(newSchemaPath, schema)) {
1453
1465
  transformationTracer.operation(rootScopeName, {
1454
- operation: 'skipTypeDowngrade',
1466
+ operation: 'skipPrimitivePath',
1455
1467
  stage: 'merged',
1456
1468
  path: newSchemaPath,
1457
1469
  context: {
1458
- reason: 'would overwrite primitive with object/array',
1459
- existingType,
1460
- newType: postfixValue,
1470
+ reason: 'would go through primitive type',
1471
+ postfixValue,
1461
1472
  },
1462
1473
  });
1463
1474
  continue;
1464
1475
  }
1465
- // Don't overwrite a complex/union type with a primitive
1466
- // e.g., if schema has 'scenarios[]': 'Scenario | null', don't overwrite with 'string'
1467
- if (
1468
- !PRIMITIVE_TYPES.has(existingType) &&
1469
- PRIMITIVE_TYPES.has(postfixValue)
1470
- ) {
1471
- transformationTracer.operation(rootScopeName, {
1472
- operation: 'skipTypeDowngrade',
1473
- stage: 'merged',
1474
- path: newSchemaPath,
1475
- context: {
1476
- reason: 'would overwrite complex type with primitive',
1477
- existingType,
1478
- newType: postfixValue,
1479
- },
1480
- });
1476
+
1477
+ // Skip setting primitive type when there are child postfixes that indicate structure.
1478
+ // This prevents downgrading an object/array element to a primitive type.
1479
+ // Uses pre-computed postfixesWithChildren Set for O(1) lookup instead of O(n) iteration.
1480
+ const hasChildPostfixes =
1481
+ (relevantPostfix === '' || relevantPostfix.endsWith('[]')) &&
1482
+ postfixesWithChildren.has(postfixPath);
1483
+ if (PRIMITIVE_TYPES.has(postfixValue) && hasChildPostfixes) {
1481
1484
  continue;
1482
1485
  }
1486
+
1487
+ // Don't overwrite a more specific type with a less specific one
1488
+ // This can happen when nested roots share entries with their parent roots
1489
+ const existingType = schema[newSchemaPath];
1490
+ if (existingType) {
1491
+ // Don't overwrite a primitive type with 'object' or 'array'
1492
+ // e.g., if schema has 'entities[].scenarioCount': 'number', don't overwrite with 'object'
1493
+ if (
1494
+ PRIMITIVE_TYPES.has(existingType) &&
1495
+ (postfixValue === 'object' || postfixValue === 'array')
1496
+ ) {
1497
+ transformationTracer.operation(rootScopeName, {
1498
+ operation: 'skipTypeDowngrade',
1499
+ stage: 'merged',
1500
+ path: newSchemaPath,
1501
+ context: {
1502
+ reason: 'would overwrite primitive with object/array',
1503
+ existingType,
1504
+ newType: postfixValue,
1505
+ },
1506
+ });
1507
+ continue;
1508
+ }
1509
+ // Don't overwrite a complex/union type with a primitive
1510
+ // e.g., if schema has 'scenarios[]': 'Scenario | null', don't overwrite with 'string'
1511
+ if (
1512
+ !PRIMITIVE_TYPES.has(existingType) &&
1513
+ PRIMITIVE_TYPES.has(postfixValue)
1514
+ ) {
1515
+ transformationTracer.operation(rootScopeName, {
1516
+ operation: 'skipTypeDowngrade',
1517
+ stage: 'merged',
1518
+ path: newSchemaPath,
1519
+ context: {
1520
+ reason: 'would overwrite complex type with primitive',
1521
+ existingType,
1522
+ newType: postfixValue,
1523
+ },
1524
+ });
1525
+ continue;
1526
+ }
1527
+ }
1528
+
1529
+ // Log the successful postfix merge
1530
+ transformationTracer.operation(rootScopeName, {
1531
+ operation: 'mergePostfix',
1532
+ stage: 'merged',
1533
+ path: newSchemaPath,
1534
+ before: existingType,
1535
+ after: postfixValue,
1536
+ context: {
1537
+ schemaRootPath: equivalentRoot.schemaRootPath,
1538
+ postfix: relevantPostfix,
1539
+ dependency: equivalentRoot.function?.name,
1540
+ },
1541
+ });
1542
+ schema[newSchemaPath] = postfixValue;
1483
1543
  }
1484
1544
 
1485
- // Log the successful postfix merge
1486
- transformationTracer.operation(rootScopeName, {
1487
- operation: 'mergePostfix',
1488
- stage: 'merged',
1489
- path: newSchemaPath,
1490
- before: existingType,
1491
- after: postfixValue,
1492
- context: {
1493
- schemaRootPath: equivalentRoot.schemaRootPath,
1494
- postfix: relevantPostfix,
1495
- dependency: equivalentRoot.function?.name,
1496
- },
1497
- });
1498
- schema[newSchemaPath] = postfixValue;
1545
+ schemasToClean.add(schema);
1499
1546
  }
1500
-
1501
- schemasToClean.add(schema);
1502
1547
  }
1503
- }
1504
1548
 
1505
- const postfixElapsed = Date.now() - mergeStartTime;
1549
+ const postfixElapsed = Date.now() - mergeStartTime;
1506
1550
 
1507
- // Batch-clean all modified schemas once (instead of once per root per ESP entry)
1508
- for (const schema of schemasToClean) {
1509
- cleanSchema(schema, { stage: 'afterMergePostfix' });
1510
- }
1551
+ // Batch-clean all modified schemas once (instead of once per root per ESP entry)
1552
+ for (const schema of schemasToClean) {
1553
+ cleanSchema(schema, { stage: 'afterMergePostfix' });
1554
+ }
1511
1555
 
1512
- const cleanElapsed = Date.now() - mergeStartTime;
1513
-
1514
- // Propagate equivalency-derived attributes to generic function call variants.
1515
- // When attributes are traced via equivalencies (e.g., fileComparisons from buildDataMap.signature[2]),
1516
- // they get written to non-generic paths (returnValue.data.x or funcName().functionCallReturnValue.data.x).
1517
- // If the ORIGINAL input schema has generic variants (funcName<T>().functionCallReturnValue.data),
1518
- // we need to copy the attributes to those paths too.
1519
- for (const filePath in mergedDataStructure.dependencySchemas) {
1520
- for (const depName in mergedDataStructure.dependencySchemas[filePath]) {
1521
- const depSchema =
1522
- mergedDataStructure.dependencySchemas[filePath][depName];
1523
- const returnValueSchema = depSchema.returnValueSchema;
1524
-
1525
- // Look at the ORIGINAL input dependencySchemas for generic variants,
1526
- // since the merged schema may have lost them during equivalency processing
1527
- const originalSchema = dependencySchemas?.[filePath]?.[depName];
1528
- const schemaToSearchForGenericVariants =
1529
- originalSchema?.returnValueSchema || returnValueSchema;
1530
-
1531
- // Find all unique generic variants of this function
1532
- // e.g., useFetcher<BranchEntityDiffResult>() from useFetcher<BranchEntityDiffResult>().functionCallReturnValue.data
1533
- const genericVariants = new Set<string>();
1534
- const genericRegex = new RegExp(
1535
- `^${depName.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')}<[^>]+>\\(\\)`,
1536
- );
1556
+ const cleanElapsed = Date.now() - mergeStartTime;
1557
+
1558
+ // Propagate equivalency-derived attributes to generic function call variants.
1559
+ // When attributes are traced via equivalencies (e.g., fileComparisons from buildDataMap.signature[2]),
1560
+ // they get written to non-generic paths (returnValue.data.x or funcName().functionCallReturnValue.data.x).
1561
+ // If the ORIGINAL input schema has generic variants (funcName<T>().functionCallReturnValue.data),
1562
+ // we need to copy the attributes to those paths too.
1563
+ for (const filePath in mergedDataStructure.dependencySchemas) {
1564
+ for (const depName in mergedDataStructure.dependencySchemas[filePath]) {
1565
+ const depSchema =
1566
+ mergedDataStructure.dependencySchemas[filePath][depName];
1567
+ const returnValueSchema = depSchema.returnValueSchema;
1568
+
1569
+ // Look at the ORIGINAL input dependencySchemas for generic variants,
1570
+ // since the merged schema may have lost them during equivalency processing
1571
+ const originalSchema = dependencySchemas?.[filePath]?.[depName];
1572
+ const schemaToSearchForGenericVariants =
1573
+ originalSchema?.returnValueSchema || returnValueSchema;
1574
+
1575
+ // Find all unique generic variants of this function
1576
+ // e.g., useFetcher<BranchEntityDiffResult>() from useFetcher<BranchEntityDiffResult>().functionCallReturnValue.data
1577
+ const genericVariants = new Set<string>();
1578
+ const genericRegex = new RegExp(
1579
+ `^${depName.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')}<[^>]+>\\(\\)`,
1580
+ );
1537
1581
 
1538
- for (const path in schemaToSearchForGenericVariants) {
1539
- checkDeadline();
1540
- const match = path.match(genericRegex);
1541
- if (match) {
1542
- genericVariants.add(match[0]);
1582
+ for (const path in schemaToSearchForGenericVariants) {
1583
+ checkDeadline();
1584
+ const match = path.match(genericRegex);
1585
+ if (match) {
1586
+ genericVariants.add(match[0]);
1587
+ }
1543
1588
  }
1544
- }
1545
1589
 
1546
- if (genericVariants.size === 0) continue;
1590
+ if (genericVariants.size === 0) continue;
1547
1591
 
1548
- // For each returnValue. path or non-generic function call path,
1549
- // create corresponding paths for each generic variant
1550
- const pathsToAdd: [string, string][] = [];
1592
+ // For each returnValue. path or non-generic function call path,
1593
+ // create corresponding paths for each generic variant
1594
+ const pathsToAdd: [string, string][] = [];
1551
1595
 
1552
- for (const path in returnValueSchema) {
1553
- checkDeadline();
1554
- const value = returnValueSchema[path];
1555
-
1556
- // Handle returnValue. paths
1557
- if (path.startsWith('returnValue.')) {
1558
- const suffix = path.slice('returnValue.'.length);
1559
- for (const genericVariant of genericVariants) {
1560
- const genericPath = `${genericVariant}.functionCallReturnValue.${suffix}`;
1561
- if (!(genericPath in returnValueSchema)) {
1562
- pathsToAdd.push([genericPath, value]);
1596
+ for (const path in returnValueSchema) {
1597
+ checkDeadline();
1598
+ const value = returnValueSchema[path];
1599
+
1600
+ // Handle returnValue. paths
1601
+ if (path.startsWith('returnValue.')) {
1602
+ const suffix = path.slice('returnValue.'.length);
1603
+ for (const genericVariant of genericVariants) {
1604
+ const genericPath = `${genericVariant}.functionCallReturnValue.${suffix}`;
1605
+ if (!(genericPath in returnValueSchema)) {
1606
+ pathsToAdd.push([genericPath, value]);
1607
+ }
1563
1608
  }
1564
1609
  }
1565
- }
1566
- // Handle non-generic function call paths like depName().functionCallReturnValue.x
1567
- else if (path.startsWith(`${depName}().functionCallReturnValue.`)) {
1568
- const suffix = path.slice(
1569
- `${depName}().functionCallReturnValue.`.length,
1570
- );
1571
- for (const genericVariant of genericVariants) {
1572
- const genericPath = `${genericVariant}.functionCallReturnValue.${suffix}`;
1573
- if (!(genericPath in returnValueSchema)) {
1574
- pathsToAdd.push([genericPath, value]);
1610
+ // Handle non-generic function call paths like depName().functionCallReturnValue.x
1611
+ else if (path.startsWith(`${depName}().functionCallReturnValue.`)) {
1612
+ const suffix = path.slice(
1613
+ `${depName}().functionCallReturnValue.`.length,
1614
+ );
1615
+ for (const genericVariant of genericVariants) {
1616
+ const genericPath = `${genericVariant}.functionCallReturnValue.${suffix}`;
1617
+ if (!(genericPath in returnValueSchema)) {
1618
+ pathsToAdd.push([genericPath, value]);
1619
+ }
1575
1620
  }
1576
1621
  }
1577
1622
  }
1578
- }
1579
1623
 
1580
- // Add the new generic variant paths
1581
- for (const [path, value] of pathsToAdd) {
1582
- returnValueSchema[path] = value;
1624
+ // Add the new generic variant paths
1625
+ for (const [path, value] of pathsToAdd) {
1626
+ returnValueSchema[path] = value;
1627
+ }
1583
1628
  }
1584
1629
  }
1585
- }
1586
1630
 
1587
- // For mocked dependencies: copy paths from dependencySchemas (usage info) and normalize
1588
- // returnValue. paths that were created by equivalency processing.
1589
- // This ensures all paths use the consistent functionName().functionCallReturnValue. format.
1590
- for (const dependency of importedExports) {
1591
- if (!dependency.isMocked) continue;
1592
-
1593
- const srcSchema =
1594
- dependencySchemas?.[dependency.filePath]?.[dependency.name];
1595
- if (!srcSchema?.returnValueSchema) continue;
1596
-
1597
- const depSchema = findOrCreateDependentSchemas({
1598
- filePath: dependency.filePath,
1599
- name: dependency.name,
1600
- });
1601
-
1602
- // First, normalize any returnValue paths that were written by equivalency processing
1603
- // to the standard functionName().functionCallReturnValue format.
1604
- // This includes both returnValue. (dot) and returnValue[ (array) paths.
1605
- const pathsToNormalize: [string, string][] = [];
1606
- for (const path in depSchema.returnValueSchema) {
1607
- checkDeadline();
1608
- if (
1609
- path === 'returnValue' ||
1610
- path.startsWith('returnValue.') ||
1611
- path.startsWith('returnValue[')
1612
- ) {
1613
- pathsToNormalize.push([path, depSchema.returnValueSchema[path]]);
1614
- }
1615
- }
1616
- for (const [path, value] of pathsToNormalize) {
1617
- delete depSchema.returnValueSchema[path];
1618
- let normalizedPath: string;
1619
- if (path === 'returnValue') {
1620
- normalizedPath = `${dependency.name}().functionCallReturnValue`;
1621
- } else if (path.startsWith('returnValue.')) {
1622
- normalizedPath = path.replace(
1623
- /^returnValue\./,
1624
- `${dependency.name}().functionCallReturnValue.`,
1625
- );
1626
- } else {
1627
- // path.startsWith('returnValue[')
1628
- // e.g., returnValue[] -> getOptions().functionCallReturnValue[]
1629
- // e.g., returnValue[].label -> getOptions().functionCallReturnValue[].label
1630
- normalizedPath = path.replace(
1631
- /^returnValue/,
1632
- `${dependency.name}().functionCallReturnValue`,
1633
- );
1634
- }
1635
- transformationTracer.operation(rootScopeName, {
1636
- operation: 'normalizeReturnValuePath',
1637
- stage: 'merged',
1638
- path: normalizedPath,
1639
- before: path,
1640
- after: normalizedPath,
1641
- context: { dependency: dependency.name, value },
1642
- });
1643
- depSchema.returnValueSchema[normalizedPath] = value;
1644
- }
1631
+ // For mocked dependencies: copy paths from dependencySchemas (usage info) and normalize
1632
+ // returnValue. paths that were created by equivalency processing.
1633
+ // This ensures all paths use the consistent functionName().functionCallReturnValue. format.
1634
+ for (const dependency of importedExports) {
1635
+ if (!dependency.isMocked) continue;
1645
1636
 
1646
- // Now copy paths from the source schema (dependencySchemas)
1647
- for (const path in srcSchema.returnValueSchema) {
1648
- checkDeadline();
1649
- const value = srcSchema.returnValueSchema[path];
1650
-
1651
- // Normalize paths starting with 'returnValue' to use the standard format:
1652
- // 'returnValue.foo' -> 'dependencyName().functionCallReturnValue.foo'
1653
- // This ensures consistency across the codebase and allows constructMockCode
1654
- // and gatherDataForMocks to work correctly.
1655
- if (path === 'returnValue' || path.startsWith('returnValue.')) {
1656
- // Convert 'returnValue' -> 'name().functionCallReturnValue'
1657
- // Convert 'returnValue.foo' -> 'name().functionCallReturnValue.foo'
1658
- const normalizedPath =
1659
- path === 'returnValue'
1660
- ? `${dependency.name}().functionCallReturnValue`
1661
- : path.replace(
1662
- /^returnValue\./,
1663
- `${dependency.name}().functionCallReturnValue.`,
1664
- );
1637
+ const srcSchema =
1638
+ dependencySchemas?.[dependency.filePath]?.[dependency.name];
1639
+ if (!srcSchema?.returnValueSchema) continue;
1665
1640
 
1666
- // Always write srcSchema values - they take precedence over equivalency-derived values
1667
- depSchema.returnValueSchema[normalizedPath] = value;
1668
- continue;
1669
- }
1641
+ const depSchema = findOrCreateDependentSchemas({
1642
+ filePath: dependency.filePath,
1643
+ name: dependency.name,
1644
+ });
1670
1645
 
1671
- // Copy paths containing functionCallReturnValue (return value structures)
1672
- // These are needed for constructMockCode to build the proper mock data hierarchy
1673
- // Example: supabase.auth.getSession().functionCallReturnValue.data.session
1674
- if (path.includes('.functionCallReturnValue')) {
1675
- // Always write srcSchema values - they take precedence over equivalency-derived values
1676
- depSchema.returnValueSchema[path] = value;
1677
- continue;
1646
+ // First, normalize any returnValue paths that were written by equivalency processing
1647
+ // to the standard functionName().functionCallReturnValue format.
1648
+ // This includes both returnValue. (dot) and returnValue[ (array) paths.
1649
+ const pathsToNormalize: [string, string][] = [];
1650
+ for (const path in depSchema.returnValueSchema) {
1651
+ checkDeadline();
1652
+ if (
1653
+ path === 'returnValue' ||
1654
+ path.startsWith('returnValue.') ||
1655
+ path.startsWith('returnValue[')
1656
+ ) {
1657
+ pathsToNormalize.push([path, depSchema.returnValueSchema[path]]);
1658
+ }
1678
1659
  }
1679
-
1680
- // Copy function-typed paths that end with () (are function calls)
1681
- // These include:
1682
- // - Function stubs without functionCallReturnValue (like onAuthStateChange)
1683
- // - Function markers with async-function type (like getSession(): async-function)
1684
- // which are needed for constructMockCode to know to generate async functions
1685
- // Skip paths starting with 'returnValue' - they were already handled above
1686
- if (
1687
- ['function', 'async-function'].includes(value) &&
1688
- path.endsWith(')') &&
1689
- !path.startsWith('returnValue')
1690
- ) {
1691
- if (!(path in depSchema.returnValueSchema)) {
1692
- depSchema.returnValueSchema[path] = value;
1660
+ for (const [path, value] of pathsToNormalize) {
1661
+ delete depSchema.returnValueSchema[path];
1662
+ let normalizedPath: string;
1663
+ if (path === 'returnValue') {
1664
+ normalizedPath = `${dependency.name}().functionCallReturnValue`;
1665
+ } else if (path.startsWith('returnValue.')) {
1666
+ normalizedPath = path.replace(
1667
+ /^returnValue\./,
1668
+ `${dependency.name}().functionCallReturnValue.`,
1669
+ );
1670
+ } else {
1671
+ // path.startsWith('returnValue[')
1672
+ // e.g., returnValue[] -> getOptions().functionCallReturnValue[]
1673
+ // e.g., returnValue[].label -> getOptions().functionCallReturnValue[].label
1674
+ normalizedPath = path.replace(
1675
+ /^returnValue/,
1676
+ `${dependency.name}().functionCallReturnValue`,
1677
+ );
1693
1678
  }
1679
+ transformationTracer.operation(rootScopeName, {
1680
+ operation: 'normalizeReturnValuePath',
1681
+ stage: 'merged',
1682
+ path: normalizedPath,
1683
+ before: path,
1684
+ after: normalizedPath,
1685
+ context: { dependency: dependency.name, value },
1686
+ });
1687
+ depSchema.returnValueSchema[normalizedPath] = value;
1694
1688
  }
1695
1689
 
1696
- // Copy object-typed paths for chained API access patterns (like trpc.customer.getCustomersByOrg)
1697
- // These intermediate paths are needed for constructMockCode to build the nested mock structure.
1698
- // Example: for trpc.customer.getCustomersByOrg.useQuery().functionCallReturnValue.data,
1699
- // we need 'trpc', 'trpc.customer', 'trpc.customer.getCustomersByOrg' all typed as 'object'.
1700
- // Skip paths starting with 'returnValue' - they were already handled above
1701
- //
1702
- // EXCEPTION: For function-style dependencies like getSupabase(), skip intermediate object
1703
- // paths like 'getSupabase().auth' that are just property access after a function call.
1704
- // These aren't needed because constructMockCode can infer the structure from the actual
1705
- // function call paths like 'getSupabase().auth.getUser()'. We only need object paths
1706
- // for object-style dependencies like 'supabase.auth' where the dependency itself is an object.
1707
- if (value === 'object' && !path.startsWith('returnValue')) {
1708
- // Check if this is a function-style dependency (path starts with name() or name<T>())
1709
- const isFunctionStyleDependency =
1710
- path.startsWith(`${dependency.name}()`) ||
1711
- path.match(new RegExp(`^${dependency.name}<[^>]+>\\(\\)`));
1712
-
1713
- // For function-style dependencies, skip intermediate object paths
1714
- // Only keep object paths that are within functionCallReturnValue
1715
- if (
1716
- isFunctionStyleDependency &&
1717
- !path.includes('.functionCallReturnValue')
1718
- ) {
1690
+ // Now copy paths from the source schema (dependencySchemas)
1691
+ for (const path in srcSchema.returnValueSchema) {
1692
+ checkDeadline();
1693
+ const value = srcSchema.returnValueSchema[path];
1694
+
1695
+ // Normalize paths starting with 'returnValue' to use the standard format:
1696
+ // 'returnValue.foo' -> 'dependencyName().functionCallReturnValue.foo'
1697
+ // This ensures consistency across the codebase and allows constructMockCode
1698
+ // and gatherDataForMocks to work correctly.
1699
+ if (path === 'returnValue' || path.startsWith('returnValue.')) {
1700
+ // Convert 'returnValue' -> 'name().functionCallReturnValue'
1701
+ // Convert 'returnValue.foo' -> 'name().functionCallReturnValue.foo'
1702
+ const normalizedPath =
1703
+ path === 'returnValue'
1704
+ ? `${dependency.name}().functionCallReturnValue`
1705
+ : path.replace(
1706
+ /^returnValue\./,
1707
+ `${dependency.name}().functionCallReturnValue.`,
1708
+ );
1709
+
1710
+ // Always write srcSchema values - they take precedence over equivalency-derived values
1711
+ depSchema.returnValueSchema[normalizedPath] = value;
1719
1712
  continue;
1720
1713
  }
1721
1714
 
1722
- if (!(path in depSchema.returnValueSchema)) {
1715
+ // Copy paths containing functionCallReturnValue (return value structures)
1716
+ // These are needed for constructMockCode to build the proper mock data hierarchy
1717
+ // Example: supabase.auth.getSession().functionCallReturnValue.data.session
1718
+ if (path.includes('.functionCallReturnValue')) {
1719
+ // Always write srcSchema values - they take precedence over equivalency-derived values
1723
1720
  depSchema.returnValueSchema[path] = value;
1721
+ continue;
1724
1722
  }
1725
- }
1726
- }
1727
1723
 
1728
- cleanSchema(depSchema.returnValueSchema, {
1729
- stage: 'afterMockedDependencyMerge',
1730
- dependency: dependency.name,
1731
- });
1732
-
1733
- // Pull signature requirements from downstream functions into the mocked return value.
1734
- // When a mocked function's return flows into another function's signature (via usageEquivalencies),
1735
- // we need to include that function's signature requirements in the mock.
1736
- //
1737
- // Example: fromE5() returns a currency object that flows to calculateTotalPrice(price, quantity).
1738
- // calculateTotalPrice's signatureSchema shows signature[0].multiply() is required.
1739
- // We need to add multiply() to fromE5's mock return value.
1740
- const usageEquivalencies = srcSchema.usageEquivalencies ?? {};
1741
- for (const [returnPath, equivalencies] of Object.entries(
1742
- usageEquivalencies,
1743
- )) {
1744
- // Only process return value paths (functionCallReturnValue)
1745
- if (!returnPath.includes('.functionCallReturnValue')) continue;
1746
-
1747
- for (const equiv of equivalencies) {
1748
- // Check if this equivalency points to a signature path
1749
- const signatureMatch = equiv.schemaPath.match(/\.signature\[(\d+)\]$/);
1750
- if (!signatureMatch) continue;
1751
-
1752
- const targetFunctionName = cleanFunctionName(equiv.scopeNodeName);
1753
- const signatureIndex = signatureMatch[1];
1754
-
1755
- // Look up the target function's analysis to get its signature requirements
1756
- // First try dependentAnalyses, then dependencySchemas
1757
- let targetSignatureSchema: Record<string, string> | undefined;
1758
-
1759
- // Check dependentAnalyses first (has the full merged analysis)
1760
- for (const depFilePath in dependentAnalyses) {
1761
- const analysis = dependentAnalyses[depFilePath]?.[targetFunctionName];
1762
- if (analysis?.metadata?.mergedDataStructure?.signatureSchema) {
1763
- targetSignatureSchema =
1764
- analysis.metadata.mergedDataStructure.signatureSchema;
1765
- break;
1724
+ // Copy function-typed paths that end with () (are function calls)
1725
+ // These include:
1726
+ // - Function stubs without functionCallReturnValue (like onAuthStateChange)
1727
+ // - Function markers with async-function type (like getSession(): async-function)
1728
+ // which are needed for constructMockCode to know to generate async functions
1729
+ // Skip paths starting with 'returnValue' - they were already handled above
1730
+ if (
1731
+ ['function', 'async-function'].includes(value) &&
1732
+ path.endsWith(')') &&
1733
+ !path.startsWith('returnValue')
1734
+ ) {
1735
+ if (!(path in depSchema.returnValueSchema)) {
1736
+ depSchema.returnValueSchema[path] = value;
1766
1737
  }
1767
1738
  }
1768
1739
 
1769
- // Fallback to dependencySchemas if not found
1770
- if (!targetSignatureSchema) {
1771
- for (const depFilePath in dependencySchemas) {
1772
- const schema = dependencySchemas[depFilePath]?.[targetFunctionName];
1773
- if (schema?.signatureSchema) {
1774
- targetSignatureSchema = schema.signatureSchema;
1775
- break;
1776
- }
1740
+ // Copy object-typed paths for chained API access patterns (like trpc.customer.getCustomersByOrg)
1741
+ // These intermediate paths are needed for constructMockCode to build the nested mock structure.
1742
+ // Example: for trpc.customer.getCustomersByOrg.useQuery().functionCallReturnValue.data,
1743
+ // we need 'trpc', 'trpc.customer', 'trpc.customer.getCustomersByOrg' all typed as 'object'.
1744
+ // Skip paths starting with 'returnValue' - they were already handled above
1745
+ //
1746
+ // EXCEPTION: For function-style dependencies like getSupabase(), skip intermediate object
1747
+ // paths like 'getSupabase().auth' that are just property access after a function call.
1748
+ // These aren't needed because constructMockCode can infer the structure from the actual
1749
+ // function call paths like 'getSupabase().auth.getUser()'. We only need object paths
1750
+ // for object-style dependencies like 'supabase.auth' where the dependency itself is an object.
1751
+ if (value === 'object' && !path.startsWith('returnValue')) {
1752
+ // Check if this is a function-style dependency (path starts with name() or name<T>())
1753
+ const isFunctionStyleDependency =
1754
+ path.startsWith(`${dependency.name}()`) ||
1755
+ path.match(new RegExp(`^${dependency.name}<[^>]+>\\(\\)`));
1756
+
1757
+ // For function-style dependencies, skip intermediate object paths
1758
+ // Only keep object paths that are within functionCallReturnValue
1759
+ if (
1760
+ isFunctionStyleDependency &&
1761
+ !path.includes('.functionCallReturnValue')
1762
+ ) {
1763
+ continue;
1777
1764
  }
1778
- }
1779
-
1780
- if (!targetSignatureSchema) continue;
1781
1765
 
1782
- // Find all paths in the target's signatureSchema that extend from signature[N]
1783
- // e.g., signature[0].multiply(quantity) -> .multiply(quantity)
1784
- const signaturePrefix = `signature[${signatureIndex}]`;
1785
- for (const [sigPath, sigType] of Object.entries(
1786
- targetSignatureSchema,
1787
- )) {
1788
- if (!sigPath.startsWith(signaturePrefix)) continue;
1766
+ if (!(path in depSchema.returnValueSchema)) {
1767
+ depSchema.returnValueSchema[path] = value;
1768
+ }
1769
+ }
1770
+ }
1789
1771
 
1790
- // Skip the base signature[N] path itself - we only want the method/property extensions
1791
- if (sigPath === signaturePrefix) continue;
1772
+ cleanSchema(depSchema.returnValueSchema, {
1773
+ stage: 'afterMockedDependencyMerge',
1774
+ dependency: dependency.name,
1775
+ });
1792
1776
 
1793
- // Extract the suffix after signature[N] (e.g., ".multiply(quantity)")
1794
- const suffix = sigPath.slice(signaturePrefix.length);
1777
+ // Pull signature requirements from downstream functions into the mocked return value.
1778
+ // When a mocked function's return flows into another function's signature (via usageEquivalencies),
1779
+ // we need to include that function's signature requirements in the mock.
1780
+ //
1781
+ // Example: fromE5() returns a currency object that flows to calculateTotalPrice(price, quantity).
1782
+ // calculateTotalPrice's signatureSchema shows signature[0].multiply() is required.
1783
+ // We need to add multiply() to fromE5's mock return value.
1784
+ const usageEquivalencies = srcSchema.usageEquivalencies ?? {};
1785
+ for (const [returnPath, equivalencies] of Object.entries(
1786
+ usageEquivalencies,
1787
+ )) {
1788
+ // Only process return value paths (functionCallReturnValue)
1789
+ if (!returnPath.includes('.functionCallReturnValue')) continue;
1795
1790
 
1796
- // Build the path for the mocked return value
1797
- // e.g., fromE5(priceE5).functionCallReturnValue.multiply(quantity)
1798
- const returnValuePath = returnPath + suffix;
1791
+ for (const equiv of equivalencies) {
1792
+ // Check if this equivalency points to a signature path
1793
+ const signatureMatch = equiv.schemaPath.match(
1794
+ /\.signature\[(\d+)\]$/,
1795
+ );
1796
+ if (!signatureMatch) continue;
1797
+
1798
+ const targetFunctionName = cleanFunctionName(equiv.scopeNodeName);
1799
+ const signatureIndex = signatureMatch[1];
1800
+
1801
+ // Look up the target function's analysis to get its signature requirements
1802
+ // First try dependentAnalyses, then dependencySchemas
1803
+ let targetSignatureSchema: Record<string, string> | undefined;
1804
+
1805
+ // Check dependentAnalyses first (has the full merged analysis)
1806
+ for (const depFilePath in dependentAnalyses) {
1807
+ const analysis =
1808
+ dependentAnalyses[depFilePath]?.[targetFunctionName];
1809
+ if (analysis?.metadata?.mergedDataStructure?.signatureSchema) {
1810
+ targetSignatureSchema =
1811
+ analysis.metadata.mergedDataStructure.signatureSchema;
1812
+ break;
1813
+ }
1814
+ }
1799
1815
 
1800
- // Add to the mocked dependency's return value schema if not already present
1801
- if (!(returnValuePath in depSchema.returnValueSchema)) {
1802
- depSchema.returnValueSchema[returnValuePath] = sigType;
1816
+ // Fallback to dependencySchemas if not found
1817
+ if (!targetSignatureSchema) {
1818
+ for (const depFilePath in dependencySchemas) {
1819
+ const schema =
1820
+ dependencySchemas[depFilePath]?.[targetFunctionName];
1821
+ if (schema?.signatureSchema) {
1822
+ targetSignatureSchema = schema.signatureSchema;
1823
+ break;
1824
+ }
1825
+ }
1803
1826
  }
1804
- }
1805
- }
1806
- }
1807
1827
 
1808
- cleanSchema(depSchema.returnValueSchema, {
1809
- stage: 'afterSignatureRequirementsMerge',
1810
- dependency: dependency.name,
1811
- });
1812
- }
1828
+ if (!targetSignatureSchema) continue;
1813
1829
 
1814
- // Process the input dependencySchemas FIRST (before child dependentAnalyses).
1815
- // This ensures the parent entity's direct usage of dependencies takes precedence.
1816
- // When both parent and child use the same dependency (e.g., useLoaderData),
1817
- // the parent's schema paths are preserved, and child's paths are merged in later.
1818
- //
1819
- // Some dependencies (like .d.ts type declaration files) may not have:
1820
- // - Equivalencies with the root scope
1821
- // - A dependent analysis (they're just type declarations)
1822
- // - Be marked as mocked
1823
- // Without this, their schemas would be lost entirely.
1824
- for (const filePath in dependencySchemas) {
1825
- for (const name in dependencySchemas[filePath]) {
1826
- const srcSchema = dependencySchemas[filePath][name];
1827
- if (!srcSchema) continue;
1828
-
1829
- // Skip mocked dependencies - they were already processed above with path normalization
1830
- if (mockedDependencies.has(`${filePath}::${name}`)) {
1831
- continue;
1832
- }
1830
+ // Find all paths in the target's signatureSchema that extend from signature[N]
1831
+ // e.g., signature[0].multiply(quantity) -> .multiply(quantity)
1832
+ const signaturePrefix = `signature[${signatureIndex}]`;
1833
+ for (const [sigPath, sigType] of Object.entries(
1834
+ targetSignatureSchema,
1835
+ )) {
1836
+ if (!sigPath.startsWith(signaturePrefix)) continue;
1833
1837
 
1834
- // Check if this dependency was already processed by equivalencies
1835
- const existingSchema =
1836
- mergedDataStructure.dependencySchemas[filePath]?.[name];
1838
+ // Skip the base signature[N] path itself - we only want the method/property extensions
1839
+ if (sigPath === signaturePrefix) continue;
1837
1840
 
1838
- // Only add if no existing schema (equivalencies didn't process it)
1839
- if (!existingSchema) {
1840
- const depSchema = findOrCreateDependentSchemas({ filePath, name });
1841
- for (const path in srcSchema.returnValueSchema) {
1842
- checkDeadline();
1843
- depSchema.returnValueSchema[path] = srcSchema.returnValueSchema[path];
1844
- }
1845
- for (const path in srcSchema.signatureSchema) {
1846
- checkDeadline();
1847
- depSchema.signatureSchema[path] = srcSchema.signatureSchema[path];
1848
- }
1841
+ // Extract the suffix after signature[N] (e.g., ".multiply(quantity)")
1842
+ const suffix = sigPath.slice(signaturePrefix.length);
1849
1843
 
1850
- // Clean known object functions (like String.prototype.replace, Array.prototype.map)
1851
- // from the copied schema. Without this, method call paths on primitives like
1852
- // "projectSlug.replace(...)" would cause convertDotNotation to create nested
1853
- // object structures instead of preserving the primitive type.
1854
- cleanSchema(depSchema.returnValueSchema, {
1855
- stage: 'afterDependencySchemaCopy',
1856
- filePath,
1857
- dependency: name,
1858
- });
1859
- }
1844
+ // Build the path for the mocked return value
1845
+ // e.g., fromE5(priceE5).functionCallReturnValue.multiply(quantity)
1846
+ const returnValuePath = returnPath + suffix;
1860
1847
 
1861
- // TYPE REFINEMENT: Check if dependentAnalyses has a more specific type for this dependency.
1862
- // When a parent passes `entity.filePath` (string | undefined) to a child component
1863
- // that requires `filePath: string`, we should use the child's more specific type.
1864
- // This prevents mock data from having undefined values for required props.
1865
- //
1866
- // This runs REGARDLESS of whether equivalencies already processed the schema,
1867
- // because equivalencies copy the parent's type (string | undefined), not the child's
1868
- // required type (string).
1869
- const depSchema = findOrCreateDependentSchemas({ filePath, name });
1870
- const childAnalysis = dependentAnalyses[filePath]?.[name];
1871
- const childSignatureSchema =
1872
- childAnalysis?.metadata?.mergedDataStructure?.signatureSchema;
1873
-
1874
- if (childSignatureSchema) {
1875
- for (const path in depSchema.signatureSchema) {
1876
- checkDeadline();
1877
- const parentType = depSchema.signatureSchema[path];
1878
- const childType = childSignatureSchema[path];
1879
-
1880
- if (parentType && childType) {
1881
- // Check if parent has optional type and child has required type
1882
- const parentIsOptional =
1883
- parentType.includes('| undefined') ||
1884
- parentType.includes('| null');
1885
- const childIsOptional =
1886
- childType.includes('| undefined') || childType.includes('| null');
1887
-
1888
- // If child requires a more specific type (not optional), use it
1889
- if (parentIsOptional && !childIsOptional) {
1890
- depSchema.signatureSchema[path] = childType;
1848
+ // Add to the mocked dependency's return value schema if not already present
1849
+ if (!(returnValuePath in depSchema.returnValueSchema)) {
1850
+ depSchema.returnValueSchema[returnValuePath] = sigType;
1891
1851
  }
1892
1852
  }
1893
1853
  }
1894
1854
  }
1895
1855
 
1896
- // For functions with multiple different type parameters, also create separate entries
1897
- // for each type-parameterized variant. This allows gatherDataForMocks to look up
1898
- // the specific schema for each call signature.
1899
- // This runs regardless of whether the base entry already existed, since we need
1900
- // the separate variant entries for proper schema lookup.
1901
- const baseName = cleanFunctionName(name);
1902
- if (functionsWithMultipleTypeParams.has(baseName)) {
1903
- // Find all unique type-parameterized call signatures in the schema
1904
- const typeParamVariants = new Set<string>();
1905
- for (const path of Object.keys(srcSchema.returnValueSchema)) {
1906
- const parts = splitOutsideParenthesesAndArrays(path);
1907
- if (
1908
- parts.length > 0 &&
1909
- parts[0].includes('<') &&
1910
- parts[0].endsWith(')')
1911
- ) {
1912
- typeParamVariants.add(parts[0]);
1913
- }
1856
+ cleanSchema(depSchema.returnValueSchema, {
1857
+ stage: 'afterSignatureRequirementsMerge',
1858
+ dependency: dependency.name,
1859
+ });
1860
+ }
1861
+
1862
+ // Process the input dependencySchemas FIRST (before child dependentAnalyses).
1863
+ // This ensures the parent entity's direct usage of dependencies takes precedence.
1864
+ // When both parent and child use the same dependency (e.g., useLoaderData),
1865
+ // the parent's schema paths are preserved, and child's paths are merged in later.
1866
+ //
1867
+ // Some dependencies (like .d.ts type declaration files) may not have:
1868
+ // - Equivalencies with the root scope
1869
+ // - A dependent analysis (they're just type declarations)
1870
+ // - Be marked as mocked
1871
+ // Without this, their schemas would be lost entirely.
1872
+ for (const filePath in dependencySchemas) {
1873
+ for (const name in dependencySchemas[filePath]) {
1874
+ const srcSchema = dependencySchemas[filePath][name];
1875
+ if (!srcSchema) continue;
1876
+
1877
+ // Skip mocked dependencies - they were already processed above with path normalization
1878
+ if (mockedDependencies.has(`${filePath}::${name}`)) {
1879
+ continue;
1914
1880
  }
1915
1881
 
1916
- // Create a separate entry for each type-parameterized variant
1917
- for (const variant of typeParamVariants) {
1918
- const variantSchema = findOrCreateDependentSchemas({
1919
- filePath,
1920
- name: variant,
1921
- });
1882
+ // Check if this dependency was already processed by equivalencies
1883
+ const existingSchema =
1884
+ mergedDataStructure.dependencySchemas[filePath]?.[name];
1922
1885
 
1923
- // Copy only paths that belong to this variant
1886
+ // Only add if no existing schema (equivalencies didn't process it)
1887
+ if (!existingSchema) {
1888
+ const depSchema = findOrCreateDependentSchemas({ filePath, name });
1924
1889
  for (const path in srcSchema.returnValueSchema) {
1925
1890
  checkDeadline();
1926
- if (path.startsWith(variant)) {
1927
- variantSchema.returnValueSchema[path] =
1928
- srcSchema.returnValueSchema[path];
1929
- }
1891
+ depSchema.returnValueSchema[path] =
1892
+ srcSchema.returnValueSchema[path];
1893
+ }
1894
+ for (const path in srcSchema.signatureSchema) {
1895
+ checkDeadline();
1896
+ depSchema.signatureSchema[path] = srcSchema.signatureSchema[path];
1930
1897
  }
1931
- cleanSchema(variantSchema.returnValueSchema, {
1932
- stage: 'afterTypeVariantCopy',
1898
+
1899
+ // Clean known object functions (like String.prototype.replace, Array.prototype.map)
1900
+ // from the copied schema. Without this, method call paths on primitives like
1901
+ // "projectSlug.replace(...)" would cause convertDotNotation to create nested
1902
+ // object structures instead of preserving the primitive type.
1903
+ cleanSchema(depSchema.returnValueSchema, {
1904
+ stage: 'afterDependencySchemaCopy',
1933
1905
  filePath,
1934
1906
  dependency: name,
1935
- variant,
1936
1907
  });
1937
1908
  }
1938
- }
1939
- }
1940
- }
1941
-
1942
- // Ensure ALL dependencies from dependentAnalyses are included in dependencySchemas,
1943
- // even if they have no equivalencies with the root scope.
1944
- // This preserves nested functionCallReturnValue paths that would otherwise be lost.
1945
- // EXCEPT: Skip mocked dependencies - we don't want their internal implementation details.
1946
- for (const filePath in dependentAnalyses) {
1947
- for (const name in dependentAnalyses[filePath]) {
1948
- checkDeadline();
1949
- const dependentMergedDataStructure =
1950
- dependentAnalyses[filePath][name].metadata?.mergedDataStructure;
1951
-
1952
- if (!dependentMergedDataStructure) continue;
1953
1909
 
1954
- const isMocked = mockedDependencies.has(`${filePath}::${name}`);
1955
-
1956
- // For mocked dependencies: ONLY copy nested dependencySchemas (skip internal implementation)
1957
- // For non-mocked dependencies: copy everything (signature, returnValue, and nested dependencySchemas)
1958
- if (!isMocked) {
1959
- // Create the dependency schema entry if it doesn't exist
1910
+ // TYPE REFINEMENT: Check if dependentAnalyses has a more specific type for this dependency.
1911
+ // When a parent passes `entity.filePath` (string | undefined) to a child component
1912
+ // that requires `filePath: string`, we should use the child's more specific type.
1913
+ // This prevents mock data from having undefined values for required props.
1914
+ //
1915
+ // This runs REGARDLESS of whether equivalencies already processed the schema,
1916
+ // because equivalencies copy the parent's type (string | undefined), not the child's
1917
+ // required type (string).
1960
1918
  const depSchema = findOrCreateDependentSchemas({ filePath, name });
1919
+ const childAnalysis = dependentAnalyses[filePath]?.[name];
1920
+ const childSignatureSchema =
1921
+ childAnalysis?.metadata?.mergedDataStructure?.signatureSchema;
1961
1922
 
1962
- // Copy over all paths from the dependent's returnValueSchema
1963
- // Only add paths that don't already exist (don't overwrite values set by equivalencies)
1964
- for (const path in dependentMergedDataStructure.returnValueSchema) {
1965
- checkDeadline();
1966
- const translatedPath = translatePath(path, name);
1967
- if (!(translatedPath in depSchema.returnValueSchema)) {
1968
- depSchema.returnValueSchema[translatedPath] =
1969
- dependentMergedDataStructure.returnValueSchema[path];
1923
+ if (childSignatureSchema) {
1924
+ for (const path in depSchema.signatureSchema) {
1925
+ checkDeadline();
1926
+ const parentType = depSchema.signatureSchema[path];
1927
+ const childType = childSignatureSchema[path];
1928
+
1929
+ if (parentType && childType) {
1930
+ // Check if parent has optional type and child has required type
1931
+ const parentIsOptional =
1932
+ parentType.includes('| undefined') ||
1933
+ parentType.includes('| null');
1934
+ const childIsOptional =
1935
+ childType.includes('| undefined') ||
1936
+ childType.includes('| null');
1937
+
1938
+ // If child requires a more specific type (not optional), use it
1939
+ if (parentIsOptional && !childIsOptional) {
1940
+ depSchema.signatureSchema[path] = childType;
1941
+ }
1942
+ }
1970
1943
  }
1971
1944
  }
1972
1945
 
1973
- // Copy over signature schema as well
1974
- for (const path in dependentMergedDataStructure.signatureSchema) {
1975
- checkDeadline();
1976
- const translatedPath = translatePath(path, name);
1977
- if (!(translatedPath in depSchema.signatureSchema)) {
1978
- depSchema.signatureSchema[translatedPath] =
1979
- dependentMergedDataStructure.signatureSchema[path];
1946
+ // For functions with multiple different type parameters, also create separate entries
1947
+ // for each type-parameterized variant. This allows gatherDataForMocks to look up
1948
+ // the specific schema for each call signature.
1949
+ // This runs regardless of whether the base entry already existed, since we need
1950
+ // the separate variant entries for proper schema lookup.
1951
+ const baseName = cleanFunctionName(name);
1952
+ if (functionsWithMultipleTypeParams.has(baseName)) {
1953
+ // Find all unique type-parameterized call signatures in the schema
1954
+ const typeParamVariants = new Set<string>();
1955
+ for (const path of Object.keys(srcSchema.returnValueSchema)) {
1956
+ const parts = splitOutsideParenthesesAndArrays(path);
1957
+ if (
1958
+ parts.length > 0 &&
1959
+ parts[0].includes('<') &&
1960
+ parts[0].endsWith(')')
1961
+ ) {
1962
+ typeParamVariants.add(parts[0]);
1963
+ }
1980
1964
  }
1981
- }
1982
- }
1983
1965
 
1984
- // Copy nested dependencySchemas for ALL entities (including mocked ones)
1985
- // This represents what dependencies THIS entity uses, not its internal implementation
1986
- if (dependentMergedDataStructure.dependencySchemas) {
1987
- for (const depFilePath in dependentMergedDataStructure.dependencySchemas) {
1988
- for (const depName in dependentMergedDataStructure.dependencySchemas[
1989
- depFilePath
1990
- ]) {
1991
- const nestedDepSchema =
1992
- dependentMergedDataStructure.dependencySchemas[depFilePath][
1993
- depName
1994
- ];
1995
- const targetDepSchema = findOrCreateDependentSchemas({
1996
- filePath: depFilePath,
1997
- name: depName,
1966
+ // Create a separate entry for each type-parameterized variant
1967
+ for (const variant of typeParamVariants) {
1968
+ const variantSchema = findOrCreateDependentSchemas({
1969
+ filePath,
1970
+ name: variant,
1998
1971
  });
1999
1972
 
2000
- // Merge in the nested dependency schemas
2001
- for (const path in nestedDepSchema.returnValueSchema) {
1973
+ // Copy only paths that belong to this variant
1974
+ for (const path in srcSchema.returnValueSchema) {
2002
1975
  checkDeadline();
2003
- if (!(path in targetDepSchema.returnValueSchema)) {
2004
- const value = nestedDepSchema.returnValueSchema[path];
2005
- targetDepSchema.returnValueSchema[path] = value;
1976
+ if (path.startsWith(variant)) {
1977
+ variantSchema.returnValueSchema[path] =
1978
+ srcSchema.returnValueSchema[path];
2006
1979
  }
2007
1980
  }
1981
+ cleanSchema(variantSchema.returnValueSchema, {
1982
+ stage: 'afterTypeVariantCopy',
1983
+ filePath,
1984
+ dependency: name,
1985
+ variant,
1986
+ });
1987
+ }
1988
+ }
1989
+ }
1990
+ }
2008
1991
 
2009
- for (const path in nestedDepSchema.signatureSchema) {
2010
- checkDeadline();
2011
- if (!(path in targetDepSchema.signatureSchema)) {
2012
- targetDepSchema.signatureSchema[path] =
2013
- nestedDepSchema.signatureSchema[path];
1992
+ // Ensure ALL dependencies from dependentAnalyses are included in dependencySchemas,
1993
+ // even if they have no equivalencies with the root scope.
1994
+ // This preserves nested functionCallReturnValue paths that would otherwise be lost.
1995
+ // EXCEPT: Skip mocked dependencies - we don't want their internal implementation details.
1996
+ for (const filePath in dependentAnalyses) {
1997
+ for (const name in dependentAnalyses[filePath]) {
1998
+ checkDeadline();
1999
+ const dependentMergedDataStructure =
2000
+ dependentAnalyses[filePath][name].metadata?.mergedDataStructure;
2001
+
2002
+ if (!dependentMergedDataStructure) continue;
2003
+
2004
+ const isMocked = mockedDependencies.has(`${filePath}::${name}`);
2005
+
2006
+ // For mocked dependencies: ONLY copy nested dependencySchemas (skip internal implementation)
2007
+ // For non-mocked dependencies: copy everything (signature, returnValue, and nested dependencySchemas)
2008
+ if (!isMocked) {
2009
+ // Create the dependency schema entry if it doesn't exist
2010
+ const depSchema = findOrCreateDependentSchemas({ filePath, name });
2011
+
2012
+ // Copy over paths from the dependent's returnValueSchema.
2013
+ // Only add paths that don't already exist (don't overwrite values set by equivalencies).
2014
+ // Skip if either source or target exceeds the cap — copying 2,531 paths from
2015
+ // ArticleTable with translatePath on each takes ~1.5s for one entity.
2016
+ const srcRetSize = Object.keys(
2017
+ dependentMergedDataStructure.returnValueSchema || {},
2018
+ ).length;
2019
+ if (
2020
+ srcRetSize > SCHEMA_KEY_CAP ||
2021
+ Object.keys(depSchema.returnValueSchema).length > SCHEMA_KEY_CAP
2022
+ )
2023
+ continue;
2024
+ for (const path in dependentMergedDataStructure.returnValueSchema) {
2025
+ // Fast path: only call translatePath when the path starts with the
2026
+ // dependency name (e.g., "ArticleTable().functionCallReturnValue.x").
2027
+ // Most paths start with "returnValue" or "signature" and don't need translation.
2028
+ const translatedPath = path.startsWith(name)
2029
+ ? translatePath(path, name)
2030
+ : path;
2031
+ if (!(translatedPath in depSchema.returnValueSchema)) {
2032
+ depSchema.returnValueSchema[translatedPath] =
2033
+ dependentMergedDataStructure.returnValueSchema[path];
2034
+ }
2035
+ }
2036
+
2037
+ // Copy over signature schema as well
2038
+ for (const path in dependentMergedDataStructure.signatureSchema) {
2039
+ const translatedPath = path.startsWith(name)
2040
+ ? translatePath(path, name)
2041
+ : path;
2042
+ if (!(translatedPath in depSchema.signatureSchema)) {
2043
+ depSchema.signatureSchema[translatedPath] =
2044
+ dependentMergedDataStructure.signatureSchema[path];
2045
+ }
2046
+ }
2047
+ }
2048
+
2049
+ // Copy nested dependencySchemas for ALL entities (including mocked ones)
2050
+ // This represents what dependencies THIS entity uses, not its internal implementation
2051
+ if (dependentMergedDataStructure.dependencySchemas) {
2052
+ for (const depFilePath in dependentMergedDataStructure.dependencySchemas) {
2053
+ for (const depName in dependentMergedDataStructure
2054
+ .dependencySchemas[depFilePath]) {
2055
+ const nestedDepSchema =
2056
+ dependentMergedDataStructure.dependencySchemas[depFilePath][
2057
+ depName
2058
+ ];
2059
+ const targetDepSchema = findOrCreateDependentSchemas({
2060
+ filePath: depFilePath,
2061
+ name: depName,
2062
+ });
2063
+
2064
+ // Merge in the nested dependency schemas
2065
+ for (const path in nestedDepSchema.returnValueSchema) {
2066
+ checkDeadline();
2067
+ if (!(path in targetDepSchema.returnValueSchema)) {
2068
+ const value = nestedDepSchema.returnValueSchema[path];
2069
+ targetDepSchema.returnValueSchema[path] = value;
2070
+ }
2071
+ }
2072
+
2073
+ for (const path in nestedDepSchema.signatureSchema) {
2074
+ checkDeadline();
2075
+ if (!(path in targetDepSchema.signatureSchema)) {
2076
+ targetDepSchema.signatureSchema[path] =
2077
+ nestedDepSchema.signatureSchema[path];
2078
+ }
2014
2079
  }
2015
2080
  }
2016
2081
  }
2017
2082
  }
2018
2083
  }
2019
2084
  }
2020
- }
2021
2085
 
2022
- const totalElapsed = Date.now() - mergeStartTime;
2023
- const retKeys = Object.keys(mergedDataStructure.returnValueSchema).length;
2086
+ const totalElapsed = Date.now() - mergeStartTime;
2087
+ const retKeys = Object.keys(mergedDataStructure.returnValueSchema).length;
2024
2088
 
2025
- // Only log phase breakdown for slow merges (>2s)
2026
- if (totalElapsed > 2000) {
2027
- console.log(
2028
- `CodeYam Log Level 2: ${rootScopeName} merge phases: gather=${gatherElapsed}ms mergeESP=${mergeEspElapsed - gatherElapsed}ms postfix=${postfixElapsed - mergeEspElapsed}ms clean=${cleanElapsed - postfixElapsed}ms depCopy=${totalElapsed - cleanElapsed}ms total=${totalElapsed}ms ret=${retKeys}`,
2029
- );
2030
- }
2089
+ // Only log phase breakdown for slow merges (>2s)
2090
+ if (totalElapsed > 2000) {
2091
+ console.log(
2092
+ `CodeYam Log Level 2: ${rootScopeName} merge phases: gather=${gatherElapsed}ms mergeESP=${mergeEspElapsed - gatherElapsed}ms postfix=${postfixElapsed - mergeEspElapsed}ms clean=${cleanElapsed - postfixElapsed}ms depCopy=${totalElapsed - cleanElapsed}ms total=${totalElapsed}ms ret=${retKeys}`,
2093
+ );
2094
+ }
2031
2095
 
2032
- return mergedDataStructure;
2096
+ return mergedDataStructure;
2097
+ } catch (error) {
2098
+ if (error instanceof DataStructureTimeoutError) {
2099
+ // Return partial results instead of propagating the timeout.
2100
+ // By this point, mergedDataStructure has valid data from completed phases
2101
+ // (gather + mergeESP complete in <1s, postfix/clean/depCopy may be partial).
2102
+ const retKeys = Object.keys(mergedDataStructure.returnValueSchema).length;
2103
+ console.log(
2104
+ `CodeYam Log Level 1: ${rootScopeName} merge timed out — returning partial results (${retKeys} ret keys, ${Math.round((Date.now() - mergeStartTime) / 1000)}s)`,
2105
+ );
2106
+ (mergedDataStructure as any).timedOut = true;
2107
+ return mergedDataStructure;
2108
+ }
2109
+ throw error;
2110
+ }
2033
2111
  }