@codeyam/codeyam-cli 0.1.21 → 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 (122) 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/ai/src/lib/dataStructure/ScopeDataStructure.ts +31 -8
  7. package/analyzer-template/packages/ai/src/lib/dataStructure/equivalencyManagers/ParentScopeManager.ts +10 -3
  8. package/analyzer-template/packages/ai/src/lib/dataStructure/helpers/cleanKnownObjectFunctions.ts +16 -6
  9. package/analyzer-template/packages/analyze/index.ts +4 -1
  10. package/analyzer-template/packages/analyze/src/lib/files/analyze/analyzeEntities/prepareDataStructures.ts +28 -2
  11. package/analyzer-template/packages/analyze/src/lib/files/analyze/analyzeEntities.ts +5 -36
  12. package/analyzer-template/packages/analyze/src/lib/files/analyze/trackEntityCircularDependencies.ts +21 -0
  13. package/analyzer-template/packages/analyze/src/lib/files/analyze/validateDependencyAnalyses.ts +82 -10
  14. package/analyzer-template/packages/analyze/src/lib/files/analyzeNextRoute.ts +8 -3
  15. package/analyzer-template/packages/analyze/src/lib/files/scenarios/generateDataStructure.ts +239 -58
  16. package/analyzer-template/packages/analyze/src/lib/files/scenarios/mergeInDependentDataStructure.ts +1684 -1462
  17. package/codeyam-cli/src/commands/__tests__/editor.analyzeImportsArgs.test.js +47 -0
  18. package/codeyam-cli/src/commands/__tests__/editor.analyzeImportsArgs.test.js.map +1 -0
  19. package/codeyam-cli/src/commands/__tests__/editor.auditNoAutoAnalysis.test.js +71 -0
  20. package/codeyam-cli/src/commands/__tests__/editor.auditNoAutoAnalysis.test.js.map +1 -0
  21. package/codeyam-cli/src/commands/editor.js +545 -94
  22. package/codeyam-cli/src/commands/editor.js.map +1 -1
  23. package/codeyam-cli/src/commands/editorAnalyzeImportsArgs.js +23 -0
  24. package/codeyam-cli/src/commands/editorAnalyzeImportsArgs.js.map +1 -0
  25. package/codeyam-cli/src/utils/__tests__/editorAudit.test.js +456 -1
  26. package/codeyam-cli/src/utils/__tests__/editorAudit.test.js.map +1 -1
  27. package/codeyam-cli/src/utils/__tests__/editorPreview.test.js +11 -3
  28. package/codeyam-cli/src/utils/__tests__/editorPreview.test.js.map +1 -1
  29. package/codeyam-cli/src/utils/__tests__/editorScenarios.test.js +140 -1
  30. package/codeyam-cli/src/utils/__tests__/editorScenarios.test.js.map +1 -1
  31. package/codeyam-cli/src/utils/__tests__/editorSeedAdapter.test.js +50 -1
  32. package/codeyam-cli/src/utils/__tests__/editorSeedAdapter.test.js.map +1 -1
  33. package/codeyam-cli/src/utils/__tests__/entityChangeStatus.test.js +33 -1
  34. package/codeyam-cli/src/utils/__tests__/entityChangeStatus.test.js.map +1 -1
  35. package/codeyam-cli/src/utils/__tests__/manualEntityAnalysis.test.js +302 -0
  36. package/codeyam-cli/src/utils/__tests__/manualEntityAnalysis.test.js.map +1 -0
  37. package/codeyam-cli/src/utils/__tests__/testRunner.test.js +217 -0
  38. package/codeyam-cli/src/utils/__tests__/testRunner.test.js.map +1 -0
  39. package/codeyam-cli/src/utils/analysisRunner.js +28 -1
  40. package/codeyam-cli/src/utils/analysisRunner.js.map +1 -1
  41. package/codeyam-cli/src/utils/analyzer.js +4 -2
  42. package/codeyam-cli/src/utils/analyzer.js.map +1 -1
  43. package/codeyam-cli/src/utils/editorAudit.js +136 -5
  44. package/codeyam-cli/src/utils/editorAudit.js.map +1 -1
  45. package/codeyam-cli/src/utils/editorPreview.js +5 -3
  46. package/codeyam-cli/src/utils/editorPreview.js.map +1 -1
  47. package/codeyam-cli/src/utils/editorScenarios.js +60 -0
  48. package/codeyam-cli/src/utils/editorScenarios.js.map +1 -1
  49. package/codeyam-cli/src/utils/editorSeedAdapter.js +42 -2
  50. package/codeyam-cli/src/utils/editorSeedAdapter.js.map +1 -1
  51. package/codeyam-cli/src/utils/entityChangeStatus.server.js +16 -0
  52. package/codeyam-cli/src/utils/entityChangeStatus.server.js.map +1 -1
  53. package/codeyam-cli/src/utils/manualEntityAnalysis.js +196 -0
  54. package/codeyam-cli/src/utils/manualEntityAnalysis.js.map +1 -0
  55. package/codeyam-cli/src/utils/queue/job.js +20 -2
  56. package/codeyam-cli/src/utils/queue/job.js.map +1 -1
  57. package/codeyam-cli/src/utils/testRunner.js +199 -1
  58. package/codeyam-cli/src/utils/testRunner.js.map +1 -1
  59. package/codeyam-cli/src/webserver/__tests__/editorProxy.test.js +30 -11
  60. package/codeyam-cli/src/webserver/__tests__/editorProxy.test.js.map +1 -1
  61. package/codeyam-cli/src/webserver/__tests__/idleDetector.test.js +35 -0
  62. package/codeyam-cli/src/webserver/__tests__/idleDetector.test.js.map +1 -1
  63. package/codeyam-cli/src/webserver/build/client/assets/MiniClaudeChat-CQENLSrF.js +36 -0
  64. package/codeyam-cli/src/webserver/build/client/assets/cy-logo-cli-Coe5NhbS.js +1 -0
  65. package/codeyam-cli/src/webserver/build/client/assets/{cy-logo-cli-CJzc4vOH.svg → cy-logo-cli-DoA97ML3.svg} +2 -2
  66. package/codeyam-cli/src/webserver/build/client/assets/editor.entity.(_sha)-DMv5ESGo.js +96 -0
  67. package/codeyam-cli/src/webserver/build/client/assets/{editorPreview-NTuLi4Xg.js → editorPreview-CluPkvXJ.js} +6 -6
  68. package/codeyam-cli/src/webserver/build/client/assets/{entity._sha._-Blfy9UlN.js → entity._sha._-ByHz6rAQ.js} +13 -12
  69. package/codeyam-cli/src/webserver/build/client/assets/{entity._sha.scenarios._scenarioId.dev-BA5L8bU-.js → entity._sha.scenarios._scenarioId.dev-CmLO432x.js} +1 -1
  70. package/codeyam-cli/src/webserver/build/client/assets/{entity._sha.scenarios._scenarioId.fullscreen-D4dmRgvO.js → entity._sha.scenarios._scenarioId.fullscreen-Bz9sCUF_.js} +1 -1
  71. package/codeyam-cli/src/webserver/build/client/assets/globals-oyPmV37k.css +1 -0
  72. package/codeyam-cli/src/webserver/build/client/assets/{manifest-5025e428.js → manifest-1a45e154.js} +1 -1
  73. package/codeyam-cli/src/webserver/build/client/assets/{root-BCx1S8Z3.js → root-D2_tktnk.js} +6 -6
  74. package/codeyam-cli/src/webserver/build/server/assets/analysisRunner-By5slFjw.js +16 -0
  75. package/codeyam-cli/src/webserver/build/server/assets/{index-C91yWWCI.js → index-DXaOwBnm.js} +1 -1
  76. package/codeyam-cli/src/webserver/build/server/assets/{init-Dkas-RUS.js → init-CLG1LjQM.js} +1 -1
  77. package/codeyam-cli/src/webserver/build/server/assets/server-build-NZmUqQv6.js +688 -0
  78. package/codeyam-cli/src/webserver/build/server/index.js +1 -1
  79. package/codeyam-cli/src/webserver/build-info.json +5 -5
  80. package/codeyam-cli/src/webserver/editorProxy.js +55 -3
  81. package/codeyam-cli/src/webserver/editorProxy.js.map +1 -1
  82. package/codeyam-cli/src/webserver/idleDetector.js +15 -0
  83. package/codeyam-cli/src/webserver/idleDetector.js.map +1 -1
  84. package/codeyam-cli/src/webserver/terminalServer.js +8 -2
  85. package/codeyam-cli/src/webserver/terminalServer.js.map +1 -1
  86. package/codeyam-cli/templates/codeyam-editor-reference.md +8 -6
  87. package/codeyam-cli/templates/nextjs-prisma-sqlite/seed-adapter.ts +42 -34
  88. package/codeyam-cli/templates/skills/codeyam-editor/SKILL.md +2 -2
  89. package/package.json +1 -1
  90. package/packages/ai/src/lib/astScopes/methodSemantics.js +99 -0
  91. package/packages/ai/src/lib/astScopes/methodSemantics.js.map +1 -1
  92. package/packages/ai/src/lib/astScopes/nodeToSource.js +16 -0
  93. package/packages/ai/src/lib/astScopes/nodeToSource.js.map +1 -1
  94. package/packages/ai/src/lib/astScopes/paths.js +12 -3
  95. package/packages/ai/src/lib/astScopes/paths.js.map +1 -1
  96. package/packages/ai/src/lib/dataStructure/ScopeDataStructure.js +23 -9
  97. package/packages/ai/src/lib/dataStructure/ScopeDataStructure.js.map +1 -1
  98. package/packages/ai/src/lib/dataStructure/equivalencyManagers/ParentScopeManager.js +9 -2
  99. package/packages/ai/src/lib/dataStructure/equivalencyManagers/ParentScopeManager.js.map +1 -1
  100. package/packages/ai/src/lib/dataStructure/helpers/cleanKnownObjectFunctions.js +14 -4
  101. package/packages/ai/src/lib/dataStructure/helpers/cleanKnownObjectFunctions.js.map +1 -1
  102. package/packages/analyze/index.js +1 -1
  103. package/packages/analyze/index.js.map +1 -1
  104. package/packages/analyze/src/lib/files/analyze/analyzeEntities/prepareDataStructures.js +16 -2
  105. package/packages/analyze/src/lib/files/analyze/analyzeEntities/prepareDataStructures.js.map +1 -1
  106. package/packages/analyze/src/lib/files/analyze/analyzeEntities.js +6 -26
  107. package/packages/analyze/src/lib/files/analyze/analyzeEntities.js.map +1 -1
  108. package/packages/analyze/src/lib/files/analyze/trackEntityCircularDependencies.js +14 -0
  109. package/packages/analyze/src/lib/files/analyze/trackEntityCircularDependencies.js.map +1 -1
  110. package/packages/analyze/src/lib/files/analyze/validateDependencyAnalyses.js +44 -11
  111. package/packages/analyze/src/lib/files/analyze/validateDependencyAnalyses.js.map +1 -1
  112. package/packages/analyze/src/lib/files/analyzeNextRoute.js +5 -1
  113. package/packages/analyze/src/lib/files/analyzeNextRoute.js.map +1 -1
  114. package/packages/analyze/src/lib/files/scenarios/generateDataStructure.js +120 -28
  115. package/packages/analyze/src/lib/files/scenarios/generateDataStructure.js.map +1 -1
  116. package/packages/analyze/src/lib/files/scenarios/mergeInDependentDataStructure.js +1368 -1193
  117. package/packages/analyze/src/lib/files/scenarios/mergeInDependentDataStructure.js.map +1 -1
  118. package/codeyam-cli/src/webserver/build/client/assets/cy-logo-cli-DODLxLcw.js +0 -1
  119. package/codeyam-cli/src/webserver/build/client/assets/editor.entity.(_sha)-Dx-h1rJK.js +0 -130
  120. package/codeyam-cli/src/webserver/build/client/assets/globals-BrPXT1iR.css +0 -1
  121. package/codeyam-cli/src/webserver/build/server/assets/analysisRunner-C1kjC9UJ.js +0 -13
  122. package/codeyam-cli/src/webserver/build/server/assets/server-build-pulXLTrG.js +0 -640
@@ -25,6 +25,34 @@ function getTypeParameter(functionName) {
25
25
  }
26
26
  return null;
27
27
  }
28
+ /**
29
+ * Check if a schema path contains a Set/Map collection method call.
30
+ * Paths like `.has(articleId)`, `.delete(articleId)`, `.add(articleId)` represent
31
+ * membership checks on Sets/Maps, not meaningful data flow for schema generation.
32
+ * These create massive combinatorial explosions when every filter field (filterRatings,
33
+ * filterPublications, filterAuthors, etc.) × every method (has, delete, add) gets
34
+ * tracked as a separate equivalency.
35
+ */
36
+ const COLLECTION_METHOD_PATTERN = /\.(?:has|delete|add|clear|get|set)\(/;
37
+ function isCollectionMethodPath(path) {
38
+ return COLLECTION_METHOD_PATTERN.test(path);
39
+ }
40
+ /**
41
+ * Check if a path contains an inline object literal inside a function call argument.
42
+ * e.g., setUndoEntry({ label: '...', undo: () => {} }) has '{' inside '(' ')'.
43
+ * These paths are call-site snapshots where the source code text was captured
44
+ * as the path. They don't represent schema structure and are very expensive to
45
+ * parse (avg 319 chars). They account for ~55% of equivalencies in complex entities.
46
+ */
47
+ const INLINE_OBJECT_ARG_PATTERN = /\([^)]*\{[^}]*:/;
48
+ function hasInlineObjectArg(path) {
49
+ // Match function calls containing object literals with key-value pairs.
50
+ // Pattern: open paren, then { with a : inside before closing }.
51
+ // e.g., setUndoEntry({ label: '...' }) matches
52
+ // e.g., find(item) does NOT match
53
+ // e.g., fn({a:1, b:2}) matches
54
+ return INLINE_OBJECT_ARG_PATTERN.test(path);
55
+ }
28
56
  // Primitive types that should not have child paths
29
57
  const PRIMITIVE_TYPES = new Set([
30
58
  'number',
@@ -107,7 +135,31 @@ function bestValueFromOptions(options) {
107
135
  }
108
136
  return options[0] ?? 'unknown';
109
137
  }
110
- export default function mergeInDependentDataStructure({ importedExports, dependentAnalyses, rootScopeName, dataStructure, dependencySchemas, }) {
138
+ /** Timeout (ms) for the merge operation. Throws DataStructureTimeoutError if exceeded.
139
+ * All successful merges complete in <300ms. Anything exceeding 2s is pathological. */
140
+ const MERGE_TIMEOUT_MS = 2000;
141
+ /** Cap for schema size during postfix application and dep copy.
142
+ * Successful merges produce <3K ret keys. Beyond 5K, further entries
143
+ * are cross-products of unrelated equivalencies — noise, not signal. */
144
+ const SCHEMA_KEY_CAP = 5000;
145
+ export class DataStructureTimeoutError extends Error {
146
+ constructor(entityName, elapsedMs) {
147
+ super(`Data structure merge timed out for ${entityName} after ${Math.round(elapsedMs / 1000)}s (limit: ${MERGE_TIMEOUT_MS / 1000}s)`);
148
+ this.name = 'DataStructureTimeoutError';
149
+ }
150
+ }
151
+ export default function mergeInDependentDataStructure({ importedExports, dependentAnalyses, rootScopeName, dataStructure, dependencySchemas, timeoutMs = MERGE_TIMEOUT_MS, }) {
152
+ const mergeStartTime = Date.now();
153
+ const mergeDeadline = timeoutMs > 0 ? mergeStartTime + timeoutMs : 0;
154
+ /** Call in hot loops. Throws DataStructureTimeoutError if deadline exceeded.
155
+ * Date.now() is ~20ns — negligible vs the ms-scale string ops in each iteration. */
156
+ const checkDeadline = () => {
157
+ if (!mergeDeadline)
158
+ return;
159
+ if (Date.now() > mergeDeadline) {
160
+ throw new DataStructureTimeoutError(rootScopeName, Date.now() - mergeStartTime);
161
+ }
162
+ };
111
163
  const mergedDataStructure = {
112
164
  signatureSchema: { ...(dataStructure.signatureSchema ?? {}) },
113
165
  returnValueSchema: { ...(dataStructure.returnValueSchema ?? {}) },
@@ -116,550 +168,591 @@ export default function mergeInDependentDataStructure({ importedExports, depende
116
168
  dependencySchemas: {},
117
169
  environmentVariables: [...(dataStructure.environmentVariables || [])],
118
170
  };
119
- // Build a set of functions that have multiple DIFFERENT type parameters.
120
- // For these functions, we must NOT normalize paths to avoid merging different schemas.
121
- // e.g., if we have both useFetcher<{ data: UserData }>() and useFetcher<{ data: ConfigData }>(),
122
- // they must stay separate and not both become 'returnValue'.
123
- const functionsWithMultipleTypeParams = new Set();
124
- const typeParamsByFunction = {};
125
- // Helper to scan a schema for type parameters
126
- const scanSchemaForTypeParams = (schema) => {
127
- for (const schemaPath of Object.keys(schema ?? {})) {
128
- const parts = splitOutsideParenthesesAndArrays(schemaPath);
129
- if (parts.length > 0) {
130
- const firstPart = parts[0];
131
- const typeParam = getTypeParameter(firstPart);
132
- if (typeParam) {
133
- const baseName = cleanFunctionName(firstPart);
134
- typeParamsByFunction[baseName] || (typeParamsByFunction[baseName] = new Set());
135
- typeParamsByFunction[baseName].add(typeParam);
136
- if (typeParamsByFunction[baseName].size > 1) {
137
- functionsWithMultipleTypeParams.add(baseName);
171
+ try {
172
+ // Build a set of functions that have multiple DIFFERENT type parameters.
173
+ // For these functions, we must NOT normalize paths to avoid merging different schemas.
174
+ // e.g., if we have both useFetcher<{ data: UserData }>() and useFetcher<{ data: ConfigData }>(),
175
+ // they must stay separate and not both become 'returnValue'.
176
+ const functionsWithMultipleTypeParams = new Set();
177
+ const typeParamsByFunction = {};
178
+ // Helper to scan a schema for type parameters
179
+ const scanSchemaForTypeParams = (schema) => {
180
+ for (const schemaPath of Object.keys(schema ?? {})) {
181
+ const parts = splitOutsideParenthesesAndArrays(schemaPath);
182
+ if (parts.length > 0) {
183
+ const firstPart = parts[0];
184
+ const typeParam = getTypeParameter(firstPart);
185
+ if (typeParam) {
186
+ const baseName = cleanFunctionName(firstPart);
187
+ typeParamsByFunction[baseName] || (typeParamsByFunction[baseName] = new Set());
188
+ typeParamsByFunction[baseName].add(typeParam);
189
+ if (typeParamsByFunction[baseName].size > 1) {
190
+ functionsWithMultipleTypeParams.add(baseName);
191
+ }
138
192
  }
139
193
  }
140
194
  }
195
+ };
196
+ // Scan the root entity's schema
197
+ scanSchemaForTypeParams(dataStructure.returnValueSchema);
198
+ // Also scan all dependency schemas for type parameters
199
+ for (const filePath of Object.keys(dependencySchemas ?? {})) {
200
+ for (const name of Object.keys(dependencySchemas[filePath] ?? {})) {
201
+ scanSchemaForTypeParams(dependencySchemas[filePath][name]?.returnValueSchema);
202
+ }
141
203
  }
142
- };
143
- // Scan the root entity's schema
144
- scanSchemaForTypeParams(dataStructure.returnValueSchema);
145
- // Also scan all dependency schemas for type parameters
146
- for (const filePath of Object.keys(dependencySchemas ?? {})) {
147
- for (const name of Object.keys(dependencySchemas[filePath] ?? {})) {
148
- scanSchemaForTypeParams(dependencySchemas[filePath][name]?.returnValueSchema);
149
- }
150
- }
151
- let equivalentSchemaPaths = [];
152
- // O(1) index for findOrCreateEquivalentSchemaPathsEntry.
153
- // Maps "(rootPath)::(normalizedFuncName)" the entry containing that root.
154
- // This replaces the O(E) linear search that was causing O(E²) gather performance.
155
- const espIndex = new Map();
156
- const espIndexKey = (path, functionName) => {
157
- const normalized = cleanFunctionName(functionName);
158
- const funcKey = normalized === rootScopeName ? '__self__' : normalized || '__self__';
159
- return `${path}::${funcKey}`;
160
- };
161
- const updateEspIndex = (entry) => {
162
- for (const root of entry.equivalentRoots) {
163
- const funcName = root.function?.name ?? rootScopeName;
164
- espIndex.set(espIndexKey(root.schemaRootPath, funcName), entry);
165
- }
166
- };
167
- // Pre-build a lookup map from cleaned function name to dependency for O(1) lookups.
168
- // This avoids O(n) linear search in findRelevantDependency which was causing O(n²) performance.
169
- const dependencyByCleanedName = new Map();
170
- for (const dep of importedExports) {
171
- const cleanedName = cleanFunctionName(dep.name);
172
- if (!dependencyByCleanedName.has(cleanedName)) {
173
- dependencyByCleanedName.set(cleanedName, dep);
174
- }
175
- }
176
- const findRelevantDependency = (functionName) => {
177
- return dependencyByCleanedName.get(cleanFunctionName(functionName));
178
- };
179
- const findRelevantDependentDataStructure = (functionName) => {
180
- const dependency = findRelevantDependency(functionName);
181
- if (!dependency)
182
- return;
183
- return dependencySchemas?.[dependency.filePath]?.[dependency.name];
184
- };
185
- const findRelevantDependentAnalysisDataStructure = (functionName) => {
186
- const dependency = findRelevantDependency(functionName);
187
- if (!dependency)
188
- return;
189
- const dependentAnalysis = dependentAnalyses[dependency.filePath]?.[dependency.name];
190
- if (!dependentAnalysis?.metadata?.mergedDataStructure) {
191
- return;
192
- }
193
- return dependentAnalysis.metadata.mergedDataStructure;
194
- };
195
- const findOrCreateDependentSchemas = (dependency) => {
196
- var _a, _b;
197
- const { filePath, name } = dependency;
198
- (_a = mergedDataStructure.dependencySchemas)[filePath] || (_a[filePath] = {});
199
- (_b = mergedDataStructure.dependencySchemas[filePath])[name] || (_b[name] = {
200
- signatureSchema: {},
201
- returnValueSchema: {},
202
- });
203
- return mergedDataStructure.dependencySchemas[filePath][name];
204
- };
205
- const cleanSchema = (schema, context) => {
206
- transformationTracer.traceSchemaTransform(rootScopeName, 'cleanKnownObjectFunctionsFromMapping', schema, cleanKnownObjectFunctionsFromMapping, context);
207
- };
208
- const translatePath = (path, dependencyName) => {
209
- if (path.startsWith(dependencyName)) {
210
- const pathParts = splitOutsideParenthesesAndArrays(path);
211
- if (pathParts.length > 1) {
212
- if (pathParts[1].startsWith('functionCallReturnValue')) {
213
- // Check if this function has multiple DIFFERENT type parameters.
214
- // If so, DON'T normalize to returnValue - keep the full path to avoid
215
- // merging different type-parameterized variants together.
216
- // e.g., useFetcher<{ data: UserData }>().functionCallReturnValue.data
217
- // should NOT be merged with useFetcher<{ data: ConfigData }>().functionCallReturnValue.data
218
- const baseName = cleanFunctionName(pathParts[0]);
219
- if (functionsWithMultipleTypeParams.has(baseName)) {
220
- return path; // Keep the original path with type parameters
221
- }
222
- // functionCallReturnValue immediately follows - normalize to returnValue
223
- // e.g., useAuth().functionCallReturnValue.user -> returnValue.user
224
- return joinParenthesesAndArrays([
225
- 'returnValue',
226
- ...pathParts.slice(2),
227
- ]);
228
- }
229
- else if (pathParts[0].endsWith(')') &&
230
- pathParts[1].startsWith('signature[')) {
231
- // Hook-style with signature access (e.g., BranchChangesTab().signature[0]...)
232
- // Strip the function name for signature equivalency matching
233
- return joinParenthesesAndArrays(pathParts.slice(1));
234
- }
235
- // For all other cases (object-style APIs like getSupabase().auth and
236
- // direct object references like supabase.from), preserve the path as-is.
237
- // The prefix must be kept for proper schema lookups in constructMockCode
238
- // and gatherDataForMocks.
204
+ let equivalentSchemaPaths = [];
205
+ // O(1) index for findOrCreateEquivalentSchemaPathsEntry.
206
+ // Maps "(rootPath)::(normalizedFuncName)" → the entry containing that root.
207
+ // This replaces the O(E) linear search that was causing O(E²) gather performance.
208
+ const espIndex = new Map();
209
+ const espIndexKey = (path, functionName) => {
210
+ const normalized = cleanFunctionName(functionName);
211
+ const funcKey = normalized === rootScopeName ? '__self__' : normalized || '__self__';
212
+ return `${path}::${funcKey}`;
213
+ };
214
+ const updateEspIndex = (entry) => {
215
+ for (const root of entry.equivalentRoots) {
216
+ const funcName = root.function?.name ?? rootScopeName;
217
+ espIndex.set(espIndexKey(root.schemaRootPath, funcName), entry);
218
+ }
219
+ };
220
+ // Pre-build a lookup map from cleaned function name to dependency for O(1) lookups.
221
+ // This avoids O(n) linear search in findRelevantDependency which was causing O(n²) performance.
222
+ const dependencyByCleanedName = new Map();
223
+ for (const dep of importedExports) {
224
+ const cleanedName = cleanFunctionName(dep.name);
225
+ if (!dependencyByCleanedName.has(cleanedName)) {
226
+ dependencyByCleanedName.set(cleanedName, dep);
239
227
  }
240
228
  }
241
- return path;
242
- };
243
- const gatherAllEquivalentSchemaPaths = (functionName, sourceAndUsageEquivalencies, dataStructure) => {
244
- if (!sourceAndUsageEquivalencies)
245
- return;
246
- const normalizedSchemaCache = new Map();
247
- const getSchemaIndex = (schema) => {
248
- if (!schema)
249
- return { byFirstPart: new Map() };
250
- const cached = normalizedSchemaCache.get(schema);
251
- if (cached)
229
+ const findRelevantDependency = (functionName) => {
230
+ return dependencyByCleanedName.get(cleanFunctionName(functionName));
231
+ };
232
+ const findRelevantDependentDataStructure = (functionName) => {
233
+ const dependency = findRelevantDependency(functionName);
234
+ if (!dependency)
235
+ return;
236
+ return dependencySchemas?.[dependency.filePath]?.[dependency.name];
237
+ };
238
+ const findRelevantDependentAnalysisDataStructure = (functionName) => {
239
+ const dependency = findRelevantDependency(functionName);
240
+ if (!dependency)
241
+ return;
242
+ const dependentAnalysis = dependentAnalyses[dependency.filePath]?.[dependency.name];
243
+ if (!dependentAnalysis?.metadata?.mergedDataStructure) {
244
+ return;
245
+ }
246
+ return dependentAnalysis.metadata.mergedDataStructure;
247
+ };
248
+ const findOrCreateDependentSchemas = (dependency) => {
249
+ var _a, _b;
250
+ const { filePath, name } = dependency;
251
+ (_a = mergedDataStructure.dependencySchemas)[filePath] || (_a[filePath] = {});
252
+ (_b = mergedDataStructure.dependencySchemas[filePath])[name] || (_b[name] = {
253
+ signatureSchema: {},
254
+ returnValueSchema: {},
255
+ });
256
+ return mergedDataStructure.dependencySchemas[filePath][name];
257
+ };
258
+ const cleanSchema = (schema, context) => {
259
+ transformationTracer.traceSchemaTransform(rootScopeName, 'cleanKnownObjectFunctionsFromMapping', schema, cleanKnownObjectFunctionsFromMapping, context);
260
+ };
261
+ // Cache translatePath results — the same path is often translated multiple times
262
+ // (once per equivalency entry that references it). Avoids redundant
263
+ // splitOutsideParenthesesAndArrays calls on long paths.
264
+ const translatePathCache = new Map();
265
+ const translatePath = (path, dependencyName) => {
266
+ const cacheKey = `${dependencyName}\0${path}`;
267
+ const cached = translatePathCache.get(cacheKey);
268
+ if (cached !== undefined)
252
269
  return cached;
253
- const byFirstPart = new Map();
254
- for (const path in schema) {
255
- let parts = splitOutsideParenthesesAndArrays(path);
256
- if (parts[0].startsWith(functionName)) {
257
- const baseName = cleanFunctionName(parts[0]);
258
- if (!functionsWithMultipleTypeParams.has(baseName)) {
259
- parts =
260
- parts[1] === 'functionCallReturnValue'
261
- ? ['returnValue', ...parts.slice(2)]
262
- : parts.slice(1);
270
+ let result = path;
271
+ if (path.startsWith(dependencyName)) {
272
+ const pathParts = splitOutsideParenthesesAndArrays(path);
273
+ if (pathParts.length > 1) {
274
+ if (pathParts[1].startsWith('functionCallReturnValue')) {
275
+ // Check if this function has multiple DIFFERENT type parameters.
276
+ // If so, DON'T normalize to returnValue - keep the full path to avoid
277
+ // merging different type-parameterized variants together.
278
+ const baseName = cleanFunctionName(pathParts[0]);
279
+ if (!functionsWithMultipleTypeParams.has(baseName)) {
280
+ // functionCallReturnValue immediately follows - normalize to returnValue
281
+ result = joinParenthesesAndArrays([
282
+ 'returnValue',
283
+ ...pathParts.slice(2),
284
+ ]);
285
+ }
286
+ }
287
+ else if (pathParts[0].endsWith(')') &&
288
+ pathParts[1].startsWith('signature[')) {
289
+ // Hook-style with signature access
290
+ result = joinParenthesesAndArrays(pathParts.slice(1));
263
291
  }
264
292
  }
265
- const entry = { path, parts };
266
- // Index by the base of the first part (before any function call args)
267
- const firstPart = parts[0] ?? '';
268
- const parenIdx = firstPart.indexOf('(');
269
- const firstPartBase = parenIdx >= 0 ? firstPart.slice(0, parenIdx) : firstPart;
270
- let bucket = byFirstPart.get(firstPartBase);
271
- if (!bucket) {
272
- bucket = [];
273
- byFirstPart.set(firstPartBase, bucket);
274
- }
275
- bucket.push(entry);
276
293
  }
277
- const result = { byFirstPart };
278
- normalizedSchemaCache.set(schema, result);
294
+ translatePathCache.set(cacheKey, result);
279
295
  return result;
280
296
  };
281
- const findOrCreateEquivalentSchemaPathsEntry = (allPaths) => {
282
- const equivalentRoots = allPaths
283
- .filter((p) => p.functionName === rootScopeName ||
284
- !!findRelevantDependency(p.functionName))
285
- .map((p) => ({
286
- schemaRootPath: p.path,
287
- function: p.functionName === rootScopeName
288
- ? undefined
289
- : findRelevantDependency(p.functionName),
290
- }));
291
- let equivalentSchemaPathsEntry;
292
- // Collect the signature indices from the new roots we want to add
293
- const newRootSignatureIndices = new Set();
294
- for (const root of equivalentRoots) {
295
- const idx = extractSignatureIndex(root.schemaRootPath);
296
- if (idx !== undefined) {
297
- newRootSignatureIndices.add(idx);
297
+ const gatherAllEquivalentSchemaPaths = (functionName, sourceAndUsageEquivalencies, dataStructure) => {
298
+ checkDeadline();
299
+ if (!sourceAndUsageEquivalencies)
300
+ return;
301
+ const normalizedSchemaCache = new Map();
302
+ const getSchemaIndex = (schema) => {
303
+ if (!schema)
304
+ return { byFirstPart: new Map() };
305
+ const cached = normalizedSchemaCache.get(schema);
306
+ if (cached)
307
+ return cached;
308
+ const byFirstPart = new Map();
309
+ for (const path in schema) {
310
+ checkDeadline();
311
+ let parts = splitOutsideParenthesesAndArrays(path);
312
+ if (parts[0].startsWith(functionName)) {
313
+ const baseName = cleanFunctionName(parts[0]);
314
+ if (!functionsWithMultipleTypeParams.has(baseName)) {
315
+ parts =
316
+ parts[1] === 'functionCallReturnValue'
317
+ ? ['returnValue', ...parts.slice(2)]
318
+ : parts.slice(1);
319
+ }
320
+ }
321
+ const entry = { path, parts };
322
+ // Index by the base of the first part (before any function call args)
323
+ const firstPart = parts[0] ?? '';
324
+ const parenIdx = firstPart.indexOf('(');
325
+ const firstPartBase = parenIdx >= 0 ? firstPart.slice(0, parenIdx) : firstPart;
326
+ let bucket = byFirstPart.get(firstPartBase);
327
+ if (!bucket) {
328
+ bucket = [];
329
+ byFirstPart.set(firstPartBase, bucket);
330
+ }
331
+ bucket.push(entry);
298
332
  }
299
- }
300
- // Use espIndex Map for O(1) lookup instead of O(E) linear search.
301
- // Falls back to linear search only when Map hit has a signature index conflict.
302
- for (const pathInfo of allPaths) {
303
- if (equivalentSchemaPathsEntry)
304
- break;
305
- const candidate = espIndex.get(espIndexKey(pathInfo.path, pathInfo.functionName));
306
- if (!candidate)
307
- continue;
308
- // Verify no signature index conflict with the candidate entry
309
- if (newRootSignatureIndices.size > 0) {
310
- const existingIndicesByFunction = new Map();
311
- for (const er of candidate.equivalentRoots) {
312
- const funcKey = er.function
313
- ? `${er.function.name}::${er.function.filePath}`
314
- : '__self__';
315
- const idx = extractSignatureIndex(er.schemaRootPath);
316
- if (idx !== undefined) {
317
- if (!existingIndicesByFunction.has(funcKey)) {
318
- existingIndicesByFunction.set(funcKey, new Set());
333
+ const result = { byFirstPart };
334
+ normalizedSchemaCache.set(schema, result);
335
+ return result;
336
+ };
337
+ const findOrCreateEquivalentSchemaPathsEntry = (allPaths) => {
338
+ const equivalentRoots = allPaths
339
+ .filter((p) => p.functionName === rootScopeName ||
340
+ !!findRelevantDependency(p.functionName))
341
+ .map((p) => ({
342
+ schemaRootPath: p.path,
343
+ function: p.functionName === rootScopeName
344
+ ? undefined
345
+ : findRelevantDependency(p.functionName),
346
+ }));
347
+ let equivalentSchemaPathsEntry;
348
+ // Collect the signature indices from the new roots we want to add
349
+ const newRootSignatureIndices = new Set();
350
+ for (const root of equivalentRoots) {
351
+ const idx = extractSignatureIndex(root.schemaRootPath);
352
+ if (idx !== undefined) {
353
+ newRootSignatureIndices.add(idx);
354
+ }
355
+ }
356
+ // Use espIndex Map for O(1) lookup instead of O(E) linear search.
357
+ // Falls back to linear search only when Map hit has a signature index conflict.
358
+ for (const pathInfo of allPaths) {
359
+ checkDeadline();
360
+ if (equivalentSchemaPathsEntry)
361
+ break;
362
+ const candidate = espIndex.get(espIndexKey(pathInfo.path, pathInfo.functionName));
363
+ if (!candidate)
364
+ continue;
365
+ // Verify no signature index conflict with the candidate entry
366
+ if (newRootSignatureIndices.size > 0) {
367
+ const existingIndicesByFunction = new Map();
368
+ for (const er of candidate.equivalentRoots) {
369
+ const funcKey = er.function
370
+ ? `${er.function.name}::${er.function.filePath}`
371
+ : '__self__';
372
+ const idx = extractSignatureIndex(er.schemaRootPath);
373
+ if (idx !== undefined) {
374
+ if (!existingIndicesByFunction.has(funcKey)) {
375
+ existingIndicesByFunction.set(funcKey, new Set());
376
+ }
377
+ existingIndicesByFunction.get(funcKey).add(idx);
319
378
  }
320
- existingIndicesByFunction.get(funcKey).add(idx);
321
379
  }
322
- }
323
- let hasConflict = false;
324
- for (const newRoot of equivalentRoots) {
325
- const funcKey = newRoot.function
326
- ? `${newRoot.function.name}::${newRoot.function.filePath}`
327
- : '__self__';
328
- const newIdx = extractSignatureIndex(newRoot.schemaRootPath);
329
- if (newIdx !== undefined) {
330
- const existingIndices = existingIndicesByFunction.get(funcKey);
331
- if (existingIndices && existingIndices.size > 0) {
332
- if (!existingIndices.has(newIdx)) {
333
- hasConflict = true;
334
- break;
380
+ let hasConflict = false;
381
+ for (const newRoot of equivalentRoots) {
382
+ const funcKey = newRoot.function
383
+ ? `${newRoot.function.name}::${newRoot.function.filePath}`
384
+ : '__self__';
385
+ const newIdx = extractSignatureIndex(newRoot.schemaRootPath);
386
+ if (newIdx !== undefined) {
387
+ const existingIndices = existingIndicesByFunction.get(funcKey);
388
+ if (existingIndices && existingIndices.size > 0) {
389
+ if (!existingIndices.has(newIdx)) {
390
+ hasConflict = true;
391
+ break;
392
+ }
335
393
  }
336
394
  }
337
395
  }
396
+ if (hasConflict)
397
+ continue;
338
398
  }
339
- if (hasConflict)
340
- continue;
399
+ equivalentSchemaPathsEntry = candidate;
341
400
  }
342
- equivalentSchemaPathsEntry = candidate;
343
- }
344
- if (!equivalentSchemaPathsEntry) {
345
- // Before creating a new entry, filter out roots that have conflicting
346
- // signature indices from the same function. An entry should never contain
347
- // roots with different signature indices from the same function.
348
- // This prevents the bug where signature[1], signature[2], signature[4]
349
- // all get merged together due to incorrect sourceEquivalencies.
350
- let filteredRoots = equivalentRoots;
351
- if (newRootSignatureIndices.size > 1) {
352
- // There are multiple signature indices - we need to filter to keep only
353
- // one consistent set. We'll keep the roots that match the PRIMARY index
354
- // (the first signature index we encounter from self, or the lowest index).
355
- // First, determine the primary index - prefer the self root's index
356
- let primaryIndex;
357
- for (const root of equivalentRoots) {
358
- if (!root.function) {
359
- // This is a self root
360
- const idx = extractSignatureIndex(root.schemaRootPath);
361
- if (idx !== undefined) {
362
- primaryIndex = idx;
363
- break;
401
+ if (!equivalentSchemaPathsEntry) {
402
+ // Before creating a new entry, filter out roots that have conflicting
403
+ // signature indices from the same function. An entry should never contain
404
+ // roots with different signature indices from the same function.
405
+ // This prevents the bug where signature[1], signature[2], signature[4]
406
+ // all get merged together due to incorrect sourceEquivalencies.
407
+ let filteredRoots = equivalentRoots;
408
+ if (newRootSignatureIndices.size > 1) {
409
+ // There are multiple signature indices - we need to filter to keep only
410
+ // one consistent set. We'll keep the roots that match the PRIMARY index
411
+ // (the first signature index we encounter from self, or the lowest index).
412
+ // First, determine the primary index - prefer the self root's index
413
+ let primaryIndex;
414
+ for (const root of equivalentRoots) {
415
+ if (!root.function) {
416
+ // This is a self root
417
+ const idx = extractSignatureIndex(root.schemaRootPath);
418
+ if (idx !== undefined) {
419
+ primaryIndex = idx;
420
+ break;
421
+ }
364
422
  }
365
423
  }
424
+ // If no self root has a signature index, use the lowest index
425
+ if (primaryIndex === undefined) {
426
+ primaryIndex = Math.min(...newRootSignatureIndices);
427
+ }
428
+ // Filter roots: keep if no signature index OR signature index matches primary
429
+ filteredRoots = equivalentRoots.filter((root) => {
430
+ const idx = extractSignatureIndex(root.schemaRootPath);
431
+ return idx === undefined || idx === primaryIndex;
432
+ });
366
433
  }
367
- // If no self root has a signature index, use the lowest index
368
- if (primaryIndex === undefined) {
369
- primaryIndex = Math.min(...newRootSignatureIndices);
370
- }
371
- // Filter roots: keep if no signature index OR signature index matches primary
372
- filteredRoots = equivalentRoots.filter((root) => {
373
- const idx = extractSignatureIndex(root.schemaRootPath);
374
- return idx === undefined || idx === primaryIndex;
375
- });
434
+ equivalentSchemaPathsEntry = {
435
+ equivalentRoots: filteredRoots,
436
+ equivalentPostfixes: {},
437
+ };
438
+ equivalentSchemaPaths.push(equivalentSchemaPathsEntry);
376
439
  }
377
- equivalentSchemaPathsEntry = {
378
- equivalentRoots: filteredRoots,
379
- equivalentPostfixes: {},
380
- };
381
- equivalentSchemaPaths.push(equivalentSchemaPathsEntry);
382
- }
383
- else {
384
- equivalentSchemaPathsEntry.equivalentRoots.push(...equivalentRoots);
385
- }
386
- // Deduplicate roots using a Set for O(n) instead of O(n²)
387
- const seenRoots = new Set();
388
- equivalentSchemaPathsEntry.equivalentRoots =
389
- equivalentSchemaPathsEntry.equivalentRoots.filter((er) => {
390
- const key = er.schemaRootPath + '::' + (er.function?.name ?? '');
391
- if (seenRoots.has(key))
392
- return false;
393
- seenRoots.add(key);
394
- return true;
395
- });
396
- // Keep the espIndex in sync after adding/deduplicating roots
397
- updateEspIndex(equivalentSchemaPathsEntry);
398
- return equivalentSchemaPathsEntry;
399
- };
400
- // Helper to extract function name from a path that starts with a function call.
401
- // e.g., 'ScenarioViewer().signature[0].scenario' -> 'ScenarioViewer'
402
- // Returns undefined if the path doesn't start with a function call or the function isn't a dependency.
403
- const extractFunctionNameFromPath = (path) => {
404
- const parts = splitOutsideParenthesesAndArrays(path);
405
- if (parts.length > 0 && parts[0].endsWith(')')) {
406
- // Extract the function name without the () suffix and type params
407
- const funcCallPart = parts[0];
408
- const funcName = cleanFunctionName(funcCallPart.replace(/\(\)$/, ''));
409
- // Check if this function is a dependency
410
- if (findRelevantDependency(funcName)) {
411
- return funcName;
440
+ else {
441
+ equivalentSchemaPathsEntry.equivalentRoots.push(...equivalentRoots);
412
442
  }
413
- }
414
- return undefined;
415
- };
416
- const allEquivalencies = [
417
- sourceAndUsageEquivalencies.usageEquivalencies,
418
- sourceAndUsageEquivalencies.sourceEquivalencies,
419
- ].filter(Boolean);
420
- for (const equivalencies of allEquivalencies) {
421
- const schemaPathEntries = Object.entries(equivalencies);
422
- for (const [schemaPath, usages] of schemaPathEntries) {
423
- // First, check if the raw schemaPath starts with a function call to a dependency.
424
- // If so, use that dependency name for translation (so translatePath can strip the prefix).
425
- const extractedFuncName = extractFunctionNameFromPath(schemaPath);
426
- const effectiveFunctionName = extractedFuncName || functionName;
427
- const translatedPath = translatePath(schemaPath, effectiveFunctionName);
428
- const allPaths = [
429
- { path: translatedPath, functionName: effectiveFunctionName },
430
- ...usages.map((u) => ({
431
- path: translatePath(u.schemaPath, u.scopeNodeName),
432
- functionName: u.scopeNodeName,
433
- })),
434
- ].filter((pathInfo) => !pathInfo.path.includes('.map('));
435
- // Fix 38: Derive base paths from property access paths.
436
- // When we have equivalent paths like:
437
- // Parent: signature[0].scenarios[].name
438
- // Child: signature[0].selectedScenario.name
439
- // We want to derive the base paths by finding the common suffix:
440
- // Common suffix: .name
441
- // Parent base: signature[0].scenarios[]
442
- // Child base: signature[0].selectedScenario
443
- // This allows the merge to find nested child schema fields under the base prop.
444
- // Find child signature paths (paths from child components)
445
- const childPaths = allPaths.filter((p) => p.functionName &&
446
- p.functionName !== rootScopeName &&
447
- p.functionName !== effectiveFunctionName);
448
- // Find parent paths (paths from this component)
449
- const parentPaths = allPaths.filter((p) => !p.functionName ||
450
- p.functionName === rootScopeName ||
451
- p.functionName === effectiveFunctionName);
452
- const derivedBasePaths = [];
453
- const allPathSet = new Set(allPaths.map((p) => p.path));
454
- const derivedBasePathSet = new Set();
455
- // For each child path, find its equivalent parent path and derive bases
456
- for (const childPathInfo of childPaths) {
457
- const childParts = splitOutsideParenthesesAndArrays(childPathInfo.path);
458
- // Look for a parent path that shares a common suffix with this child path
459
- for (const parentPathInfo of parentPaths) {
460
- const parentParts = splitOutsideParenthesesAndArrays(parentPathInfo.path);
461
- // Find the common suffix (from the end)
462
- let commonSuffixLength = 0;
463
- const minLen = Math.min(childParts.length, parentParts.length);
464
- for (let i = 1; i <= minLen; i++) {
465
- if (childParts[childParts.length - i] ===
466
- parentParts[parentParts.length - i]) {
467
- commonSuffixLength = i;
468
- }
469
- else {
470
- break;
443
+ // Deduplicate roots using a Set for O(n) instead of O(n²)
444
+ const seenRoots = new Set();
445
+ equivalentSchemaPathsEntry.equivalentRoots =
446
+ equivalentSchemaPathsEntry.equivalentRoots.filter((er) => {
447
+ const key = er.schemaRootPath + '::' + (er.function?.name ?? '');
448
+ if (seenRoots.has(key))
449
+ return false;
450
+ seenRoots.add(key);
451
+ return true;
452
+ });
453
+ // Keep the espIndex in sync after adding/deduplicating roots
454
+ updateEspIndex(equivalentSchemaPathsEntry);
455
+ return equivalentSchemaPathsEntry;
456
+ };
457
+ // Helper to extract function name from a path that starts with a function call.
458
+ // e.g., 'ScenarioViewer().signature[0].scenario' -> 'ScenarioViewer'
459
+ // Returns undefined if the path doesn't start with a function call or the function isn't a dependency.
460
+ const extractFunctionNameFromPath = (path) => {
461
+ const parts = splitOutsideParenthesesAndArrays(path);
462
+ if (parts.length > 0 && parts[0].endsWith(')')) {
463
+ // Extract the function name without the () suffix and type params
464
+ const funcCallPart = parts[0];
465
+ const funcName = cleanFunctionName(funcCallPart.replace(/\(\)$/, ''));
466
+ // Check if this function is a dependency
467
+ if (findRelevantDependency(funcName)) {
468
+ return funcName;
469
+ }
470
+ }
471
+ return undefined;
472
+ };
473
+ const allEquivalencies = [
474
+ sourceAndUsageEquivalencies.usageEquivalencies,
475
+ sourceAndUsageEquivalencies.sourceEquivalencies,
476
+ ].filter(Boolean);
477
+ // Global dedup across ALL equivalency entries. The same (scope, targetPath)
478
+ // pair often appears in 30-50 different source entries (e.g., every variable
479
+ // that flows through loadView references the same 50 target paths).
480
+ // Processing these redundantly accounts for 96% of work in the gather phase.
481
+ const globalSeenTargets = new Set();
482
+ for (const equivalencies of allEquivalencies) {
483
+ const schemaPathEntries = Object.entries(equivalencies);
484
+ for (const [schemaPath, usages] of schemaPathEntries) {
485
+ checkDeadline();
486
+ // Skip equivalency entries whose source path is a Set/Map membership operation.
487
+ // Patterns like `.has(articleId)`, `.delete(articleId)`, `.add(articleId)` on
488
+ // Sets/Maps represent membership checks, not meaningful data flow for schema generation.
489
+ // In the Margo LibraryPage case, these account for 74% of all equivalency targets
490
+ // (19,444 of 26,340) and cause a combinatorial explosion in the merge.
491
+ if (isCollectionMethodPath(schemaPath))
492
+ continue;
493
+ // Skip paths with inline object literals in function call arguments.
494
+ // These are call-site snapshots (e.g., setUndoEntry({ label: '...', undo: ... }))
495
+ // that embed source code text as path strings. They're expensive to parse
496
+ // and don't contribute useful schema information.
497
+ if (hasInlineObjectArg(schemaPath))
498
+ continue;
499
+ // First, check if the raw schemaPath starts with a function call to a dependency.
500
+ // If so, use that dependency name for translation (so translatePath can strip the prefix).
501
+ const extractedFuncName = extractFunctionNameFromPath(schemaPath);
502
+ const effectiveFunctionName = extractedFuncName || functionName;
503
+ const translatedPath = translatePath(schemaPath, effectiveFunctionName);
504
+ const allPathsRaw = [
505
+ { path: translatedPath, functionName: effectiveFunctionName },
506
+ ...usages
507
+ .filter((u) => !isCollectionMethodPath(u.schemaPath))
508
+ .map((u) => ({
509
+ path: translatePath(u.schemaPath, u.scopeNodeName),
510
+ functionName: u.scopeNodeName,
511
+ })),
512
+ ].filter((pathInfo) => !pathInfo.path.includes('.map('));
513
+ // Deduplicate by translated path + function name, both within this entry
514
+ // AND across all entries. The same target path appears in 30-50 different
515
+ // source entries (every variable flowing through loadView references the same
516
+ // 50 target paths). Without global dedup, we process 5,533 targets instead of 217.
517
+ const allPaths = allPathsRaw.filter((p) => {
518
+ const key = `${p.functionName ?? ''}::${p.path}`;
519
+ if (globalSeenTargets.has(key))
520
+ return false;
521
+ globalSeenTargets.add(key);
522
+ return true;
523
+ });
524
+ // Fix 38: Derive base paths from property access paths.
525
+ // When we have equivalent paths like:
526
+ // Parent: signature[0].scenarios[].name
527
+ // Child: signature[0].selectedScenario.name
528
+ // We want to derive the base paths by finding the common suffix:
529
+ // Common suffix: .name
530
+ // Parent base: signature[0].scenarios[]
531
+ // Child base: signature[0].selectedScenario
532
+ // This allows the merge to find nested child schema fields under the base prop.
533
+ // Find child signature paths (paths from child components)
534
+ const childPaths = allPaths.filter((p) => p.functionName &&
535
+ p.functionName !== rootScopeName &&
536
+ p.functionName !== effectiveFunctionName);
537
+ // Find parent paths (paths from this component)
538
+ const parentPaths = allPaths.filter((p) => !p.functionName ||
539
+ p.functionName === rootScopeName ||
540
+ p.functionName === effectiveFunctionName);
541
+ const derivedBasePaths = [];
542
+ const allPathSet = new Set(allPaths.map((p) => p.path));
543
+ const derivedBasePathSet = new Set();
544
+ // For each child path, find its equivalent parent path and derive bases
545
+ for (const childPathInfo of childPaths) {
546
+ checkDeadline();
547
+ const childParts = splitOutsideParenthesesAndArrays(childPathInfo.path);
548
+ // Look for a parent path that shares a common suffix with this child path
549
+ for (const parentPathInfo of parentPaths) {
550
+ const parentParts = splitOutsideParenthesesAndArrays(parentPathInfo.path);
551
+ // Find the common suffix (from the end)
552
+ let commonSuffixLength = 0;
553
+ const minLen = Math.min(childParts.length, parentParts.length);
554
+ for (let i = 1; i <= minLen; i++) {
555
+ if (childParts[childParts.length - i] ===
556
+ parentParts[parentParts.length - i]) {
557
+ commonSuffixLength = i;
558
+ }
559
+ else {
560
+ break;
561
+ }
471
562
  }
472
- }
473
- // If there's a common suffix and both paths have more parts than the suffix
474
- if (commonSuffixLength > 0 &&
475
- childParts.length > commonSuffixLength &&
476
- parentParts.length > commonSuffixLength) {
477
- const childBaseParts = childParts.slice(0, childParts.length - commonSuffixLength);
478
- const parentBaseParts = parentParts.slice(0, parentParts.length - commonSuffixLength);
479
- // Only derive if BOTH paths look like signature paths.
480
- // This ensures we're handling JSX child-to-parent prop mappings,
481
- // not other complex equivalencies like function call returns.
482
- const isChildSignaturePath = childBaseParts[0]?.startsWith('signature[') ||
483
- (childBaseParts[0]?.endsWith(')') &&
484
- childBaseParts[1]?.startsWith('signature['));
485
- const isParentSignaturePath = parentBaseParts[0]?.startsWith('signature[');
486
- if (isChildSignaturePath && isParentSignaturePath) {
487
- const childBase = joinParenthesesAndArrays(childBaseParts);
488
- const parentBase = joinParenthesesAndArrays(parentBaseParts);
489
- // Only derive if:
490
- // 1. Parent has array iteration (e.g., scenarios[]) and child does NOT
491
- // 2. Bases are different
492
- // 3. Child base is NOT just "signature[N]" (too generic - every component has this)
493
- // We only want specific prop paths like "signature[0].selectedScenario"
494
- // This targets array-to-object mappings like scenarios[] -> selectedScenario
495
- const parentHasArrayIterator = parentBase.includes('[]');
496
- const childHasArrayIterator = childBase.includes('[]');
497
- // Skip if child base is just the generic signature marker (e.g., "signature[0]")
498
- const childBaseIsGenericSignature = /^signature\[\d+\]$/.test(childBase);
499
- if (childBase !== parentBase &&
500
- parentHasArrayIterator &&
501
- !childHasArrayIterator &&
502
- !childBaseIsGenericSignature) {
503
- // Add child base if not already present (O(1) Set lookup)
504
- if (!allPathSet.has(childBase) &&
505
- !derivedBasePathSet.has(childBase)) {
506
- derivedBasePaths.push({
507
- path: childBase,
508
- functionName: childPathInfo.functionName,
509
- });
510
- derivedBasePathSet.add(childBase);
511
- }
512
- // Add parent base if not already present (O(1) Set lookup)
513
- if (!allPathSet.has(parentBase) &&
514
- !derivedBasePathSet.has(parentBase)) {
515
- derivedBasePaths.push({
516
- path: parentBase,
517
- functionName: parentPathInfo.functionName,
518
- });
519
- derivedBasePathSet.add(parentBase);
563
+ // If there's a common suffix and both paths have more parts than the suffix
564
+ if (commonSuffixLength > 0 &&
565
+ childParts.length > commonSuffixLength &&
566
+ parentParts.length > commonSuffixLength) {
567
+ const childBaseParts = childParts.slice(0, childParts.length - commonSuffixLength);
568
+ const parentBaseParts = parentParts.slice(0, parentParts.length - commonSuffixLength);
569
+ // Only derive if BOTH paths look like signature paths.
570
+ // This ensures we're handling JSX child-to-parent prop mappings,
571
+ // not other complex equivalencies like function call returns.
572
+ const isChildSignaturePath = childBaseParts[0]?.startsWith('signature[') ||
573
+ (childBaseParts[0]?.endsWith(')') &&
574
+ childBaseParts[1]?.startsWith('signature['));
575
+ const isParentSignaturePath = parentBaseParts[0]?.startsWith('signature[');
576
+ if (isChildSignaturePath && isParentSignaturePath) {
577
+ const childBase = joinParenthesesAndArrays(childBaseParts);
578
+ const parentBase = joinParenthesesAndArrays(parentBaseParts);
579
+ // Only derive if:
580
+ // 1. Parent has array iteration (e.g., scenarios[]) and child does NOT
581
+ // 2. Bases are different
582
+ // 3. Child base is NOT just "signature[N]" (too generic - every component has this)
583
+ // We only want specific prop paths like "signature[0].selectedScenario"
584
+ // This targets array-to-object mappings like scenarios[] -> selectedScenario
585
+ const parentHasArrayIterator = parentBase.includes('[]');
586
+ const childHasArrayIterator = childBase.includes('[]');
587
+ // Skip if child base is just the generic signature marker (e.g., "signature[0]")
588
+ const childBaseIsGenericSignature = /^signature\[\d+\]$/.test(childBase);
589
+ if (childBase !== parentBase &&
590
+ parentHasArrayIterator &&
591
+ !childHasArrayIterator &&
592
+ !childBaseIsGenericSignature) {
593
+ // Add child base if not already present (O(1) Set lookup)
594
+ if (!allPathSet.has(childBase) &&
595
+ !derivedBasePathSet.has(childBase)) {
596
+ derivedBasePaths.push({
597
+ path: childBase,
598
+ functionName: childPathInfo.functionName,
599
+ });
600
+ derivedBasePathSet.add(childBase);
601
+ }
602
+ // Add parent base if not already present (O(1) Set lookup)
603
+ if (!allPathSet.has(parentBase) &&
604
+ !derivedBasePathSet.has(parentBase)) {
605
+ derivedBasePaths.push({
606
+ path: parentBase,
607
+ functionName: parentPathInfo.functionName,
608
+ });
609
+ derivedBasePathSet.add(parentBase);
610
+ }
520
611
  }
521
612
  }
522
613
  }
523
614
  }
524
615
  }
525
- }
526
- allPaths.push(...derivedBasePaths);
527
- const entry = findOrCreateEquivalentSchemaPathsEntry(allPaths);
528
- // Trace equivalency gathering - helps debug why paths may not be connected
529
- if (allPaths.length > 1) {
530
- transformationTracer.operation(rootScopeName, {
531
- operation: 'gatherEquivalency',
532
- stage: 'gathering',
533
- path: translatedPath,
534
- context: {
535
- sourceFunction: functionName,
536
- equivalentPaths: allPaths.map((p) => ({
537
- path: p.path,
538
- function: p.functionName,
539
- })),
540
- equivalentRoots: entry.equivalentRoots.map((r) => ({
541
- path: r.schemaRootPath,
542
- function: r.function?.name,
543
- })),
544
- },
545
- });
546
- }
547
- for (const equivalentRoot of entry.equivalentRoots) {
548
- const dataStructures = equivalentRoot.function &&
549
- equivalentRoot.function.name !== rootScopeName
550
- ? [
551
- findRelevantDependentDataStructure(equivalentRoot.function.name),
552
- findRelevantDependentAnalysisDataStructure(equivalentRoot.function.name),
553
- ]
554
- : [dataStructure];
555
- // Determine if this is a signature schema path.
556
- // The path might be 'signature[0]...' directly, or 'FuncName().signature[0]...' if it has a function prefix.
557
- const schemaRootParts = splitOutsideParenthesesAndArrays(equivalentRoot.schemaRootPath);
558
- const isSignaturePath = equivalentRoot.schemaRootPath.startsWith('signature[') ||
559
- (schemaRootParts[0]?.endsWith(')') &&
560
- schemaRootParts[1]?.startsWith('signature['));
561
- const schemas = dataStructures.map((dataStructure) => isSignaturePath
562
- ? dataStructure?.signatureSchema
563
- : dataStructure?.returnValueSchema);
564
- let pathParts = splitOutsideParenthesesAndArrays(equivalentRoot.schemaRootPath);
565
- // Fix: When processing a child component's schema, the schemaRootPath has the function
566
- // prefix (e.g., 'ScenarioViewer().signature[0].scenario'), but the child's schema paths
567
- // don't have that prefix (e.g., 'signature[0].scenario.metadata.screenshotPaths').
568
- // Strip the function prefix from pathParts so they can match.
569
- if (equivalentRoot.function &&
570
- pathParts[0].endsWith(')') &&
571
- pathParts[1]?.startsWith('signature[')) {
572
- pathParts = pathParts.slice(1);
616
+ allPaths.push(...derivedBasePaths);
617
+ const entry = findOrCreateEquivalentSchemaPathsEntry(allPaths);
618
+ // Trace equivalency gathering - helps debug why paths may not be connected
619
+ if (allPaths.length > 1) {
620
+ transformationTracer.operation(rootScopeName, {
621
+ operation: 'gatherEquivalency',
622
+ stage: 'gathering',
623
+ path: translatedPath,
624
+ context: {
625
+ sourceFunction: functionName,
626
+ equivalentPaths: allPaths.map((p) => ({
627
+ path: p.path,
628
+ function: p.functionName,
629
+ })),
630
+ equivalentRoots: entry.equivalentRoots.map((r) => ({
631
+ path: r.schemaRootPath,
632
+ function: r.function?.name,
633
+ })),
634
+ },
635
+ });
573
636
  }
574
- for (const schema of schemas) {
575
- // Use pre-computed index to only iterate schema entries whose
576
- // normalized first part matches pathParts[0], instead of all entries.
577
- const schemaIndex = getSchemaIndex(schema);
578
- const lookupPart = pathParts[0] ?? '';
579
- const lookupParenIdx = lookupPart.indexOf('(');
580
- const lookupBase = lookupParenIdx >= 0
581
- ? lookupPart.slice(0, lookupParenIdx)
582
- : lookupPart;
583
- const candidates = schemaIndex.byFirstPart.get(lookupBase) || [];
584
- for (const { path: schemaPath, parts: schemaPathParts, } of candidates) {
585
- if (schemaPathParts.length < pathParts.length)
586
- continue;
587
- // Check if all path parts match (allowing function call variants)
588
- let allMatch = true;
589
- let matchedUpToIndex = pathParts.length;
590
- for (let i = 0; i < pathParts.length; i++) {
591
- if (!pathPartMatches(pathParts[i], schemaPathParts[i])) {
592
- allMatch = false;
593
- break;
594
- }
595
- // If the last pathPart matched a function call variant,
596
- // we need to include it in the postfix calculation
597
- if (i === pathParts.length - 1 &&
598
- schemaPathParts[i] !== pathParts[i] &&
599
- schemaPathParts[i].startsWith(pathParts[i] + '(')) {
600
- // The schemaPathPart is a function call variant (e.g., 'isEntityBeingAnalyzed(entity.sha)')
601
- // We want to include this as part of the postfix
602
- matchedUpToIndex = i;
637
+ for (const equivalentRoot of entry.equivalentRoots) {
638
+ checkDeadline();
639
+ const dataStructures = equivalentRoot.function &&
640
+ equivalentRoot.function.name !== rootScopeName
641
+ ? [
642
+ findRelevantDependentDataStructure(equivalentRoot.function.name),
643
+ findRelevantDependentAnalysisDataStructure(equivalentRoot.function.name),
644
+ ]
645
+ : [dataStructure];
646
+ // Determine if this is a signature schema path.
647
+ // The path might be 'signature[0]...' directly, or 'FuncName().signature[0]...' if it has a function prefix.
648
+ const schemaRootParts = splitOutsideParenthesesAndArrays(equivalentRoot.schemaRootPath);
649
+ const isSignaturePath = equivalentRoot.schemaRootPath.startsWith('signature[') ||
650
+ (schemaRootParts[0]?.endsWith(')') &&
651
+ schemaRootParts[1]?.startsWith('signature['));
652
+ const schemas = dataStructures.map((dataStructure) => isSignaturePath
653
+ ? dataStructure?.signatureSchema
654
+ : dataStructure?.returnValueSchema);
655
+ let pathParts = splitOutsideParenthesesAndArrays(equivalentRoot.schemaRootPath);
656
+ // Fix: When processing a child component's schema, the schemaRootPath has the function
657
+ // prefix (e.g., 'ScenarioViewer().signature[0].scenario'), but the child's schema paths
658
+ // don't have that prefix (e.g., 'signature[0].scenario.metadata.screenshotPaths').
659
+ // Strip the function prefix from pathParts so they can match.
660
+ if (equivalentRoot.function &&
661
+ pathParts[0].endsWith(')') &&
662
+ pathParts[1]?.startsWith('signature[')) {
663
+ pathParts = pathParts.slice(1);
664
+ }
665
+ for (const schema of schemas) {
666
+ // Use pre-computed index to only iterate schema entries whose
667
+ // normalized first part matches pathParts[0], instead of all entries.
668
+ const schemaIndex = getSchemaIndex(schema);
669
+ const lookupPart = pathParts[0] ?? '';
670
+ const lookupParenIdx = lookupPart.indexOf('(');
671
+ const lookupBase = lookupParenIdx >= 0
672
+ ? lookupPart.slice(0, lookupParenIdx)
673
+ : lookupPart;
674
+ const candidates = schemaIndex.byFirstPart.get(lookupBase) || [];
675
+ for (const { path: schemaPath, parts: schemaPathParts, } of candidates) {
676
+ checkDeadline();
677
+ if (schemaPathParts.length < pathParts.length)
678
+ continue;
679
+ // Check if all path parts match (allowing function call variants)
680
+ let allMatch = true;
681
+ let matchedUpToIndex = pathParts.length;
682
+ for (let i = 0; i < pathParts.length; i++) {
683
+ if (!pathPartMatches(pathParts[i], schemaPathParts[i])) {
684
+ allMatch = false;
685
+ break;
686
+ }
687
+ // If the last pathPart matched a function call variant,
688
+ // we need to include it in the postfix calculation
689
+ if (i === pathParts.length - 1 &&
690
+ schemaPathParts[i] !== pathParts[i] &&
691
+ schemaPathParts[i].startsWith(pathParts[i] + '(')) {
692
+ // The schemaPathPart is a function call variant (e.g., 'isEntityBeingAnalyzed(entity.sha)')
693
+ // We want to include this as part of the postfix
694
+ matchedUpToIndex = i;
695
+ }
603
696
  }
604
- }
605
- if (allMatch) {
606
- // When we matched a function call variant at the end (e.g., 'foo' matched 'foo(args)'),
607
- // the base itself should be marked as a function, and the function call details
608
- // should be included as sub-paths
609
- if (matchedUpToIndex < pathParts.length) {
610
- // This is a function call variant match at the last position
611
- // Mark the base as a function (empty postfix = the base path itself)
612
- entry.equivalentPostfixes[''] = bestValueFromOptions([
613
- entry.equivalentPostfixes[''],
614
- 'function',
615
- ]);
616
- // Also capture the function call and any remaining parts
617
- // e.g., 'isEntityBeingAnalyzed(entity.sha)' or 'isEntityBeingAnalyzed(entity.sha).functionCallReturnValue'
618
- const funcCallPart = schemaPathParts[matchedUpToIndex];
619
- const baseName = pathParts[matchedUpToIndex]; // e.g., 'isEntityBeingAnalyzed'
620
- const argsMatch = funcCallPart.match(/\(.*\)$/);
621
- if (argsMatch) {
622
- // Create postfix using just the args portion: (entity.sha) instead of isEntityBeingAnalyzed(entity.sha)
623
- // This avoids duplicating the base name in the final path
624
- const argsPortion = argsMatch[0]; // e.g., '(entity.sha)'
625
- const remainingParts = schemaPathParts.slice(matchedUpToIndex + 1);
626
- // Build the postfix as: (args).remaining.parts
627
- const funcPostfix = joinParenthesesAndArrays([
628
- argsPortion,
629
- ...remainingParts,
697
+ if (allMatch) {
698
+ // When we matched a function call variant at the end (e.g., 'foo' matched 'foo(args)'),
699
+ // the base itself should be marked as a function, and the function call details
700
+ // should be included as sub-paths
701
+ if (matchedUpToIndex < pathParts.length) {
702
+ // This is a function call variant match at the last position
703
+ // Mark the base as a function (empty postfix = the base path itself)
704
+ entry.equivalentPostfixes[''] = bestValueFromOptions([
705
+ entry.equivalentPostfixes[''],
706
+ 'function',
630
707
  ]);
631
- entry.equivalentPostfixes[funcPostfix] = entry
632
- .equivalentPostfixes[funcPostfix]
633
- ? bestValueFromOptions([
634
- entry.equivalentPostfixes[funcPostfix],
635
- schema[schemaPath],
636
- ])
637
- : schema[schemaPath];
708
+ // Also capture the function call and any remaining parts
709
+ // e.g., 'isEntityBeingAnalyzed(entity.sha)' or 'isEntityBeingAnalyzed(entity.sha).functionCallReturnValue'
710
+ const funcCallPart = schemaPathParts[matchedUpToIndex];
711
+ const baseName = pathParts[matchedUpToIndex]; // e.g., 'isEntityBeingAnalyzed'
712
+ const argsMatch = funcCallPart.match(/\(.*\)$/);
713
+ if (argsMatch) {
714
+ // Create postfix using just the args portion: (entity.sha) instead of isEntityBeingAnalyzed(entity.sha)
715
+ // This avoids duplicating the base name in the final path
716
+ const argsPortion = argsMatch[0]; // e.g., '(entity.sha)'
717
+ const remainingParts = schemaPathParts.slice(matchedUpToIndex + 1);
718
+ // Build the postfix as: (args).remaining.parts
719
+ const funcPostfix = joinParenthesesAndArrays([
720
+ argsPortion,
721
+ ...remainingParts,
722
+ ]);
723
+ entry.equivalentPostfixes[funcPostfix] = entry
724
+ .equivalentPostfixes[funcPostfix]
725
+ ? bestValueFromOptions([
726
+ entry.equivalentPostfixes[funcPostfix],
727
+ schema[schemaPath],
728
+ ])
729
+ : schema[schemaPath];
730
+ }
638
731
  }
639
- }
640
- else {
641
- // Regular exact match - use the standard postfix logic
642
- const postfix = joinParenthesesAndArrays(schemaPathParts.slice(matchedUpToIndex));
643
- const previousValue = entry.equivalentPostfixes[postfix];
644
- const newValue = schema[schemaPath];
645
- entry.equivalentPostfixes[postfix] = previousValue
646
- ? bestValueFromOptions([previousValue, newValue])
647
- : newValue;
648
- // Trace postfix gathering - shows where type info comes from
649
- if (entry.equivalentPostfixes[postfix] !== previousValue) {
650
- transformationTracer.operation(rootScopeName, {
651
- operation: 'gatherPostfix',
652
- stage: 'gathering',
653
- path: postfix || '(root)',
654
- before: previousValue,
655
- after: entry.equivalentPostfixes[postfix],
656
- context: {
657
- sourceSchemaPath: schemaPath,
658
- sourceFunction: equivalentRoot.function?.name || rootScopeName,
659
- equivalentRootPath: equivalentRoot.schemaRootPath,
660
- rawValue: newValue,
661
- },
662
- });
732
+ else {
733
+ // Regular exact match - use the standard postfix logic
734
+ const postfix = joinParenthesesAndArrays(schemaPathParts.slice(matchedUpToIndex));
735
+ const previousValue = entry.equivalentPostfixes[postfix];
736
+ const newValue = schema[schemaPath];
737
+ entry.equivalentPostfixes[postfix] = previousValue
738
+ ? bestValueFromOptions([previousValue, newValue])
739
+ : newValue;
740
+ // Trace postfix gathering - shows where type info comes from
741
+ if (entry.equivalentPostfixes[postfix] !== previousValue) {
742
+ transformationTracer.operation(rootScopeName, {
743
+ operation: 'gatherPostfix',
744
+ stage: 'gathering',
745
+ path: postfix || '(root)',
746
+ before: previousValue,
747
+ after: entry.equivalentPostfixes[postfix],
748
+ context: {
749
+ sourceSchemaPath: schemaPath,
750
+ sourceFunction: equivalentRoot.function?.name || rootScopeName,
751
+ equivalentRootPath: equivalentRoot.schemaRootPath,
752
+ rawValue: newValue,
753
+ },
754
+ });
755
+ }
663
756
  }
664
757
  }
665
758
  }
@@ -667,782 +760,864 @@ export default function mergeInDependentDataStructure({ importedExports, depende
667
760
  }
668
761
  }
669
762
  }
670
- }
671
- if (Object.keys(dataStructure?.returnValueSchema ?? {}).length > 0) {
672
- // Find all paths that contain functionCallReturnValue and extract unique base paths
673
- // For each path containing functionCallReturnValue, find the FIRST occurrence and use
674
- // that as a base path. This handles nested cases like:
675
- // X().functionCallReturnValue.A.B.Y().functionCallReturnValue
676
- // where we want both X().functionCallReturnValue and Y().functionCallReturnValue as bases
677
- const allBasePaths = new Set();
678
- for (const path of Object.keys(dataStructure.returnValueSchema)) {
679
- const parts = splitOutsideParenthesesAndArrays(path);
680
- // Find all positions of functionCallReturnValue and create base paths for each
681
- for (let i = 0; i < parts.length; i++) {
682
- if (parts[i] === 'functionCallReturnValue') {
683
- const basePath = joinParenthesesAndArrays(parts.slice(0, i + 1));
684
- allBasePaths.add(basePath);
763
+ if (Object.keys(dataStructure?.returnValueSchema ?? {}).length > 0) {
764
+ // Find all paths that contain functionCallReturnValue and extract unique base paths
765
+ // For each path containing functionCallReturnValue, find the FIRST occurrence and use
766
+ // that as a base path. This handles nested cases like:
767
+ // X().functionCallReturnValue.A.B.Y().functionCallReturnValue
768
+ // where we want both X().functionCallReturnValue and Y().functionCallReturnValue as bases
769
+ const allBasePaths = new Set();
770
+ for (const path of Object.keys(dataStructure.returnValueSchema)) {
771
+ checkDeadline();
772
+ const parts = splitOutsideParenthesesAndArrays(path);
773
+ // Find all positions of functionCallReturnValue and create base paths for each
774
+ for (let i = 0; i < parts.length; i++) {
775
+ if (parts[i] === 'functionCallReturnValue') {
776
+ const basePath = joinParenthesesAndArrays(parts.slice(0, i + 1));
777
+ allBasePaths.add(basePath);
778
+ }
685
779
  }
686
780
  }
687
- }
688
- // Sort by length so shorter paths are processed first
689
- const sortedBasePaths = [...allBasePaths].sort((a, b) => a.length - b.length);
690
- for (const basePath of sortedBasePaths) {
691
- const translatedBasePath = translatePath(basePath, functionName);
692
- const entry = findOrCreateEquivalentSchemaPathsEntry([
693
- { path: translatedBasePath, functionName: functionName },
694
- ]);
695
- const newRoot = {
696
- schemaRootPath: translatedBasePath,
697
- function: findRelevantDependency(functionName),
698
- };
699
- entry.equivalentRoots.push(newRoot);
700
- // Update index for the newly added root
701
- const newRootFuncName = newRoot.function?.name ?? rootScopeName;
702
- espIndex.set(espIndexKey(newRoot.schemaRootPath, newRootFuncName), entry);
703
- const basePathParts = splitOutsideParenthesesAndArrays(basePath);
704
- for (const schemaPath in dataStructure.returnValueSchema) {
705
- const schemaPathParts = splitOutsideParenthesesAndArrays(schemaPath);
706
- if (schemaPathParts.length < basePathParts.length)
707
- continue;
708
- // Check if this schemaPath actually starts with this basePath
709
- // (not just has the same length prefix)
710
- const prefixParts = schemaPathParts.slice(0, basePathParts.length);
711
- if (joinParenthesesAndArrays(prefixParts) !==
712
- joinParenthesesAndArrays(basePathParts)) {
713
- continue;
781
+ // Sort by length so shorter paths are processed first
782
+ const sortedBasePaths = [...allBasePaths].sort((a, b) => a.length - b.length);
783
+ for (const basePath of sortedBasePaths) {
784
+ const translatedBasePath = translatePath(basePath, functionName);
785
+ const entry = findOrCreateEquivalentSchemaPathsEntry([
786
+ { path: translatedBasePath, functionName: functionName },
787
+ ]);
788
+ const newRoot = {
789
+ schemaRootPath: translatedBasePath,
790
+ function: findRelevantDependency(functionName),
791
+ };
792
+ entry.equivalentRoots.push(newRoot);
793
+ // Update index for the newly added root
794
+ const newRootFuncName = newRoot.function?.name ?? rootScopeName;
795
+ espIndex.set(espIndexKey(newRoot.schemaRootPath, newRootFuncName), entry);
796
+ const basePathParts = splitOutsideParenthesesAndArrays(basePath);
797
+ for (const schemaPath in dataStructure.returnValueSchema) {
798
+ checkDeadline();
799
+ const schemaPathParts = splitOutsideParenthesesAndArrays(schemaPath);
800
+ if (schemaPathParts.length < basePathParts.length)
801
+ continue;
802
+ // Check if this schemaPath actually starts with this basePath
803
+ // (not just has the same length prefix)
804
+ const prefixParts = schemaPathParts.slice(0, basePathParts.length);
805
+ if (joinParenthesesAndArrays(prefixParts) !==
806
+ joinParenthesesAndArrays(basePathParts)) {
807
+ continue;
808
+ }
809
+ const postfix = joinParenthesesAndArrays(schemaPathParts.slice(basePathParts.length));
810
+ const newValue = entry.equivalentPostfixes[postfix]
811
+ ? bestValueFromOptions([
812
+ entry.equivalentPostfixes[postfix],
813
+ dataStructure.returnValueSchema[schemaPath],
814
+ ])
815
+ : dataStructure.returnValueSchema[schemaPath];
816
+ entry.equivalentPostfixes[postfix] = newValue;
714
817
  }
715
- const postfix = joinParenthesesAndArrays(schemaPathParts.slice(basePathParts.length));
716
- const newValue = entry.equivalentPostfixes[postfix]
717
- ? bestValueFromOptions([
718
- entry.equivalentPostfixes[postfix],
719
- dataStructure.returnValueSchema[schemaPath],
720
- ])
721
- : dataStructure.returnValueSchema[schemaPath];
722
- entry.equivalentPostfixes[postfix] = newValue;
723
818
  }
724
819
  }
725
- }
726
- };
727
- const mergeAllEquivalentSchemaPaths = () => {
728
- const mergedEquivalentSchemaPaths = [];
729
- // Pre-pass: Connect entries with array/array-element relationships.
730
- // This handles cases like:
731
- // - Entry A has root 'surveys' (array)
732
- // - Entry B has root 'surveys[]' (array element)
733
- // These need to be connected so Entry B's field postfixes flow to Entry A.
734
- // We do this before the main merge to ensure the connection happens regardless
735
- // of processing order.
736
- for (const esp of equivalentSchemaPaths) {
737
- for (const root of esp.equivalentRoots) {
738
- if (root.schemaRootPath.endsWith('[]')) {
739
- // Find a matching parent entry with the base array path
740
- const baseArrayPath = root.schemaRootPath.slice(0, -2);
741
- const parentEntry = equivalentSchemaPaths.find((other) => other !== esp &&
742
- other.equivalentRoots.some((otherRoot) => otherRoot.schemaRootPath === baseArrayPath &&
743
- otherRoot.function?.name === root.function?.name &&
744
- otherRoot.function?.filePath === root.function?.filePath));
745
- if (parentEntry) {
746
- // Add transformed postfixes from child (array element) to parent (array)
747
- // so they can be applied with [] prefix to parent paths
748
- for (const [postfixPath, postfixValue] of Object.entries(esp.equivalentPostfixes)) {
749
- const transformedPostfix = joinParenthesesAndArrays(['[]', postfixPath].filter(Boolean));
750
- if (!(transformedPostfix in parentEntry.equivalentPostfixes)) {
751
- parentEntry.equivalentPostfixes[transformedPostfix] =
752
- postfixValue;
820
+ };
821
+ const mergeAllEquivalentSchemaPaths = () => {
822
+ const mergedEquivalentSchemaPaths = [];
823
+ // Pre-pass: Connect entries with array/array-element relationships.
824
+ // This handles cases like:
825
+ // - Entry A has root 'surveys' (array)
826
+ // - Entry B has root 'surveys[]' (array element)
827
+ // These need to be connected so Entry B's field postfixes flow to Entry A.
828
+ // We do this before the main merge to ensure the connection happens regardless
829
+ // of processing order.
830
+ for (const esp of equivalentSchemaPaths) {
831
+ checkDeadline();
832
+ for (const root of esp.equivalentRoots) {
833
+ if (root.schemaRootPath.endsWith('[]')) {
834
+ // Find a matching parent entry with the base array path
835
+ const baseArrayPath = root.schemaRootPath.slice(0, -2);
836
+ const parentEntry = equivalentSchemaPaths.find((other) => other !== esp &&
837
+ other.equivalentRoots.some((otherRoot) => otherRoot.schemaRootPath === baseArrayPath &&
838
+ otherRoot.function?.name === root.function?.name &&
839
+ otherRoot.function?.filePath === root.function?.filePath));
840
+ if (parentEntry) {
841
+ // Add transformed postfixes from child (array element) to parent (array)
842
+ // so they can be applied with [] prefix to parent paths
843
+ for (const [postfixPath, postfixValue] of Object.entries(esp.equivalentPostfixes)) {
844
+ checkDeadline();
845
+ const transformedPostfix = joinParenthesesAndArrays(['[]', postfixPath].filter(Boolean));
846
+ if (!(transformedPostfix in parentEntry.equivalentPostfixes)) {
847
+ parentEntry.equivalentPostfixes[transformedPostfix] =
848
+ postfixValue;
849
+ }
753
850
  }
754
851
  }
755
852
  }
756
853
  }
757
854
  }
758
- }
759
- const findEquivalentSchemaPathEntry = (schemaSubPath, equivalentRootFunction) => {
760
- let postfix;
761
- // Get the signature index we're looking for (if any)
762
- const lookingForSignatureIndex = extractSignatureIndex(schemaSubPath);
763
- const equivalentEntry = mergedEquivalentSchemaPaths.find((esp) => esp.equivalentRoots.some((er) => {
764
- if ((schemaSubPath.startsWith('returnValue') ||
765
- schemaSubPath.startsWith('signature[')) &&
766
- (er.function?.name !== equivalentRootFunction?.name ||
767
- er.function?.filePath !== equivalentRootFunction?.filePath)) {
768
- return false;
769
- }
770
- if (schemaSubPath === er.schemaRootPath) {
771
- // Additional check: if we're looking for a signature path, make sure
772
- // the entry doesn't already have DIFFERENT signature indices.
773
- // This prevents entries with signature[1], signature[2], signature[4]
774
- // from all being merged together.
775
- if (lookingForSignatureIndex !== undefined) {
776
- const hasConflictingSignatureIndex = esp.equivalentRoots.some((otherRoot) => {
777
- // Only check roots from the same function
778
- if (otherRoot.function?.name !== equivalentRootFunction?.name ||
779
- otherRoot.function?.filePath !==
780
- equivalentRootFunction?.filePath) {
855
+ const findEquivalentSchemaPathEntry = (schemaSubPath, equivalentRootFunction) => {
856
+ let postfix;
857
+ // Get the signature index we're looking for (if any)
858
+ const lookingForSignatureIndex = extractSignatureIndex(schemaSubPath);
859
+ const equivalentEntry = mergedEquivalentSchemaPaths.find((esp) => esp.equivalentRoots.some((er) => {
860
+ if ((schemaSubPath.startsWith('returnValue') ||
861
+ schemaSubPath.startsWith('signature[')) &&
862
+ (er.function?.name !== equivalentRootFunction?.name ||
863
+ er.function?.filePath !== equivalentRootFunction?.filePath)) {
864
+ return false;
865
+ }
866
+ if (schemaSubPath === er.schemaRootPath) {
867
+ // Additional check: if we're looking for a signature path, make sure
868
+ // the entry doesn't already have DIFFERENT signature indices.
869
+ // This prevents entries with signature[1], signature[2], signature[4]
870
+ // from all being merged together.
871
+ if (lookingForSignatureIndex !== undefined) {
872
+ const hasConflictingSignatureIndex = esp.equivalentRoots.some((otherRoot) => {
873
+ // Only check roots from the same function
874
+ if (otherRoot.function?.name !==
875
+ equivalentRootFunction?.name ||
876
+ otherRoot.function?.filePath !==
877
+ equivalentRootFunction?.filePath) {
878
+ return false;
879
+ }
880
+ const otherIndex = extractSignatureIndex(otherRoot.schemaRootPath);
881
+ return (otherIndex !== undefined &&
882
+ otherIndex !== lookingForSignatureIndex);
883
+ });
884
+ if (hasConflictingSignatureIndex) {
781
885
  return false;
782
886
  }
783
- const otherIndex = extractSignatureIndex(otherRoot.schemaRootPath);
784
- return (otherIndex !== undefined &&
785
- otherIndex !== lookingForSignatureIndex);
786
- });
787
- if (hasConflictingSignatureIndex) {
788
- return false;
789
887
  }
888
+ postfix = er.postfix;
889
+ return true;
790
890
  }
791
- postfix = er.postfix;
792
- return true;
793
- }
794
- return false;
795
- }));
796
- return { equivalentEntry, postfix };
797
- };
798
- const sortedEquivalentSchemaPaths = equivalentSchemaPaths.sort((a, b) => Math.max(...a.equivalentRoots.map((er) => splitOutsideParenthesesAndArrays(er.schemaRootPath).length)) -
799
- Math.max(...b.equivalentRoots.map((er) => splitOutsideParenthesesAndArrays(er.schemaRootPath).length)));
800
- for (const esp of sortedEquivalentSchemaPaths) {
801
- if (esp.equivalentRoots.length === 0)
802
- continue;
803
- let bestCandidateLength;
804
- let bestCandidate;
805
- let postfix;
806
- for (const equivalentRoot of esp.equivalentRoots) {
807
- const rootSchemaPath = equivalentRoot.schemaRootPath;
808
- const schemaPathParts = splitOutsideParenthesesAndArrays(rootSchemaPath);
809
- for (let i = 0; i < schemaPathParts.length; i++) {
810
- const subPath = joinParenthesesAndArrays(schemaPathParts.slice(0, i + 1));
811
- const { equivalentEntry, postfix: equivalentEntryPostfix } = findEquivalentSchemaPathEntry(subPath, equivalentRoot.function);
812
- if (equivalentEntry &&
813
- (!bestCandidateLength || bestCandidateLength > i + 1)) {
814
- bestCandidate = equivalentEntry;
815
- bestCandidateLength = i + 1;
816
- postfix = joinParenthesesAndArrays([equivalentEntryPostfix, ...schemaPathParts.slice(i + 1)].filter(Boolean));
891
+ return false;
892
+ }));
893
+ return { equivalentEntry, postfix };
894
+ };
895
+ const sortedEquivalentSchemaPaths = equivalentSchemaPaths.sort((a, b) => Math.max(...a.equivalentRoots.map((er) => splitOutsideParenthesesAndArrays(er.schemaRootPath).length)) -
896
+ Math.max(...b.equivalentRoots.map((er) => splitOutsideParenthesesAndArrays(er.schemaRootPath).length)));
897
+ for (const esp of sortedEquivalentSchemaPaths) {
898
+ checkDeadline();
899
+ if (esp.equivalentRoots.length === 0)
900
+ continue;
901
+ let bestCandidateLength;
902
+ let bestCandidate;
903
+ let postfix;
904
+ for (const equivalentRoot of esp.equivalentRoots) {
905
+ const rootSchemaPath = equivalentRoot.schemaRootPath;
906
+ const schemaPathParts = splitOutsideParenthesesAndArrays(rootSchemaPath);
907
+ for (let i = 0; i < schemaPathParts.length; i++) {
908
+ const subPath = joinParenthesesAndArrays(schemaPathParts.slice(0, i + 1));
909
+ const { equivalentEntry, postfix: equivalentEntryPostfix } = findEquivalentSchemaPathEntry(subPath, equivalentRoot.function);
910
+ if (equivalentEntry &&
911
+ (!bestCandidateLength || bestCandidateLength > i + 1)) {
912
+ bestCandidate = equivalentEntry;
913
+ bestCandidateLength = i + 1;
914
+ postfix = joinParenthesesAndArrays([
915
+ equivalentEntryPostfix,
916
+ ...schemaPathParts.slice(i + 1),
917
+ ].filter(Boolean));
918
+ }
817
919
  }
818
920
  }
819
- }
820
- if (bestCandidate) {
821
- for (const root of esp.equivalentRoots) {
822
- if (postfix.length > 0) {
823
- root.postfix = postfix;
921
+ if (bestCandidate) {
922
+ for (const root of esp.equivalentRoots) {
923
+ if (postfix.length > 0) {
924
+ root.postfix = postfix;
925
+ }
926
+ bestCandidate.equivalentRoots.push(root);
824
927
  }
825
- bestCandidate.equivalentRoots.push(root);
928
+ const postfixesToMerge = postfix.length > 0
929
+ ? Object.keys(esp.equivalentPostfixes).reduce((acc, postfixPath) => {
930
+ const fullPath = joinParenthesesAndArrays([
931
+ postfix,
932
+ postfixPath,
933
+ ]);
934
+ acc[fullPath] = esp.equivalentPostfixes[postfixPath];
935
+ return acc;
936
+ }, {})
937
+ : esp.equivalentPostfixes;
938
+ bestCandidate.equivalentPostfixes = {
939
+ ...bestCandidate.equivalentPostfixes,
940
+ ...postfixesToMerge,
941
+ };
942
+ }
943
+ else {
944
+ mergedEquivalentSchemaPaths.push(esp);
826
945
  }
827
- const postfixesToMerge = postfix.length > 0
828
- ? Object.keys(esp.equivalentPostfixes).reduce((acc, postfixPath) => {
829
- const fullPath = joinParenthesesAndArrays([
830
- postfix,
831
- postfixPath,
832
- ]);
833
- acc[fullPath] = esp.equivalentPostfixes[postfixPath];
834
- return acc;
835
- }, {})
836
- : esp.equivalentPostfixes;
837
- bestCandidate.equivalentPostfixes = {
838
- ...bestCandidate.equivalentPostfixes,
839
- ...postfixesToMerge,
840
- };
841
946
  }
842
- else {
843
- mergedEquivalentSchemaPaths.push(esp);
947
+ return mergedEquivalentSchemaPaths;
948
+ };
949
+ // Build a lookup of mocked dependencies to skip their internal implementation
950
+ const mockedDependencies = new Set();
951
+ for (const dep of importedExports) {
952
+ if (dep.isMocked) {
953
+ mockedDependencies.add(`${dep.filePath}::${dep.name}`);
844
954
  }
845
955
  }
846
- return mergedEquivalentSchemaPaths;
847
- };
848
- // Build a lookup of mocked dependencies to skip their internal implementation
849
- const mockedDependencies = new Set();
850
- for (const dep of importedExports) {
851
- if (dep.isMocked) {
852
- mockedDependencies.add(`${dep.filePath}::${dep.name}`);
853
- }
854
- }
855
- gatherAllEquivalentSchemaPaths(rootScopeName, dataStructure);
856
- // Process dependencySchemas for all dependencies (including mocked ones)
857
- // dependencySchemas contains usage information (how dependencies are called),
858
- // not internal implementation, so we want this for mocked dependencies too
859
- for (const dependency of importedExports) {
860
- const dependentDataStructure = dependencySchemas?.[dependency.filePath]?.[dependency.name];
861
- if (!dependentDataStructure)
862
- continue;
863
- gatherAllEquivalentSchemaPaths(dependency.name, dependentDataStructure, dependentDataStructure);
864
- }
865
- for (const filePath in dependentAnalyses) {
866
- for (const name in dependentAnalyses[filePath]) {
867
- // Skip mocked dependencies - we don't want to merge in their internal implementation
868
- if (mockedDependencies.has(`${filePath}::${name}`)) {
956
+ gatherAllEquivalentSchemaPaths(rootScopeName, dataStructure);
957
+ // Process dependencySchemas for all dependencies (including mocked ones)
958
+ // dependencySchemas contains usage information (how dependencies are called),
959
+ // not internal implementation, so we want this for mocked dependencies too
960
+ for (const dependency of importedExports) {
961
+ checkDeadline();
962
+ const dependentDataStructure = dependencySchemas?.[dependency.filePath]?.[dependency.name];
963
+ if (!dependentDataStructure)
869
964
  continue;
870
- }
871
- const childMergedDataStructure = dependentAnalyses[filePath][name].metadata?.mergedDataStructure || {};
872
- gatherAllEquivalentSchemaPaths(name, childMergedDataStructure);
873
- }
874
- }
875
- equivalentSchemaPaths = mergeAllEquivalentSchemaPaths();
876
- // Collect schemas that need cleaning — batch the calls for the end instead of
877
- // calling cleanSchema inside the inner root loop (which was O(roots * schemaSize)).
878
- const schemasToClean = new Set();
879
- for (const esp of equivalentSchemaPaths) {
880
- // Pre-compute which postfixes have children to avoid O(n²) lookups in the inner loop.
881
- // A postfix "has children" if there are other postfixes that extend it.
882
- const postfixesWithChildren = new Set();
883
- const postfixKeys = Object.keys(esp.equivalentPostfixes);
884
- // Check for empty postfix having children (any other postfixes exist)
885
- if (postfixKeys.length > 1 && '' in esp.equivalentPostfixes) {
886
- postfixesWithChildren.add('');
965
+ gatherAllEquivalentSchemaPaths(dependency.name, dependentDataStructure, dependentDataStructure);
887
966
  }
888
- // Check for array element postfixes having children using a prefix set.
889
- // This avoids O(n²) scans across large postfix lists.
890
- // e.g., 'currentEntities[]' has children if a path like 'currentEntities[].sha' exists.
891
- const postfixPrefixSet = new Set();
892
- for (const postfixPath of postfixKeys) {
893
- if (!postfixPath)
894
- continue;
895
- const parts = splitOutsideParenthesesAndArrays(postfixPath);
896
- for (let i = 1; i < parts.length; i++) {
897
- postfixPrefixSet.add(joinParenthesesAndArrays(parts.slice(0, i)));
967
+ for (const filePath in dependentAnalyses) {
968
+ for (const name in dependentAnalyses[filePath]) {
969
+ // Skip mocked dependencies - we don't want to merge in their internal implementation
970
+ if (mockedDependencies.has(`${filePath}::${name}`)) {
971
+ continue;
972
+ }
973
+ const childMergedDataStructure = dependentAnalyses[filePath][name].metadata?.mergedDataStructure || {};
974
+ gatherAllEquivalentSchemaPaths(name, childMergedDataStructure);
898
975
  }
899
976
  }
900
- for (const postfixPath of postfixKeys) {
901
- if (postfixPath.endsWith('[]') && postfixPrefixSet.has(postfixPath)) {
902
- postfixesWithChildren.add(postfixPath);
977
+ const gatherElapsed = Date.now() - mergeStartTime;
978
+ equivalentSchemaPaths = mergeAllEquivalentSchemaPaths();
979
+ const mergeEspElapsed = Date.now() - mergeStartTime;
980
+ // Collect schemas that need cleaning — batch the calls for the end instead of
981
+ // calling cleanSchema inside the inner root loop (which was O(roots * schemaSize)).
982
+ const schemasToClean = new Set();
983
+ for (const esp of equivalentSchemaPaths) {
984
+ checkDeadline();
985
+ // Pre-compute which postfixes have children to avoid O(n²) lookups in the inner loop.
986
+ // A postfix "has children" if there are other postfixes that extend it.
987
+ const postfixesWithChildren = new Set();
988
+ const postfixKeys = Object.keys(esp.equivalentPostfixes);
989
+ // Pre-parse ALL postfix paths once. These parsed parts are reused in:
990
+ // 1. The children detection loop below
991
+ // 2. The inner postfix application loop (lines that split postfixPath and equivalentRoot.postfix)
992
+ // This eliminates thousands of redundant splitOutsideParenthesesAndArrays calls.
993
+ const postfixPartsCache = new Map();
994
+ for (const postfixPath of postfixKeys) {
995
+ if (!postfixPath)
996
+ continue;
997
+ postfixPartsCache.set(postfixPath, splitOutsideParenthesesAndArrays(postfixPath));
903
998
  }
904
- }
905
- // Deduplicate equivalentRoots that would write to the same schema paths.
906
- // Roots with the same (function, schemaRootPath, postfix) are redundant.
907
- const seenRootKeys = new Set();
908
- const uniqueRoots = esp.equivalentRoots.filter((root) => {
909
- const key = `${root.function?.filePath ?? ''}::${root.function?.name ?? ''}::${root.schemaRootPath}::${root.postfix ?? ''}`;
910
- if (seenRootKeys.has(key))
911
- return false;
912
- seenRootKeys.add(key);
913
- return true;
914
- });
915
- for (const equivalentRoot of uniqueRoots) {
916
- let merged;
917
- if (equivalentRoot.function) {
918
- merged = findOrCreateDependentSchemas(equivalentRoot.function);
999
+ // Check for empty postfix having children (any other postfixes exist)
1000
+ if (postfixKeys.length > 1 && '' in esp.equivalentPostfixes) {
1001
+ postfixesWithChildren.add('');
919
1002
  }
920
- else {
921
- merged = mergedDataStructure;
1003
+ // Check for array element postfixes having children using a prefix set.
1004
+ // This avoids O(n²) scans across large postfix lists.
1005
+ // e.g., 'currentEntities[]' has children if a path like 'currentEntities[].sha' exists.
1006
+ const postfixPrefixSet = new Set();
1007
+ for (const postfixPath of postfixKeys) {
1008
+ if (!postfixPath)
1009
+ continue;
1010
+ const parts = postfixPartsCache.get(postfixPath);
1011
+ for (let i = 1; i < parts.length; i++) {
1012
+ postfixPrefixSet.add(joinParenthesesAndArrays(parts.slice(0, i)));
1013
+ }
922
1014
  }
923
- if (!merged)
924
- continue;
925
- const schema = equivalentRoot.schemaRootPath.startsWith('signature[')
926
- ? merged.signatureSchema
927
- : merged.returnValueSchema;
928
- for (const [postfixPath, postfixValue] of Object.entries(esp.equivalentPostfixes)) {
929
- let relevantPostfix = postfixPath;
930
- if (equivalentRoot.postfix) {
931
- // Check if postfixPath starts with equivalentRoot.postfix at a path boundary.
932
- // Must ensure exact path part match - "entityCode" should NOT match "entity" prefix.
933
- // Valid: "entity.foo" starts with "entity" (boundary at '.')
934
- // Valid: "entity[0]" starts with "entity" (boundary at '[')
935
- // Invalid: "entityCode" starts with "entity" (no boundary, different property)
936
- if (!postfixPath.startsWith(equivalentRoot.postfix)) {
937
- continue;
938
- }
939
- // Additional check: ensure the match is at a path boundary
940
- const nextChar = postfixPath[equivalentRoot.postfix.length];
941
- if (nextChar !== undefined && nextChar !== '.' && nextChar !== '[') {
942
- // The postfixPath continues with more characters that aren't a path separator.
943
- // This means "entity" matched "entityCode" which is wrong - they're different properties.
944
- continue;
945
- }
946
- const postFixPathParts = splitOutsideParenthesesAndArrays(postfixPath);
947
- const equivalentRootPostFixParts = splitOutsideParenthesesAndArrays(equivalentRoot.postfix);
948
- relevantPostfix = joinParenthesesAndArrays(postFixPathParts.slice(equivalentRootPostFixParts.length));
1015
+ for (const postfixPath of postfixKeys) {
1016
+ if (postfixPath.endsWith('[]') && postfixPrefixSet.has(postfixPath)) {
1017
+ postfixesWithChildren.add(postfixPath);
949
1018
  }
950
- const newSchemaPath = joinParenthesesAndArrays([
951
- equivalentRoot.schemaRootPath,
952
- relevantPostfix,
953
- ]);
954
- // Skip paths that would go through a primitive type
955
- // e.g., if schema has 'entities[].scenarioCount': 'number', skip 'entities[].scenarioCount.sha'
956
- if (wouldGoThroughPrimitive(newSchemaPath, schema)) {
957
- transformationTracer.operation(rootScopeName, {
958
- operation: 'skipPrimitivePath',
959
- stage: 'merged',
960
- path: newSchemaPath,
961
- context: {
962
- reason: 'would go through primitive type',
963
- postfixValue,
964
- },
965
- });
966
- continue;
1019
+ }
1020
+ // Deduplicate equivalentRoots that would write to the same schema paths.
1021
+ // Roots with the same (function, schemaRootPath, postfix) are redundant.
1022
+ const seenRootKeys = new Set();
1023
+ const uniqueRoots = esp.equivalentRoots.filter((root) => {
1024
+ const key = `${root.function?.filePath ?? ''}::${root.function?.name ?? ''}::${root.schemaRootPath}::${root.postfix ?? ''}`;
1025
+ if (seenRootKeys.has(key))
1026
+ return false;
1027
+ seenRootKeys.add(key);
1028
+ return true;
1029
+ });
1030
+ for (const equivalentRoot of uniqueRoots) {
1031
+ checkDeadline();
1032
+ let merged;
1033
+ if (equivalentRoot.function) {
1034
+ merged = findOrCreateDependentSchemas(equivalentRoot.function);
967
1035
  }
968
- // Skip setting primitive type when there are child postfixes that indicate structure.
969
- // This prevents downgrading an object/array element to a primitive type.
970
- // Uses pre-computed postfixesWithChildren Set for O(1) lookup instead of O(n) iteration.
971
- const hasChildPostfixes = (relevantPostfix === '' || relevantPostfix.endsWith('[]')) &&
972
- postfixesWithChildren.has(postfixPath);
973
- if (PRIMITIVE_TYPES.has(postfixValue) && hasChildPostfixes) {
974
- continue;
1036
+ else {
1037
+ merged = mergedDataStructure;
975
1038
  }
976
- // Don't overwrite a more specific type with a less specific one
977
- // This can happen when nested roots share entries with their parent roots
978
- const existingType = schema[newSchemaPath];
979
- if (existingType) {
980
- // Don't overwrite a primitive type with 'object' or 'array'
981
- // e.g., if schema has 'entities[].scenarioCount': 'number', don't overwrite with 'object'
982
- if (PRIMITIVE_TYPES.has(existingType) &&
983
- (postfixValue === 'object' || postfixValue === 'array')) {
1039
+ if (!merged)
1040
+ continue;
1041
+ const schema = equivalentRoot.schemaRootPath.startsWith('signature[')
1042
+ ? merged.signatureSchema
1043
+ : merged.returnValueSchema;
1044
+ // Skip if this schema has already grown past the cap
1045
+ if (Object.keys(schema).length > SCHEMA_KEY_CAP)
1046
+ continue;
1047
+ for (const [postfixPath, postfixValue] of Object.entries(esp.equivalentPostfixes)) {
1048
+ checkDeadline();
1049
+ let relevantPostfix = postfixPath;
1050
+ if (equivalentRoot.postfix) {
1051
+ // Check if postfixPath starts with equivalentRoot.postfix at a path boundary.
1052
+ // Must ensure exact path part match - "entityCode" should NOT match "entity" prefix.
1053
+ // Valid: "entity.foo" starts with "entity" (boundary at '.')
1054
+ // Valid: "entity[0]" starts with "entity" (boundary at '[')
1055
+ // Invalid: "entityCode" starts with "entity" (no boundary, different property)
1056
+ if (!postfixPath.startsWith(equivalentRoot.postfix)) {
1057
+ continue;
1058
+ }
1059
+ // Additional check: ensure the match is at a path boundary
1060
+ const nextChar = postfixPath[equivalentRoot.postfix.length];
1061
+ if (nextChar !== undefined &&
1062
+ nextChar !== '.' &&
1063
+ nextChar !== '[') {
1064
+ // The postfixPath continues with more characters that aren't a path separator.
1065
+ // This means "entity" matched "entityCode" which is wrong - they're different properties.
1066
+ continue;
1067
+ }
1068
+ const postFixPathParts = postfixPartsCache.get(postfixPath) ??
1069
+ splitOutsideParenthesesAndArrays(postfixPath);
1070
+ // Cache equivalentRoot.postfix parts — same root reused across all postfixes
1071
+ if (!postfixPartsCache.has(equivalentRoot.postfix)) {
1072
+ postfixPartsCache.set(equivalentRoot.postfix, splitOutsideParenthesesAndArrays(equivalentRoot.postfix));
1073
+ }
1074
+ const equivalentRootPostFixParts = postfixPartsCache.get(equivalentRoot.postfix);
1075
+ relevantPostfix = joinParenthesesAndArrays(postFixPathParts.slice(equivalentRootPostFixParts.length));
1076
+ }
1077
+ const newSchemaPath = joinParenthesesAndArrays([
1078
+ equivalentRoot.schemaRootPath,
1079
+ relevantPostfix,
1080
+ ]);
1081
+ // Skip paths that would go through a primitive type
1082
+ // e.g., if schema has 'entities[].scenarioCount': 'number', skip 'entities[].scenarioCount.sha'
1083
+ if (wouldGoThroughPrimitive(newSchemaPath, schema)) {
984
1084
  transformationTracer.operation(rootScopeName, {
985
- operation: 'skipTypeDowngrade',
1085
+ operation: 'skipPrimitivePath',
986
1086
  stage: 'merged',
987
1087
  path: newSchemaPath,
988
1088
  context: {
989
- reason: 'would overwrite primitive with object/array',
990
- existingType,
991
- newType: postfixValue,
1089
+ reason: 'would go through primitive type',
1090
+ postfixValue,
992
1091
  },
993
1092
  });
994
1093
  continue;
995
1094
  }
996
- // Don't overwrite a complex/union type with a primitive
997
- // e.g., if schema has 'scenarios[]': 'Scenario | null', don't overwrite with 'string'
998
- if (!PRIMITIVE_TYPES.has(existingType) &&
999
- PRIMITIVE_TYPES.has(postfixValue)) {
1000
- transformationTracer.operation(rootScopeName, {
1001
- operation: 'skipTypeDowngrade',
1002
- stage: 'merged',
1003
- path: newSchemaPath,
1004
- context: {
1005
- reason: 'would overwrite complex type with primitive',
1006
- existingType,
1007
- newType: postfixValue,
1008
- },
1009
- });
1095
+ // Skip setting primitive type when there are child postfixes that indicate structure.
1096
+ // This prevents downgrading an object/array element to a primitive type.
1097
+ // Uses pre-computed postfixesWithChildren Set for O(1) lookup instead of O(n) iteration.
1098
+ const hasChildPostfixes = (relevantPostfix === '' || relevantPostfix.endsWith('[]')) &&
1099
+ postfixesWithChildren.has(postfixPath);
1100
+ if (PRIMITIVE_TYPES.has(postfixValue) && hasChildPostfixes) {
1010
1101
  continue;
1011
1102
  }
1103
+ // Don't overwrite a more specific type with a less specific one
1104
+ // This can happen when nested roots share entries with their parent roots
1105
+ const existingType = schema[newSchemaPath];
1106
+ if (existingType) {
1107
+ // Don't overwrite a primitive type with 'object' or 'array'
1108
+ // e.g., if schema has 'entities[].scenarioCount': 'number', don't overwrite with 'object'
1109
+ if (PRIMITIVE_TYPES.has(existingType) &&
1110
+ (postfixValue === 'object' || postfixValue === 'array')) {
1111
+ transformationTracer.operation(rootScopeName, {
1112
+ operation: 'skipTypeDowngrade',
1113
+ stage: 'merged',
1114
+ path: newSchemaPath,
1115
+ context: {
1116
+ reason: 'would overwrite primitive with object/array',
1117
+ existingType,
1118
+ newType: postfixValue,
1119
+ },
1120
+ });
1121
+ continue;
1122
+ }
1123
+ // Don't overwrite a complex/union type with a primitive
1124
+ // e.g., if schema has 'scenarios[]': 'Scenario | null', don't overwrite with 'string'
1125
+ if (!PRIMITIVE_TYPES.has(existingType) &&
1126
+ PRIMITIVE_TYPES.has(postfixValue)) {
1127
+ transformationTracer.operation(rootScopeName, {
1128
+ operation: 'skipTypeDowngrade',
1129
+ stage: 'merged',
1130
+ path: newSchemaPath,
1131
+ context: {
1132
+ reason: 'would overwrite complex type with primitive',
1133
+ existingType,
1134
+ newType: postfixValue,
1135
+ },
1136
+ });
1137
+ continue;
1138
+ }
1139
+ }
1140
+ // Log the successful postfix merge
1141
+ transformationTracer.operation(rootScopeName, {
1142
+ operation: 'mergePostfix',
1143
+ stage: 'merged',
1144
+ path: newSchemaPath,
1145
+ before: existingType,
1146
+ after: postfixValue,
1147
+ context: {
1148
+ schemaRootPath: equivalentRoot.schemaRootPath,
1149
+ postfix: relevantPostfix,
1150
+ dependency: equivalentRoot.function?.name,
1151
+ },
1152
+ });
1153
+ schema[newSchemaPath] = postfixValue;
1012
1154
  }
1013
- // Log the successful postfix merge
1014
- transformationTracer.operation(rootScopeName, {
1015
- operation: 'mergePostfix',
1016
- stage: 'merged',
1017
- path: newSchemaPath,
1018
- before: existingType,
1019
- after: postfixValue,
1020
- context: {
1021
- schemaRootPath: equivalentRoot.schemaRootPath,
1022
- postfix: relevantPostfix,
1023
- dependency: equivalentRoot.function?.name,
1024
- },
1025
- });
1026
- schema[newSchemaPath] = postfixValue;
1155
+ schemasToClean.add(schema);
1027
1156
  }
1028
- schemasToClean.add(schema);
1029
1157
  }
1030
- }
1031
- // Batch-clean all modified schemas once (instead of once per root per ESP entry)
1032
- for (const schema of schemasToClean) {
1033
- cleanSchema(schema, { stage: 'afterMergePostfix' });
1034
- }
1035
- // Propagate equivalency-derived attributes to generic function call variants.
1036
- // When attributes are traced via equivalencies (e.g., fileComparisons from buildDataMap.signature[2]),
1037
- // they get written to non-generic paths (returnValue.data.x or funcName().functionCallReturnValue.data.x).
1038
- // If the ORIGINAL input schema has generic variants (funcName<T>().functionCallReturnValue.data),
1039
- // we need to copy the attributes to those paths too.
1040
- for (const filePath in mergedDataStructure.dependencySchemas) {
1041
- for (const depName in mergedDataStructure.dependencySchemas[filePath]) {
1042
- const depSchema = mergedDataStructure.dependencySchemas[filePath][depName];
1043
- const returnValueSchema = depSchema.returnValueSchema;
1044
- // Look at the ORIGINAL input dependencySchemas for generic variants,
1045
- // since the merged schema may have lost them during equivalency processing
1046
- const originalSchema = dependencySchemas?.[filePath]?.[depName];
1047
- const schemaToSearchForGenericVariants = originalSchema?.returnValueSchema || returnValueSchema;
1048
- // Find all unique generic variants of this function
1049
- // e.g., useFetcher<BranchEntityDiffResult>() from useFetcher<BranchEntityDiffResult>().functionCallReturnValue.data
1050
- const genericVariants = new Set();
1051
- const genericRegex = new RegExp(`^${depName.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')}<[^>]+>\\(\\)`);
1052
- for (const path in schemaToSearchForGenericVariants) {
1053
- const match = path.match(genericRegex);
1054
- if (match) {
1055
- genericVariants.add(match[0]);
1158
+ const postfixElapsed = Date.now() - mergeStartTime;
1159
+ // Batch-clean all modified schemas once (instead of once per root per ESP entry)
1160
+ for (const schema of schemasToClean) {
1161
+ cleanSchema(schema, { stage: 'afterMergePostfix' });
1162
+ }
1163
+ const cleanElapsed = Date.now() - mergeStartTime;
1164
+ // Propagate equivalency-derived attributes to generic function call variants.
1165
+ // When attributes are traced via equivalencies (e.g., fileComparisons from buildDataMap.signature[2]),
1166
+ // they get written to non-generic paths (returnValue.data.x or funcName().functionCallReturnValue.data.x).
1167
+ // If the ORIGINAL input schema has generic variants (funcName<T>().functionCallReturnValue.data),
1168
+ // we need to copy the attributes to those paths too.
1169
+ for (const filePath in mergedDataStructure.dependencySchemas) {
1170
+ for (const depName in mergedDataStructure.dependencySchemas[filePath]) {
1171
+ const depSchema = mergedDataStructure.dependencySchemas[filePath][depName];
1172
+ const returnValueSchema = depSchema.returnValueSchema;
1173
+ // Look at the ORIGINAL input dependencySchemas for generic variants,
1174
+ // since the merged schema may have lost them during equivalency processing
1175
+ const originalSchema = dependencySchemas?.[filePath]?.[depName];
1176
+ const schemaToSearchForGenericVariants = originalSchema?.returnValueSchema || returnValueSchema;
1177
+ // Find all unique generic variants of this function
1178
+ // e.g., useFetcher<BranchEntityDiffResult>() from useFetcher<BranchEntityDiffResult>().functionCallReturnValue.data
1179
+ const genericVariants = new Set();
1180
+ const genericRegex = new RegExp(`^${depName.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')}<[^>]+>\\(\\)`);
1181
+ for (const path in schemaToSearchForGenericVariants) {
1182
+ checkDeadline();
1183
+ const match = path.match(genericRegex);
1184
+ if (match) {
1185
+ genericVariants.add(match[0]);
1186
+ }
1056
1187
  }
1057
- }
1058
- if (genericVariants.size === 0)
1059
- continue;
1060
- // For each returnValue. path or non-generic function call path,
1061
- // create corresponding paths for each generic variant
1062
- const pathsToAdd = [];
1063
- for (const path in returnValueSchema) {
1064
- const value = returnValueSchema[path];
1065
- // Handle returnValue. paths
1066
- if (path.startsWith('returnValue.')) {
1067
- const suffix = path.slice('returnValue.'.length);
1068
- for (const genericVariant of genericVariants) {
1069
- const genericPath = `${genericVariant}.functionCallReturnValue.${suffix}`;
1070
- if (!(genericPath in returnValueSchema)) {
1071
- pathsToAdd.push([genericPath, value]);
1188
+ if (genericVariants.size === 0)
1189
+ continue;
1190
+ // For each returnValue. path or non-generic function call path,
1191
+ // create corresponding paths for each generic variant
1192
+ const pathsToAdd = [];
1193
+ for (const path in returnValueSchema) {
1194
+ checkDeadline();
1195
+ const value = returnValueSchema[path];
1196
+ // Handle returnValue. paths
1197
+ if (path.startsWith('returnValue.')) {
1198
+ const suffix = path.slice('returnValue.'.length);
1199
+ for (const genericVariant of genericVariants) {
1200
+ const genericPath = `${genericVariant}.functionCallReturnValue.${suffix}`;
1201
+ if (!(genericPath in returnValueSchema)) {
1202
+ pathsToAdd.push([genericPath, value]);
1203
+ }
1072
1204
  }
1073
1205
  }
1074
- }
1075
- // Handle non-generic function call paths like depName().functionCallReturnValue.x
1076
- else if (path.startsWith(`${depName}().functionCallReturnValue.`)) {
1077
- const suffix = path.slice(`${depName}().functionCallReturnValue.`.length);
1078
- for (const genericVariant of genericVariants) {
1079
- const genericPath = `${genericVariant}.functionCallReturnValue.${suffix}`;
1080
- if (!(genericPath in returnValueSchema)) {
1081
- pathsToAdd.push([genericPath, value]);
1206
+ // Handle non-generic function call paths like depName().functionCallReturnValue.x
1207
+ else if (path.startsWith(`${depName}().functionCallReturnValue.`)) {
1208
+ const suffix = path.slice(`${depName}().functionCallReturnValue.`.length);
1209
+ for (const genericVariant of genericVariants) {
1210
+ const genericPath = `${genericVariant}.functionCallReturnValue.${suffix}`;
1211
+ if (!(genericPath in returnValueSchema)) {
1212
+ pathsToAdd.push([genericPath, value]);
1213
+ }
1082
1214
  }
1083
1215
  }
1084
1216
  }
1217
+ // Add the new generic variant paths
1218
+ for (const [path, value] of pathsToAdd) {
1219
+ returnValueSchema[path] = value;
1220
+ }
1085
1221
  }
1086
- // Add the new generic variant paths
1087
- for (const [path, value] of pathsToAdd) {
1088
- returnValueSchema[path] = value;
1089
- }
1090
- }
1091
- }
1092
- // For mocked dependencies: copy paths from dependencySchemas (usage info) and normalize
1093
- // returnValue. paths that were created by equivalency processing.
1094
- // This ensures all paths use the consistent functionName().functionCallReturnValue. format.
1095
- for (const dependency of importedExports) {
1096
- if (!dependency.isMocked)
1097
- continue;
1098
- const srcSchema = dependencySchemas?.[dependency.filePath]?.[dependency.name];
1099
- if (!srcSchema?.returnValueSchema)
1100
- continue;
1101
- const depSchema = findOrCreateDependentSchemas({
1102
- filePath: dependency.filePath,
1103
- name: dependency.name,
1104
- });
1105
- // First, normalize any returnValue paths that were written by equivalency processing
1106
- // to the standard functionName().functionCallReturnValue format.
1107
- // This includes both returnValue. (dot) and returnValue[ (array) paths.
1108
- const pathsToNormalize = [];
1109
- for (const path in depSchema.returnValueSchema) {
1110
- if (path === 'returnValue' ||
1111
- path.startsWith('returnValue.') ||
1112
- path.startsWith('returnValue[')) {
1113
- pathsToNormalize.push([path, depSchema.returnValueSchema[path]]);
1114
- }
1115
- }
1116
- for (const [path, value] of pathsToNormalize) {
1117
- delete depSchema.returnValueSchema[path];
1118
- let normalizedPath;
1119
- if (path === 'returnValue') {
1120
- normalizedPath = `${dependency.name}().functionCallReturnValue`;
1121
- }
1122
- else if (path.startsWith('returnValue.')) {
1123
- normalizedPath = path.replace(/^returnValue\./, `${dependency.name}().functionCallReturnValue.`);
1124
- }
1125
- else {
1126
- // path.startsWith('returnValue[')
1127
- // e.g., returnValue[] -> getOptions().functionCallReturnValue[]
1128
- // e.g., returnValue[].label -> getOptions().functionCallReturnValue[].label
1129
- normalizedPath = path.replace(/^returnValue/, `${dependency.name}().functionCallReturnValue`);
1130
- }
1131
- transformationTracer.operation(rootScopeName, {
1132
- operation: 'normalizeReturnValuePath',
1133
- stage: 'merged',
1134
- path: normalizedPath,
1135
- before: path,
1136
- after: normalizedPath,
1137
- context: { dependency: dependency.name, value },
1138
- });
1139
- depSchema.returnValueSchema[normalizedPath] = value;
1140
1222
  }
1141
- // Now copy paths from the source schema (dependencySchemas)
1142
- for (const path in srcSchema.returnValueSchema) {
1143
- const value = srcSchema.returnValueSchema[path];
1144
- // Normalize paths starting with 'returnValue' to use the standard format:
1145
- // 'returnValue.foo' -> 'dependencyName().functionCallReturnValue.foo'
1146
- // This ensures consistency across the codebase and allows constructMockCode
1147
- // and gatherDataForMocks to work correctly.
1148
- if (path === 'returnValue' || path.startsWith('returnValue.')) {
1149
- // Convert 'returnValue' -> 'name().functionCallReturnValue'
1150
- // Convert 'returnValue.foo' -> 'name().functionCallReturnValue.foo'
1151
- const normalizedPath = path === 'returnValue'
1152
- ? `${dependency.name}().functionCallReturnValue`
1153
- : path.replace(/^returnValue\./, `${dependency.name}().functionCallReturnValue.`);
1154
- // Always write srcSchema values - they take precedence over equivalency-derived values
1155
- depSchema.returnValueSchema[normalizedPath] = value;
1223
+ // For mocked dependencies: copy paths from dependencySchemas (usage info) and normalize
1224
+ // returnValue. paths that were created by equivalency processing.
1225
+ // This ensures all paths use the consistent functionName().functionCallReturnValue. format.
1226
+ for (const dependency of importedExports) {
1227
+ if (!dependency.isMocked)
1156
1228
  continue;
1157
- }
1158
- // Copy paths containing functionCallReturnValue (return value structures)
1159
- // These are needed for constructMockCode to build the proper mock data hierarchy
1160
- // Example: supabase.auth.getSession().functionCallReturnValue.data.session
1161
- if (path.includes('.functionCallReturnValue')) {
1162
- // Always write srcSchema values - they take precedence over equivalency-derived values
1163
- depSchema.returnValueSchema[path] = value;
1229
+ const srcSchema = dependencySchemas?.[dependency.filePath]?.[dependency.name];
1230
+ if (!srcSchema?.returnValueSchema)
1164
1231
  continue;
1232
+ const depSchema = findOrCreateDependentSchemas({
1233
+ filePath: dependency.filePath,
1234
+ name: dependency.name,
1235
+ });
1236
+ // First, normalize any returnValue paths that were written by equivalency processing
1237
+ // to the standard functionName().functionCallReturnValue format.
1238
+ // This includes both returnValue. (dot) and returnValue[ (array) paths.
1239
+ const pathsToNormalize = [];
1240
+ for (const path in depSchema.returnValueSchema) {
1241
+ checkDeadline();
1242
+ if (path === 'returnValue' ||
1243
+ path.startsWith('returnValue.') ||
1244
+ path.startsWith('returnValue[')) {
1245
+ pathsToNormalize.push([path, depSchema.returnValueSchema[path]]);
1246
+ }
1165
1247
  }
1166
- // Copy function-typed paths that end with () (are function calls)
1167
- // These include:
1168
- // - Function stubs without functionCallReturnValue (like onAuthStateChange)
1169
- // - Function markers with async-function type (like getSession(): async-function)
1170
- // which are needed for constructMockCode to know to generate async functions
1171
- // Skip paths starting with 'returnValue' - they were already handled above
1172
- if (['function', 'async-function'].includes(value) &&
1173
- path.endsWith(')') &&
1174
- !path.startsWith('returnValue')) {
1175
- if (!(path in depSchema.returnValueSchema)) {
1176
- depSchema.returnValueSchema[path] = value;
1248
+ for (const [path, value] of pathsToNormalize) {
1249
+ delete depSchema.returnValueSchema[path];
1250
+ let normalizedPath;
1251
+ if (path === 'returnValue') {
1252
+ normalizedPath = `${dependency.name}().functionCallReturnValue`;
1253
+ }
1254
+ else if (path.startsWith('returnValue.')) {
1255
+ normalizedPath = path.replace(/^returnValue\./, `${dependency.name}().functionCallReturnValue.`);
1177
1256
  }
1257
+ else {
1258
+ // path.startsWith('returnValue[')
1259
+ // e.g., returnValue[] -> getOptions().functionCallReturnValue[]
1260
+ // e.g., returnValue[].label -> getOptions().functionCallReturnValue[].label
1261
+ normalizedPath = path.replace(/^returnValue/, `${dependency.name}().functionCallReturnValue`);
1262
+ }
1263
+ transformationTracer.operation(rootScopeName, {
1264
+ operation: 'normalizeReturnValuePath',
1265
+ stage: 'merged',
1266
+ path: normalizedPath,
1267
+ before: path,
1268
+ after: normalizedPath,
1269
+ context: { dependency: dependency.name, value },
1270
+ });
1271
+ depSchema.returnValueSchema[normalizedPath] = value;
1178
1272
  }
1179
- // Copy object-typed paths for chained API access patterns (like trpc.customer.getCustomersByOrg)
1180
- // These intermediate paths are needed for constructMockCode to build the nested mock structure.
1181
- // Example: for trpc.customer.getCustomersByOrg.useQuery().functionCallReturnValue.data,
1182
- // we need 'trpc', 'trpc.customer', 'trpc.customer.getCustomersByOrg' all typed as 'object'.
1183
- // Skip paths starting with 'returnValue' - they were already handled above
1184
- //
1185
- // EXCEPTION: For function-style dependencies like getSupabase(), skip intermediate object
1186
- // paths like 'getSupabase().auth' that are just property access after a function call.
1187
- // These aren't needed because constructMockCode can infer the structure from the actual
1188
- // function call paths like 'getSupabase().auth.getUser()'. We only need object paths
1189
- // for object-style dependencies like 'supabase.auth' where the dependency itself is an object.
1190
- if (value === 'object' && !path.startsWith('returnValue')) {
1191
- // Check if this is a function-style dependency (path starts with name() or name<T>())
1192
- const isFunctionStyleDependency = path.startsWith(`${dependency.name}()`) ||
1193
- path.match(new RegExp(`^${dependency.name}<[^>]+>\\(\\)`));
1194
- // For function-style dependencies, skip intermediate object paths
1195
- // Only keep object paths that are within functionCallReturnValue
1196
- if (isFunctionStyleDependency &&
1197
- !path.includes('.functionCallReturnValue')) {
1273
+ // Now copy paths from the source schema (dependencySchemas)
1274
+ for (const path in srcSchema.returnValueSchema) {
1275
+ checkDeadline();
1276
+ const value = srcSchema.returnValueSchema[path];
1277
+ // Normalize paths starting with 'returnValue' to use the standard format:
1278
+ // 'returnValue.foo' -> 'dependencyName().functionCallReturnValue.foo'
1279
+ // This ensures consistency across the codebase and allows constructMockCode
1280
+ // and gatherDataForMocks to work correctly.
1281
+ if (path === 'returnValue' || path.startsWith('returnValue.')) {
1282
+ // Convert 'returnValue' -> 'name().functionCallReturnValue'
1283
+ // Convert 'returnValue.foo' -> 'name().functionCallReturnValue.foo'
1284
+ const normalizedPath = path === 'returnValue'
1285
+ ? `${dependency.name}().functionCallReturnValue`
1286
+ : path.replace(/^returnValue\./, `${dependency.name}().functionCallReturnValue.`);
1287
+ // Always write srcSchema values - they take precedence over equivalency-derived values
1288
+ depSchema.returnValueSchema[normalizedPath] = value;
1198
1289
  continue;
1199
1290
  }
1200
- if (!(path in depSchema.returnValueSchema)) {
1291
+ // Copy paths containing functionCallReturnValue (return value structures)
1292
+ // These are needed for constructMockCode to build the proper mock data hierarchy
1293
+ // Example: supabase.auth.getSession().functionCallReturnValue.data.session
1294
+ if (path.includes('.functionCallReturnValue')) {
1295
+ // Always write srcSchema values - they take precedence over equivalency-derived values
1201
1296
  depSchema.returnValueSchema[path] = value;
1202
- }
1203
- }
1204
- }
1205
- cleanSchema(depSchema.returnValueSchema, {
1206
- stage: 'afterMockedDependencyMerge',
1207
- dependency: dependency.name,
1208
- });
1209
- // Pull signature requirements from downstream functions into the mocked return value.
1210
- // When a mocked function's return flows into another function's signature (via usageEquivalencies),
1211
- // we need to include that function's signature requirements in the mock.
1212
- //
1213
- // Example: fromE5() returns a currency object that flows to calculateTotalPrice(price, quantity).
1214
- // calculateTotalPrice's signatureSchema shows signature[0].multiply() is required.
1215
- // We need to add multiply() to fromE5's mock return value.
1216
- const usageEquivalencies = srcSchema.usageEquivalencies ?? {};
1217
- for (const [returnPath, equivalencies] of Object.entries(usageEquivalencies)) {
1218
- // Only process return value paths (functionCallReturnValue)
1219
- if (!returnPath.includes('.functionCallReturnValue'))
1220
- continue;
1221
- for (const equiv of equivalencies) {
1222
- // Check if this equivalency points to a signature path
1223
- const signatureMatch = equiv.schemaPath.match(/\.signature\[(\d+)\]$/);
1224
- if (!signatureMatch)
1225
1297
  continue;
1226
- const targetFunctionName = cleanFunctionName(equiv.scopeNodeName);
1227
- const signatureIndex = signatureMatch[1];
1228
- // Look up the target function's analysis to get its signature requirements
1229
- // First try dependentAnalyses, then dependencySchemas
1230
- let targetSignatureSchema;
1231
- // Check dependentAnalyses first (has the full merged analysis)
1232
- for (const depFilePath in dependentAnalyses) {
1233
- const analysis = dependentAnalyses[depFilePath]?.[targetFunctionName];
1234
- if (analysis?.metadata?.mergedDataStructure?.signatureSchema) {
1235
- targetSignatureSchema =
1236
- analysis.metadata.mergedDataStructure.signatureSchema;
1237
- break;
1238
- }
1239
1298
  }
1240
- // Fallback to dependencySchemas if not found
1241
- if (!targetSignatureSchema) {
1242
- for (const depFilePath in dependencySchemas) {
1243
- const schema = dependencySchemas[depFilePath]?.[targetFunctionName];
1244
- if (schema?.signatureSchema) {
1245
- targetSignatureSchema = schema.signatureSchema;
1246
- break;
1247
- }
1299
+ // Copy function-typed paths that end with () (are function calls)
1300
+ // These include:
1301
+ // - Function stubs without functionCallReturnValue (like onAuthStateChange)
1302
+ // - Function markers with async-function type (like getSession(): async-function)
1303
+ // which are needed for constructMockCode to know to generate async functions
1304
+ // Skip paths starting with 'returnValue' - they were already handled above
1305
+ if (['function', 'async-function'].includes(value) &&
1306
+ path.endsWith(')') &&
1307
+ !path.startsWith('returnValue')) {
1308
+ if (!(path in depSchema.returnValueSchema)) {
1309
+ depSchema.returnValueSchema[path] = value;
1248
1310
  }
1249
1311
  }
1250
- if (!targetSignatureSchema)
1251
- continue;
1252
- // Find all paths in the target's signatureSchema that extend from signature[N]
1253
- // e.g., signature[0].multiply(quantity) -> .multiply(quantity)
1254
- const signaturePrefix = `signature[${signatureIndex}]`;
1255
- for (const [sigPath, sigType] of Object.entries(targetSignatureSchema)) {
1256
- if (!sigPath.startsWith(signaturePrefix))
1257
- continue;
1258
- // Skip the base signature[N] path itself - we only want the method/property extensions
1259
- if (sigPath === signaturePrefix)
1312
+ // Copy object-typed paths for chained API access patterns (like trpc.customer.getCustomersByOrg)
1313
+ // These intermediate paths are needed for constructMockCode to build the nested mock structure.
1314
+ // Example: for trpc.customer.getCustomersByOrg.useQuery().functionCallReturnValue.data,
1315
+ // we need 'trpc', 'trpc.customer', 'trpc.customer.getCustomersByOrg' all typed as 'object'.
1316
+ // Skip paths starting with 'returnValue' - they were already handled above
1317
+ //
1318
+ // EXCEPTION: For function-style dependencies like getSupabase(), skip intermediate object
1319
+ // paths like 'getSupabase().auth' that are just property access after a function call.
1320
+ // These aren't needed because constructMockCode can infer the structure from the actual
1321
+ // function call paths like 'getSupabase().auth.getUser()'. We only need object paths
1322
+ // for object-style dependencies like 'supabase.auth' where the dependency itself is an object.
1323
+ if (value === 'object' && !path.startsWith('returnValue')) {
1324
+ // Check if this is a function-style dependency (path starts with name() or name<T>())
1325
+ const isFunctionStyleDependency = path.startsWith(`${dependency.name}()`) ||
1326
+ path.match(new RegExp(`^${dependency.name}<[^>]+>\\(\\)`));
1327
+ // For function-style dependencies, skip intermediate object paths
1328
+ // Only keep object paths that are within functionCallReturnValue
1329
+ if (isFunctionStyleDependency &&
1330
+ !path.includes('.functionCallReturnValue')) {
1260
1331
  continue;
1261
- // Extract the suffix after signature[N] (e.g., ".multiply(quantity)")
1262
- const suffix = sigPath.slice(signaturePrefix.length);
1263
- // Build the path for the mocked return value
1264
- // e.g., fromE5(priceE5).functionCallReturnValue.multiply(quantity)
1265
- const returnValuePath = returnPath + suffix;
1266
- // Add to the mocked dependency's return value schema if not already present
1267
- if (!(returnValuePath in depSchema.returnValueSchema)) {
1268
- depSchema.returnValueSchema[returnValuePath] = sigType;
1332
+ }
1333
+ if (!(path in depSchema.returnValueSchema)) {
1334
+ depSchema.returnValueSchema[path] = value;
1269
1335
  }
1270
1336
  }
1271
1337
  }
1272
- }
1273
- cleanSchema(depSchema.returnValueSchema, {
1274
- stage: 'afterSignatureRequirementsMerge',
1275
- dependency: dependency.name,
1276
- });
1277
- }
1278
- // Process the input dependencySchemas FIRST (before child dependentAnalyses).
1279
- // This ensures the parent entity's direct usage of dependencies takes precedence.
1280
- // When both parent and child use the same dependency (e.g., useLoaderData),
1281
- // the parent's schema paths are preserved, and child's paths are merged in later.
1282
- //
1283
- // Some dependencies (like .d.ts type declaration files) may not have:
1284
- // - Equivalencies with the root scope
1285
- // - A dependent analysis (they're just type declarations)
1286
- // - Be marked as mocked
1287
- // Without this, their schemas would be lost entirely.
1288
- for (const filePath in dependencySchemas) {
1289
- for (const name in dependencySchemas[filePath]) {
1290
- const srcSchema = dependencySchemas[filePath][name];
1291
- if (!srcSchema)
1292
- continue;
1293
- // Skip mocked dependencies - they were already processed above with path normalization
1294
- if (mockedDependencies.has(`${filePath}::${name}`)) {
1295
- continue;
1296
- }
1297
- // Check if this dependency was already processed by equivalencies
1298
- const existingSchema = mergedDataStructure.dependencySchemas[filePath]?.[name];
1299
- // Only add if no existing schema (equivalencies didn't process it)
1300
- if (!existingSchema) {
1301
- const depSchema = findOrCreateDependentSchemas({ filePath, name });
1302
- for (const path in srcSchema.returnValueSchema) {
1303
- depSchema.returnValueSchema[path] = srcSchema.returnValueSchema[path];
1304
- }
1305
- for (const path in srcSchema.signatureSchema) {
1306
- depSchema.signatureSchema[path] = srcSchema.signatureSchema[path];
1307
- }
1308
- // Clean known object functions (like String.prototype.replace, Array.prototype.map)
1309
- // from the copied schema. Without this, method call paths on primitives like
1310
- // "projectSlug.replace(...)" would cause convertDotNotation to create nested
1311
- // object structures instead of preserving the primitive type.
1312
- cleanSchema(depSchema.returnValueSchema, {
1313
- stage: 'afterDependencySchemaCopy',
1314
- filePath,
1315
- dependency: name,
1316
- });
1317
- }
1318
- // TYPE REFINEMENT: Check if dependentAnalyses has a more specific type for this dependency.
1319
- // When a parent passes `entity.filePath` (string | undefined) to a child component
1320
- // that requires `filePath: string`, we should use the child's more specific type.
1321
- // This prevents mock data from having undefined values for required props.
1338
+ cleanSchema(depSchema.returnValueSchema, {
1339
+ stage: 'afterMockedDependencyMerge',
1340
+ dependency: dependency.name,
1341
+ });
1342
+ // Pull signature requirements from downstream functions into the mocked return value.
1343
+ // When a mocked function's return flows into another function's signature (via usageEquivalencies),
1344
+ // we need to include that function's signature requirements in the mock.
1322
1345
  //
1323
- // This runs REGARDLESS of whether equivalencies already processed the schema,
1324
- // because equivalencies copy the parent's type (string | undefined), not the child's
1325
- // required type (string).
1326
- const depSchema = findOrCreateDependentSchemas({ filePath, name });
1327
- const childAnalysis = dependentAnalyses[filePath]?.[name];
1328
- const childSignatureSchema = childAnalysis?.metadata?.mergedDataStructure?.signatureSchema;
1329
- if (childSignatureSchema) {
1330
- for (const path in depSchema.signatureSchema) {
1331
- const parentType = depSchema.signatureSchema[path];
1332
- const childType = childSignatureSchema[path];
1333
- if (parentType && childType) {
1334
- // Check if parent has optional type and child has required type
1335
- const parentIsOptional = parentType.includes('| undefined') ||
1336
- parentType.includes('| null');
1337
- const childIsOptional = childType.includes('| undefined') || childType.includes('| null');
1338
- // If child requires a more specific type (not optional), use it
1339
- if (parentIsOptional && !childIsOptional) {
1340
- depSchema.signatureSchema[path] = childType;
1346
+ // Example: fromE5() returns a currency object that flows to calculateTotalPrice(price, quantity).
1347
+ // calculateTotalPrice's signatureSchema shows signature[0].multiply() is required.
1348
+ // We need to add multiply() to fromE5's mock return value.
1349
+ const usageEquivalencies = srcSchema.usageEquivalencies ?? {};
1350
+ for (const [returnPath, equivalencies] of Object.entries(usageEquivalencies)) {
1351
+ // Only process return value paths (functionCallReturnValue)
1352
+ if (!returnPath.includes('.functionCallReturnValue'))
1353
+ continue;
1354
+ for (const equiv of equivalencies) {
1355
+ // Check if this equivalency points to a signature path
1356
+ const signatureMatch = equiv.schemaPath.match(/\.signature\[(\d+)\]$/);
1357
+ if (!signatureMatch)
1358
+ continue;
1359
+ const targetFunctionName = cleanFunctionName(equiv.scopeNodeName);
1360
+ const signatureIndex = signatureMatch[1];
1361
+ // Look up the target function's analysis to get its signature requirements
1362
+ // First try dependentAnalyses, then dependencySchemas
1363
+ let targetSignatureSchema;
1364
+ // Check dependentAnalyses first (has the full merged analysis)
1365
+ for (const depFilePath in dependentAnalyses) {
1366
+ const analysis = dependentAnalyses[depFilePath]?.[targetFunctionName];
1367
+ if (analysis?.metadata?.mergedDataStructure?.signatureSchema) {
1368
+ targetSignatureSchema =
1369
+ analysis.metadata.mergedDataStructure.signatureSchema;
1370
+ break;
1371
+ }
1372
+ }
1373
+ // Fallback to dependencySchemas if not found
1374
+ if (!targetSignatureSchema) {
1375
+ for (const depFilePath in dependencySchemas) {
1376
+ const schema = dependencySchemas[depFilePath]?.[targetFunctionName];
1377
+ if (schema?.signatureSchema) {
1378
+ targetSignatureSchema = schema.signatureSchema;
1379
+ break;
1380
+ }
1381
+ }
1382
+ }
1383
+ if (!targetSignatureSchema)
1384
+ continue;
1385
+ // Find all paths in the target's signatureSchema that extend from signature[N]
1386
+ // e.g., signature[0].multiply(quantity) -> .multiply(quantity)
1387
+ const signaturePrefix = `signature[${signatureIndex}]`;
1388
+ for (const [sigPath, sigType] of Object.entries(targetSignatureSchema)) {
1389
+ if (!sigPath.startsWith(signaturePrefix))
1390
+ continue;
1391
+ // Skip the base signature[N] path itself - we only want the method/property extensions
1392
+ if (sigPath === signaturePrefix)
1393
+ continue;
1394
+ // Extract the suffix after signature[N] (e.g., ".multiply(quantity)")
1395
+ const suffix = sigPath.slice(signaturePrefix.length);
1396
+ // Build the path for the mocked return value
1397
+ // e.g., fromE5(priceE5).functionCallReturnValue.multiply(quantity)
1398
+ const returnValuePath = returnPath + suffix;
1399
+ // Add to the mocked dependency's return value schema if not already present
1400
+ if (!(returnValuePath in depSchema.returnValueSchema)) {
1401
+ depSchema.returnValueSchema[returnValuePath] = sigType;
1341
1402
  }
1342
1403
  }
1343
1404
  }
1344
1405
  }
1345
- // For functions with multiple different type parameters, also create separate entries
1346
- // for each type-parameterized variant. This allows gatherDataForMocks to look up
1347
- // the specific schema for each call signature.
1348
- // This runs regardless of whether the base entry already existed, since we need
1349
- // the separate variant entries for proper schema lookup.
1350
- const baseName = cleanFunctionName(name);
1351
- if (functionsWithMultipleTypeParams.has(baseName)) {
1352
- // Find all unique type-parameterized call signatures in the schema
1353
- const typeParamVariants = new Set();
1354
- for (const path of Object.keys(srcSchema.returnValueSchema)) {
1355
- const parts = splitOutsideParenthesesAndArrays(path);
1356
- if (parts.length > 0 &&
1357
- parts[0].includes('<') &&
1358
- parts[0].endsWith(')')) {
1359
- typeParamVariants.add(parts[0]);
1360
- }
1406
+ cleanSchema(depSchema.returnValueSchema, {
1407
+ stage: 'afterSignatureRequirementsMerge',
1408
+ dependency: dependency.name,
1409
+ });
1410
+ }
1411
+ // Process the input dependencySchemas FIRST (before child dependentAnalyses).
1412
+ // This ensures the parent entity's direct usage of dependencies takes precedence.
1413
+ // When both parent and child use the same dependency (e.g., useLoaderData),
1414
+ // the parent's schema paths are preserved, and child's paths are merged in later.
1415
+ //
1416
+ // Some dependencies (like .d.ts type declaration files) may not have:
1417
+ // - Equivalencies with the root scope
1418
+ // - A dependent analysis (they're just type declarations)
1419
+ // - Be marked as mocked
1420
+ // Without this, their schemas would be lost entirely.
1421
+ for (const filePath in dependencySchemas) {
1422
+ for (const name in dependencySchemas[filePath]) {
1423
+ const srcSchema = dependencySchemas[filePath][name];
1424
+ if (!srcSchema)
1425
+ continue;
1426
+ // Skip mocked dependencies - they were already processed above with path normalization
1427
+ if (mockedDependencies.has(`${filePath}::${name}`)) {
1428
+ continue;
1361
1429
  }
1362
- // Create a separate entry for each type-parameterized variant
1363
- for (const variant of typeParamVariants) {
1364
- const variantSchema = findOrCreateDependentSchemas({
1365
- filePath,
1366
- name: variant,
1367
- });
1368
- // Copy only paths that belong to this variant
1430
+ // Check if this dependency was already processed by equivalencies
1431
+ const existingSchema = mergedDataStructure.dependencySchemas[filePath]?.[name];
1432
+ // Only add if no existing schema (equivalencies didn't process it)
1433
+ if (!existingSchema) {
1434
+ const depSchema = findOrCreateDependentSchemas({ filePath, name });
1369
1435
  for (const path in srcSchema.returnValueSchema) {
1370
- if (path.startsWith(variant)) {
1371
- variantSchema.returnValueSchema[path] =
1372
- srcSchema.returnValueSchema[path];
1373
- }
1436
+ checkDeadline();
1437
+ depSchema.returnValueSchema[path] =
1438
+ srcSchema.returnValueSchema[path];
1374
1439
  }
1375
- cleanSchema(variantSchema.returnValueSchema, {
1376
- stage: 'afterTypeVariantCopy',
1440
+ for (const path in srcSchema.signatureSchema) {
1441
+ checkDeadline();
1442
+ depSchema.signatureSchema[path] = srcSchema.signatureSchema[path];
1443
+ }
1444
+ // Clean known object functions (like String.prototype.replace, Array.prototype.map)
1445
+ // from the copied schema. Without this, method call paths on primitives like
1446
+ // "projectSlug.replace(...)" would cause convertDotNotation to create nested
1447
+ // object structures instead of preserving the primitive type.
1448
+ cleanSchema(depSchema.returnValueSchema, {
1449
+ stage: 'afterDependencySchemaCopy',
1377
1450
  filePath,
1378
1451
  dependency: name,
1379
- variant,
1380
1452
  });
1381
1453
  }
1382
- }
1383
- }
1384
- }
1385
- // Ensure ALL dependencies from dependentAnalyses are included in dependencySchemas,
1386
- // even if they have no equivalencies with the root scope.
1387
- // This preserves nested functionCallReturnValue paths that would otherwise be lost.
1388
- // EXCEPT: Skip mocked dependencies - we don't want their internal implementation details.
1389
- for (const filePath in dependentAnalyses) {
1390
- for (const name in dependentAnalyses[filePath]) {
1391
- const dependentMergedDataStructure = dependentAnalyses[filePath][name].metadata?.mergedDataStructure;
1392
- if (!dependentMergedDataStructure)
1393
- continue;
1394
- const isMocked = mockedDependencies.has(`${filePath}::${name}`);
1395
- // For mocked dependencies: ONLY copy nested dependencySchemas (skip internal implementation)
1396
- // For non-mocked dependencies: copy everything (signature, returnValue, and nested dependencySchemas)
1397
- if (!isMocked) {
1398
- // Create the dependency schema entry if it doesn't exist
1454
+ // TYPE REFINEMENT: Check if dependentAnalyses has a more specific type for this dependency.
1455
+ // When a parent passes `entity.filePath` (string | undefined) to a child component
1456
+ // that requires `filePath: string`, we should use the child's more specific type.
1457
+ // This prevents mock data from having undefined values for required props.
1458
+ //
1459
+ // This runs REGARDLESS of whether equivalencies already processed the schema,
1460
+ // because equivalencies copy the parent's type (string | undefined), not the child's
1461
+ // required type (string).
1399
1462
  const depSchema = findOrCreateDependentSchemas({ filePath, name });
1400
- // Copy over all paths from the dependent's returnValueSchema
1401
- // Only add paths that don't already exist (don't overwrite values set by equivalencies)
1402
- for (const path in dependentMergedDataStructure.returnValueSchema) {
1403
- const translatedPath = translatePath(path, name);
1404
- if (!(translatedPath in depSchema.returnValueSchema)) {
1405
- depSchema.returnValueSchema[translatedPath] =
1406
- dependentMergedDataStructure.returnValueSchema[path];
1463
+ const childAnalysis = dependentAnalyses[filePath]?.[name];
1464
+ const childSignatureSchema = childAnalysis?.metadata?.mergedDataStructure?.signatureSchema;
1465
+ if (childSignatureSchema) {
1466
+ for (const path in depSchema.signatureSchema) {
1467
+ checkDeadline();
1468
+ const parentType = depSchema.signatureSchema[path];
1469
+ const childType = childSignatureSchema[path];
1470
+ if (parentType && childType) {
1471
+ // Check if parent has optional type and child has required type
1472
+ const parentIsOptional = parentType.includes('| undefined') ||
1473
+ parentType.includes('| null');
1474
+ const childIsOptional = childType.includes('| undefined') ||
1475
+ childType.includes('| null');
1476
+ // If child requires a more specific type (not optional), use it
1477
+ if (parentIsOptional && !childIsOptional) {
1478
+ depSchema.signatureSchema[path] = childType;
1479
+ }
1480
+ }
1407
1481
  }
1408
1482
  }
1409
- // Copy over signature schema as well
1410
- for (const path in dependentMergedDataStructure.signatureSchema) {
1411
- const translatedPath = translatePath(path, name);
1412
- if (!(translatedPath in depSchema.signatureSchema)) {
1413
- depSchema.signatureSchema[translatedPath] =
1414
- dependentMergedDataStructure.signatureSchema[path];
1483
+ // For functions with multiple different type parameters, also create separate entries
1484
+ // for each type-parameterized variant. This allows gatherDataForMocks to look up
1485
+ // the specific schema for each call signature.
1486
+ // This runs regardless of whether the base entry already existed, since we need
1487
+ // the separate variant entries for proper schema lookup.
1488
+ const baseName = cleanFunctionName(name);
1489
+ if (functionsWithMultipleTypeParams.has(baseName)) {
1490
+ // Find all unique type-parameterized call signatures in the schema
1491
+ const typeParamVariants = new Set();
1492
+ for (const path of Object.keys(srcSchema.returnValueSchema)) {
1493
+ const parts = splitOutsideParenthesesAndArrays(path);
1494
+ if (parts.length > 0 &&
1495
+ parts[0].includes('<') &&
1496
+ parts[0].endsWith(')')) {
1497
+ typeParamVariants.add(parts[0]);
1498
+ }
1415
1499
  }
1416
- }
1417
- }
1418
- // Copy nested dependencySchemas for ALL entities (including mocked ones)
1419
- // This represents what dependencies THIS entity uses, not its internal implementation
1420
- if (dependentMergedDataStructure.dependencySchemas) {
1421
- for (const depFilePath in dependentMergedDataStructure.dependencySchemas) {
1422
- for (const depName in dependentMergedDataStructure.dependencySchemas[depFilePath]) {
1423
- const nestedDepSchema = dependentMergedDataStructure.dependencySchemas[depFilePath][depName];
1424
- const targetDepSchema = findOrCreateDependentSchemas({
1425
- filePath: depFilePath,
1426
- name: depName,
1500
+ // Create a separate entry for each type-parameterized variant
1501
+ for (const variant of typeParamVariants) {
1502
+ const variantSchema = findOrCreateDependentSchemas({
1503
+ filePath,
1504
+ name: variant,
1427
1505
  });
1428
- // Merge in the nested dependency schemas
1429
- for (const path in nestedDepSchema.returnValueSchema) {
1430
- if (!(path in targetDepSchema.returnValueSchema)) {
1431
- const value = nestedDepSchema.returnValueSchema[path];
1432
- targetDepSchema.returnValueSchema[path] = value;
1506
+ // Copy only paths that belong to this variant
1507
+ for (const path in srcSchema.returnValueSchema) {
1508
+ checkDeadline();
1509
+ if (path.startsWith(variant)) {
1510
+ variantSchema.returnValueSchema[path] =
1511
+ srcSchema.returnValueSchema[path];
1433
1512
  }
1434
1513
  }
1435
- for (const path in nestedDepSchema.signatureSchema) {
1436
- if (!(path in targetDepSchema.signatureSchema)) {
1437
- targetDepSchema.signatureSchema[path] =
1438
- nestedDepSchema.signatureSchema[path];
1514
+ cleanSchema(variantSchema.returnValueSchema, {
1515
+ stage: 'afterTypeVariantCopy',
1516
+ filePath,
1517
+ dependency: name,
1518
+ variant,
1519
+ });
1520
+ }
1521
+ }
1522
+ }
1523
+ }
1524
+ // Ensure ALL dependencies from dependentAnalyses are included in dependencySchemas,
1525
+ // even if they have no equivalencies with the root scope.
1526
+ // This preserves nested functionCallReturnValue paths that would otherwise be lost.
1527
+ // EXCEPT: Skip mocked dependencies - we don't want their internal implementation details.
1528
+ for (const filePath in dependentAnalyses) {
1529
+ for (const name in dependentAnalyses[filePath]) {
1530
+ checkDeadline();
1531
+ const dependentMergedDataStructure = dependentAnalyses[filePath][name].metadata?.mergedDataStructure;
1532
+ if (!dependentMergedDataStructure)
1533
+ continue;
1534
+ const isMocked = mockedDependencies.has(`${filePath}::${name}`);
1535
+ // For mocked dependencies: ONLY copy nested dependencySchemas (skip internal implementation)
1536
+ // For non-mocked dependencies: copy everything (signature, returnValue, and nested dependencySchemas)
1537
+ if (!isMocked) {
1538
+ // Create the dependency schema entry if it doesn't exist
1539
+ const depSchema = findOrCreateDependentSchemas({ filePath, name });
1540
+ // Copy over paths from the dependent's returnValueSchema.
1541
+ // Only add paths that don't already exist (don't overwrite values set by equivalencies).
1542
+ // Skip if either source or target exceeds the cap — copying 2,531 paths from
1543
+ // ArticleTable with translatePath on each takes ~1.5s for one entity.
1544
+ const srcRetSize = Object.keys(dependentMergedDataStructure.returnValueSchema || {}).length;
1545
+ if (srcRetSize > SCHEMA_KEY_CAP ||
1546
+ Object.keys(depSchema.returnValueSchema).length > SCHEMA_KEY_CAP)
1547
+ continue;
1548
+ for (const path in dependentMergedDataStructure.returnValueSchema) {
1549
+ // Fast path: only call translatePath when the path starts with the
1550
+ // dependency name (e.g., "ArticleTable().functionCallReturnValue.x").
1551
+ // Most paths start with "returnValue" or "signature" and don't need translation.
1552
+ const translatedPath = path.startsWith(name)
1553
+ ? translatePath(path, name)
1554
+ : path;
1555
+ if (!(translatedPath in depSchema.returnValueSchema)) {
1556
+ depSchema.returnValueSchema[translatedPath] =
1557
+ dependentMergedDataStructure.returnValueSchema[path];
1558
+ }
1559
+ }
1560
+ // Copy over signature schema as well
1561
+ for (const path in dependentMergedDataStructure.signatureSchema) {
1562
+ const translatedPath = path.startsWith(name)
1563
+ ? translatePath(path, name)
1564
+ : path;
1565
+ if (!(translatedPath in depSchema.signatureSchema)) {
1566
+ depSchema.signatureSchema[translatedPath] =
1567
+ dependentMergedDataStructure.signatureSchema[path];
1568
+ }
1569
+ }
1570
+ }
1571
+ // Copy nested dependencySchemas for ALL entities (including mocked ones)
1572
+ // This represents what dependencies THIS entity uses, not its internal implementation
1573
+ if (dependentMergedDataStructure.dependencySchemas) {
1574
+ for (const depFilePath in dependentMergedDataStructure.dependencySchemas) {
1575
+ for (const depName in dependentMergedDataStructure
1576
+ .dependencySchemas[depFilePath]) {
1577
+ const nestedDepSchema = dependentMergedDataStructure.dependencySchemas[depFilePath][depName];
1578
+ const targetDepSchema = findOrCreateDependentSchemas({
1579
+ filePath: depFilePath,
1580
+ name: depName,
1581
+ });
1582
+ // Merge in the nested dependency schemas
1583
+ for (const path in nestedDepSchema.returnValueSchema) {
1584
+ checkDeadline();
1585
+ if (!(path in targetDepSchema.returnValueSchema)) {
1586
+ const value = nestedDepSchema.returnValueSchema[path];
1587
+ targetDepSchema.returnValueSchema[path] = value;
1588
+ }
1589
+ }
1590
+ for (const path in nestedDepSchema.signatureSchema) {
1591
+ checkDeadline();
1592
+ if (!(path in targetDepSchema.signatureSchema)) {
1593
+ targetDepSchema.signatureSchema[path] =
1594
+ nestedDepSchema.signatureSchema[path];
1595
+ }
1439
1596
  }
1440
1597
  }
1441
1598
  }
1442
1599
  }
1443
1600
  }
1444
1601
  }
1602
+ const totalElapsed = Date.now() - mergeStartTime;
1603
+ const retKeys = Object.keys(mergedDataStructure.returnValueSchema).length;
1604
+ // Only log phase breakdown for slow merges (>2s)
1605
+ if (totalElapsed > 2000) {
1606
+ console.log(`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}`);
1607
+ }
1608
+ return mergedDataStructure;
1609
+ }
1610
+ catch (error) {
1611
+ if (error instanceof DataStructureTimeoutError) {
1612
+ // Return partial results instead of propagating the timeout.
1613
+ // By this point, mergedDataStructure has valid data from completed phases
1614
+ // (gather + mergeESP complete in <1s, postfix/clean/depCopy may be partial).
1615
+ const retKeys = Object.keys(mergedDataStructure.returnValueSchema).length;
1616
+ console.log(`CodeYam Log Level 1: ${rootScopeName} merge timed out — returning partial results (${retKeys} ret keys, ${Math.round((Date.now() - mergeStartTime) / 1000)}s)`);
1617
+ mergedDataStructure.timedOut = true;
1618
+ return mergedDataStructure;
1619
+ }
1620
+ throw error;
1445
1621
  }
1446
- return mergedDataStructure;
1447
1622
  }
1448
1623
  //# sourceMappingURL=mergeInDependentDataStructure.js.map