@codeyam/codeyam-cli 0.1.0-staging.28f73cf → 0.1.0-staging.596f0eb

package/packages/analyze/src/lib/asts/sourceFiles/getAllExports.js

@@ -77,17 +77,18 @@ export function getAllExports(sourceFile) {
         }
     }
     function processNamedExports(node) {
+        const parent = node.parent;
         node.elements.forEach((element) => {
             const name = element.name.text;
             const propertyName = element.propertyName
                 ? element.propertyName.text
                 : name;
-            const importedSource = node.parent.moduleSpecifier
-                ? node.parent.moduleSpecifier.getText(sourceFile)
+            const importedSource = parent?.moduleSpecifier
+                ? parent.moduleSpecifier.getText(sourceFile)
                 : undefined;
             if (importedSource) {
                 const directImportPath = importedSource?.replace(/['"]/g, '');
-                const directImport = node.parent.exportClause?.elements?.find((e) => (e.propertyName || e.name).text === propertyName);
+                const directImport = parent?.exportClause?.elements?.find((e) => (e.propertyName || e.name).text === propertyName);
                 const directImportName = (directImport?.propertyName ?? directImport?.name)?.text;
                 // Safely get node text - use custom serialization if getText fails
                 let nodeText;
@@ -150,13 +151,38 @@ export function getAllExports(sourceFile) {
         else if (ts.isExportAssignment(node) && !node.isExportEquals) {
             processExportAssignment(node);
         }
-        else if (ts.isExportDeclaration(node) && node.exportClause) {
-            if (ts.isNamedExports(node.exportClause)) {
-                processNamedExports(node.exportClause);
+        else if (ts.isExportDeclaration(node)) {
+            if (node.exportClause) {
+                if (ts.isNamedExports(node.exportClause)) {
+                    processNamedExports(node.exportClause);
+                }
+                else if (ts.isNamespaceExport(node.exportClause)) {
+                    const name = node.exportClause.name.text;
+                    addExport(name, node, false);
+                }
             }
-            else if (ts.isNamespaceExport(node.exportClause)) {
-                const name = node.exportClause.name.text;
-                addExport(name, node, false);
+            else if (node.moduleSpecifier) {
+                // Handle star exports: export * from "./path.js"
+                // The exportClause is undefined for star exports
+                const moduleSpecifier = node.moduleSpecifier.getText(sourceFile);
+                const directImportPath = moduleSpecifier.replace(/['"]/g, '');
+                // Use a synthetic name based on the path to track this star export
+                // The * symbol indicates this is a star export that needs to be followed
+                const starExportName = `*:${directImportPath}`;
+                let code;
+                try {
+                    code = node.getFullText(sourceFile).trim();
+                }
+                catch (error) {
+                    code = `export * from '${directImportPath}'`;
+                }
+                exportsCodeMap[starExportName] = {
+                    code,
+                    isDefaultExport: false,
+                    isNamespaceExport: false,
+                    isStarExport: true,
+                    directImportPath,
+                };
             }
         }
     });

package/packages/analyze/src/lib/asts/sourceFiles/getAllExports.js.map

@@ -1 +1 @@
-{"version":3,"file":"getAllExports.js","sourceRoot":"","sources":["../../../../../../../../packages/analyze/src/lib/asts/sourceFiles/getAllExports.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,YAAY,CAAC;AAC5B,OAAO,KAAK,GAAG,MAAM,aAAa,CAAC;AAUnC,MAAM,UAAU,aAAa,CAC3B,UAAyB;IAEzB,MAAM,cAAc,GAAqC,EAAE,CAAC;IAE5D,SAAS,SAAS,CAChB,IAAY,EACZ,IAAa,EACb,SAAkB,EAClB,gBAAyB,EACzB,gBAAyB;QAEzB,IAAI,cAAc,CAAC,IAAI,CAAC;YAAE,OAAO;QAEjC,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,IAAI,EAAE,CAAC;QACjD,cAAc,CAAC,IAAI,CAAC,GAAG;YACrB,IAAI;YACJ,eAAe,EAAE,SAAS;YAC1B,iBAAiB,EAAE,iBAAiB,CAAC,IAAI,CAAC;YAC1C,gBAAgB;YAChB,gBAAgB;SACjB,CAAC;IACJ,CAAC;IAED,SAAS,iBAAiB,CAAC,IAAa;QACtC,IAAI,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtD,OAAO,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACjD,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,SAAS,kBAAkB,CAAC,IAAa,EAAE,eAAwB;QACjE,IAAI,IAAI,GAAG,SAAS,CAAC;QAErB,IAAI,EAAE,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YAChD,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;QACxB,CAAC;aAAM,IAAI,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;YACxC,MAAM,CAAC,WAAW,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC;YACxD,IAAI,EAAE,CAAC,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;gBACtC,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC;YAC/B,CAAC;QACH,CAAC;QAED,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,eAAe,CAAC,CAAC;IACzC,CAAC;IAED,SAAS,uBAAuB,CAAC,IAAyB;QACxD,IAAI,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;YACrC,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;YAClC,UAAU,CAAC,YAAY,CAAC,CAAC,KAAK,EAAE,EAAE;gBAChC,IAAI,EAAE,CAAC,kBAAkB,CAAC,KAAK,CAAC,EAAE,CAAC;oBACjC,IAAI,gBAAoC,CAAC;oBACzC,IAAI,gBAAwB,CAAC;oBAC7B,IAAI,CAAC;wBACH,gBAAgB,GAAG,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;oBAC1D,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,4CAA4C;wBAC5C,gBAAgB,GAAI,KAAK,CAAC,UAAkB,CAAC,IAAI,IAAI,IAAI,CAAC;oBAC5D,CAAC;oBACD,MAAM,eAAe,GACnB,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC;oBACtD,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,EAAE,WAAW,CAAC,EAAE,EAAE;wBAC9D,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;4BACrC,IAAI,UAAU,CAAC,WAAW,KAAK,IAAI,EAAE,CAAC;gCACpC,gBAAgB,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gCAC7C,gBAAgB,GAAG,UAAU,CAAC,UAAU,CAAC;4BAC3C,CAAC;wBACH,CAAC;oBACH,CAAC,CAAC,CAAC;oBAEH,SAAS,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,gBAAgB,EAAE,gBAAgB,CAAC,CAAC;gBACnE,CAAC;qBAAM,IACL,CAAC,EAAE,CAAC,qBAAqB,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;oBAClE,CAAC,CAAC,EAAE,CAAC,qBAAqB,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,IAAI,EAAE,IAAI,KAAK,IAAI,CAAC;wBAC7D,CAAC,EAAE,CAAC,mBAAmB,CAAC,KAAK,CAAC;4BAC5B,KAAK,CAAC,eAAe,CAAC,YAAY,CAAC,IAAI,CACrC,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,IAAI,KAAK,IAAI,CAC7D,CAAC,CAAC,EACP,CAAC;oBACD,SAAS,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;gBAC/B,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,MAAM,mBAAmB,GAAG,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAC;YACxE,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,IAAI,EAAE,CAAC;YACjD,cAAc,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC,GAAG;gBACpD,IAAI;gBACJ,eAAe,EAAE,IAAI;gBACrB,iBAAiB,EAAE,KAAK;aACzB,CAAC;QACJ,CAAC;IACH,CAAC;IAED,SAAS,mBAAmB,CAAC,IAAqB;QAChD,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAChC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC;YAC/B,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY;gBACvC,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI;gBAC3B,CAAC,CAAC,IAAI,CAAC;YACT,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe;gBAChD,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,UAAU,CAAC;gBACjD,CAAC,CAAC,SAAS,CAAC;YAEd,IAAI,cAAc,EAAE,CAAC;gBACnB,MAAM,gBAAgB,GAAG,cAAc,EAAE,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gBAC9D,MAAM,YAAY,GAChB,IAAI,CAAC,MAAM,CAAC,YACb,EAAE,QAAQ,EAAE,IAAI,CACf,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,YAAY,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,YAAY,CACxD,CAAC;gBACF,MAAM,gBAAgB,GAAG,CACvB,YAAY,EAAE,YAAY,IAAI,YAAY,EAAE,IAAI,CACjD,EAAE,IAAI,CAAC;gBAER,mEAAmE;gBACnE,IAAI,QAAgB,CAAC;gBACrB,IAAI,CAAC;oBACH,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;gBACtC,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,+CAA+C;oBAC/C,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ;yBAC3B,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE;wBACV,IAAI,EAAE,CAAC,YAAY,EAAE,CAAC;4BACpB,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,OAAO,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;wBACtD,CAAC;wBACD,OAAO,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;oBACtB,CAAC,CAAC;yBACD,IAAI,CAAC,IAAI,CAAC,CAAC;oBACd,QAAQ,GAAG,KAAK,QAAQ,IAAI,CAAC;gBAC/B,CAAC;gBACD,MAAM,eAAe,GAAG,UAAU,QAAQ,SAAS,cAAc,EAAE,CAAC;gBAEpE,cAAc,CAAC,IAAI,CAAC,GAAG;oBACrB,IAAI,EAAE,eAAe;oBACrB,eAAe,EAAE,IAAI,KAAK,SAAS;oBACnC,iBAAiB,EAAE,KAAK;oBACxB,gBAAgB;oBAChB,gBAAgB;iBACjB,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,MAAM,YAAY,GAAG,UAAU,CAAC,UAAU,CAAC,MAAM,CAC/C,CACE,IAAI,EAImB,EAAE,CACzB,CAAC,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC;oBAC3B,EAAE,CAAC,qBAAqB,CAAC,IAAI,CAAC;oBAC9B,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;oBAC/B,CAAC,CAAC,EAAE,CAAC,qBAAqB,CAAC,IAAI,CAAC;wBAC9B,IAAI,CAAC,IAAI,EAAE,IAAI,KAAK,YAAY,CAAC;wBACjC,CAAC,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC;4BAC3B,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,IAAI,CACpC,CAAC,GAAG,EAAE,EAAE,CACN,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,IAAI,KAAK,YAAY,CAC9D,CAAC;wBACJ,CAAC,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC;4BAC3B,CAAC,IAAI,CAAC,YAAY,EAAE,OAAO,CAAC,UAAU,CAAC,KAAK,YAAY;gCACtD,CAAC,IAAI,CAAC,YAAY,EAAE,aAAa;oCAC/B,EAAE,CAAC,cAAc,CAAC,IAAI,CAAC,YAAY,EAAE,aAAa,CAAC;oCACnD,IAAI,CAAC,YAAY,EAAE,aAAa,CAAC,QAAQ,CAAC,IAAI,CAC5C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,YAAY,CACpC,CAAC,CAAC,CAAC,CAAC,CAChB,CAAC;gBAEF,YAAY,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;oBAC5B,IAAI,gBAAoC,CAAC;oBACzC,IAAI,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;wBACjC,gBAAgB,GAAG,IAAI,CAAC,eAAe;4BACrC,EAAE,OAAO,CAAC,UAAU,CAAC;6BACpB,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;oBAC1B,CAAC;oBACD,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,gBAAgB,CAAC,CAAC;gBACjD,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,UAAU,CAAC,YAAY,CAAC,CAAC,IAAI,EAAE,EAAE;QAC/B,IAAI,EAAE,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;YACnE,IACE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,EAAE,CAAC,UAAU,CAAC,aAAa,CAAC,EACvE,CAAC;gBACD,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,EAAE,IAAI,CAC1C,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,EAAE,CAAC,UAAU,CAAC,cAAc,CACnD,CAAC;gBACF,kBAAkB,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;YAC5C,CAAC;QACH,CAAC;aAAM,IAAI,EAAE,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YAC/D,uBAAuB,CAAC,IAAI,CAAC,CAAC;QAChC,CAAC;aAAM,IAAI,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YAC7D,IAAI,EAAE,CAAC,cAAc,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;gBACzC,mBAAmB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACzC,CAAC;iBAAM,IAAI,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;gBACnD,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;gBACzC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;YAC/B,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,cAAc,CAAC;AACxB,CAAC"}
+{"version":3,"file":"getAllExports.js","sourceRoot":"","sources":["../../../../../../../../packages/analyze/src/lib/asts/sourceFiles/getAllExports.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,YAAY,CAAC;AAC5B,OAAO,KAAK,GAAG,MAAM,aAAa,CAAC;AAWnC,MAAM,UAAU,aAAa,CAC3B,UAAyB;IAEzB,MAAM,cAAc,GAAqC,EAAE,CAAC;IAE5D,SAAS,SAAS,CAChB,IAAY,EACZ,IAAa,EACb,SAAkB,EAClB,gBAAyB,EACzB,gBAAyB;QAEzB,IAAI,cAAc,CAAC,IAAI,CAAC;YAAE,OAAO;QAEjC,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,IAAI,EAAE,CAAC;QACjD,cAAc,CAAC,IAAI,CAAC,GAAG;YACrB,IAAI;YACJ,eAAe,EAAE,SAAS;YAC1B,iBAAiB,EAAE,iBAAiB,CAAC,IAAI,CAAC;YAC1C,gBAAgB;YAChB,gBAAgB;SACjB,CAAC;IACJ,CAAC;IAED,SAAS,iBAAiB,CAAC,IAAa;QACtC,IAAI,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtD,OAAO,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACjD,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,SAAS,kBAAkB,CAAC,IAAa,EAAE,eAAwB;QACjE,IAAI,IAAI,GAAG,SAAS,CAAC;QAErB,IAAI,EAAE,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YAChD,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;QACxB,CAAC;aAAM,IAAI,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;YACxC,MAAM,CAAC,WAAW,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC;YACxD,IAAI,EAAE,CAAC,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;gBACtC,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC;YAC/B,CAAC;QACH,CAAC;QAED,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,eAAe,CAAC,CAAC;IACzC,CAAC;IAED,SAAS,uBAAuB,CAAC,IAAyB;QACxD,IAAI,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;YACrC,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;YAClC,UAAU,CAAC,YAAY,CAAC,CAAC,KAAK,EAAE,EAAE;gBAChC,IAAI,EAAE,CAAC,kBAAkB,CAAC,KAAK,CAAC,EAAE,CAAC;oBACjC,IAAI,gBAAoC,CAAC;oBACzC,IAAI,gBAAwB,CAAC;oBAC7B,IAAI,CAAC;wBACH,gBAAgB,GAAG,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;oBAC1D,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,4CAA4C;wBAC5C,gBAAgB,GAAI,KAAK,CAAC,UAAkB,CAAC,IAAI,IAAI,IAAI,CAAC;oBAC5D,CAAC;oBACD,MAAM,eAAe,GACnB,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC;oBACtD,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,EAAE,WAAW,CAAC,EAAE,EAAE;wBAC9D,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;4BACrC,IAAI,UAAU,CAAC,WAAW,KAAK,IAAI,EAAE,CAAC;gCACpC,gBAAgB,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gCAC7C,gBAAgB,GAAG,UAAU,CAAC,UAAU,CAAC;4BAC3C,CAAC;wBACH,CAAC;oBACH,CAAC,CAAC,CAAC;oBAEH,SAAS,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,gBAAgB,EAAE,gBAAgB,CAAC,CAAC;gBACnE,CAAC;qBAAM,IACL,CAAC,EAAE,CAAC,qBAAqB,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;oBAClE,CAAC,CAAC,EAAE,CAAC,qBAAqB,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,IAAI,EAAE,IAAI,KAAK,IAAI,CAAC;wBAC7D,CAAC,EAAE,CAAC,mBAAmB,CAAC,KAAK,CAAC;4BAC5B,KAAK,CAAC,eAAe,CAAC,YAAY,CAAC,IAAI,CACrC,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,IAAI,KAAK,IAAI,CAC7D,CAAC,CAAC,EACP,CAAC;oBACD,SAAS,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;gBAC/B,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,MAAM,mBAAmB,GAAG,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAC;YACxE,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,IAAI,EAAE,CAAC;YACjD,cAAc,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC,GAAG;gBACpD,IAAI;gBACJ,eAAe,EAAE,IAAI;gBACrB,iBAAiB,EAAE,KAAK;aACzB,CAAC;QACJ,CAAC;IACH,CAAC;IAED,SAAS,mBAAmB,CAAC,IAAqB;QAChD,MAAM,MAAM,GAAG,IAAI,CAAC,MAA0C,CAAC;QAE/D,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAChC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC;YAC/B,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY;gBACvC,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI;gBAC3B,CAAC,CAAC,IAAI,CAAC;YACT,MAAM,cAAc,GAAG,MAAM,EAAE,eAAe;gBAC5C,CAAC,CAAC,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,UAAU,CAAC;gBAC5C,CAAC,CAAC,SAAS,CAAC;YAEd,IAAI,cAAc,EAAE,CAAC;gBACnB,MAAM,gBAAgB,GAAG,cAAc,EAAE,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gBAC9D,MAAM,YAAY,GAChB,MAAM,EAAE,YACT,EAAE,QAAQ,EAAE,IAAI,CACf,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,YAAY,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,YAAY,CACxD,CAAC;gBACF,MAAM,gBAAgB,GAAG,CACvB,YAAY,EAAE,YAAY,IAAI,YAAY,EAAE,IAAI,CACjD,EAAE,IAAI,CAAC;gBAER,mEAAmE;gBACnE,IAAI,QAAgB,CAAC;gBACrB,IAAI,CAAC;oBACH,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;gBACtC,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,+CAA+C;oBAC/C,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ;yBAC3B,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE;wBACV,IAAI,EAAE,CAAC,YAAY,EAAE,CAAC;4BACpB,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,OAAO,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;wBACtD,CAAC;wBACD,OAAO,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;oBACtB,CAAC,CAAC;yBACD,IAAI,CAAC,IAAI,CAAC,CAAC;oBACd,QAAQ,GAAG,KAAK,QAAQ,IAAI,CAAC;gBAC/B,CAAC;gBACD,MAAM,eAAe,GAAG,UAAU,QAAQ,SAAS,cAAc,EAAE,CAAC;gBAEpE,cAAc,CAAC,IAAI,CAAC,GAAG;oBACrB,IAAI,EAAE,eAAe;oBACrB,eAAe,EAAE,IAAI,KAAK,SAAS;oBACnC,iBAAiB,EAAE,KAAK;oBACxB,gBAAgB;oBAChB,gBAAgB;iBACjB,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,MAAM,YAAY,GAAG,UAAU,CAAC,UAAU,CAAC,MAAM,CAC/C,CACE,IAAI,EAImB,EAAE,CACzB,CAAC,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC;oBAC3B,EAAE,CAAC,qBAAqB,CAAC,IAAI,CAAC;oBAC9B,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;oBAC/B,CAAC,CAAC,EAAE,CAAC,qBAAqB,CAAC,IAAI,CAAC;wBAC9B,IAAI,CAAC,IAAI,EAAE,IAAI,KAAK,YAAY,CAAC;wBACjC,CAAC,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC;4BAC3B,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,IAAI,CACpC,CAAC,GAAG,EAAE,EAAE,CACN,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,IAAI,KAAK,YAAY,CAC9D,CAAC;wBACJ,CAAC,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC;4BAC3B,CAAC,IAAI,CAAC,YAAY,EAAE,OAAO,CAAC,UAAU,CAAC,KAAK,YAAY;gCACtD,CAAC,IAAI,CAAC,YAAY,EAAE,aAAa;oCAC/B,EAAE,CAAC,cAAc,CAAC,IAAI,CAAC,YAAY,EAAE,aAAa,CAAC;oCACnD,IAAI,CAAC,YAAY,EAAE,aAAa,CAAC,QAAQ,CAAC,IAAI,CAC5C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,YAAY,CACpC,CAAC,CAAC,CAAC,CAAC,CAChB,CAAC;gBAEF,YAAY,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;oBAC5B,IAAI,gBAAoC,CAAC;oBACzC,IAAI,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;wBACjC,gBAAgB,GAAG,IAAI,CAAC,eAAe;4BACrC,EAAE,OAAO,CAAC,UAAU,CAAC;6BACpB,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;oBAC1B,CAAC;oBACD,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,gBAAgB,CAAC,CAAC;gBACjD,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,UAAU,CAAC,YAAY,CAAC,CAAC,IAAI,EAAE,EAAE;QAC/B,IAAI,EAAE,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;YACnE,IACE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,EAAE,CAAC,UAAU,CAAC,aAAa,CAAC,EACvE,CAAC;gBACD,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,EAAE,IAAI,CAC1C,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,EAAE,CAAC,UAAU,CAAC,cAAc,CACnD,CAAC;gBACF,kBAAkB,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;YAC5C,CAAC;QACH,CAAC;aAAM,IAAI,EAAE,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YAC/D,uBAAuB,CAAC,IAAI,CAAC,CAAC;QAChC,CAAC;aAAM,IAAI,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;YACxC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACtB,IAAI,EAAE,CAAC,cAAc,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;oBACzC,mBAAmB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;gBACzC,CAAC;qBAAM,IAAI,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;oBACnD,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;oBACzC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;gBAC/B,CAAC;YACH,CAAC;iBAAM,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;gBAChC,8CAA8C;gBAC9C,iDAAiD;gBACjD,MAAM,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;gBACjE,MAAM,gBAAgB,GAAG,eAAe,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gBAE9D,mEAAmE;gBACnE,yEAAyE;gBACzE,MAAM,cAAc,GAAG,KAAK,gBAAgB,EAAE,CAAC;gBAE/C,IAAI,IAAY,CAAC;gBACjB,IAAI,CAAC;oBACH,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,IAAI,EAAE,CAAC;gBAC7C,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,GAAG,kBAAkB,gBAAgB,GAAG,CAAC;gBAC/C,CAAC;gBAED,cAAc,CAAC,cAAc,CAAC,GAAG;oBAC/B,IAAI;oBACJ,eAAe,EAAE,KAAK;oBACtB,iBAAiB,EAAE,KAAK;oBACxB,YAAY,EAAE,IAAI;oBAClB,gBAAgB;iBACjB,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,cAAc,CAAC;AACxB,CAAC"}

package/packages/analyze/src/lib/files/analyze/analyzeEntities/prepareDataStructures.js

@@ -1,107 +1,176 @@
 import { awsLog, awsLogDebugLevel } from "../../../../../../../packages/utils/index.js";
 import measureAndReportExecutionTime from "../../../utils/measureAndReportExecutionTime.js";
 import { fillInDirectSchemaGapsAndUnknowns, generateEntityDataStructure, deduplicateFunctionSchemas, getEquivalentSignatureVariables, getFunctionSignature, getReturnValue, getSourceEquivalencies, getUsageEquivalencies, getConditionalUsages, detectWrapperRequirements, } from "../../../../../../../packages/ai/index.js";
+// VERSION MARKER: server-only-detection-v2 (2025-12-28)
+// This version includes automatic detection of "use server" and "server-only" files
+// and runs server-only detection even for cached entities
 /**
  * PHASE 2: Prepare entity data structures
  * Generate data structures for all entities upfront, before graph traversal.
  * This ensures that data structure generation failures don't corrupt the traversal state.
  */
 export default async function prepareEntityDataStructures(allEntities, projectAnalyzer, options) {
+    console.log('CodeYam: prepareEntityDataStructures RUNNING (server-only-detection-v3)');
+    console.log(`CodeYam: prepareEntityDataStructures processing ${allEntities.size} entities`);
+    // Log all entity names for debugging
+    const entityNames = Array.from(allEntities).map((e) => e.name);
+    console.log(`CodeYam: prepareEntityDataStructures entity names: ${entityNames.join(', ')}`);
     // Process entities in parallel for speed
     await Promise.all(Array.from(allEntities).map(async (entity) => {
-        // Skip if already has cached data structure (unless force is set)
+        // Check if already has cached data structure (unless force is set)
         const hasCache = !!entity.metadata?.isolatedDataStructure;
-        if (hasCache && !options.force && !options.forceAll) {
-            return;
-        }
-        try {
-            const fileAnalyzer = projectAnalyzer.getFileAnalyzerByPath(entity.resolvedFilePath ?? entity.filePath);
-            if (!fileAnalyzer) {
-                awsLog(`CodeYam: prepareEntityDataStructures: FileAnalyzer not found`, {
-                    entityFilePath: entity.filePath,
-                    entityName: entity.name,
-                });
-                return;
-            }
-            const dataStructureInfo = await getEntityDataStructureAndFunctionCalls({
-                entity,
-                fileAnalyzer,
-            });
-            if (!dataStructureInfo?.isolatedDataStructure) {
-                // Data structure generation failed - entity will proceed with empty schemas
-                // This is okay - the entity can still be analyzed, just without detailed type info
-                if (['visual', 'library'].includes(entity.entityType)) {
-                    // we'd really expect these to have data structures
-                    awsLog(`CodeYam: prepareEntityDataStructures: Failed to generate data structure for ${entity.filePath}:${entity.name}`);
+        const shouldRegenerateDataStructure = !hasCache || options.force || options.forceAll;
+        // PHASE 1: Generate data structure (only for non-cached entities or when forced)
+        // Wrapped in its own try-catch so PHASE 2 always runs even if this fails
+        if (shouldRegenerateDataStructure) {
+            try {
+                const fileAnalyzer = projectAnalyzer.getFileAnalyzerByPath(entity.resolvedFilePath ?? entity.filePath);
+                if (!fileAnalyzer) {
+                    awsLog(`CodeYam: prepareEntityDataStructures: FileAnalyzer not found`, {
+                        entityFilePath: entity.filePath,
+                        entityName: entity.name,
+                    });
+                    // Still continue to PHASE 2 for server-only detection
                 }
-                return;
-            }
-            const { isolatedDataStructure, functionCalls } = dataStructureInfo;
-            // Store data structure in entity metadata
-            entity.metadata || (entity.metadata = {});
-            // Deep clone dependencySchemas to avoid shared references
-            // We need to clone because getReturnValue() returns direct references to schema objects
-            // and multiple entities might reference the same dependency
-            const clonedDependencySchemas = {};
-            for (const filePath in isolatedDataStructure.dependencySchemas) {
-                clonedDependencySchemas[filePath] = {};
-                for (const entityName in isolatedDataStructure.dependencySchemas[filePath]) {
-                    const depSchema = isolatedDataStructure.dependencySchemas[filePath][entityName];
-                    clonedDependencySchemas[filePath][entityName] = {
-                        signatureSchema: { ...depSchema.signatureSchema },
-                        returnValueSchema: { ...depSchema.returnValueSchema },
-                        usageEquivalencies: depSchema.usageEquivalencies,
-                        sourceEquivalencies: depSchema.sourceEquivalencies,
-                    };
+                else {
+                    const dataStructureInfo = await getEntityDataStructureAndFunctionCalls({
+                        entity,
+                        fileAnalyzer,
+                    });
+                    if (!dataStructureInfo?.isolatedDataStructure) {
+                        // Data structure generation failed - entity will proceed with empty schemas
+                        // This is okay - the entity can still be analyzed, just without detailed type info
+                        if (['visual', 'library'].includes(entity.entityType)) {
+                            // we'd really expect these to have data structures
+                            awsLog(`CodeYam: prepareEntityDataStructures: Failed to generate data structure for ${entity.filePath}:${entity.name}`);
+                        }
+                        // Still continue to PHASE 2 for server-only detection
+                    }
+                    else {
+                        const { isolatedDataStructure, functionCalls, callVariableNames, } = dataStructureInfo;
+                        // Store data structure in entity metadata
+                        entity.metadata || (entity.metadata = {});
+                        // Deep clone dependencySchemas to avoid shared references
+                        // We need to clone because getReturnValue() returns direct references to schema objects
+                        // and multiple entities might reference the same dependency
+                        const clonedDependencySchemas = {};
+                        for (const filePath in isolatedDataStructure.dependencySchemas) {
+                            clonedDependencySchemas[filePath] = {};
+                            for (const entityName in isolatedDataStructure
+                                .dependencySchemas[filePath]) {
+                                const depSchema = isolatedDataStructure.dependencySchemas[filePath][entityName];
+                                clonedDependencySchemas[filePath][entityName] = {
+                                    signatureSchema: { ...depSchema.signatureSchema },
+                                    returnValueSchema: { ...depSchema.returnValueSchema },
+                                    usageEquivalencies: depSchema.usageEquivalencies,
+                                    sourceEquivalencies: depSchema.sourceEquivalencies,
+                                };
+                            }
+                        }
+                        entity.metadata.isolatedDataStructure = {
+                            signatureSchema: deduplicateFunctionSchemas(isolatedDataStructure.signatureSchema),
+                            returnValueSchema: deduplicateFunctionSchemas(isolatedDataStructure.returnValueSchema),
+                            equivalentSignatureVariables: isolatedDataStructure.equivalentSignatureVariables,
+                            dependencySchemas: clonedDependencySchemas,
+                            environmentVariables: isolatedDataStructure.environmentVariables,
+                            // Use enriched conditional usages with source tracing from analysis
+                            conditionalUsages: isolatedDataStructure.conditionalUsages,
+                        };
+                        // Detect wrapper requirements (Context providers, etc.)
+                        const wrapperRequirements = detectWrapperRequirements({
+                            entity,
+                            fileAnalyzer,
+                        });
+                        if (wrapperRequirements.length > 0) {
+                            entity.metadata.wrapperRequirements = wrapperRequirements;
+                        }
+                        // Annotate imported exports with function call information
+                        const importedExports = entity.metadata.importedExports ?? [];
+                        for (const importedExport of importedExports) {
+                            const relevantCalls = functionCalls[importedExport.name];
+                            if (relevantCalls) {
+                                importedExport.calls = relevantCalls;
+                            }
+                            const relevantVariableNames = callVariableNames[importedExport.name];
+                            if (relevantVariableNames) {
+                                importedExport.callVariableNames = relevantVariableNames;
+                            }
+                        }
+                        // Annotate node module imports with function call information
+                        for (const [path, infos] of Object.entries(entity.metadata.nodeModuleImports ?? {})) {
+                            for (const info of infos) {
+                                const relevantCalls = functionCalls[info.name];
+                                if (relevantCalls) {
+                                    info.calls = relevantCalls;
+                                }
+                                const relevantVariableNames = callVariableNames[info.name];
+                                if (relevantVariableNames) {
+                                    info.callVariableNames = relevantVariableNames;
+                                }
+                            }
+                        }
+                    }
                 }
             }
-            entity.metadata.isolatedDataStructure = {
-                signatureSchema: deduplicateFunctionSchemas(isolatedDataStructure.signatureSchema),
-                returnValueSchema: deduplicateFunctionSchemas(isolatedDataStructure.returnValueSchema),
-                equivalentSignatureVariables: isolatedDataStructure.equivalentSignatureVariables,
-                dependencySchemas: clonedDependencySchemas,
-                environmentVariables: isolatedDataStructure.environmentVariables,
-                // Use enriched conditional usages with source tracing from analysis
-                conditionalUsages: isolatedDataStructure.conditionalUsages,
-            };
-            // Detect wrapper requirements (Context providers, etc.)
-            const wrapperRequirements = detectWrapperRequirements({
-                entity,
-                fileAnalyzer,
-            });
-            if (wrapperRequirements.length > 0) {
-                entity.metadata.wrapperRequirements = wrapperRequirements;
+            catch (phase1Error) {
+                console.warn(`CodeYam: prepareEntityDataStructures: PHASE 1 error for ${entity.filePath}:${entity.name}:`, phase1Error);
+                // Continue to PHASE 2 - server-only detection should still run
             }
-            // Annotate imported exports with function call information
-            const importedExports = entity.metadata.importedExports ?? [];
+        }
+        try {
+            // PHASE 2: Server-only detection and isMocked determination
+            // This runs for ALL entities, even those with cached data structures
+            // because server-only detection may have been added after the entity was cached
+            const importedExports = entity.metadata?.importedExports ?? [];
+            console.log(`CodeYam: [${entity.name}] Processing ${importedExports.length} importedExports for isMocked determination (cached: ${hasCache})`);
+            const mockedDependencies = determineMockedDependencies(entity);
             for (const importedExport of importedExports) {
-                const relevantCalls = functionCalls[importedExport.name];
-                if (relevantCalls) {
-                    importedExport.calls = relevantCalls;
+                // Skip if already has a definitive isMocked value (true or false, not null)
+                // But re-check if isMocked is null or undefined - this handles previously cached entities
+                if (importedExport.isMocked === true ||
+                    importedExport.isMocked === false) {
+                    console.log(`CodeYam: [${entity.name}] SKIP ${importedExport.name} - already has isMocked=${importedExport.isMocked}`);
+                    continue;
                 }
-            }
-            // Annotate node module imports with function call information
-            for (const [path, infos] of Object.entries(entity.metadata.nodeModuleImports ?? {})) {
-                for (const info of infos) {
-                    const relevantCalls = functionCalls[info.name];
-                    if (relevantCalls) {
-                        info.calls = relevantCalls;
-                    }
+                // Never mock visual components - they render UI and should use real implementation
+                if (importedExport.entityType === 'visual') {
+                    console.log(`CodeYam: [${entity.name}] SKIP ${importedExport.name} - visual component, setting isMocked=false`);
+                    importedExport.isMocked = false;
+                    continue;
                 }
-            }
-            // Determine and mark mocked dependencies
-            const mockedDependencies = determineMockedDependencies(entity);
-            for (const importedExport of importedExports) {
-                if (importedExport.isMocked === undefined) {
-                    importedExport.isMocked = mockedDependencies.some((dep) => dep.filePath === importedExport.filePath &&
-                        dep.entityName === importedExport.name);
+                // Always mock imports from server-only files ("use server" or import "server-only")
+                // These files contain Server Actions or server-only code that cannot run on the client
+                const importedFilePath = importedExport.resolvedFilePath ?? importedExport.filePath;
+                const importedFileAnalyzer = projectAnalyzer.getFileAnalyzerByPath(importedFilePath);
+                // Use sourceFile.getText() which is always available, unlike file.content which is lazy-loaded
+                const importedFileContent = importedFileAnalyzer?.sourceFile?.getText();
+                const isServerOnly = importedFileContent
+                    ? isServerOnlyFile(importedFileContent)
+                    : false;
+                console.log(`CodeYam: [${entity.name}] CHECK ${importedExport.name} from ${importedFilePath}:`, {
+                    hasFileAnalyzer: !!importedFileAnalyzer,
+                    hasSourceFile: !!importedFileAnalyzer?.sourceFile,
+                    hasFileContent: !!importedFileContent,
+                    contentFirst50Chars: importedFileContent
+                        ?.substring(0, 50)
+                        ?.replace(/\n/g, '\\n'),
+                    isServerOnly,
+                });
+                if (importedFileContent && isServerOnly) {
+                    console.log(`CodeYam: [${entity.name}] ✓ MARKING ${importedExport.name} as isMocked=true (server-only file detected)`);
+                    importedExport.isMocked = true;
+                    continue;
                 }
+                const matchesMockedDep = mockedDependencies.some((dep) => dep.filePath === importedExport.filePath &&
+                    dep.entityName === importedExport.name);
+                importedExport.isMocked = matchesMockedDep;
+                console.log(`CodeYam: [${entity.name}] ${importedExport.name} isMocked=${matchesMockedDep} (from mockedDependencies)`);
             }
             // Browser globals that should always be mocked (they make network calls)
             const alwaysMockedBrowserGlobals = ['fetch', 'EventSource'];
-            for (const [path, infos] of Object.entries(entity.metadata.nodeModuleImports ?? {})) {
+            for (const [path, infos] of Object.entries(entity.metadata?.nodeModuleImports ?? {})) {
                 for (const info of infos) {
-                    if (info.isMocked === undefined) {
+                    if (info.isMocked === undefined || info.isMocked === null) {
                         // Browser globals should always be mocked
                         if (alwaysMockedBrowserGlobals.includes(path)) {
                             info.isMocked = true;
@@ -184,15 +253,71 @@ async function getEntityDataStructureAndFunctionCalls({ entity, fileAnalyzer, })
         conditionalUsages: Object.keys(conditionalUsages).length > 0 ? conditionalUsages : undefined,
     };
     const functionCalls = {};
+    const callVariableNames = {};
     // Note: externalFunctionCalls is needed for callSignature info, which isn't exposed through helpers
     for (const filePath in dependencySchemas) {
         for (const entityName in dependencySchemas[filePath]) {
-            functionCalls[entityName] = dataStructure.externalFunctionCalls
-                .filter((efc) => efc.callSignature.startsWith(entityName))
-                .map((efc) => efc.callSignature);
+            const matchingCalls = dataStructure.externalFunctionCalls.filter((efc) => efc.callSignature.startsWith(entityName));
+            // DEBUG: Log external function calls info for useFetcher
+            if (entityName === 'useFetcher' || entityName.includes('Fetcher')) {
+                console.log('CodeYam DEBUG prepareDataStructures:', JSON.stringify({
+                    entityName,
+                    filePath,
+                    totalExternalFunctionCalls: dataStructure.externalFunctionCalls.length,
+                    matchingCallsCount: matchingCalls.length,
+                    matchingCalls: matchingCalls.map((efc) => ({
+                        name: efc.name,
+                        callSignature: efc.callSignature,
+                        allCallSignatures: efc.allCallSignatures,
+                        receivingVariableNames: efc.receivingVariableNames,
+                        callSignatureToVariable: efc.callSignatureToVariable,
+                    })),
+                }, null, 2));
+            }
+            // Collect all call signatures from all matching efc entries
+            // Each efc may have just one callSignature, or allCallSignatures if merged
+            const allSignatures = [];
+            for (const efc of matchingCalls) {
+                if (efc.allCallSignatures) {
+                    allSignatures.push(...efc.allCallSignatures);
+                }
+                else {
+                    allSignatures.push(efc.callSignature);
+                }
+            }
+            functionCalls[entityName] = allSignatures;
+            // Aggregate callSignatureToVariable maps across all matching calls
+            // This handles the case where tracking data ends up on just one efc entry
+            const aggregatedCallSignatureToVariable = {};
+            const aggregatedReceivingVariableNames = [];
+            for (const efc of matchingCalls) {
+                if (efc.callSignatureToVariable) {
+                    Object.assign(aggregatedCallSignatureToVariable, efc.callSignatureToVariable);
+                }
+                if (efc.receivingVariableNames) {
+                    aggregatedReceivingVariableNames.push(...efc.receivingVariableNames);
+                }
+            }
+            // Extract variable names for each call signature using the aggregated map
+            const variableNames = [];
+            for (const signature of allSignatures) {
+                const variableName = aggregatedCallSignatureToVariable[signature];
+                if (variableName) {
+                    variableNames.push(variableName);
+                }
+            }
+            // If no variables found via aggregated map, fall back to aggregated receivingVariableNames
+            // This handles cases where receivingVariableNames is set but callSignatureToVariable is not
+            if (variableNames.length === 0 &&
+                aggregatedReceivingVariableNames.length > 0) {
+                variableNames.push(...aggregatedReceivingVariableNames);
+            }
+            if (variableNames.length > 0) {
+                callVariableNames[entityName] = variableNames;
+            }
         }
     }
-    return { isolatedDataStructure, functionCalls };
+    return { isolatedDataStructure, functionCalls, callVariableNames };
 }
 /**
  * Helper: Check if a schema contains actual function calls
@@ -208,6 +333,46 @@ function schemaContainsFunctions(schema) {
     }
     return false;
 }
+/**
+ * Helper: Check if ALL function paths in the schema are jsx-components
+ *
+ * A function path is considered a jsx-component if:
+ * - The value is 'jsx-component', OR
+ * - The functionCallReturnValue for that path is 'jsx-component'
+ *
+ * Returns true if ALL function paths are jsx-components (entity should NOT be mocked)
+ * Returns false if there are any real functions (entity may need mocking)
+ */
+function allFunctionsAreJsxComponents(schema) {
+    // Get all function paths (paths ending with () that aren't functionCallReturnValue)
+    const functionPaths = Object.entries(schema ?? {}).filter(([path, value]) => {
+        // Only consider paths that are function calls
+        if (!path.includes('('))
+            return false;
+        // Skip functionCallReturnValue entries - we check those separately
+        if (path.includes('functionCallReturnValue'))
+            return false;
+        // This is a function call path
+        return value === 'function' || value === 'jsx-component';
+    });
+    // If no function paths at all, return false (no functions to check)
+    if (functionPaths.length === 0)
+        return false;
+    // Check if ALL function paths are jsx-components
+    for (const [path, value] of functionPaths) {
+        // If the value itself is jsx-component, skip it
+        if (value === 'jsx-component')
+            continue;
+        // Check if the functionCallReturnValue for this path is jsx-component
+        const returnValuePath = path + '.functionCallReturnValue';
+        if (schema[returnValuePath] === 'jsx-component')
+            continue;
+        // This is a real function that's NOT a jsx-component
+        return false;
+    }
+    // All function paths are jsx-components
+    return true;
+}
 /**
  * Helper: Check if a return value schema has actual return value data (not just function type markers)
  *
@@ -230,12 +395,73 @@ function hasActualReturnValueData(schema) {
     }
     return false;
 }
+/**
+ * Helper: Check if a return value schema contains chained method calls (not just simple function calls)
+ *
+ * A chained method call has a pattern like:
+ * - `z.string()` - method call on object z
+ * - `z.string().cuid2()` - chained method calls
+ * - `client.schema().action()` - chained method calls
+ *
+ * A simple function call is just:
+ * - `FileProvider()` - standalone function call (no dot before parens)
+ *
+ * We detect chained calls by looking for patterns where there's a `.` followed by an identifier
+ * and then `()`. The pattern `.identifier()` indicates a method call, not a standalone function.
+ *
+ * EXCEPTION: React Context's .Provider and .Consumer are built-in components that
+ * look like method calls but aren't. When a component uses <SomeContext.Provider>,
+ * the AST creates paths like "SomeContext.Provider()" which should NOT trigger
+ * the chained method call detection.
+ */
+function hasChainedMethodCalls(schema) {
+    // Pattern: look for ".identifier(" which indicates a method call
+    // This distinguishes "z.string()" (chained) from "FileProvider()" (simple)
+    const chainedMethodPattern = /\.\w+\(/;
+    for (const [path, value] of Object.entries(schema ?? {})) {
+        // Skip JSX components - they look like chained method calls (e.g., Context.Provider())
+        // but they're actually React components used in JSX, not method chains to bypass
+        if (value === 'jsx-component') {
+            continue;
+        }
+        // Only look at paths that contain function calls
+        if (path.includes('(') || value === 'function') {
+            if (chainedMethodPattern.test(path)) {
+                return true;
+            }
+        }
+    }
+    return false;
+}
+/**
+ * Validation function name prefixes that should always be mocked.
+ * These functions validate data using schemas (e.g., Zod) that we can't easily
+ * generate data for. Instead of trying to generate data that passes validation,
+ * we mock the validation function itself to return its input unchanged.
+ */
+const VALIDATION_FUNCTION_PREFIXES = ['validate', 'check', 'verify', 'assert'];
+/**
+ * Check if a function name indicates it's a validation function that should be mocked.
+ * Validation functions are identified by their name starting with common validation prefixes.
+ */
+export function isValidationFunction(name) {
+    const lowerName = name.toLowerCase();
+    return VALIDATION_FUNCTION_PREFIXES.some((prefix) => lowerName.startsWith(prefix));
+}
 /**
  * Helper: Determine which dependencies should be mocked based on whether they contain functions
  *
  * Only entities that are functions or contain functions in their return values need to be mocked.
  * Data constants (like arrays, objects, strings) don't need mocking because they don't have
  * behavior that needs to be replaced.
+ *
+ * Additionally, validation functions (names starting with validate, check, verify, assert)
+ * are always mocked because they use validation schemas we can't easily generate data for.
+ *
+ * Functions/chains with no return value are also mocked because:
+ * - If a function takes parameters but returns nothing, it's doing side effects (validation, logging)
+ * - If a chain like z.string().cuid2() has no functionCallReturnValue, the chain is doing
+ *   operations (like schema building/validation) whose return is meaningless to us
  */
 export function determineMockedDependencies(entity) {
     const mockedEntities = [];
@@ -245,6 +471,7 @@ export function determineMockedDependencies(entity) {
             const dataStructure = dependencySchemas[filePath][entityName];
             const entitySignatureSchema = dataStructure.signatureSchema ?? {};
             const emptySignature = Object.keys(entitySignatureSchema).length === 0;
+            const hasSignature = !emptySignature;
             const complexSignature = Object.values(entitySignatureSchema).some((value) => value === 'object' || value === 'array');
             // Clone before passing to fillInDirectSchemaGapsAndUnknowns because it mutates its input
             const entityReturnValueSchema = fillInDirectSchemaGapsAndUnknowns({
@@ -260,10 +487,25 @@ export function determineMockedDependencies(entity) {
             // it doesn't mean there's actual return data to mock
             const hasReturnValue = hasActualReturnValueData(entityReturnValueSchema);
             const complexReturnValue = Object.values(entityReturnValueSchema ?? {}).some((value) => value === 'object' || value === 'array');
-            const entityShouldBeMocked = (signatureHasFunctions || returnValueHasFunctions) &&
-                hasReturnValue &&
-                ((emptySignature && hasReturnValue) ||
-                    (!complexSignature && complexReturnValue));
+            // Check if this is a validation function that should always be mocked
+            const isValidationFn = isValidationFunction(entityName);
+            // Check if entity has chained method calls but no return value
+            // e.g., z.string().cuid2() or authenticatedActionClient.schema().action()
+            // These chains are doing operations (validation, schema building) we want to bypass
+            // NOTE: Simple function calls like FileProvider() are NOT chained - only method calls
+            // with a pattern like .method() count as chained
+            const hasChainedCallsWithNoReturn = hasChainedMethodCalls(entityReturnValueSchema) && !hasReturnValue;
+            // Check if ALL functions in the return value schema are jsx-components
+            // If so, we should NOT mock this entity - jsx-components like React Context's
+            // Provider/Consumer should use the original, not be mocked
+            const onlyHasJsxComponents = allFunctionsAreJsxComponents(entityReturnValueSchema);
+            const entityShouldBeMocked = !onlyHasJsxComponents &&
+                (isValidationFn ||
+                    hasChainedCallsWithNoReturn ||
+                    ((signatureHasFunctions || returnValueHasFunctions) &&
+                        hasReturnValue &&
+                        ((emptySignature && hasReturnValue) ||
+                            (!complexSignature && complexReturnValue))));
             if (entityShouldBeMocked) {
                 mockedEntities.push({
                     filePath,
@@ -274,4 +516,27 @@ export function determineMockedDependencies(entity) {
     }
     return mockedEntities;
 }
+/**
+ * Check if a file is server-only based on its content.
+ *
+ * A file is considered server-only if it:
+ * 1. Starts with "use server" directive (Server Actions in Next.js)
+ * 2. Contains import "server-only" (explicit server-only marker)
+ *
+ * Server-only files should always be mocked when imported by client components
+ * because they contain code that cannot run in the browser.
+ */
+export function isServerOnlyFile(fileContent) {
+    const trimmed = fileContent.trimStart();
+    // Check for "use server" directive at the start of the file
+    if (trimmed.startsWith('"use server"') ||
+        trimmed.startsWith("'use server'")) {
+        return true;
+    }
+    // Check for server-only import anywhere in the file
+    if (/import\s+["']server-only["']/.test(fileContent)) {
+        return true;
+    }
+    return false;
+}
 //# sourceMappingURL=prepareDataStructures.js.map