@codeyam/codeyam-cli 0.0.0 → 0.1.0-staging.036391e

package/background/src/lib/virtualized/project/getFilesWithEntitiesFromRepo.js

@@ -0,0 +1,161 @@
+import { loadFiles, saveFiles, generateSha, loadEntities, } from "../../../../../packages/database/index.js";
+export default async function getFilesWithEntitiesFromRepo({ project, projectAnalyzer, filePath, silent, requestedEntityShas, }) {
+    const validFiles = [];
+    const sourceFiles = projectAnalyzer.getAllSourceFiles();
+    if (!silent) {
+        console.log(`CodeYam: Enumerating ${sourceFiles.length} source files for ${project.slug}`);
+    }
+    // Filter valid source files
+    const validSourceFiles = sourceFiles.filter((sourceFile) => {
+        if (filePath && sourceFile.path !== filePath)
+            return false;
+        if (sourceFile.absolutePath.indexOf('node_modules') > -1)
+            return false;
+        if (sourceFile.content.length === 0)
+            return false;
+        if (project.metadata.approvedPaths) {
+            const approved = project.metadata.approvedPaths.find((path) => sourceFile.absolutePath.includes(path));
+            if (!approved)
+                return false;
+        }
+        return true;
+    });
+    // Separate files that exist in project vs need to be loaded from DB
+    const filesNeedingDbLookup = [];
+    const sourceFileMap = new Map();
+    for (const sourceFile of validSourceFiles) {
+        sourceFileMap.set(sourceFile.path, sourceFile);
+        const existingFile = project.files.find((f) => f.path === sourceFile.path);
+        if (!existingFile) {
+            filesNeedingDbLookup.push(sourceFile);
+        }
+        else {
+            existingFile.content = sourceFile.content;
+            validFiles.push(existingFile);
+        }
+    }
+    // Bulk load missing files from DB
+    let filesFromDb = [];
+    if (filesNeedingDbLookup.length > 0) {
+        const filePaths = filesNeedingDbLookup.map((sf) => sf.path);
+        filesFromDb = await loadFiles({
+            projectId: project.id,
+            filePaths,
+        });
+    }
+    // Identify files that still need to be created
+    const filesToCreate = [];
+    for (const sourceFile of filesNeedingDbLookup) {
+        const existingDbFile = filesFromDb.find((f) => f.path === sourceFile.path);
+        if (existingDbFile) {
+            existingDbFile.content = sourceFile.content;
+            validFiles.push(existingDbFile);
+        }
+        else {
+            const name = sourceFile.path.split('/').pop();
+            filesToCreate.push({
+                name,
+                path: sourceFile.path,
+                project_id: project.id.toString(),
+            });
+        }
+    }
+    // Bulk save new files
+    let newFiles = [];
+    if (filesToCreate.length > 0) {
+        newFiles = await saveFiles(filesToCreate);
+        for (const newFile of newFiles) {
+            const sourceFile = sourceFileMap.get(newFile.path);
+            if (sourceFile) {
+                newFile.content = sourceFile.content;
+                validFiles.push(newFile);
+            }
+        }
+    }
+    for (const file of validFiles) {
+        const fileAnalyzer = projectAnalyzer.getFileAnalyzer(file);
+        file.entities = [];
+        const allEntityNodes = fileAnalyzer.getAllEntityNodesOnlyAliases();
+        for (const entityName in allEntityNodes) {
+            const entityType = fileAnalyzer.getEntityType(entityName);
+            const entityCode = fileAnalyzer.getEntityCode(entityName);
+            const sha = generateSha(file.path, entityName, entityCode);
+            file.entities.push({
+                name: entityName,
+                entityType,
+                code: entityCode,
+                sha,
+                fileId: file.id,
+                projectId: project.id,
+                filePath: file.path,
+                metadata: {
+                    notExported: !fileAnalyzer.isExported(entityName),
+                    namedExport: !fileAnalyzer.isDefaultExport(entityName),
+                },
+            });
+        }
+    }
+    // If specific entity SHAs were requested, load them from the database
+    // This handles cases where the file has been modified since the entity was created
+    if (requestedEntityShas && requestedEntityShas.length > 0) {
+        const generatedShas = new Set(validFiles.flatMap((f) => f.entities.map((e) => e.sha)));
+        const missingShas = requestedEntityShas.filter((sha) => !generatedShas.has(sha));
+        if (missingShas.length > 0) {
+            if (!silent) {
+                console.log(`CodeYam: Loading ${missingShas.length} requested entities from database that don't match current file content`);
+            }
+            const dbEntities = await loadEntities({
+                projectId: project.id,
+                shas: missingShas,
+            });
+            if (dbEntities && dbEntities.length > 0) {
+                // Group entities by file path
+                const entitiesByFilePath = new Map();
+                for (const entity of dbEntities) {
+                    if (!entity.filePath)
+                        continue;
+                    if (!entitiesByFilePath.has(entity.filePath)) {
+                        entitiesByFilePath.set(entity.filePath, []);
+                    }
+                    entitiesByFilePath.get(entity.filePath).push(entity);
+                }
+                // Add entities to their files
+                for (const [filePath, entities] of entitiesByFilePath.entries()) {
+                    let file = validFiles.find((f) => f.path === filePath);
+                    // If file doesn't exist in validFiles, we may need to load it
+                    if (!file) {
+                        const filesFromDb = await loadFiles({
+                            projectId: project.id,
+                            filePaths: [filePath],
+                        });
+                        if (filesFromDb && filesFromDb.length > 0) {
+                            file = filesFromDb[0];
+                            const sourceFile = sourceFileMap.get(filePath);
+                            if (sourceFile) {
+                                file.content = sourceFile.content;
+                            }
+                            file.entities = [];
+                            validFiles.push(file);
+                        }
+                    }
+                    if (file) {
+                        // Add the DB entities to the file
+                        file.entities = file.entities || [];
+                        for (const entity of entities) {
+                            // Only add if not already present
+                            if (!file.entities.some((e) => e.sha === entity.sha)) {
+                                file.entities.push(entity);
+                            }
+                        }
+                    }
+                }
+            }
+        }
+    }
+    project.files = validFiles;
+    if (!silent) {
+        console.log('CodeYam: Valid Files', project.files.length, 'Valid Entities', project.files.reduce((acc, file) => acc + file.entities.length, 0));
+    }
+    return validFiles;
+}
+//# sourceMappingURL=getFilesWithEntitiesFromRepo.js.map

package/background/src/lib/virtualized/project/getFilesWithEntitiesFromRepo.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"getFilesWithEntitiesFromRepo.js","sourceRoot":"","sources":["../../../../../../../background/src/lib/virtualized/project/getFilesWithEntitiesFromRepo.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,SAAS,EACT,SAAS,EACT,WAAW,EACX,YAAY,GACb,MAAM,mBAAmB,CAAC;AAY3B,MAAM,CAAC,OAAO,CAAC,KAAK,UAAU,4BAA4B,CAAC,EACzD,OAAO,EACP,eAAe,EACf,QAAQ,EACR,MAAM,EACN,mBAAmB,GACc;IACjC,MAAM,UAAU,GAAW,EAAE,CAAC;IAE9B,MAAM,WAAW,GAAG,eAAe,CAAC,iBAAiB,EAAE,CAAC;IACxD,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,CAAC,GAAG,CACT,wBAAwB,WAAW,CAAC,MAAM,qBAAqB,OAAO,CAAC,IAAI,EAAE,CAC9E,CAAC;IACJ,CAAC;IAED,4BAA4B;IAC5B,MAAM,gBAAgB,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,UAAU,EAAE,EAAE;QACzD,IAAI,QAAQ,IAAI,UAAU,CAAC,IAAI,KAAK,QAAQ;YAAE,OAAO,KAAK,CAAC;QAC3D,IAAI,UAAU,CAAC,YAAY,CAAC,OAAO,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;YAAE,OAAO,KAAK,CAAC;QACvE,IAAI,UAAU,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QAClD,IAAI,OAAO,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;YACnC,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAC5D,UAAU,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,CACvC,CAAC;YACF,IAAI,CAAC,QAAQ;gBAAE,OAAO,KAAK,CAAC;QAC9B,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;IAEH,oEAAoE;IACpE,MAAM,oBAAoB,GAA4B,EAAE,CAAC;IACzD,MAAM,aAAa,GAAG,IAAI,GAAG,EAAwC,CAAC;IAEtE,KAAK,MAAM,UAAU,IAAI,gBAAgB,EAAE,CAAC;QAC1C,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAC/C,MAAM,YAAY,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,IAAI,CAAC,CAAC;QAC3E,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,oBAAoB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACxC,CAAC;aAAM,CAAC;YACN,YAAY,CAAC,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;YAC1C,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAED,kCAAkC;IAClC,IAAI,WAAW,GAAW,EAAE,CAAC;IAC7B,IAAI,oBAAoB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACpC,MAAM,SAAS,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;QAC5D,WAAW,GAAG,MAAM,SAAS,CAAC;YAC5B,SAAS,EAAE,OAAO,CAAC,EAAE;YACrB,SAAS;SACV,CAAC,CAAC;IACL,CAAC;IAED,+CAA+C;IAC/C,MAAM,aAAa,GAId,EAAE,CAAC;IACR,KAAK,MAAM,UAAU,IAAI,oBAAoB,EAAE,CAAC;QAC9C,MAAM,cAAc,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,IAAI,CAAC,CAAC;QAC3E,IAAI,cAAc,EAAE,CAAC;YACnB,cAAc,CAAC,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;YAC5C,UAAU,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAClC,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC;YAC9C,aAAa,CAAC,IAAI,CAAC;gBACjB,IAAI;gBACJ,IAAI,EAAE,UAAU,CAAC,IAAI;gBACrB,UAAU,EAAE,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE;aAClC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,sBAAsB;IACtB,IAAI,QAAQ,GAAW,EAAE,CAAC;IAC1B,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7B,QAAQ,GAAG,MAAM,SAAS,CAAC,aAAa,CAAC,CAAC;QAC1C,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,MAAM,UAAU,GAAG,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACnD,IAAI,UAAU,EAAE,CAAC;gBACf,OAAO,CAAC,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;gBACrC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QAC9B,MAAM,YAAY,GAAG,eAAe,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAE3D,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC;QACnB,MAAM,cAAc,GAAG,YAAY,CAAC,4BAA4B,EAAE,CAAC;QACnE,KAAK,MAAM,UAAU,IAAI,cAAc,EAAE,CAAC;YACxC,MAAM,UAAU,GAAG,YAAY,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;YAC1D,MAAM,UAAU,GAAG,YAAY,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;YAC1D,MAAM,GAAG,GAAG,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;YAC3D,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACjB,IAAI,EAAE,UAAU;gBAChB,UAAU;gBACV,IAAI,EAAE,UAAU;gBAChB,GAAG;gBACH,MAAM,EAAE,IAAI,CAAC,EAAE;gBACf,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,QAAQ,EAAE,IAAI,CAAC,IAAI;gBACnB,QAAQ,EAAE;oBACR,WAAW,EAAE,CAAC,YAAY,CAAC,UAAU,CAAC,UAAU,CAAC;oBACjD,WAAW,EAAE,CAAC,YAAY,CAAC,eAAe,CAAC,UAAU,CAAC;iBACvD;aACF,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,sEAAsE;IACtE,mFAAmF;IACnF,IAAI,mBAAmB,IAAI,mBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC1D,MAAM,aAAa,GAAG,IAAI,GAAG,CAC3B,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CACxD,CAAC;QACF,MAAM,WAAW,GAAG,mBAAmB,CAAC,MAAM,CAC5C,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,CACjC,CAAC;QAEF,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO,CAAC,GAAG,CACT,oBAAoB,WAAW,CAAC,MAAM,yEAAyE,CAChH,CAAC;YACJ,CAAC;YAED,MAAM,UAAU,GAAG,MAAM,YAAY,CAAC;gBACpC,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,IAAI,EAAE,WAAW;aAClB,CAAC,CAAC;YAEH,IAAI,UAAU,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxC,8BAA8B;gBAC9B,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAA6B,CAAC;gBAChE,KAAK,MAAM,MAAM,IAAI,UAAU,EAAE,CAAC;oBAChC,IAAI,CAAC,MAAM,CAAC,QAAQ;wBAAE,SAAS;oBAC/B,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;wBAC7C,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;oBAC9C,CAAC;oBACD,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACxD,CAAC;gBAED,8BAA8B;gBAC9B,KAAK,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,IAAI,kBAAkB,CAAC,OAAO,EAAE,EAAE,CAAC;oBAChE,IAAI,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;oBAEvD,8DAA8D;oBAC9D,IAAI,CAAC,IAAI,EAAE,CAAC;wBACV,MAAM,WAAW,GAAG,MAAM,SAAS,CAAC;4BAClC,SAAS,EAAE,OAAO,CAAC,EAAE;4BACrB,SAAS,EAAE,CAAC,QAAQ,CAAC;yBACtB,CAAC,CAAC;wBAEH,IAAI,WAAW,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;4BAC1C,IAAI,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;4BACtB,MAAM,UAAU,GAAG,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;4BAC/C,IAAI,UAAU,EAAE,CAAC;gCACf,IAAI,CAAC,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;4BACpC,CAAC;4BACD,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC;4BACnB,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;wBACxB,CAAC;oBACH,CAAC;oBAED,IAAI,IAAI,EAAE,CAAC;wBACT,kCAAkC;wBAClC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC;wBACpC,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;4BAC9B,kCAAkC;4BAClC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,KAAK,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;gCACrD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;4BAC7B,CAAC;wBACH,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,CAAC,KAAK,GAAG,UAAU,CAAC;IAC3B,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,CAAC,GAAG,CACT,sBAAsB,EACtB,OAAO,CAAC,KAAK,CAAC,MAAM,EACpB,gBAAgB,EAChB,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,CACnE,CAAC;IACJ,CAAC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC"}

package/background/src/lib/virtualized/project/getScenarioPagePath.js

@@ -0,0 +1,62 @@
+import { ProjectFramework } from "../../../../../packages/types/index.js";
+import { findWebAppForFile, PROJECT_RELATIVE_PATH, safeFileName, } from "../../../../../packages/utils/index.js";
+import * as path from 'path';
+export default function getScenarioPagePath({ project, analysis, scenarioName, framework, }) {
+    const scenarioSlug = safeFileName(scenarioName);
+    const relativePath = PROJECT_RELATIVE_PATH;
+    switch (framework) {
+        case ProjectFramework.CodeYam:
+            return getPagePathForCodeYam(project, analysis, scenarioSlug, relativePath);
+        case ProjectFramework.Remix:
+        case ProjectFramework.CRA:
+            return getPagePathForRemix(project, analysis, scenarioSlug, relativePath);
+        case ProjectFramework.Next:
+            return getPagePathForNext(project, analysis, scenarioSlug, relativePath);
+        case ProjectFramework.NextPages:
+            return getPagePathForNextPages(project, analysis, scenarioSlug, relativePath);
+        default:
+            throw new Error(`Unsupported framework ${framework}`);
+    }
+}
+// Helper function to get the full app path (webapp path + appDirectory)
+const getFullAppPath = (project, analysisFilePath) => {
+    const webapp = findWebAppForFile(analysisFilePath, project.metadata?.webapps ?? []);
+    if (webapp) {
+        const webappPath = webapp.path === '.' ? '' : webapp.path;
+        const appDirectory = webapp.appDirectory ?? 'app';
+        return webappPath ? path.join(webappPath, appDirectory) : appDirectory;
+    }
+    // Fallback
+    return 'app';
+};
+const getPagePathForRemix = (project, analysis, scenarioSlug, relativePath) => {
+    const entitySlug = safeFileName(analysis.entityName);
+    const fullAppPath = getFullAppPath(project, analysis.filePath);
+    const routesDir = `${relativePath}/${fullAppPath}/routes`;
+    const pageName = `static.${project.slug}.${analysis.id}.${entitySlug}.${scenarioSlug}.default`; // e.g. static.123.button-123-314-342.19202928282289.default
+    const pagePath = `${routesDir}/${pageName}.tsx`; // e.g. ./codeyam-cli/src/webserver/app/routes/static.button-123-314-342.19202928282289.default.tsx
+    return pagePath;
+};
+const getPagePathForCodeYam = (project, analysis, scenarioSlug, relativePath) => {
+    const entitySlug = safeFileName(analysis.entityName);
+    const fullAppPath = getFullAppPath(project, analysis.filePath);
+    const routesDir = `${relativePath}/${fullAppPath}/routes`;
+    const pageName = `static.${project.slug}.${analysis.id}.${entitySlug}.${scenarioSlug}.default`; // e.g. static.123.button-123-314-342.19202928282289.default
+    const pagePath = `${routesDir}/${pageName}.tsx`; // e.g. ./codeyam-cli/src/webserver/app/routes/static.button-123-314-342.19202928282289.default.tsx
+    return pagePath;
+};
+const getPagePathForNext = (project, analysis, scenarioSlug, relativePath) => {
+    const entitySlug = safeFileName(analysis.entityName);
+    const fullAppPath = getFullAppPath(project, analysis.filePath);
+    const pageName = `${analysis.id}/${entitySlug}/${scenarioSlug}`; // e.g. button-123-314-342/19202928282289/123
+    const pagePath = `${relativePath}/${fullAppPath}/static/${project.slug}/${pageName}/default/page.tsx`; // e.g. ./apps/web/app/static/123/button-123-314-342/19202928282289/default/page.tsx
+    return pagePath;
+};
+const getPagePathForNextPages = (project, analysis, scenarioSlug, relativePath) => {
+    const entitySlug = safeFileName(analysis.entityName);
+    const fullAppPath = getFullAppPath(project, analysis.filePath);
+    const pageName = `${analysis.id}/${entitySlug}/${scenarioSlug}`; // e.g. button-123-314-342/19202928282289/123
+    const pagePath = `${relativePath}/${fullAppPath}/static/${project.slug}/${pageName}/default/index.tsx`; // e.g. ./frontend/pages/static/123/button-123-314-342/19202928282289/default/index.tsx
+    return pagePath;
+};
+//# sourceMappingURL=getScenarioPagePath.js.map

package/background/src/lib/virtualized/project/getScenarioPagePath.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"getScenarioPagePath.js","sourceRoot":"","sources":["../../../../../../../background/src/lib/virtualized/project/getScenarioPagePath.ts"],"names":[],"mappings":"AAAA,OAAO,EAAqB,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AACrE,OAAO,EACL,iBAAiB,EACjB,qBAAqB,EACrB,YAAY,GACb,MAAM,gBAAgB,CAAC;AACxB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAS7B,MAAM,CAAC,OAAO,UAAU,mBAAmB,CAAC,EAC1C,OAAO,EACP,QAAQ,EACR,YAAY,EACZ,SAAS,GACe;IACxB,MAAM,YAAY,GAAG,YAAY,CAAC,YAAY,CAAC,CAAC;IAEhD,MAAM,YAAY,GAAG,qBAAqB,CAAC;IAC3C,QAAQ,SAAS,EAAE,CAAC;QAClB,KAAK,gBAAgB,CAAC,OAAO;YAC3B,OAAO,qBAAqB,CAC1B,OAAO,EACP,QAAQ,EACR,YAAY,EACZ,YAAY,CACb,CAAC;QACJ,KAAK,gBAAgB,CAAC,KAAK,CAAC;QAC5B,KAAK,gBAAgB,CAAC,GAAG;YACvB,OAAO,mBAAmB,CAAC,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,YAAY,CAAC,CAAC;QAC5E,KAAK,gBAAgB,CAAC,IAAI;YACxB,OAAO,kBAAkB,CAAC,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,YAAY,CAAC,CAAC;QAC3E,KAAK,gBAAgB,CAAC,SAAS;YAC7B,OAAO,uBAAuB,CAC5B,OAAO,EACP,QAAQ,EACR,YAAY,EACZ,YAAY,CACb,CAAC;QACJ;YACE,MAAM,IAAI,KAAK,CAAC,yBAAyB,SAAS,EAAE,CAAC,CAAC;IAC1D,CAAC;AACH,CAAC;AAED,wEAAwE;AACxE,MAAM,cAAc,GAAG,CAAC,OAAgB,EAAE,gBAAwB,EAAU,EAAE;IAC5E,MAAM,MAAM,GAAG,iBAAiB,CAC9B,gBAAgB,EAChB,OAAO,CAAC,QAAQ,EAAE,OAAO,IAAI,EAAE,CAChC,CAAC;IACF,IAAI,MAAM,EAAE,CAAC;QACX,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,KAAK,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC;QAC1D,MAAM,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,KAAK,CAAC;QAClD,OAAO,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC;IACzE,CAAC;IACD,WAAW;IACX,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AAEF,MAAM,mBAAmB,GAAG,CAC1B,OAAgB,EAChB,QAAkB,EAClB,YAAoB,EACpB,YAAoB,EACZ,EAAE;IACV,MAAM,UAAU,GAAG,YAAY,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;IACrD,MAAM,WAAW,GAAG,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAC/D,MAAM,SAAS,GAAG,GAAG,YAAY,IAAI,WAAW,SAAS,CAAC;IAE1D,MAAM,QAAQ,GAAG,UAAU,OAAO,CAAC,IAAI,IAAI,QAAQ,CAAC,EAAE,IAAI,UAAU,IAAI,YAAY,UAAU,CAAC,CAAC,4DAA4D;IAC5J,MAAM,QAAQ,GAAG,GAAG,SAAS,IAAI,QAAQ,MAAM,CAAC,CAAC,mGAAmG;IACpJ,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC;AAEF,MAAM,qBAAqB,GAAG,CAC5B,OAAgB,EAChB,QAAkB,EAClB,YAAoB,EACpB,YAAoB,EACZ,EAAE;IACV,MAAM,UAAU,GAAG,YAAY,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;IACrD,MAAM,WAAW,GAAG,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAC/D,MAAM,SAAS,GAAG,GAAG,YAAY,IAAI,WAAW,SAAS,CAAC;IAE1D,MAAM,QAAQ,GAAG,UAAU,OAAO,CAAC,IAAI,IAAI,QAAQ,CAAC,EAAE,IAAI,UAAU,IAAI,YAAY,UAAU,CAAC,CAAC,4DAA4D;IAC5J,MAAM,QAAQ,GAAG,GAAG,SAAS,IAAI,QAAQ,MAAM,CAAC,CAAC,mGAAmG;IACpJ,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC;AAEF,MAAM,kBAAkB,GAAG,CACzB,OAAgB,EAChB,QAAkB,EAClB,YAAoB,EACpB,YAAoB,EACZ,EAAE;IACV,MAAM,UAAU,GAAG,YAAY,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;IACrD,MAAM,WAAW,GAAG,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAC/D,MAAM,QAAQ,GAAG,GAAG,QAAQ,CAAC,EAAE,IAAI,UAAU,IAAI,YAAY,EAAE,CAAC,CAAC,6CAA6C;IAC9G,MAAM,QAAQ,GAAG,GAAG,YAAY,IAAI,WAAW,WAAW,OAAO,CAAC,IAAI,IAAI,QAAQ,mBAAmB,CAAC,CAAC,oFAAoF;IAC3L,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC;AAEF,MAAM,uBAAuB,GAAG,CAC9B,OAAgB,EAChB,QAAkB,EAClB,YAAoB,EACpB,YAAoB,EACZ,EAAE;IACV,MAAM,UAAU,GAAG,YAAY,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;IACrD,MAAM,WAAW,GAAG,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAC/D,MAAM,QAAQ,GAAG,GAAG,QAAQ,CAAC,EAAE,IAAI,UAAU,IAAI,YAAY,EAAE,CAAC,CAAC,6CAA6C;IAC9G,MAAM,QAAQ,GAAG,GAAG,YAAY,IAAI,WAAW,WAAW,OAAO,CAAC,IAAI,IAAI,QAAQ,oBAAoB,CAAC,CAAC,uFAAuF;IAC/L,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC"}

package/background/src/lib/virtualized/project/getScenarioUrl.js

@@ -0,0 +1,87 @@
+import { ProjectFramework, } from "../../../../../packages/types/index.js";
+import getScenarioPagePaths from "./getScenarioPagePath.js";
+import { frameworkRouteFileNameToRoute, isFrameworkRoute, PROJECT_RELATIVE_PATH, getAppDirectoryForFile, } from "../../../../../packages/utils/index.js";
+/**
+ * Extract route params from scenario data for URL generation.
+ * Handles array values (for catch-all routes) by joining them.
+ *
+ * Scenario data can be stored in two locations:
+ * - scenario.data.argumentsData (primary)
+ * - scenario.metadata.data.argumentsData (fallback)
+ *
+ * @param scenario - The scenario containing mock data
+ * @returns A record of param names to string values, or undefined if no params
+ */
+function extractRouteParamsFromScenario(scenario) {
+    // Check both possible locations for argumentsData
+    const argumentsData = scenario.data?.argumentsData ?? scenario.metadata?.data?.argumentsData;
+    if (!Array.isArray(argumentsData) || argumentsData.length === 0) {
+        return undefined;
+    }
+    const params = argumentsData[0]?.params;
+    if (!params || typeof params !== 'object') {
+        return undefined;
+    }
+    // Convert all values to strings or string arrays
+    const result = {};
+    for (const [key, value] of Object.entries(params)) {
+        if (Array.isArray(value)) {
+            result[key] = value.map(String);
+        }
+        else if (value !== null && value !== undefined) {
+            result[key] = String(value);
+        }
+    }
+    return Object.keys(result).length > 0 ? result : undefined;
+}
+/**
+ * Get the URL path for a scenario (the path that goes in the iframe src)
+ * This is the same logic used in generateCodeYamPage.ts to determine the iframe src
+ */
+export default function getScenarioUrl({ project, file, analysis, scenario, framework, }) {
+    const routePath = getScenarioPagePaths({
+        project,
+        analysis,
+        scenarioName: scenario.name,
+        framework,
+    });
+    let path = '';
+    switch (framework) {
+        case ProjectFramework.CodeYam:
+        case ProjectFramework.CRA:
+        case ProjectFramework.Remix:
+            // the routePath looks like: relative/path/dashboard/app/routes/static.123.c054b7ee-3961-4697-8354-886062a95a9d.default.default.tsx
+            // the path should look like: static/123/c054b7ee-3961-4697-8354-886062a95a9d/default/default
+            path = routePath.split('/').pop().split('.').slice(0, -1).join('/');
+            break;
+        case ProjectFramework.Next:
+            // the routePath looks like: relative/path/app/123/static/c054b7ee-3961-4697-8354-886062a95a9d/default/page.tsx
+            // the path should look like: static/123/c054b7ee-3961-4697-8354-886062a95a9d/default/0
+            const appDirectory = getAppDirectoryForFile(file.path, project.metadata?.webapps);
+            path = routePath
+                .replace(PROJECT_RELATIVE_PATH + '/' + appDirectory + '/', '')
+                .split('/')
+                .slice(0, -1)
+                .join('/');
+            break;
+        case ProjectFramework.NextPages:
+            // the routePath looks like: relative/path/app/123/static/c054b7ee-3961-4697-8354-886062a95a9d/default/page.tsx
+            // the path should look like: static/123/c054b7ee-3961-4697-8354-886062a95a9d/default/0
+            const pagesDirectory = getAppDirectoryForFile(file.path, project.metadata?.webapps);
+            path = routePath
+                .replace(PROJECT_RELATIVE_PATH + '/' + pagesDirectory + '/', '')
+                .split('/')
+                .slice(0, -1)
+                .join('/');
+            break;
+        default:
+            break;
+    }
+    if (isFrameworkRoute(file, analysis.entity, framework)) {
+        // Extract route params from scenario mock data for URL generation
+        const routeParams = extractRouteParamsFromScenario(scenario);
+        path = frameworkRouteFileNameToRoute(project.slug, file, analysis, framework, scenario.name, routeParams);
+    }
+    return `/${path}`;
+}
+//# sourceMappingURL=getScenarioUrl.js.map

package/background/src/lib/virtualized/project/getScenarioUrl.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"getScenarioUrl.js","sourceRoot":"","sources":["../../../../../../../background/src/lib/virtualized/project/getScenarioUrl.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,gBAAgB,GAGjB,MAAM,gBAAgB,CAAC;AACxB,OAAO,oBAAoB,MAAM,uBAAuB,CAAC;AACzD,OAAO,EACL,6BAA6B,EAC7B,gBAAgB,EAChB,qBAAqB,EACrB,sBAAsB,GACvB,MAAM,gBAAgB,CAAC;AAUxB;;;;;;;;;;GAUG;AACH,SAAS,8BAA8B,CACrC,QAAkB;IAElB,kDAAkD;IAClD,MAAM,aAAa,GACjB,QAAQ,CAAC,IAAI,EAAE,aAAa,IAAI,QAAQ,CAAC,QAAQ,EAAE,IAAI,EAAE,aAAa,CAAC;IAEzE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAChE,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,MAAM,GAAI,aAAa,CAAC,CAAC,CAA6B,EAAE,MAAM,CAAC;IACrE,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC1C,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,iDAAiD;IACjD,MAAM,MAAM,GAAsC,EAAE,CAAC;IACrD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CACvC,MAAiC,CAClC,EAAE,CAAC;QACF,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzB,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAClC,CAAC;aAAM,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACjD,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;AAC7D,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,OAAO,UAAU,cAAc,CAAC,EACrC,OAAO,EACP,IAAI,EACJ,QAAQ,EACR,QAAQ,EACR,SAAS,GACU;IACnB,MAAM,SAAS,GAAG,oBAAoB,CAAC;QACrC,OAAO;QACP,QAAQ;QACR,YAAY,EAAE,QAAQ,CAAC,IAAI;QAC3B,SAAS;KACV,CAAC,CAAC;IAEH,IAAI,IAAI,GAAG,EAAE,CAAC;IACd,QAAQ,SAAS,EAAE,CAAC;QAClB,KAAK,gBAAgB,CAAC,OAAO,CAAC;QAC9B,KAAK,gBAAgB,CAAC,GAAG,CAAC;QAC1B,KAAK,gBAAgB,CAAC,KAAK;YACzB,mIAAmI;YACnI,6FAA6F;YAC7F,IAAI,GAAG,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrE,MAAM;QACR,KAAK,gBAAgB,CAAC,IAAI;YACxB,+GAA+G;YAC/G,uFAAuF;YACvF,MAAM,YAAY,GAAG,sBAAsB,CACzC,IAAI,CAAC,IAAI,EACT,OAAO,CAAC,QAAQ,EAAE,OAAO,CAC1B,CAAC;YACF,IAAI,GAAG,SAAS;iBACb,OAAO,CAAC,qBAAqB,GAAG,GAAG,GAAG,YAAY,GAAG,GAAG,EAAE,EAAE,CAAC;iBAC7D,KAAK,CAAC,GAAG,CAAC;iBACV,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;iBACZ,IAAI,CAAC,GAAG,CAAC,CAAC;YACb,MAAM;QACR,KAAK,gBAAgB,CAAC,SAAS;YAC7B,+GAA+G;YAC/G,uFAAuF;YACvF,MAAM,cAAc,GAAG,sBAAsB,CAC3C,IAAI,CAAC,IAAI,EACT,OAAO,CAAC,QAAQ,EAAE,OAAO,CAC1B,CAAC;YACF,IAAI,GAAG,SAAS;iBACb,OAAO,CAAC,qBAAqB,GAAG,GAAG,GAAG,cAAc,GAAG,GAAG,EAAE,EAAE,CAAC;iBAC/D,KAAK,CAAC,GAAG,CAAC;iBACV,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;iBACZ,IAAI,CAAC,GAAG,CAAC,CAAC;YACb,MAAM;QACR;YACE,MAAM;IACV,CAAC;IAED,IAAI,gBAAgB,CAAC,IAAI,EAAE,QAAQ,CAAC,MAAM,EAAE,SAAS,CAAC,EAAE,CAAC;QACvD,kEAAkE;QAClE,MAAM,WAAW,GAAG,8BAA8B,CAAC,QAAQ,CAAC,CAAC;QAC7D,IAAI,GAAG,6BAA6B,CAClC,OAAO,CAAC,IAAI,EACZ,IAAI,EACJ,QAAQ,EACR,SAAS,EACT,QAAQ,CAAC,IAAI,EACb,WAAW,CACZ,CAAC;IACJ,CAAC;IAED,OAAO,IAAI,IAAI,EAAE,CAAC;AACpB,CAAC"}

package/background/src/lib/virtualized/project/loadReadyToBeCaptured.js

@@ -0,0 +1,217 @@
+import { discoverDirectDependencies } from "../../../../../packages/analyze/index.js";
+import { loadAnalysis, loadEntities, loadReadyToBeCapturedAnalyses, } from "../../../../../packages/database/index.js";
+const DB_CONCURRENCY = 1;
+async function mapWithConcurrency(items, fn, concurrency) {
+    const results = new Array(items.length);
+    let index = 0;
+    const workers = Array.from({ length: Math.min(concurrency, items.length) }, async () => {
+        while (index < items.length) {
+            const i = index++;
+            results[i] = await fn(items[i]);
+        }
+    });
+    await Promise.all(workers);
+    return results;
+}
+/**
+ * Recursively gather all entities referenced in importedExports of entities already in context.
+ * This fills the gap where discoverDirectDependencies skips mocked/type/implicit imports,
+ * but writeScenarioComponents still needs those entities in the context.
+ */
+async function gatherAllReferencedEntities(context, project) {
+    const processedKeys = new Set();
+    let iterationCount = 0;
+    const MAX_ITERATIONS = 20; // Safety limit to prevent infinite loops
+    const getAllEntities = () => Object.values(context.getMutableEntityMap()).flatMap(Object.values);
+    while (iterationCount < MAX_ITERATIONS) {
+        iterationCount++;
+        // Collect all (filePath, entityName) pairs from importedExports of entities in context
+        const missingEntities = new Map();
+        for (const entity of getAllEntities()) {
+            for (const ie of entity.metadata?.importedExports ?? []) {
+                const filePath = ie.resolvedFilePath ?? ie.filePath;
+                const entityName = ie.resolvedName ?? ie.name;
+                const key = `${filePath}:${entityName}`;
+                // Skip if already in context or already processed
+                if (context.getEntity(filePath, entityName) || processedKeys.has(key)) {
+                    continue;
+                }
+                missingEntities.set(key, { filePath, name: entityName });
+            }
+        }
+        if (missingEntities.size === 0) {
+            break; // No more missing entities
+        }
+        console.log(`CodeYam: Entity gathering iteration ${iterationCount}: Loading ${missingEntities.size} missing entities`);
+        // Mark all as processed before loading to prevent re-processing
+        for (const key of missingEntities.keys()) {
+            processedKeys.add(key);
+        }
+        // Group by file path for batch loading
+        const byFilePath = new Map();
+        for (const { filePath, name } of missingEntities.values()) {
+            if (!byFilePath.has(filePath)) {
+                byFilePath.set(filePath, []);
+            }
+            byFilePath.get(filePath).push(name);
+        }
+        // Batch load entities by file path (concurrency-limited to avoid exhausting DB connections)
+        const results = await mapWithConcurrency(Array.from(byFilePath.entries()), async ([filePath, names]) => {
+            const entities = await loadEntities({
+                projectId: project.id,
+                filePaths: [filePath],
+            });
+            if (!entities)
+                return [];
+            // Filter to only the entities we need (loadEntities returns all for the file)
+            // Special handling: if we requested 'default' and got a default export entity,
+            // include it even though the name doesn't match (entities are stored by actual name, not 'default')
+            return entities.filter((e) => {
+                if (names.includes(e.name)) {
+                    return true;
+                }
+                // If we requested 'default' and this is a default export, include it
+                if (names.includes('default') &&
+                    e.metadata &&
+                    !e.metadata.notExported &&
+                    !e.metadata.namedExport) {
+                    return true;
+                }
+                return false;
+            });
+        }, DB_CONCURRENCY);
+        const loadedEntities = results.flat();
+        // Add loaded entities to context
+        for (const entity of loadedEntities) {
+            context.addEntity(entity);
+            // Also register under 'default' alias if it's a default export
+            if (entity.metadata &&
+                !entity.metadata.notExported &&
+                !entity.metadata.namedExport) {
+                context.addEntity(entity, 'default');
+            }
+        }
+        console.log(`CodeYam: Loaded ${loadedEntities.length} entities in iteration ${iterationCount}`);
+    }
+    if (iterationCount >= MAX_ITERATIONS) {
+        console.log(`CodeYam Warning: Entity gathering reached max iterations (${MAX_ITERATIONS}), some entities may be missing`);
+    }
+    console.log(`CodeYam: Entity gathering complete after ${iterationCount} iteration(s), context has ${getAllEntities().length} entities`);
+}
+/**
+ * Load the minimal set of analyses required to successfully run scenario capture.
+ * This includes the root analyses (with scenarios) plus all recursive mock dependencies.
+ */
+export default async function loadReadyToBeCaptured({ project, branchCommitSha, analysisIds, context, fileStore, }) {
+    console.log('CodeYam: Loading ready to be captured analyses...');
+    // Discovery call: Load the initial set of ready-to-be-captured analyses (with scenarios but no mocks)
+    const readyToBeCapturedAnalyses = await loadReadyToBeCapturedAnalyses({
+        projectId: project.id,
+        branchCommitSha: branchCommitSha,
+        ids: analysisIds,
+    });
+    if (!readyToBeCapturedAnalyses || readyToBeCapturedAnalyses.length === 0) {
+        return {
+            readyToBeCaptured: [],
+        };
+    }
+    console.log('CodeYam: Found ready to be captured analyses', readyToBeCapturedAnalyses.length, readyToBeCapturedAnalyses.map((a) => ({
+        id: a.id,
+        filePath: a.filePath,
+        entityName: a.entityName,
+        scenarioCount: a.scenarios?.length ?? 0,
+    })));
+    const allRequiredAnalyses = new Map();
+    // Start with just the discovered analyses - these need to be loaded with mocks
+    const stillNeeded = new Set(readyToBeCapturedAnalyses.map((a) => `${a.filePath}:${a.entityName}`));
+    console.log(`CodeYam: Loading ${stillNeeded.size} analyses with mock dependencies...`);
+    // Load analyses one by one, discovering new dependencies as we go
+    let foundNewDependencies = true;
+    let isFirstBatch = true;
+    while (foundNewDependencies && stillNeeded.size > 0) {
+        foundNewDependencies = false;
+        const currentBatch = Array.from(stillNeeded);
+        // Load analyses for current batch (concurrency-limited to avoid exhausting DB connections)
+        const results = await mapWithConcurrency(currentBatch, async (keyString) => {
+            const [filePath, entityName] = keyString.split(':');
+            // Find the file ID for this file path - use fileStore for O(1) lookup when available
+            const file = fileStore
+                ? fileStore.getByPath(filePath)
+                : project.files.find((f) => f.path === filePath);
+            if (!file) {
+                console.log(`CodeYam Warning: File not found in project: ${filePath} for key ${keyString}`);
+                return null;
+            }
+            try {
+                const analysis = await loadAnalysis({
+                    projectId: project.id,
+                    fileId: file.id,
+                    entityName,
+                    includeScenarios: isFirstBatch, // Only include scenarios for the root analyses
+                });
+                if (analysis) {
+                    console.log(`CodeYam: Loaded analysis ${filePath}:${entityName}`);
+                    return { key: keyString, analysis };
+                }
+                else {
+                    console.log(`CodeYam Warning: Analysis not found: ${filePath}:${entityName}`);
+                    return null;
+                }
+            }
+            catch (error) {
+                console.log(`CodeYam Error loading analysis ${filePath}:${entityName}:`, error);
+                return null;
+            }
+        }, DB_CONCURRENCY);
+        // Process loaded analyses and find new dependencies
+        for (const result of results) {
+            if (!result)
+                continue;
+            const { key, analysis } = result;
+            allRequiredAnalyses.set(key, analysis);
+            stillNeeded.delete(key);
+            // Use shared dependency discovery logic
+            const newDeps = discoverDirectDependencies(analysis);
+            for (const dep of newDeps) {
+                if (!allRequiredAnalyses.has(dep) && !stillNeeded.has(dep)) {
+                    stillNeeded.add(dep);
+                    foundNewDependencies = true;
+                }
+            }
+        }
+        // After first batch, we don't need scenarios for dependency analyses
+        isFirstBatch = false;
+    }
+    // Final check - warn about any missing analyses but don't fail
+    if (stillNeeded.size > 0) {
+        console.log('CodeYam Warning: Some required analyses could not be loaded:', Array.from(stillNeeded));
+    }
+    console.log(`CodeYam: Loaded ${allRequiredAnalyses.size} total analyses for scenario capture`);
+    // Populate the provided analysis context with analyses and their direct entities
+    for (const analysis of allRequiredAnalyses.values()) {
+        // Add the entity if it exists
+        if (analysis.entity) {
+            context.addEntity(analysis.entity);
+        }
+        // Add the analysis
+        context.addAnalysis(analysis, analysis.filePath, analysis.entityName);
+    }
+    // Recursively gather ALL entities referenced in importedExports.
+    // This is necessary because discoverDirectDependencies only finds entities that need
+    // ANALYSES loaded (skipping mocked imports, type imports, etc.), but writeScenarioComponents
+    // needs ALL entities in the context for import path rewriting and recursion.
+    await gatherAllReferencedEntities(context, project);
+    // Extract the root analyses with scenarios from allRequiredAnalyses (they were reloaded with mocks)
+    const finalReadyToBeCaptured = readyToBeCapturedAnalyses.map((originalAnalysis) => {
+        const key = `${originalAnalysis.filePath}:${originalAnalysis.entityName}`;
+        const reloadedAnalysis = allRequiredAnalyses.get(key);
+        if (!reloadedAnalysis) {
+            throw new Error(`Internal error: Failed to reload analysis with mocks for ${key}. This indicates a bug in loadReadyToBeCaptured.`);
+        }
+        return reloadedAnalysis;
+    });
+    return {
+        readyToBeCaptured: finalReadyToBeCaptured,
+    };
+}
+//# sourceMappingURL=loadReadyToBeCaptured.js.map

package/background/src/lib/virtualized/project/loadReadyToBeCaptured.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"loadReadyToBeCaptured.js","sourceRoot":"","sources":["../../../../../../../background/src/lib/virtualized/project/loadReadyToBeCaptured.ts"],"names":[],"mappings":"AAAA,OAAO,EAAmB,0BAA0B,EAAE,MAAM,kBAAkB,CAAC;AAC/E,OAAO,EACL,YAAY,EACZ,YAAY,EACZ,6BAA6B,GAC9B,MAAM,mBAAmB,CAAC;AAI3B,MAAM,cAAc,GAAG,CAAC,CAAC;AAEzB,KAAK,UAAU,kBAAkB,CAC/B,KAAU,EACV,EAA2B,EAC3B,WAAmB;IAEnB,MAAM,OAAO,GAAQ,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAC7C,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CACxB,EAAE,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,KAAK,CAAC,MAAM,CAAC,EAAE,EAC/C,KAAK,IAAI,EAAE;QACT,OAAO,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;YAC5B,MAAM,CAAC,GAAG,KAAK,EAAE,CAAC;YAClB,OAAO,CAAC,CAAC,CAAC,GAAG,MAAM,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC;IACH,CAAC,CACF,CAAC;IACF,MAAM,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAC3B,OAAO,OAAO,CAAC;AACjB,CAAC;AAUD;;;;GAIG;AACH,KAAK,UAAU,2BAA2B,CACxC,OAAwB,EACxB,OAAgB;IAEhB,MAAM,aAAa,GAAG,IAAI,GAAG,EAAU,CAAC;IACxC,IAAI,cAAc,GAAG,CAAC,CAAC;IACvB,MAAM,cAAc,GAAG,EAAE,CAAC,CAAC,yCAAyC;IAEpE,MAAM,cAAc,GAAG,GAAG,EAAE,CAC1B,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAEtE,OAAO,cAAc,GAAG,cAAc,EAAE,CAAC;QACvC,cAAc,EAAE,CAAC;QAEjB,uFAAuF;QACvF,MAAM,eAAe,GAAG,IAAI,GAAG,EAG5B,CAAC;QAEJ,KAAK,MAAM,MAAM,IAAI,cAAc,EAAE,EAAE,CAAC;YACtC,KAAK,MAAM,EAAE,IAAI,MAAM,CAAC,QAAQ,EAAE,eAAe,IAAI,EAAE,EAAE,CAAC;gBACxD,MAAM,QAAQ,GAAG,EAAE,CAAC,gBAAgB,IAAI,EAAE,CAAC,QAAQ,CAAC;gBACpD,MAAM,UAAU,GAAG,EAAE,CAAC,YAAY,IAAI,EAAE,CAAC,IAAI,CAAC;gBAC9C,MAAM,GAAG,GAAG,GAAG,QAAQ,IAAI,UAAU,EAAE,CAAC;gBAExC,kDAAkD;gBAClD,IAAI,OAAO,CAAC,SAAS,CAAC,QAAQ,EAAE,UAAU,CAAC,IAAI,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;oBACtE,SAAS;gBACX,CAAC;gBAED,eAAe,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC;QAED,IAAI,eAAe,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YAC/B,MAAM,CAAC,2BAA2B;QACpC,CAAC;QAED,OAAO,CAAC,GAAG,CACT,uCAAuC,cAAc,aAAa,eAAe,CAAC,IAAI,mBAAmB,CAC1G,CAAC;QAEF,gEAAgE;QAChE,KAAK,MAAM,GAAG,IAAI,eAAe,CAAC,IAAI,EAAE,EAAE,CAAC;YACzC,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACzB,CAAC;QAED,uCAAuC;QACvC,MAAM,UAAU,GAAG,IAAI,GAAG,EAAoB,CAAC;QAC/C,KAAK,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,eAAe,CAAC,MAAM,EAAE,EAAE,CAAC;YAC1D,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC9B,UAAU,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;YAC/B,CAAC;YACD,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvC,CAAC;QAED,4FAA4F;QAC5F,MAAM,OAAO,GAAG,MAAM,kBAAkB,CACtC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,EAChC,KAAK,EAAE,CAAC,QAAQ,EAAE,KAAK,CAAC,EAAE,EAAE;YAC1B,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC;gBAClC,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,SAAS,EAAE,CAAC,QAAQ,CAAC;aACtB,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ;gBAAE,OAAO,EAAE,CAAC;YAEzB,8EAA8E;YAC9E,+EAA+E;YAC/E,oGAAoG;YACpG,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;gBAC3B,IAAI,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC3B,OAAO,IAAI,CAAC;gBACd,CAAC;gBACD,qEAAqE;gBACrE,IACE,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC;oBACzB,CAAC,CAAC,QAAQ;oBACV,CAAC,CAAC,CAAC,QAAQ,CAAC,WAAW;oBACvB,CAAC,CAAC,CAAC,QAAQ,CAAC,WAAW,EACvB,CAAC;oBACD,OAAO,IAAI,CAAC;gBACd,CAAC;gBACD,OAAO,KAAK,CAAC;YACf,CAAC,CAAC,CAAC;QACL,CAAC,EACD,cAAc,CACf,CAAC;QACF,MAAM,cAAc,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;QAEtC,iCAAiC;QACjC,KAAK,MAAM,MAAM,IAAI,cAAc,EAAE,CAAC;YACpC,OAAO,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAE1B,+DAA+D;YAC/D,IACE,MAAM,CAAC,QAAQ;gBACf,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW;gBAC5B,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,EAC5B,CAAC;gBACD,OAAO,CAAC,SAAS,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;YACvC,CAAC;QACH,CAAC;QAED,OAAO,CAAC,GAAG,CACT,mBAAmB,cAAc,CAAC,MAAM,0BAA0B,cAAc,EAAE,CACnF,CAAC;IACJ,CAAC;IAED,IAAI,cAAc,IAAI,cAAc,EAAE,CAAC;QACrC,OAAO,CAAC,GAAG,CACT,6DAA6D,cAAc,iCAAiC,CAC7G,CAAC;IACJ,CAAC;IAED,OAAO,CAAC,GAAG,CACT,4CAA4C,cAAc,8BAA8B,cAAc,EAAE,CAAC,MAAM,WAAW,CAC3H,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,OAAO,CAAC,KAAK,UAAU,qBAAqB,CAAC,EAClD,OAAO,EACP,eAAe,EACf,WAAW,EACX,OAAO,EACP,SAAS,GACiB;IAG1B,OAAO,CAAC,GAAG,CAAC,mDAAmD,CAAC,CAAC;IAEjE,sGAAsG;IACtG,MAAM,yBAAyB,GAAG,MAAM,6BAA6B,CAAC;QACpE,SAAS,EAAE,OAAO,CAAC,EAAE;QACrB,eAAe,EAAE,eAAe;QAChC,GAAG,EAAE,WAAW;KACjB,CAAC,CAAC;IAEH,IAAI,CAAC,yBAAyB,IAAI,yBAAyB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzE,OAAO;YACL,iBAAiB,EAAE,EAAE;SACtB,CAAC;IACJ,CAAC;IAED,OAAO,CAAC,GAAG,CACT,8CAA8C,EAC9C,yBAAyB,CAAC,MAAM,EAChC,yBAAyB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACpC,EAAE,EAAE,CAAC,CAAC,EAAE;QACR,QAAQ,EAAE,CAAC,CAAC,QAAQ;QACpB,UAAU,EAAE,CAAC,CAAC,UAAU;QACxB,aAAa,EAAE,CAAC,CAAC,SAAS,EAAE,MAAM,IAAI,CAAC;KACxC,CAAC,CAAC,CACJ,CAAC;IAEF,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAoB,CAAC;IAExD,+EAA+E;IAC/E,MAAM,WAAW,GAAG,IAAI,GAAG,CACzB,yBAAyB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,UAAU,EAAE,CAAC,CACtE,CAAC;IAEF,OAAO,CAAC,GAAG,CACT,oBAAoB,WAAW,CAAC,IAAI,qCAAqC,CAC1E,CAAC;IAEF,kEAAkE;IAClE,IAAI,oBAAoB,GAAG,IAAI,CAAC;IAChC,IAAI,YAAY,GAAG,IAAI,CAAC;IACxB,OAAO,oBAAoB,IAAI,WAAW,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;QACpD,oBAAoB,GAAG,KAAK,CAAC;QAC7B,MAAM,YAAY,GAAG,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAE7C,2FAA2F;QAC3F,MAAM,OAAO,GAAG,MAAM,kBAAkB,CACtC,YAAY,EACZ,KAAK,EAAE,SAAS,EAAE,EAAE;YAClB,MAAM,CAAC,QAAQ,EAAE,UAAU,CAAC,GAAG,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAEpD,qFAAqF;YACrF,MAAM,IAAI,GAAG,SAAS;gBACpB,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,QAAQ,CAAC;gBAC/B,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;YACnD,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,OAAO,CAAC,GAAG,CACT,+CAA+C,QAAQ,YAAY,SAAS,EAAE,CAC/E,CAAC;gBACF,OAAO,IAAI,CAAC;YACd,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC;oBAClC,SAAS,EAAE,OAAO,CAAC,EAAE;oBACrB,MAAM,EAAE,IAAI,CAAC,EAAE;oBACf,UAAU;oBACV,gBAAgB,EAAE,YAAY,EAAE,+CAA+C;iBAChF,CAAC,CAAC;gBAEH,IAAI,QAAQ,EAAE,CAAC;oBACb,OAAO,CAAC,GAAG,CAAC,4BAA4B,QAAQ,IAAI,UAAU,EAAE,CAAC,CAAC;oBAClE,OAAO,EAAE,GAAG,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC;gBACtC,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,GAAG,CACT,wCAAwC,QAAQ,IAAI,UAAU,EAAE,CACjE,CAAC;oBACF,OAAO,IAAI,CAAC;gBACd,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,GAAG,CACT,kCAAkC,QAAQ,IAAI,UAAU,GAAG,EAC3D,KAAK,CACN,CAAC;gBACF,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC,EACD,cAAc,CACf,CAAC;QAEF,oDAAoD;QACpD,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,IAAI,CAAC,MAAM;gBAAE,SAAS;YAEtB,MAAM,EAAE,GAAG,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC;YACjC,mBAAmB,CAAC,GAAG,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;YACvC,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAExB,wCAAwC;YACxC,MAAM,OAAO,GAAG,0BAA0B,CAAC,QAAQ,CAAC,CAAC;YACrD,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;gBAC1B,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;oBAC3D,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;oBACrB,oBAAoB,GAAG,IAAI,CAAC;gBAC9B,CAAC;YACH,CAAC;QACH,CAAC;QAED,qEAAqE;QACrE,YAAY,GAAG,KAAK,CAAC;IACvB,CAAC;IAED,+DAA+D;IAC/D,IAAI,WAAW,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;QACzB,OAAO,CAAC,GAAG,CACT,8DAA8D,EAC9D,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CACxB,CAAC;IACJ,CAAC;IAED,OAAO,CAAC,GAAG,CACT,mBAAmB,mBAAmB,CAAC,IAAI,sCAAsC,CAClF,CAAC;IAEF,iFAAiF;IACjF,KAAK,MAAM,QAAQ,IAAI,mBAAmB,CAAC,MAAM,EAAE,EAAE,CAAC;QACpD,8BAA8B;QAC9B,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;YACpB,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACrC,CAAC;QAED,mBAAmB;QACnB,OAAO,CAAC,WAAW,CAAC,QAAQ,EAAE,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC;IACxE,CAAC;IAED,iEAAiE;IACjE,qFAAqF;IACrF,6FAA6F;IAC7F,6EAA6E;IAC7E,MAAM,2BAA2B,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAEpD,oGAAoG;IACpG,MAAM,sBAAsB,GAAG,yBAAyB,CAAC,GAAG,CAC1D,CAAC,gBAAgB,EAAE,EAAE;QACnB,MAAM,GAAG,GAAG,GAAG,gBAAgB,CAAC,QAAQ,IAAI,gBAAgB,CAAC,UAAU,EAAE,CAAC;QAC1E,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACtD,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CACb,4DAA4D,GAAG,kDAAkD,CAClH,CAAC;QACJ,CAAC;QACD,OAAO,gBAAgB,CAAC;IAC1B,CAAC,CACF,CAAC;IAEF,OAAO;QACL,iBAAiB,EAAE,sBAAsB;KAC1C,CAAC;AACJ,CAAC"}