@codeyam/codeyam-cli 0.1.0-staging.bbe4da9 → 0.1.0-staging.d0ad4ae

package/packages/analyze/src/lib/files/getImportedExports.js

@@ -1,4 +1,4 @@
-// import { generateSha } from "../../../../../packages/supabase/index.js";
+// import { generateSha } from "../../../../../packages/database/index.js";
 export default function getImportedExports(file, functionName, fileAnalyzer, projectAnalyzer, project, resolvedImports) {
     let relevantLocalEntities;
     try {
@@ -46,16 +46,35 @@ export default function getImportedExports(file, functionName, fileAnalyzer, pro
             // Workspace packages often resolve to dist/ paths (from package.json "types" or "main"),
             // but our project.files contains source files in src/. Try mapping dist -> src.
             if (!importedExportFile && searchPath.includes('/dist/')) {
-                const srcPath = searchPath.replace('/dist/', '/src/');
-                // Try exact match first
-                importedExportFile = project.files.find((f) => f.path === srcPath);
-                if (importedExportFile) {
-                    actualResolvedPath = srcPath;
-                    console.log(`CodeYam: Mapped dist->src path: ${searchPath} -> ${srcPath}`);
+                // Some build tools create nested structures like dist/packageName/src/file.d.ts
+                // when the source is at packages/packageName/src/file.ts
+                // Pattern: packages/X/dist/X/src/Y -> packages/X/src/Y
+                const nestedDistMatch = searchPath.match(/^(packages\/[^/]+)\/dist\/[^/]+\/src\/(.+)$/);
+                if (nestedDistMatch) {
+                    const [, packagePath, relativePath] = nestedDistMatch;
+                    const srcPath = `${packagePath}/src/${relativePath}`
+                        .replace(/\.d\.ts$/, '.ts')
+                        .replace(/\.js$/, '.ts');
+                    importedExportFile = project.files.find((f) => f.path === srcPath);
+                    if (importedExportFile) {
+                        actualResolvedPath = srcPath;
+                        console.log(`CodeYam: Mapped nested dist->src path: ${searchPath} -> ${srcPath}`);
+                    }
+                }
+                // Try simple dist -> src replacement if nested pattern didn't match
+                if (!importedExportFile) {
+                    const srcPath = searchPath.replace('/dist/', '/src/');
+                    // Try exact match first
+                    importedExportFile = project.files.find((f) => f.path === srcPath);
+                    if (importedExportFile) {
+                        actualResolvedPath = srcPath;
+                        console.log(`CodeYam: Mapped dist->src path: ${searchPath} -> ${srcPath}`);
+                    }
                 }
                 // If not found, the dist file might be index.d.ts but src might be index.ts
                 // or the export might be re-exported from another file
                 if (!importedExportFile) {
+                    const srcPath = searchPath.replace('/dist/', '/src/');
                     // Try removing .d.ts extension and finding any matching src file
                     const basePath = srcPath
                         .replace(/\.d\.ts$/, '')
@@ -71,31 +90,35 @@ export default function getImportedExports(file, functionName, fileAnalyzer, pro
                 }
             }
             if (!importedExportFile) {
-                // Check if this is a workspace package import (scoped npm package like @scope/package)
-                // but NOT a path alias like @/path or @components/...
-                // Scoped packages have format: @scope/package where scope is alphanumeric with hyphens
-                const isScopedPackage = filePath.startsWith('@') &&
-                    !filePath.startsWith('@/') && // Not a path alias like @/modules/...
-                    !filePath.startsWith('@types/') &&
-                    /^@[a-z0-9-]+\/[a-z0-9-]/.test(filePath); // Matches @scope/package pattern
-                if (isScopedPackage) {
-                    // Treat unresolved scoped packages like node modules so they get mocked
-                    // instead of being silently ignored
+                // Check if this looks like a path alias (starts with @/ or similar patterns)
+                // Path aliases should resolve to project files - if they don't, it's a config issue
+                // Also check for ~packageName/ patterns like ~codeyam/supabase
+                const isPathAlias = filePath.startsWith('@/') ||
+                    filePath.startsWith('~/') ||
+                    filePath.startsWith('#/') ||
+                    /^~[a-zA-Z]/.test(filePath); // ~packageName patterns like ~codeyam/
+                const name = importedExportInfo.importName ?? importedExportInfo.importAlias;
+                if (isPathAlias) {
+                    // Path aliases should resolve - if they don't, it's a configuration issue
+                    // Don't treat as node module, just log and skip
+                    console.log('CodeYam Warning: Path alias import not resolved (file may be missing or path alias misconfigured):', file.path, filePath, importedExportInfo);
+                    return;
+                }
+                // If still not found, track as unresolved import
+                // Don't set isMocked here - let later processing determine based on mockedDependencies
+                // This ensures the import is tracked (not dropped) while deferring the mocking decision
+                if (!importedExportFile) {
                     const name = importedExportInfo.importName ?? importedExportInfo.importAlias;
-                    console.log(`CodeYam: Treating unresolved scoped package as node module: ${filePath} -> ${name}`);
+                    console.log(`CodeYam: Tracking unresolved import (mocking decision deferred): ${filePath} -> ${name}`);
                     return {
                         filePath,
                         name,
                         isDefault: importedExportInfo.isImportDefault,
-                        isNodeModule: true, // Mark as node module so it gets mocked
-                        entityType: 'library', // Default to library since we can't determine the actual type
-                        // No code since we couldn't resolve the source file
+                        // isMocked intentionally undefined - will be determined later by prepareDataStructures
+                        // based on mockedDependencies analysis
+                        entityType: 'library',
                     };
                 }
-                else {
-                    console.log('CodeYam Error: File not found for imported export', file.path, filePath, importedExportInfo, imports[filePath]);
-                }
-                return;
             }
             try {
                 const importedExportFileAnalyzer = projectAnalyzer.getFileAnalyzer(importedExportFile);
@@ -106,8 +129,24 @@ export default function getImportedExports(file, functionName, fileAnalyzer, pro
                 // const sha = generateSha(importedExportFile.path, exportName, code);
                 const entityType = importedExportFileAnalyzer.getEntityType(exportName);
                 const name = importedExportInfo.importName ?? importedExportInfo.importAlias;
-                const resolvedFilePath = importedExportInfo.resolvedPath;
+                // Map nested dist paths in resolvedPath to src paths
+                // Pattern: packages/X/dist/X/src/Y -> packages/X/src/Y
+                let resolvedFilePath = importedExportInfo.resolvedPath;
+                if (resolvedFilePath) {
+                    const nestedDistMatch = resolvedFilePath.match(/^(packages\/[^/]+)\/dist\/[^/]+\/src\/(.+)$/);
+                    if (nestedDistMatch) {
+                        const [, packagePath, relativePath] = nestedDistMatch;
+                        resolvedFilePath = `${packagePath}/src/${relativePath}`
+                            .replace(/\.d\.ts$/, '.ts')
+                            .replace(/\.js$/, '.ts');
+                    }
+                }
                 const resolvedName = importedExportInfo.exportName;
+                // When a default export is re-exported as named (e.g., `export { default as X }`),
+                // isExportDefault will be false (it's exported as named from the barrel),
+                // but resolvedName will be 'default'. In this case, we need resolvedIsDefault
+                // to be true so that writeScenarioComponents adds the named re-export.
+                const resolvedIsDefault = importedExportInfo.isExportDefault || resolvedName === 'default';
                 return {
                     fileId: importedExportFile.id,
                     filePath: actualResolvedPath,
@@ -116,7 +155,7 @@ export default function getImportedExports(file, functionName, fileAnalyzer, pro
                         ? undefined
                         : resolvedFilePath,
                     resolvedName: resolvedName === name ? undefined : resolvedName,
-                    resolvedIsDefault: importedExportInfo.isExportDefault,
+                    resolvedIsDefault,
                     entityType,
                     name,
                     code,

package/packages/analyze/src/lib/files/getImportedExports.js.map

@@ -1 +1 @@
-{"version":3,"file":"getImportedExports.js","sourceRoot":"","sources":["../../../../../../../packages/analyze/src/lib/files/getImportedExports.ts"],"names":[],"mappings":"AAIA,mDAAmD;AAEnD,MAAM,CAAC,OAAO,UAAU,kBAAkB,CACxC,IAAU,EACV,YAAoB,EACpB,YAA0B,EAC1B,eAAgC,EAChC,OAAgB,EAChB,eAAyC;IAEzC,IAAI,qBAAqB,CAAC;IAC1B,IAAI,CAAC;QACH,qBAAqB,GAAG,YAAY;aACjC,8BAA8B,CAAC,YAAY,CAAC;aAC5C,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;YACd,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;YACzB,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;YACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC;YAC3B,MAAM,UAAU,GAAG,YAAY,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;YACpD,MAAM,SAAS,GAAG,YAAY,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YACrD,OAAO;gBACL,QAAQ;gBACR,IAAI;gBACJ,IAAI;gBACJ,UAAU;gBACV,SAAS;aACV,CAAC;QACJ,CAAC,CAAC;aACD,MAAM,CACL,CAAC,MAAM,EAAE,EAAE,CACT,MAAM,CAAC,QAAQ,KAAK,IAAI,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,KAAK,YAAY,CAChE,CAAC;IACN,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,GAAG,CACT,oEAAoE,EACpE;YACE,QAAQ,EAAE,IAAI,CAAC,IAAI;YACnB,YAAY;YACZ,KAAK,EAAE,KAAK,CAAC,OAAO;YACpB,KAAK,EAAE,KAAK,CAAC,KAAK;SACnB,CACF,CAAC;QACF,MAAM,KAAK,CAAC;IACd,CAAC;IAED,MAAM,OAAO,GACX,eAAe;QACf,YAAY,CAAC,wCAAwC,CAAC,YAAY,CAAC,CAAC;IAEtE,MAAM,eAAe,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC;SACzC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;QAChB,IAAI,OAAO,CAAC,QAAQ,CAAC,CAAC,eAAe;YAAE,OAAO;QAE9C,MAAM,mBAAmB,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,CAAC;QAE1D,OAAO,mBAAmB,CAAC,GAAG,CAAC,CAAC,kBAAkB,EAAE,EAAE;YACpD,MAAM,UAAU,GACd,kBAAkB,CAAC,YAAY;gBAC/B,OAAO,CAAC,QAAQ,CAAC,CAAC,YAAY;gBAC9B,QAAQ,CAAC;YAEX,IAAI,kBAAkB,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CACzC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAC7B,CAAC;YAEF,sFAAsF;YACtF,IAAI,kBAAkB,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC;YAExD,yFAAyF;YACzF,gFAAgF;YAChF,IAAI,CAAC,kBAAkB,IAAI,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACzD,MAAM,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;gBACtD,wBAAwB;gBACxB,kBAAkB,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC;gBACnE,IAAI,kBAAkB,EAAE,CAAC;oBACvB,kBAAkB,GAAG,OAAO,CAAC;oBAC7B,OAAO,CAAC,GAAG,CACT,mCAAmC,UAAU,OAAO,OAAO,EAAE,CAC9D,CAAC;gBACJ,CAAC;gBAED,4EAA4E;gBAC5E,uDAAuD;gBACvD,IAAI,CAAC,kBAAkB,EAAE,CAAC;oBACxB,iEAAiE;oBACjE,MAAM,QAAQ,GAAG,OAAO;yBACrB,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC;yBACvB,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;oBACxB,kBAAkB,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CACrC,CAAC,CAAC,EAAE,EAAE,CACJ,CAAC,CAAC,IAAI,KAAK,QAAQ,GAAG,KAAK;wBAC3B,CAAC,CAAC,IAAI,KAAK,QAAQ,GAAG,MAAM;wBAC5B,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;4BACjD,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAChC,CAAC;oBACF,IAAI,kBAAkB,EAAE,CAAC;wBACvB,kBAAkB,GAAG,kBAAkB,CAAC,IAAI,CAAC;wBAC7C,OAAO,CAAC,GAAG,CACT,+CAA+C,UAAU,OAAO,kBAAkB,CAAC,IAAI,EAAE,CAC1F,CAAC;oBACJ,CAAC;gBACH,CAAC;YACH,CAAC;YAED,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBACxB,uFAAuF;gBACvF,sDAAsD;gBACtD,uFAAuF;gBACvF,MAAM,eAAe,GACnB,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC;oBACxB,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,sCAAsC;oBACpE,CAAC,QAAQ,CAAC,UAAU,CAAC,SAAS,CAAC;oBAC/B,yBAAyB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,iCAAiC;gBAC7E,IAAI,eAAe,EAAE,CAAC;oBACpB,wEAAwE;oBACxE,oCAAoC;oBACpC,MAAM,IAAI,GACR,kBAAkB,CAAC,UAAU,IAAI,kBAAkB,CAAC,WAAW,CAAC;oBAClE,OAAO,CAAC,GAAG,CACT,+DAA+D,QAAQ,OAAO,IAAI,EAAE,CACrF,CAAC;oBACF,OAAO;wBACL,QAAQ;wBACR,IAAI;wBACJ,SAAS,EAAE,kBAAkB,CAAC,eAAe;wBAC7C,YAAY,EAAE,IAAI,EAAE,wCAAwC;wBAC5D,UAAU,EAAE,SAAkB,EAAE,8DAA8D;wBAC9F,oDAAoD;qBACrD,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,GAAG,CACT,mDAAmD,EACnD,IAAI,CAAC,IAAI,EACT,QAAQ,EACR,kBAAkB,EAClB,OAAO,CAAC,QAAQ,CAAC,CAClB,CAAC;gBACJ,CAAC;gBACD,OAAO;YACT,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,0BAA0B,GAC9B,eAAe,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;gBACtD,MAAM,UAAU,GAAG,kBAAkB,CAAC,eAAe;oBACnD,CAAC,CAAC,SAAS;oBACX,CAAC,CAAC,kBAAkB,CAAC,UAAU,CAAC;gBAClC,MAAM,IAAI,GAAG,0BAA0B,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;gBAClE,sEAAsE;gBACtE,MAAM,UAAU,GACd,0BAA0B,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;gBACvD,MAAM,IAAI,GACR,kBAAkB,CAAC,UAAU,IAAI,kBAAkB,CAAC,WAAW,CAAC;gBAElE,MAAM,gBAAgB,GAAG,kBAAkB,CAAC,YAAY,CAAC;gBACzD,MAAM,YAAY,GAAG,kBAAkB,CAAC,UAAU,CAAC;gBAEnD,OAAO;oBACL,MAAM,EAAE,kBAAkB,CAAC,EAAE;oBAC7B,QAAQ,EAAE,kBAAkB;oBAC5B,SAAS,EAAE,kBAAkB,CAAC,eAAe;oBAC7C,gBAAgB,EACd,gBAAgB,KAAK,kBAAkB;wBACrC,CAAC,CAAC,SAAS;wBACX,CAAC,CAAC,gBAAgB;oBACtB,YAAY,EAAE,YAAY,KAAK,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,YAAY;oBAC9D,iBAAiB,EAAE,kBAAkB,CAAC,eAAe;oBACrD,UAAU;oBACV,IAAI;oBACJ,IAAI;iBACL,CAAC;YACJ,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,iCAAiC,CAAC,EAAE,CAAC;oBAC1D,eAAe;oBACf,+DAA+D;oBAC/D,MAAM;oBACN,2BAA2B;oBAC3B,gCAAgC;oBAChC,0DAA0D;oBAC1D,0BAA0B;oBAC1B,OAAO;oBACP,KAAK;oBACL,OAAO;gBACT,CAAC;gBAED,OAAO,CAAC,GAAG,CACT,+CAA+C,EAC/C,IAAI,CAAC,SAAS,CACZ;oBACE,QAAQ,EAAE,IAAI,CAAC,IAAI;oBACnB,cAAc,EAAE,QAAQ;oBACxB,mBAAmB,EAAE,kBAAkB,CAAC,OAAO;oBAC/C,kBAAkB;iBACnB,EACD,IAAI,EACJ,CAAC,CACF,EACD,CAAC,CACF,CAAC;gBACF,MAAM,CAAC,CAAC;YACV,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;SACD,IAAI,EAAE;SACN,MAAM,CAAC,OAAO,CAAC,CAAC;IAEnB,OAAO,CAAC,GAAG,qBAAqB,EAAE,GAAG,eAAe,CAAC,CAAC;AACxD,CAAC"}
+{"version":3,"file":"getImportedExports.js","sourceRoot":"","sources":["../../../../../../../packages/analyze/src/lib/files/getImportedExports.ts"],"names":[],"mappings":"AAIA,mDAAmD;AAEnD,MAAM,CAAC,OAAO,UAAU,kBAAkB,CACxC,IAAU,EACV,YAAoB,EACpB,YAA0B,EAC1B,eAAgC,EAChC,OAAgB,EAChB,eAAyC;IAEzC,IAAI,qBAAqB,CAAC;IAC1B,IAAI,CAAC;QACH,qBAAqB,GAAG,YAAY;aACjC,8BAA8B,CAAC,YAAY,CAAC;aAC5C,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;YACd,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;YACzB,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;YACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC;YAC3B,MAAM,UAAU,GAAG,YAAY,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;YACpD,MAAM,SAAS,GAAG,YAAY,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YACrD,OAAO;gBACL,QAAQ;gBACR,IAAI;gBACJ,IAAI;gBACJ,UAAU;gBACV,SAAS;aACV,CAAC;QACJ,CAAC,CAAC;aACD,MAAM,CACL,CAAC,MAAM,EAAE,EAAE,CACT,MAAM,CAAC,QAAQ,KAAK,IAAI,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,KAAK,YAAY,CAChE,CAAC;IACN,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,GAAG,CACT,oEAAoE,EACpE;YACE,QAAQ,EAAE,IAAI,CAAC,IAAI;YACnB,YAAY;YACZ,KAAK,EAAE,KAAK,CAAC,OAAO;YACpB,KAAK,EAAE,KAAK,CAAC,KAAK;SACnB,CACF,CAAC;QACF,MAAM,KAAK,CAAC;IACd,CAAC;IAED,MAAM,OAAO,GACX,eAAe;QACf,YAAY,CAAC,wCAAwC,CAAC,YAAY,CAAC,CAAC;IAEtE,MAAM,eAAe,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC;SACzC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;QAChB,IAAI,OAAO,CAAC,QAAQ,CAAC,CAAC,eAAe;YAAE,OAAO;QAE9C,MAAM,mBAAmB,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,CAAC;QAE1D,OAAO,mBAAmB,CAAC,GAAG,CAAC,CAAC,kBAAkB,EAAE,EAAE;YACpD,MAAM,UAAU,GACd,kBAAkB,CAAC,YAAY;gBAC/B,OAAO,CAAC,QAAQ,CAAC,CAAC,YAAY;gBAC9B,QAAQ,CAAC;YAEX,IAAI,kBAAkB,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CACzC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAC7B,CAAC;YAEF,sFAAsF;YACtF,IAAI,kBAAkB,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC;YAExD,yFAAyF;YACzF,gFAAgF;YAChF,IAAI,CAAC,kBAAkB,IAAI,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACzD,gFAAgF;gBAChF,yDAAyD;gBACzD,uDAAuD;gBACvD,MAAM,eAAe,GAAG,UAAU,CAAC,KAAK,CACtC,6CAA6C,CAC9C,CAAC;gBACF,IAAI,eAAe,EAAE,CAAC;oBACpB,MAAM,CAAC,EAAE,WAAW,EAAE,YAAY,CAAC,GAAG,eAAe,CAAC;oBACtD,MAAM,OAAO,GAAG,GAAG,WAAW,QAAQ,YAAY,EAAE;yBACjD,OAAO,CAAC,UAAU,EAAE,KAAK,CAAC;yBAC1B,OAAO,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;oBAC3B,kBAAkB,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC;oBACnE,IAAI,kBAAkB,EAAE,CAAC;wBACvB,kBAAkB,GAAG,OAAO,CAAC;wBAC7B,OAAO,CAAC,GAAG,CACT,0CAA0C,UAAU,OAAO,OAAO,EAAE,CACrE,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAED,oEAAoE;gBACpE,IAAI,CAAC,kBAAkB,EAAE,CAAC;oBACxB,MAAM,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;oBACtD,wBAAwB;oBACxB,kBAAkB,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC;oBACnE,IAAI,kBAAkB,EAAE,CAAC;wBACvB,kBAAkB,GAAG,OAAO,CAAC;wBAC7B,OAAO,CAAC,GAAG,CACT,mCAAmC,UAAU,OAAO,OAAO,EAAE,CAC9D,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAED,4EAA4E;gBAC5E,uDAAuD;gBACvD,IAAI,CAAC,kBAAkB,EAAE,CAAC;oBACxB,MAAM,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;oBACtD,iEAAiE;oBACjE,MAAM,QAAQ,GAAG,OAAO;yBACrB,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC;yBACvB,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;oBACxB,kBAAkB,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CACrC,CAAC,CAAC,EAAE,EAAE,CACJ,CAAC,CAAC,IAAI,KAAK,QAAQ,GAAG,KAAK;wBAC3B,CAAC,CAAC,IAAI,KAAK,QAAQ,GAAG,MAAM;wBAC5B,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;4BACjD,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAChC,CAAC;oBACF,IAAI,kBAAkB,EAAE,CAAC;wBACvB,kBAAkB,GAAG,kBAAkB,CAAC,IAAI,CAAC;wBAC7C,OAAO,CAAC,GAAG,CACT,+CAA+C,UAAU,OAAO,kBAAkB,CAAC,IAAI,EAAE,CAC1F,CAAC;oBACJ,CAAC;gBACH,CAAC;YACH,CAAC;YAED,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBACxB,6EAA6E;gBAC7E,oFAAoF;gBACpF,+DAA+D;gBAC/D,MAAM,WAAW,GACf,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC;oBACzB,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC;oBACzB,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC;oBACzB,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,uCAAuC;gBAEtE,MAAM,IAAI,GACR,kBAAkB,CAAC,UAAU,IAAI,kBAAkB,CAAC,WAAW,CAAC;gBAElE,IAAI,WAAW,EAAE,CAAC;oBAChB,0EAA0E;oBAC1E,gDAAgD;oBAChD,OAAO,CAAC,GAAG,CACT,oGAAoG,EACpG,IAAI,CAAC,IAAI,EACT,QAAQ,EACR,kBAAkB,CACnB,CAAC;oBACF,OAAO;gBACT,CAAC;gBAED,iDAAiD;gBACjD,uFAAuF;gBACvF,wFAAwF;gBACxF,IAAI,CAAC,kBAAkB,EAAE,CAAC;oBACxB,MAAM,IAAI,GACR,kBAAkB,CAAC,UAAU,IAAI,kBAAkB,CAAC,WAAW,CAAC;oBAClE,OAAO,CAAC,GAAG,CACT,oEAAoE,QAAQ,OAAO,IAAI,EAAE,CAC1F,CAAC;oBACF,OAAO;wBACL,QAAQ;wBACR,IAAI;wBACJ,SAAS,EAAE,kBAAkB,CAAC,eAAe;wBAC7C,uFAAuF;wBACvF,uCAAuC;wBACvC,UAAU,EAAE,SAAkB;qBAC/B,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,0BAA0B,GAC9B,eAAe,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;gBACtD,MAAM,UAAU,GAAG,kBAAkB,CAAC,eAAe;oBACnD,CAAC,CAAC,SAAS;oBACX,CAAC,CAAC,kBAAkB,CAAC,UAAU,CAAC;gBAClC,MAAM,IAAI,GAAG,0BAA0B,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;gBAClE,sEAAsE;gBACtE,MAAM,UAAU,GACd,0BAA0B,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;gBACvD,MAAM,IAAI,GACR,kBAAkB,CAAC,UAAU,IAAI,kBAAkB,CAAC,WAAW,CAAC;gBAElE,qDAAqD;gBACrD,uDAAuD;gBACvD,IAAI,gBAAgB,GAAG,kBAAkB,CAAC,YAAY,CAAC;gBACvD,IAAI,gBAAgB,EAAE,CAAC;oBACrB,MAAM,eAAe,GAAG,gBAAgB,CAAC,KAAK,CAC5C,6CAA6C,CAC9C,CAAC;oBACF,IAAI,eAAe,EAAE,CAAC;wBACpB,MAAM,CAAC,EAAE,WAAW,EAAE,YAAY,CAAC,GAAG,eAAe,CAAC;wBACtD,gBAAgB,GAAG,GAAG,WAAW,QAAQ,YAAY,EAAE;6BACpD,OAAO,CAAC,UAAU,EAAE,KAAK,CAAC;6BAC1B,OAAO,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;oBAC7B,CAAC;gBACH,CAAC;gBACD,MAAM,YAAY,GAAG,kBAAkB,CAAC,UAAU,CAAC;gBAEnD,mFAAmF;gBACnF,0EAA0E;gBAC1E,8EAA8E;gBAC9E,uEAAuE;gBACvE,MAAM,iBAAiB,GACrB,kBAAkB,CAAC,eAAe,IAAI,YAAY,KAAK,SAAS,CAAC;gBAEnE,OAAO;oBACL,MAAM,EAAE,kBAAkB,CAAC,EAAE;oBAC7B,QAAQ,EAAE,kBAAkB;oBAC5B,SAAS,EAAE,kBAAkB,CAAC,eAAe;oBAC7C,gBAAgB,EACd,gBAAgB,KAAK,kBAAkB;wBACrC,CAAC,CAAC,SAAS;wBACX,CAAC,CAAC,gBAAgB;oBACtB,YAAY,EAAE,YAAY,KAAK,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,YAAY;oBAC9D,iBAAiB;oBACjB,UAAU;oBACV,IAAI;oBACJ,IAAI;iBACL,CAAC;YACJ,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,iCAAiC,CAAC,EAAE,CAAC;oBAC1D,eAAe;oBACf,+DAA+D;oBAC/D,MAAM;oBACN,2BAA2B;oBAC3B,gCAAgC;oBAChC,0DAA0D;oBAC1D,0BAA0B;oBAC1B,OAAO;oBACP,KAAK;oBACL,OAAO;gBACT,CAAC;gBAED,OAAO,CAAC,GAAG,CACT,+CAA+C,EAC/C,IAAI,CAAC,SAAS,CACZ;oBACE,QAAQ,EAAE,IAAI,CAAC,IAAI;oBACnB,cAAc,EAAE,QAAQ;oBACxB,mBAAmB,EAAE,kBAAkB,CAAC,OAAO;oBAC/C,kBAAkB;iBACnB,EACD,IAAI,EACJ,CAAC,CACF,EACD,CAAC,CACF,CAAC;gBACF,MAAM,CAAC,CAAC;YACV,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;SACD,IAAI,EAAE;SACN,MAAM,CAAC,OAAO,CAAC,CAAC;IAEnB,OAAO,CAAC,GAAG,qBAAqB,EAAE,GAAG,eAAe,CAAC,CAAC;AACxD,CAAC"}

package/packages/analyze/src/lib/files/scenarios/enrichArrayTypesFromChildSignatures.js

@@ -0,0 +1,159 @@
+/**
+ * Enriches mocked dependency array types with fields from child component signatures.
+ *
+ * When a mocked dependency (e.g., getSurveysAction) returns an array, and elements
+ * from that array are passed to child components (e.g., SurveyCard), the child's
+ * signatureSchema specifies required fields (e.g., survey.updatedAt). This function
+ * enriches the mocked dependency's returnValueSchema so the LLM knows what fields
+ * to generate for array elements.
+ *
+ * Example:
+ * - getSurveysAction returns { data: array }
+ * - SurveyCard expects signature[0].survey.updatedAt: date
+ * - After enrichment: getSurveysAction().data[]: object, data[].updatedAt: date
+ */
+export default function enrichArrayTypesFromChildSignatures(importedExports, dependencySchemas) {
+    // Find all mocked dependencies with array return types
+    const mockedArrayPaths = [];
+    for (const importedExport of importedExports) {
+        if (!importedExport.isMocked || !importedExport.filePath)
+            continue;
+        const schema = dependencySchemas?.[importedExport.filePath]?.[importedExport.name]
+            ?.returnValueSchema;
+        if (!schema)
+            continue;
+        // Find paths that are top-level data arrays (not deeply nested, not direct returns)
+        // We only want to enrich arrays that likely contain objects passed to children
+        // e.g., getSurveysAction().data but NOT:
+        //   - wrapThrows().data.createdBy (deeply nested)
+        //   - getStatusOptions().functionCallReturnValue (direct array return - usually options/primitives)
+        for (const [path, type] of Object.entries(schema)) {
+            if (type !== 'array')
+                continue;
+            // Skip paths with computed property patterns like [key], [i], [0], etc.
+            // These represent dynamic access patterns, not simple data arrays
+            // We only match brackets with content (not empty [] which indicates array element type)
+            if (/\[[^\]]+\]/.test(path)) {
+                continue;
+            }
+            // Skip direct array returns (e.g., getStatusOptions().functionCallReturnValue)
+            // These are typically specialized arrays like [{label, value}], not entity arrays
+            if (path.endsWith('.functionCallReturnValue')) {
+                continue;
+            }
+            // Check if this is a top-level array (at most one property after functionCallReturnValue)
+            // e.g., "foo().functionCallReturnValue.data" is top-level
+            // e.g., "foo().functionCallReturnValue.data.createdBy" is nested (skip)
+            const returnValueMatch = path.match(/\.functionCallReturnValue\.(.+)$/);
+            if (returnValueMatch) {
+                const afterReturnValue = returnValueMatch[1];
+                // If there's more than one property access, it's nested
+                const propertyCount = afterReturnValue.split('.').length;
+                if (propertyCount > 1) {
+                    continue; // Skip deeply nested arrays
+                }
+            }
+            mockedArrayPaths.push({
+                filePath: importedExport.filePath,
+                entityName: importedExport.name,
+                arrayPath: path,
+            });
+        }
+    }
+    if (mockedArrayPaths.length === 0)
+        return;
+    // Find all non-mocked child components and extract their signature fields
+    // We check both direct imports AND all schemas in dependencySchemas because
+    // transitive dependencies (e.g., SurveyCard imported by SurveysList imported by SurveysPage)
+    // may have signature schemas that we need to use for enrichment.
+    const childSignatureFields = [];
+    // Helper to extract prop fields from a signature schema
+    const extractPropFields = (schema) => {
+        const propFields = {};
+        for (const [path, type] of Object.entries(schema)) {
+            // Match signature[0].propName.field patterns
+            const match = path.match(/^signature\[0\]\.(\w+)\.(.+)$/);
+            if (match) {
+                const [, , fieldPath] = match;
+                // Store as "fieldPath" without the prop name prefix
+                // e.g., signature[0].survey.updatedAt -> updatedAt
+                propFields[fieldPath] = type;
+            }
+        }
+        return propFields;
+    };
+    // First, check direct imports (non-mocked)
+    const processedSchemas = new Set();
+    for (const importedExport of importedExports) {
+        // Skip mocked dependencies - we want child components
+        if (importedExport.isMocked || !importedExport.filePath)
+            continue;
+        const schemaKey = `${importedExport.filePath}::${importedExport.name}`;
+        processedSchemas.add(schemaKey);
+        const schema = dependencySchemas?.[importedExport.filePath]?.[importedExport.name]
+            ?.signatureSchema;
+        if (!schema)
+            continue;
+        const propFields = extractPropFields(schema);
+        if (Object.keys(propFields).length > 0) {
+            childSignatureFields.push({
+                filePath: importedExport.filePath,
+                entityName: importedExport.name,
+                propFields,
+            });
+        }
+    }
+    // Also check all schemas in dependencySchemas for transitive child components
+    // that may not be in importedExports but have signature schemas with prop fields
+    if (dependencySchemas) {
+        for (const [filePath, entities] of Object.entries(dependencySchemas)) {
+            for (const [entityName, schemas] of Object.entries(entities)) {
+                const schemaKey = `${filePath}::${entityName}`;
+                // Skip already processed schemas
+                if (processedSchemas.has(schemaKey))
+                    continue;
+                const signatureSchema = schemas?.signatureSchema;
+                if (!signatureSchema)
+                    continue;
+                const propFields = extractPropFields(signatureSchema);
+                if (Object.keys(propFields).length > 0) {
+                    childSignatureFields.push({
+                        filePath,
+                        entityName,
+                        propFields,
+                    });
+                }
+            }
+        }
+    }
+    if (childSignatureFields.length === 0)
+        return;
+    // Enrich mocked dependency arrays with child signature fields
+    // Only enrich arrays that don't already have element types defined
+    // A more sophisticated approach would trace the data flow to match specific arrays to specific children
+    for (const { filePath, entityName, arrayPath } of mockedArrayPaths) {
+        const schema = dependencySchemas[filePath][entityName].returnValueSchema;
+        // Check if this array already has element type definitions
+        // If so, skip it - it's likely a specialized array like [null] for refs
+        const elementPathPrefix = `${arrayPath}[`;
+        const hasExistingElementType = Object.keys(schema).some((path) => path.startsWith(elementPathPrefix));
+        if (hasExistingElementType) {
+            // Array already has element type info (e.g., useAutoAnimate()[0]: null)
+            // Don't override it with child signature fields
+            continue;
+        }
+        // Add array element type
+        const elementPath = `${arrayPath}[]`;
+        schema[elementPath] = 'object';
+        // Add fields from all child signatures to the array element
+        for (const { propFields } of childSignatureFields) {
+            for (const [fieldPath, type] of Object.entries(propFields)) {
+                const fullPath = `${elementPath}.${fieldPath}`;
+                if (!schema[fullPath]) {
+                    schema[fullPath] = type;
+                }
+            }
+        }
+    }
+}
+//# sourceMappingURL=enrichArrayTypesFromChildSignatures.js.map

package/packages/analyze/src/lib/files/scenarios/enrichArrayTypesFromChildSignatures.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"enrichArrayTypesFromChildSignatures.js","sourceRoot":"","sources":["../../../../../../../../packages/analyze/src/lib/files/scenarios/enrichArrayTypesFromChildSignatures.ts"],"names":[],"mappings":"AAEA;;;;;;;;;;;;;GAaG;AACH,MAAM,CAAC,OAAO,UAAU,mCAAmC,CACzD,eAIG,EACH,iBAAqD;IAErD,uDAAuD;IACvD,MAAM,gBAAgB,GAIhB,EAAE,CAAC;IAET,KAAK,MAAM,cAAc,IAAI,eAAe,EAAE,CAAC;QAC7C,IAAI,CAAC,cAAc,CAAC,QAAQ,IAAI,CAAC,cAAc,CAAC,QAAQ;YAAE,SAAS;QAEnE,MAAM,MAAM,GACV,iBAAiB,EAAE,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,CAAC,cAAc,CAAC,IAAI,CAAC;YACjE,EAAE,iBAAiB,CAAC;QACxB,IAAI,CAAC,MAAM;YAAE,SAAS;QAEtB,oFAAoF;QACpF,+EAA+E;QAC/E,yCAAyC;QACzC,kDAAkD;QAClD,oGAAoG;QACpG,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YAClD,IAAI,IAAI,KAAK,OAAO;gBAAE,SAAS;YAE/B,wEAAwE;YACxE,kEAAkE;YAClE,wFAAwF;YACxF,IAAI,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5B,SAAS;YACX,CAAC;YAED,+EAA+E;YAC/E,kFAAkF;YAClF,IAAI,IAAI,CAAC,QAAQ,CAAC,0BAA0B,CAAC,EAAE,CAAC;gBAC9C,SAAS;YACX,CAAC;YAED,0FAA0F;YAC1F,0DAA0D;YAC1D,wEAAwE;YACxE,MAAM,gBAAgB,GAAG,IAAI,CAAC,KAAK,CAAC,kCAAkC,CAAC,CAAC;YACxE,IAAI,gBAAgB,EAAE,CAAC;gBACrB,MAAM,gBAAgB,GAAG,gBAAgB,CAAC,CAAC,CAAC,CAAC;gBAC7C,wDAAwD;gBACxD,MAAM,aAAa,GAAG,gBAAgB,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC;gBACzD,IAAI,aAAa,GAAG,CAAC,EAAE,CAAC;oBACtB,SAAS,CAAC,4BAA4B;gBACxC,CAAC;YACH,CAAC;YAED,gBAAgB,CAAC,IAAI,CAAC;gBACpB,QAAQ,EAAE,cAAc,CAAC,QAAQ;gBACjC,UAAU,EAAE,cAAc,CAAC,IAAI;gBAC/B,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO;IAE1C,0EAA0E;IAC1E,4EAA4E;IAC5E,6FAA6F;IAC7F,iEAAiE;IACjE,MAAM,oBAAoB,GAKpB,EAAE,CAAC;IAET,wDAAwD;IACxD,MAAM,iBAAiB,GAAG,CACxB,MAA8B,EACN,EAAE;QAC1B,MAAM,UAAU,GAA2B,EAAE,CAAC;QAC9C,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YAClD,6CAA6C;YAC7C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,+BAA+B,CAAC,CAAC;YAC1D,IAAI,KAAK,EAAE,CAAC;gBACV,MAAM,CAAC,EAAE,AAAD,EAAG,SAAS,CAAC,GAAG,KAAK,CAAC;gBAC9B,oDAAoD;gBACpD,mDAAmD;gBACnD,UAAU,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC;YAC/B,CAAC;QACH,CAAC;QACD,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;IAEF,2CAA2C;IAC3C,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAU,CAAC;IAC3C,KAAK,MAAM,cAAc,IAAI,eAAe,EAAE,CAAC;QAC7C,sDAAsD;QACtD,IAAI,cAAc,CAAC,QAAQ,IAAI,CAAC,cAAc,CAAC,QAAQ;YAAE,SAAS;QAElE,MAAM,SAAS,GAAG,GAAG,cAAc,CAAC,QAAQ,KAAK,cAAc,CAAC,IAAI,EAAE,CAAC;QACvE,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAEhC,MAAM,MAAM,GACV,iBAAiB,EAAE,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,CAAC,cAAc,CAAC,IAAI,CAAC;YACjE,EAAE,eAAe,CAAC;QACtB,IAAI,CAAC,MAAM;YAAE,SAAS;QAEtB,MAAM,UAAU,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC;QAC7C,IAAI,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvC,oBAAoB,CAAC,IAAI,CAAC;gBACxB,QAAQ,EAAE,cAAc,CAAC,QAAQ;gBACjC,UAAU,EAAE,cAAc,CAAC,IAAI;gBAC/B,UAAU;aACX,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,8EAA8E;IAC9E,iFAAiF;IACjF,IAAI,iBAAiB,EAAE,CAAC;QACtB,KAAK,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACrE,KAAK,MAAM,CAAC,UAAU,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC7D,MAAM,SAAS,GAAG,GAAG,QAAQ,KAAK,UAAU,EAAE,CAAC;gBAC/C,iCAAiC;gBACjC,IAAI,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC;oBAAE,SAAS;gBAE9C,MAAM,eAAe,GAAG,OAAO,EAAE,eAAe,CAAC;gBACjD,IAAI,CAAC,eAAe;oBAAE,SAAS;gBAE/B,MAAM,UAAU,GAAG,iBAAiB,CAAC,eAAe,CAAC,CAAC;gBACtD,IAAI,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACvC,oBAAoB,CAAC,IAAI,CAAC;wBACxB,QAAQ;wBACR,UAAU;wBACV,UAAU;qBACX,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,oBAAoB,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO;IAE9C,8DAA8D;IAC9D,mEAAmE;IACnE,wGAAwG;IACxG,KAAK,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,IAAI,gBAAgB,EAAE,CAAC;QACnE,MAAM,MAAM,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC,UAAU,CAAC,CAAC,iBAAiB,CAAC;QAEzE,2DAA2D;QAC3D,wEAAwE;QACxE,MAAM,iBAAiB,GAAG,GAAG,SAAS,GAAG,CAAC;QAC1C,MAAM,sBAAsB,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAC/D,IAAI,CAAC,UAAU,CAAC,iBAAiB,CAAC,CACnC,CAAC;QAEF,IAAI,sBAAsB,EAAE,CAAC;YAC3B,wEAAwE;YACxE,gDAAgD;YAChD,SAAS;QACX,CAAC;QAED,yBAAyB;QACzB,MAAM,WAAW,GAAG,GAAG,SAAS,IAAI,CAAC;QACrC,MAAM,CAAC,WAAW,CAAC,GAAG,QAAQ,CAAC;QAE/B,4DAA4D;QAC5D,KAAK,MAAM,EAAE,UAAU,EAAE,IAAI,oBAAoB,EAAE,CAAC;YAClD,KAAK,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC3D,MAAM,QAAQ,GAAG,GAAG,WAAW,IAAI,SAAS,EAAE,CAAC;gBAC/C,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACtB,MAAM,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC;gBAC1B,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC"}

package/packages/analyze/src/lib/files/scenarios/enrichUnknownTypesFromSourceEquivalencies.js

@@ -0,0 +1,85 @@
+/**
+ * Enriches mocked dependency return value types using sourceEquivalencies.
+ *
+ * When a mocked dependency returns a value with an `unknown` type, and that value
+ * is passed to a child component, we can infer the type from the child's signature.
+ * The sourceEquivalencies track this data flow.
+ *
+ * Example:
+ * - trpc.quote.getQuoteQuestionAnswers.useQuery().functionCallReturnValue.data is `unknown`
+ * - sourceEquivalencies shows this data flows to QuoteHeaderExcel().signature[0].questionAnswers
+ * - QuoteHeaderExcel's signatureSchema says signature[0].questionAnswers is `array | undefined`
+ * - Therefore, data should be `array | undefined`
+ *
+ * Without this enrichment, fillInDirectSchemaGapsAndUnknowns will guess the type
+ * based on the property name, often incorrectly (e.g., 'data' becomes 'number'
+ * because pluralize.isPlural('data') returns true).
+ */
+export default function enrichUnknownTypesFromSourceEquivalencies(isolatedDependencySchemas, mergedDependencySchemas, mockedImports) {
+    // Build a map of mocked dependency schemas for quick lookup
+    const mockedSchemas = new Map();
+    for (const imp of mockedImports) {
+        if (!imp.isMocked || !imp.filePath)
+            continue;
+        const schema = mergedDependencySchemas?.[imp.filePath]?.[imp.name]?.returnValueSchema;
+        if (schema) {
+            mockedSchemas.set(`${imp.filePath}:${imp.name}`, schema);
+        }
+    }
+    if (mockedSchemas.size === 0)
+        return;
+    // Iterate over all sourceEquivalencies in all dependency schemas
+    for (const filePath in isolatedDependencySchemas) {
+        for (const entityName in isolatedDependencySchemas[filePath]) {
+            const depSchema = isolatedDependencySchemas[filePath][entityName];
+            const sourceEquivs = depSchema.sourceEquivalencies;
+            if (!sourceEquivs)
+                continue;
+            // For each sourceEquivalency: key is destination, value contains sources
+            for (const [destinationPath, sources] of Object.entries(sourceEquivs)) {
+                // Parse the destination to get the child entity name and signature path
+                // e.g., "QuoteHeaderExcel().signature[0].questionAnswers" ->
+                //   entityName: "QuoteHeaderExcel", sigPath: "signature[0].questionAnswers"
+                const destMatch = destinationPath.match(/^(\w+)\(\)\.(.+)$/);
+                if (!destMatch)
+                    continue;
+                const [, destEntityName, destSigPath] = destMatch;
+                // Find the child's signatureSchema to get the destination type
+                let destType;
+                for (const depFilePath in mergedDependencySchemas) {
+                    const childSchema = mergedDependencySchemas[depFilePath][destEntityName];
+                    if (childSchema?.signatureSchema?.[destSigPath]) {
+                        destType = childSchema.signatureSchema[destSigPath];
+                        break;
+                    }
+                }
+                if (!destType || destType === 'unknown')
+                    continue;
+                // Now check each source to see if it's a mocked dependency with unknown type
+                for (const source of sources) {
+                    const { scopeNodeName, schemaPath } = source;
+                    // Find which mocked dependency this belongs to
+                    for (const [key, schema] of mockedSchemas) {
+                        const [mockedFilePath, mockedName] = key.split(':');
+                        // Check if this source path is in the mocked dependency's schema
+                        // The schemaPath contains the full path like:
+                        // "trpc.quote.getQuoteQuestionAnswers.useQuery(...).functionCallReturnValue.data"
+                        // We need to find it in the returnValueSchema
+                        if (schema[schemaPath] === 'unknown') {
+                            // Found an unknown type that flows to a known type!
+                            schema[schemaPath] = destType;
+                        }
+                        // Also check without the functionCallReturnValue part
+                        // (some schemas might have the transformed path)
+                        const transformedPath = schemaPath.replace(/\.functionCallReturnValue/g, '');
+                        if (transformedPath !== schemaPath &&
+                            schema[transformedPath] === 'unknown') {
+                            schema[transformedPath] = destType;
+                        }
+                    }
+                }
+            }
+        }
+    }
+}
+//# sourceMappingURL=enrichUnknownTypesFromSourceEquivalencies.js.map

package/packages/analyze/src/lib/files/scenarios/enrichUnknownTypesFromSourceEquivalencies.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"enrichUnknownTypesFromSourceEquivalencies.js","sourceRoot":"","sources":["../../../../../../../../packages/analyze/src/lib/files/scenarios/enrichUnknownTypesFromSourceEquivalencies.ts"],"names":[],"mappings":"AAEA;;;;;;;;;;;;;;;;GAgBG;AACH,MAAM,CAAC,OAAO,UAAU,yCAAyC,CAC/D,yBAA6D,EAC7D,uBAA2D,EAC3D,aAAwE;IAExE,4DAA4D;IAC5D,MAAM,aAAa,GAAG,IAAI,GAAG,EAAkC,CAAC;IAChE,KAAK,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;QAChC,IAAI,CAAC,GAAG,CAAC,QAAQ,IAAI,CAAC,GAAG,CAAC,QAAQ;YAAE,SAAS;QAC7C,MAAM,MAAM,GACV,uBAAuB,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,iBAAiB,CAAC;QACzE,IAAI,MAAM,EAAE,CAAC;YACX,aAAa,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,QAAQ,IAAI,GAAG,CAAC,IAAI,EAAE,EAAE,MAAM,CAAC,CAAC;QAC3D,CAAC;IACH,CAAC;IAED,IAAI,aAAa,CAAC,IAAI,KAAK,CAAC;QAAE,OAAO;IAErC,iEAAiE;IACjE,KAAK,MAAM,QAAQ,IAAI,yBAAyB,EAAE,CAAC;QACjD,KAAK,MAAM,UAAU,IAAI,yBAAyB,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC7D,MAAM,SAAS,GAAG,yBAAyB,CAAC,QAAQ,CAAC,CAAC,UAAU,CAAC,CAAC;YAClE,MAAM,YAAY,GAAG,SAAS,CAAC,mBAAmB,CAAC;YACnD,IAAI,CAAC,YAAY;gBAAE,SAAS;YAE5B,yEAAyE;YACzE,KAAK,MAAM,CAAC,eAAe,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC;gBACtE,wEAAwE;gBACxE,6DAA6D;gBAC7D,4EAA4E;gBAC5E,MAAM,SAAS,GAAG,eAAe,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;gBAC7D,IAAI,CAAC,SAAS;oBAAE,SAAS;gBAEzB,MAAM,CAAC,EAAE,cAAc,EAAE,WAAW,CAAC,GAAG,SAAS,CAAC;gBAElD,+DAA+D;gBAC/D,IAAI,QAA4B,CAAC;gBACjC,KAAK,MAAM,WAAW,IAAI,uBAAuB,EAAE,CAAC;oBAClD,MAAM,WAAW,GACf,uBAAuB,CAAC,WAAW,CAAC,CAAC,cAAc,CAAC,CAAC;oBACvD,IAAI,WAAW,EAAE,eAAe,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;wBAChD,QAAQ,GAAG,WAAW,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;wBACpD,MAAM;oBACR,CAAC;gBACH,CAAC;gBAED,IAAI,CAAC,QAAQ,IAAI,QAAQ,KAAK,SAAS;oBAAE,SAAS;gBAElD,6EAA6E;gBAC7E,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;oBAC7B,MAAM,EAAE,aAAa,EAAE,UAAU,EAAE,GAAG,MAAM,CAAC;oBAE7C,+CAA+C;oBAC/C,KAAK,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,IAAI,aAAa,EAAE,CAAC;wBAC1C,MAAM,CAAC,cAAc,EAAE,UAAU,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;wBAEpD,iEAAiE;wBACjE,8CAA8C;wBAC9C,kFAAkF;wBAClF,8CAA8C;wBAC9C,IAAI,MAAM,CAAC,UAAU,CAAC,KAAK,SAAS,EAAE,CAAC;4BACrC,oDAAoD;4BACpD,MAAM,CAAC,UAAU,CAAC,GAAG,QAAQ,CAAC;wBAChC,CAAC;wBAED,sDAAsD;wBACtD,iDAAiD;wBACjD,MAAM,eAAe,GAAG,UAAU,CAAC,OAAO,CACxC,4BAA4B,EAC5B,EAAE,CACH,CAAC;wBACF,IACE,eAAe,KAAK,UAAU;4BAC9B,MAAM,CAAC,eAAe,CAAC,KAAK,SAAS,EACrC,CAAC;4BACD,MAAM,CAAC,eAAe,CAAC,GAAG,QAAQ,CAAC;wBACrC,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC"}