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

package/analyzer-template/project/writeScenarioComponents.ts

@@ -0,0 +1,3036 @@
+import {
+  Analysis,
+  Entity,
+  EntityType,
+  File,
+  Scenario,
+  Project,
+  ProjectFramework,
+  Mock,
+  ReadonlyEntity,
+  DeepReadonly,
+} from '~codeyam/types';
+import writeFile from '../common/writeFile';
+import { ProjectAnalyzer, ImmutableAnalysisContext } from '~codeyam/analyze';
+import {
+  isFrameworkRoute,
+  getFrameworkRoutePath,
+  safeFileName,
+  PROJECT_RELATIVE_PATH,
+} from '~codeyam/utils';
+import { getRelativePath, safeFolder } from '~codeyam/generate';
+import constructMockCode from './constructMockCode';
+import { splitOutsideParenthesesAndArrays } from '~codeyam/ai';
+import * as fs from 'fs';
+import * as path from 'path';
+import ts from 'typescript';
+import { LazyFileStore } from './LazyFileStore';
+import { applyServerOnlyMocks } from './serverOnlyModules';
+
+// Debug timing helper for tracking where time is spent
+const DEBUG_TIMING = process.env.DEBUG_WRITE_SCENARIO === 'true';
+let debugStartTime: number;
+let debugLastTime: number;
+
+// Timeout protection to prevent infinite hangs
+const WRITE_SCENARIO_TIMEOUT_MS = parseInt(
+  process.env.WRITE_SCENARIO_TIMEOUT_MS || '300000', // Default 5 minutes
+  10,
+);
+
+class WriteScenarioTimeoutError extends Error {
+  constructor(operation: string, timeoutMs: number) {
+    super(
+      `WriteScenarioComponents timed out after ${timeoutMs}ms during: ${operation}`,
+    );
+    this.name = 'WriteScenarioTimeoutError';
+  }
+}
+
+async function withTimeout<T>(
+  operation: string,
+  promise: Promise<T>,
+  timeoutMs: number = WRITE_SCENARIO_TIMEOUT_MS,
+): Promise<T> {
+  let timeoutId: NodeJS.Timeout | undefined;
+
+  const timeoutPromise = new Promise<never>((_, reject) => {
+    timeoutId = setTimeout(() => {
+      reject(new WriteScenarioTimeoutError(operation, timeoutMs));
+    }, timeoutMs);
+  });
+
+  try {
+    return await Promise.race([promise, timeoutPromise]);
+  } finally {
+    if (timeoutId) clearTimeout(timeoutId);
+  }
+}
+
+function debugLog(message: string, extra?: Record<string, unknown>): void {
+  if (!DEBUG_TIMING) return;
+  const now = Date.now();
+  if (!debugStartTime) {
+    debugStartTime = now;
+    debugLastTime = now;
+  }
+  const elapsed = now - debugStartTime;
+  const delta = now - debugLastTime;
+  debugLastTime = now;
+  console.log(
+    `[WriteScenario +${elapsed}ms Δ${delta}ms] ${message}`,
+    extra ? JSON.stringify(extra, null, 2) : '',
+  );
+}
+
+function resetDebugTiming(): void {
+  debugStartTime = 0;
+  debugLastTime = 0;
+}
+
+/**
+ * Find the end position of the last import/export-from statement using TypeScript AST.
+ * This is more reliable than regex for handling multiline imports, comments, etc.
+ *
+ * @param content - The file content to analyze
+ * @returns The position after the last import statement, or 0 if no imports found
+ */
+function findEndOfImports(content: string): number {
+  try {
+    // Use temp.tsx to enable JSX parsing - otherwise TypeScript may misparse
+    // JSX content containing the word "import" (e.g., "Entities that import this")
+    // as an import statement, causing mock code to be inserted in the wrong location.
+    const sourceFile = ts.createSourceFile(
+      'temp.tsx',
+      content,
+      ts.ScriptTarget.Latest,
+      true,
+      ts.ScriptKind.TSX,
+    );
+
+    let lastImportEnd = 0;
+
+    // Visit all top-level statements to find import/export declarations
+    for (const statement of sourceFile.statements) {
+      if (ts.isImportDeclaration(statement)) {
+        // Get the end position including any trailing semicolon/newline
+        lastImportEnd = statement.end;
+      } else if (
+        ts.isExportDeclaration(statement) &&
+        statement.moduleSpecifier
+      ) {
+        // Handle "export { x } from 'module'" style re-exports
+        lastImportEnd = statement.end;
+      }
+    }
+
+    return lastImportEnd;
+  } catch (error) {
+    // If parsing fails, fall back to 0 (will append at end)
+    console.warn('CodeYam: Failed to parse file for import detection:', error);
+    return 0;
+  }
+}
+
+/**
+ * Escape special regex characters in a string so it can be used as a literal pattern.
+ * This prevents characters like . * + ? ^ $ { } ( ) | [ ] \ from being interpreted as regex metacharacters.
+ */
+function escapeRegExp(str: string): string {
+  return str.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+}
+
+/**
+ * Remove a named import from file content using TypeScript AST.
+ * Handles both regular imports (`EntityName`) and type-only imports (`type EntityName`).
+ *
+ * @param fileContent - The file content to modify
+ * @param entityName - The name of the entity to remove from imports
+ * @returns The modified file content with the entity removed from imports
+ */
+function removeNamedImportAst(fileContent: string, entityName: string): string {
+  try {
+    const sourceFile = ts.createSourceFile(
+      'temp.tsx',
+      fileContent,
+      ts.ScriptTarget.Latest,
+      true,
+      ts.ScriptKind.TSX,
+    );
+
+    const replacements: { start: number; end: number; replacement: string }[] =
+      [];
+
+    for (const statement of sourceFile.statements) {
+      if (!ts.isImportDeclaration(statement)) continue;
+      if (!statement.importClause?.namedBindings) continue;
+      if (!ts.isNamedImports(statement.importClause.namedBindings)) continue;
+
+      const namedImports = statement.importClause.namedBindings;
+      const elements = namedImports.elements;
+
+      // Find the element that matches our entity name
+      const matchingIndex = elements.findIndex(
+        (el) => el.name.text === entityName,
+      );
+      if (matchingIndex === -1) continue;
+
+      // Check if there's a default import (e.g., `import DefaultName, { NamedImport } from '...'`)
+      const hasDefaultImport = !!statement.importClause.name;
+
+      // If this is the only named import AND there's no default import, remove the entire statement
+      if (elements.length === 1 && !hasDefaultImport) {
+        // Find the end including any trailing newline
+        let end = statement.getEnd();
+        const afterStatement = fileContent.slice(end);
+        const trailingNewline = afterStatement.match(/^\r?\n/);
+        if (trailingNewline) {
+          end += trailingNewline[0].length;
+        }
+        replacements.push({
+          start: statement.getStart(sourceFile),
+          end,
+          replacement: '',
+        });
+        continue;
+      }
+
+      // Otherwise, rebuild the import without this element
+      const remainingElements = elements.filter((_, i) => i !== matchingIndex);
+
+      // Get the module specifier
+      const moduleSpecifier = statement.moduleSpecifier;
+      if (!ts.isStringLiteral(moduleSpecifier)) continue;
+
+      // Preserve import type modifier if present
+      const importTypePrefix = statement.importClause.isTypeOnly ? 'type ' : '';
+
+      // Get the default import name if present
+      const defaultImportName = statement.importClause.name?.text;
+
+      let newImport: string;
+
+      if (remainingElements.length === 0) {
+        // All named imports were removed, but there's a default import to preserve
+        // (we only get here when hasDefaultImport is true, because otherwise we'd have
+        // removed the whole statement at the elements.length === 1 check above)
+        newImport = `import ${defaultImportName} from ${moduleSpecifier.getText(sourceFile)};`;
+      } else {
+        // Build the new named imports string
+        const newNamedImports = remainingElements
+          .map((el) => {
+            const isTypeOnly = el.isTypeOnly;
+            const name = el.name.text;
+            const propertyName = el.propertyName?.text;
+            if (propertyName) {
+              return isTypeOnly
+                ? `type ${propertyName} as ${name}`
+                : `${propertyName} as ${name}`;
+            }
+            return isTypeOnly ? `type ${name}` : name;
+          })
+          .join(', ');
+
+        // Build the new import statement, preserving default import if present
+        const defaultImportPrefix = defaultImportName
+          ? `${defaultImportName}, `
+          : '';
+        newImport = `import ${importTypePrefix}${defaultImportPrefix}{ ${newNamedImports} } from ${moduleSpecifier.getText(sourceFile)};`;
+      }
+
+      replacements.push({
+        start: statement.getStart(sourceFile),
+        end: statement.getEnd(),
+        replacement: newImport,
+      });
+    }
+
+    // Apply replacements in reverse order to preserve positions
+    let result = fileContent;
+    replacements.sort((a, b) => b.start - a.start);
+    for (const { start, end, replacement } of replacements) {
+      result = result.slice(0, start) + replacement + result.slice(end);
+    }
+
+    return result;
+  } catch (error) {
+    console.warn('[removeNamedImportAst] Failed to parse file:', error);
+    return fileContent; // Return original content on error
+  }
+}
+
+/**
+ * Map nested dist paths to src paths.
+ * Some build tools create nested structures like:
+ *   packages/X/dist/X/src/Y.d.ts
+ * when the source is at:
+ *   packages/X/src/Y.ts
+ * This function normalizes such paths to the src location.
+ */
+function mapNestedDistToSrc(filePath: string): string {
+  // Pattern: packages/X/dist/X/src/Y -> packages/X/src/Y
+  const nestedDistMatch = filePath.match(
+    /^(packages\/[^/]+)\/dist\/[^/]+\/src\/(.+)$/,
+  );
+  if (nestedDistMatch) {
+    const [, packagePath, relativePath] = nestedDistMatch;
+    return `${packagePath}/src/${relativePath}`
+      .replace(/\.d\.ts$/, '.ts')
+      .replace(/\.js$/, '.ts');
+  }
+  return filePath;
+}
+
+interface WriteScenarioComponentsArgs {
+  project: Project;
+  file: File;
+  entity: ReadonlyEntity;
+  rootAnalysis: Analysis;
+  scenario: Scenario;
+  context: ImmutableAnalysisContext;
+  projectAnalyzer: ProjectAnalyzer;
+  framework: ProjectFramework;
+  mocksDir: string;
+  rootFile?: File;
+  namespaceMocks?: (Mock & {
+    analysisId: string;
+    localFilePath: string;
+    resolvedFilePath: string;
+  })[];
+  writtenScenarioComponents?: { [key: string]: string[] };
+  fileStore?: LazyFileStore;
+  /**
+   * When a default export needs to be accessible as a named export (e.g., when
+   * the parent imports it as `{ Name }` via an index re-export), pass the name here.
+   * A re-export statement `export { default as Name };` will be added to the output.
+   */
+  exportAsNamed?: string;
+}
+
+type EnhancedImportedExport = Omit<
+  Entity['metadata']['importedExports'][0],
+  'calls' | 'entityType' | 'callVariableNames'
+> & {
+  analysisId: string;
+  isNodeModule?: boolean;
+  entityType?: EntityType; // Optional because nodeModuleImports don't have it
+  calls?: readonly string[]; // Optional - used for Zod schema detection heuristic
+  callVariableNames?: readonly string[]; // Variable names receiving the return value of each call
+};
+
+function hasMockStructure(
+  importedExport: EnhancedImportedExport,
+  fileAnalyses: DeepReadonly<Analysis[]>,
+) {
+  if (
+    !fileAnalyses?.some(
+      (a) => a.entity.metadata?.isolatedDataStructure?.dependencySchemas,
+    )
+  ) {
+    return false;
+  }
+
+  return !!constructMockCode(
+    importedExport.name,
+    fileAnalyses.find(
+      (a) =>
+        !!a.metadata?.mergedDataStructure?.dependencySchemas?.[
+          importedExport.filePath
+        ]?.[importedExport.name],
+    )?.metadata?.mergedDataStructure?.dependencySchemas,
+    importedExport.entityType,
+  );
+}
+
+function isIndexPath(filePath: string) {
+  if (!filePath) return false;
+  const fileName = filePath.split('/').pop();
+  return fileName.startsWith('index.');
+}
+
+/**
+ * Convert .d.ts type declaration content to stub implementations.
+ * .d.ts files only have type declarations which don't provide runtime exports.
+ * This function converts them to actual stub implementations.
+ *
+ * @param content - The .d.ts file content
+ * @param entityName - The name of the entity we're generating for (used to ensure it's exported)
+ * @returns The converted content with stub implementations
+ */
+function convertDtsToStubs(content: string, entityName: string): string {
+  let result = content;
+
+  // Handle object types FIRST (before simple types) since they span multiple lines
+  // and contain semicolons that would incorrectly terminate the simple type regex
+
+  // Convert "export declare const NAME: { ... };" (object type) to "export const NAME = {} as any;"
+  result = result.replace(
+    /export\s+declare\s+const\s+(\w+)\s*:\s*\{[^}]*\}\s*;/gs,
+    'export const $1 = {} as any;',
+  );
+
+  // Convert "export declare let NAME: { ... };" (object type) to "export let NAME = {} as any;"
+  result = result.replace(
+    /export\s+declare\s+let\s+(\w+)\s*:\s*\{[^}]*\}\s*;/gs,
+    'export let $1 = {} as any;',
+  );
+
+  // Convert "export declare var NAME: { ... };" (object type) to "export var NAME = {} as any;"
+  result = result.replace(
+    /export\s+declare\s+var\s+(\w+)\s*:\s*\{[^}]*\}\s*;/gs,
+    'export var $1 = {} as any;',
+  );
+
+  // Now handle simple types (non-object types)
+
+  // Convert "export declare const NAME: TYPE;" to "export const NAME = {} as any;"
+  result = result.replace(
+    /export\s+declare\s+const\s+(\w+)\s*:\s*[^;]+;/g,
+    'export const $1 = {} as any;',
+  );
+
+  // Convert "export declare let NAME: TYPE;" to "export let NAME = {} as any;"
+  result = result.replace(
+    /export\s+declare\s+let\s+(\w+)\s*:\s*[^;]+;/g,
+    'export let $1 = {} as any;',
+  );
+
+  // Convert "export declare var NAME: TYPE;" to "export var NAME = {} as any;"
+  result = result.replace(
+    /export\s+declare\s+var\s+(\w+)\s*:\s*[^;]+;/g,
+    'export var $1 = {} as any;',
+  );
+
+  // Convert "export declare function NAME(...): TYPE;" to "export function NAME() { return {} as any; }"
+  result = result.replace(
+    /export\s+declare\s+function\s+(\w+)\s*\([^)]*\)\s*:\s*[^;]+;/g,
+    'export function $1(...args: any[]) { return {} as any; }',
+  );
+
+  // Convert "export declare class NAME { ... }" to "export class NAME {}"
+  // This is tricky because class bodies can span multiple lines
+  result = result.replace(
+    /export\s+declare\s+class\s+(\w+)\s*(?:extends\s+[^{]+)?\{[^}]*\}/gs,
+    'export class $1 {}',
+  );
+
+  // Convert "declare const NAME: { ... }" (object type, non-exported) to "const NAME = {} as any;"
+  // Object types can span multiple lines and contain semicolons, so we handle them separately
+  // The 's' flag makes . match newlines, so this matches the entire object type block
+  result = result.replace(
+    /^declare\s+const\s+(\w+)\s*:\s*\{[^}]*\}\s*;/gms,
+    'const $1 = {} as any;',
+  );
+
+  // Convert "declare const NAME: TYPE;" (simple type, non-exported) to "const NAME = {} as any;"
+  // These are often used with "export { NAME }" at the end
+  // This must come AFTER the object type handler above
+  result = result.replace(
+    /^declare\s+const\s+(\w+)\s*:\s*[^;]+;/gm,
+    'const $1 = {} as any;',
+  );
+
+  // Convert "declare function NAME" (non-exported) to "function NAME() { return {} as any; }"
+  result = result.replace(
+    /^declare\s+function\s+(\w+)\s*\([^)]*\)\s*:\s*[^;]+;/gm,
+    'function $1(...args: any[]) { return {} as any; }',
+  );
+
+  // Convert "declare class NAME" (non-exported) to "class NAME {}"
+  result = result.replace(
+    /^declare\s+class\s+(\w+)\s*(?:extends\s+[^{]+)?\{[^}]*\}/gm,
+    'class $1 {}',
+  );
+
+  // Remove "export { }" at the end (empty export statement that marks module as having no exports)
+  result = result.replace(/export\s*\{\s*\}\s*;?\s*$/g, '');
+
+  // Keep export type and export interface statements as-is (they're valid in .ts)
+  // No transformation needed for these
+
+  return result;
+}
+
+/**
+ * Rewrite relative asset paths to correct locations when a file is moved.
+ *
+ * This function generically handles ANY string literal containing a relative path
+ * to an asset file, not just import statements. This catches patterns like:
+ * - import "./globals.css"
+ * - import styles from "./file.module.css"
+ * - localFont({ src: "../fonts/font.woff2" })
+ * - Image src="../images/logo.png"
+ * - Any other string literal with a relative asset path
+ */
+function rewriteAssetImports(
+  fileContent: string,
+  originalFilePath: string,
+  newFilePath: string,
+): string {
+  // Asset file extensions that should have their paths rewritten
+  const assetExtensions =
+    'css|scss|sass|less|styl|png|jpg|jpeg|gif|svg|webp|ico|woff|woff2|ttf|eot|otf|mp4|webm|json';
+
+  // Match ANY string literal containing a relative path to an asset file
+  // This captures both single and double quoted strings that start with ./ or ../
+  const relativeAssetPathRegex = new RegExp(
+    `(['"])(\\.{1,2}/[^'"]*\\.(?:${assetExtensions}))\\1`,
+    'g',
+  );
+
+  // Get the directory part of the original file path (relative to project root)
+  const originalDirParts = originalFilePath.split('/').slice(0, -1);
+
+  return fileContent.replace(
+    relativeAssetPathRegex,
+    (match, quote, assetPath) => {
+      // Resolve the asset path relative to the original file's directory
+      // We need to manually resolve the relative path without using path.resolve
+      // which would resolve against the actual filesystem
+      const pathParts = assetPath.split('/');
+      const resolvedParts = [...originalDirParts];
+
+      for (const part of pathParts) {
+        if (part === '..') {
+          resolvedParts.pop();
+        } else if (part !== '.') {
+          resolvedParts.push(part);
+        }
+      }
+
+      // Build the project-relative path to the asset
+      const absoluteAssetPath = resolvedParts.join('/');
+
+      // Calculate the new relative path from new file location
+      const newRelativePath = getRelativePath(newFilePath, absoluteAssetPath);
+
+      // Replace the path in the original match, preserving the quote style
+      return `${quote}${newRelativePath}${quote}`;
+    },
+  );
+}
+
+/**
+ * Rewrite relative TypeScript/JavaScript module imports when a file is moved.
+ *
+ * When a file is moved from one location to another (e.g., from [environmentId]/
+ * to _environmentId_/), relative imports like "./lib/organization" need to be
+ * rewritten to point to the correct location from the new path.
+ *
+ * This is similar to rewriteAssetImports but handles TypeScript/JavaScript modules
+ * instead of asset files (CSS, images, etc.).
+ */
+function rewriteRelativeModuleImports(
+  fileContent: string,
+  originalFilePath: string,
+  newFilePath: string,
+): string {
+  // Module file extensions that should have their relative paths rewritten
+  const moduleExtensions = 'ts|tsx|js|jsx|mjs|cjs';
+
+  // Match import statements with relative paths (starting with ./ or ../)
+  // This matches:
+  // - import { foo } from "./path"
+  // - import foo from "../path"
+  // - import * as foo from "./path"
+  // - import "./path" (side-effect imports)
+  // The path may or may not have a file extension
+  const relativeImportRegex = new RegExp(
+    `(from\\s+)(['"])(\\.\\.?\\/[^'"]+?)(?:\\.(?:${moduleExtensions}))?\\2`,
+    'g',
+  );
+
+  // Get the directory part of the original file path (relative to project root)
+  const originalDirParts = originalFilePath.split('/').slice(0, -1);
+
+  return fileContent.replace(
+    relativeImportRegex,
+    (match, fromKeyword, quote, importPath) => {
+      // Skip imports that already point to generated CodeYam files:
+      // 1. Scenario component files with SHA hashes (64 hex chars) and scenario name suffixes
+      //    e.g., "abc123def_EnvironmentLayout_Empty_Survey_No_Responses"
+      //    e.g., "abc123def_EnvironmentLayout_Survey_not_found_404" (scenario names can contain digits)
+      // 2. Mock data files in __codeyamMocks__ directory
+      //    e.g., "../__codeyamMocks__/MockData_Empty_Survey_No_Responses"
+      const scenarioFilePattern = /[a-f0-9]{64}_\w+_[A-Z]\w*$/;
+      const mockDataPattern = /__codeyamMocks__\//;
+      if (
+        scenarioFilePattern.test(importPath) ||
+        mockDataPattern.test(importPath)
+      ) {
+        // This import already points to a generated file, don't rewrite it
+        return match;
+      }
+
+      // Resolve the import path relative to the original file's directory
+      const pathParts = importPath.split('/');
+      const resolvedParts = [...originalDirParts];
+
+      for (const part of pathParts) {
+        if (part === '..') {
+          resolvedParts.pop();
+        } else if (part !== '.') {
+          resolvedParts.push(part);
+        }
+      }
+
+      // Build the project-relative path to the module
+      const absoluteModulePath = resolvedParts.join('/');
+
+      // Calculate the new relative path from new file location
+      const newRelativePath = getRelativePath(newFilePath, absoluteModulePath);
+
+      // Replace the path in the original match, preserving the quote style
+      return `${fromKeyword}${quote}${newRelativePath}${quote}`;
+    },
+  );
+}
+
+/**
+ * Strip <html> and <body> tags from root layout files for Next.js App Router.
+ *
+ * When a root layout (app/layout.tsx) is generated as a scenario component and placed
+ * under the /static/ route, it becomes nested under the real app root layout. Having
+ * two layouts with <html> and <body> tags causes hydration errors:
+ *
+ *   "In HTML, <html> cannot be a child of <body>"
+ *   "You are mounting a new html component when a previous one has not first unmounted"
+ *
+ * This function removes the <html> and <body> wrapper tags while preserving the inner
+ * content, allowing the scenario layout to work correctly when nested.
+ *
+ * Before: <html lang="en"><body className={...}><main>{children}</main></body></html>
+ * After:  <div className={...}><main>{children}</main></div>
+ */
+function stripHtmlBodyTags(
+  fileContent: string,
+  filePath: string,
+  framework: ProjectFramework,
+): string {
+  // Only apply to Next.js App Router root layouts
+  if (framework !== ProjectFramework.Next) {
+    return fileContent;
+  }
+
+  // Check if this is a root layout file (typically app/layout.tsx or apps/*/app/layout.tsx)
+  // Root layouts are at the app directory level, not in subdirectories like (app)/ or routes/
+  const pathParts = filePath.split('/');
+  const fileName = pathParts[pathParts.length - 1];
+
+  // Must be a layout file
+  if (!fileName?.startsWith('layout.')) {
+    return fileContent;
+  }
+
+  // Check if it's at the app root level (parent directory is 'app')
+  const parentDir = pathParts[pathParts.length - 2];
+  if (parentDir !== 'app') {
+    return fileContent;
+  }
+
+  // Check if this file actually contains <html> and <body> tags
+  if (!/<html[^>]*>/.test(fileContent) || !/<body[^>]*>/.test(fileContent)) {
+    return fileContent;
+  }
+
+  // Extract the body className/attributes if any, to preserve styling
+  const bodyMatch = fileContent.match(/<body([^>]*)>/);
+  const bodyAttributes = bodyMatch?.[1]?.trim() || '';
+
+  // Replace the JSX structure:
+  // <html ...><body ...>CONTENT</body></html>
+  // With: <div ...>CONTENT</div>
+  //
+  // This regex handles:
+  // 1. Opening <html> tag with any attributes
+  // 2. Opening <body> tag with any attributes (captured for the wrapper div)
+  // 3. Content between body tags (captured)
+  // 4. Closing </body> and </html> tags
+  const htmlBodyRegex =
+    /(<html[^>]*>\s*<body)([^>]*)(>)([\s\S]*?)(<\/body>\s*<\/html>)/g;
+
+  let result = fileContent.replace(
+    htmlBodyRegex,
+    (match, _htmlBody, bodyAttrs, closeBracket, content, _closing) => {
+      // Create a wrapper div with the body's attributes
+      const attrs = bodyAttrs?.trim() || '';
+      if (attrs) {
+        return `(<div${bodyAttrs}${closeBracket}${content}</div>)`;
+      } else {
+        return `(<>${content}</>)`;
+      }
+    },
+  );
+
+  // If the regex didn't match (perhaps due to different formatting),
+  // try a more lenient approach - just remove the tags
+  if (result === fileContent) {
+    // Remove <html ...> opening tag
+    result = result.replace(/<html[^>]*>\s*/g, '');
+    // Remove </html> closing tag
+    result = result.replace(/\s*<\/html>/g, '');
+    // Replace <body ...> with <div ...> to preserve attributes
+    result = result.replace(/<body([^>]*)>/g, '<div$1>');
+    // Replace </body> with </div>
+    result = result.replace(/<\/body>/g, '</div>');
+  }
+
+  return result;
+}
+
+/**
+ * Strip `import "server-only"` or `import 'server-only'` directives from file content.
+ *
+ * Next.js "server-only" package is used to mark modules that should only run on the server.
+ * When we generate scenario components for client-side rendering in the browser, importing
+ * a file with this directive causes an error:
+ *
+ *   "You're importing a component that needs 'server-only'. That only works in a Server Component"
+ *
+ * Since our scenario components are rendered client-side for capture purposes, we need to
+ * strip this import to allow the file to be imported.
+ */
+function stripServerOnlyImport(fileContent: string): string {
+  // Match import "server-only" or import 'server-only' with optional semicolon and newline
+  // Handles both double and single quotes, with or without trailing semicolon
+  return fileContent.replace(/import\s+["']server-only["'];?\s*\n?/g, '');
+}
+
+/**
+ * Extract all internal import paths from file content.
+ * Internal imports are those that start with '.', '@/', '~/', or are relative paths.
+ * Excludes node_modules imports (bare specifiers like 'react', '@prisma/client').
+ */
+function extractInternalImportPaths(fileContent: string): string[] {
+  // Always use AST parsing - regex with nested quantifiers can cause catastrophic
+  // backtracking that hangs on a single .exec() call (before iteration limits kick in)
+  return extractInternalImportPathsAst(fileContent);
+}
+
+/**
+ * Extract internal import paths using TypeScript AST - more reliable for large files
+ */
+function extractInternalImportPathsAst(fileContent: string): string[] {
+  const importPaths: string[] = [];
+
+  try {
+    // Use temp.tsx to enable JSX parsing for consistent handling of JSX files
+    const sourceFile = ts.createSourceFile(
+      'temp.tsx',
+      fileContent,
+      ts.ScriptTarget.Latest,
+      true,
+      ts.ScriptKind.TSX,
+    );
+
+    for (const statement of sourceFile.statements) {
+      if (ts.isImportDeclaration(statement) && statement.moduleSpecifier) {
+        const moduleSpecifier = statement.moduleSpecifier;
+        if (ts.isStringLiteral(moduleSpecifier)) {
+          const importPath = moduleSpecifier.text;
+          // Skip node_modules imports (bare specifiers)
+          if (
+            importPath.startsWith('.') ||
+            importPath.startsWith('@/') ||
+            importPath.startsWith('~/') ||
+            importPath.startsWith('#')
+          ) {
+            importPaths.push(importPath);
+          }
+        }
+      }
+    }
+  } catch (error) {
+    console.warn(
+      '[extractInternalImportPathsAst] Failed to parse file:',
+      error,
+    );
+  }
+
+  return importPaths;
+}
+
+/**
+ * Resolve an import path to a file path relative to the project.
+ * Handles path aliases like @/, ~/, and relative paths.
+ */
+function resolveImportPath(
+  importPath: string,
+  currentFilePath: string,
+  project: Project,
+): string | null {
+  let resolvedPath: string;
+  let appPrefix = '';
+
+  if (importPath.startsWith('./') || importPath.startsWith('../')) {
+    // Relative import - resolve relative to current file
+    const currentDir = currentFilePath.split('/').slice(0, -1).join('/');
+    const parts = [...currentDir.split('/'), ...importPath.split('/')];
+    const resolved: string[] = [];
+
+    for (const part of parts) {
+      if (part === '..') {
+        resolved.pop();
+      } else if (part !== '.' && part !== '') {
+        resolved.push(part);
+      }
+    }
+    resolvedPath = resolved.join('/');
+  } else if (importPath.startsWith('@/') || importPath.startsWith('~/')) {
+    // Path alias - strip the prefix
+    resolvedPath = importPath.slice(2);
+
+    // Infer app prefix from current file path
+    // e.g., if currentFilePath is "apps/web/lib/user/service.ts"
+    // and import is "@/modules/auth/lib/brevo", the actual file is at
+    // "apps/web/modules/auth/lib/brevo.ts"
+    // We detect this by checking if current file starts with "apps/XXX/"
+    const appMatch = currentFilePath.match(/^(apps\/[^/]+\/)/);
+    if (appMatch) {
+      appPrefix = appMatch[1];
+    }
+  } else if (importPath.startsWith('#')) {
+    // Package imports - not supported yet
+    return null;
+  } else {
+    // Unknown format
+    return null;
+  }
+
+  // Try to find the file with various extensions
+  const extensions = ['', '.ts', '.tsx', '.js', '.jsx'];
+
+  // First try with app prefix (for monorepo structures like apps/web/)
+  if (appPrefix) {
+    for (const ext of extensions) {
+      const fullPath = appPrefix + resolvedPath + ext;
+      const file = project.files?.find((f) => f.path === fullPath);
+      if (file) {
+        return file.path;
+      }
+    }
+
+    // Try index files with prefix
+    for (const ext of ['.ts', '.tsx', '.js', '.jsx']) {
+      const indexPath = `${appPrefix}${resolvedPath}/index${ext}`;
+      const file = project.files?.find((f) => f.path === indexPath);
+      if (file) {
+        return file.path;
+      }
+    }
+  }
+
+  // Then try without prefix (for simpler project structures)
+  for (const ext of extensions) {
+    const fullPath = resolvedPath + ext;
+    const file = project.files?.find((f) => f.path === fullPath);
+    if (file) {
+      return file.path;
+    }
+  }
+
+  // Try index files
+  for (const ext of ['.ts', '.tsx', '.js', '.jsx']) {
+    const indexPath = `${resolvedPath}/index${ext}`;
+    const file = project.files?.find((f) => f.path === indexPath);
+    if (file) {
+      return file.path;
+    }
+  }
+
+  return null;
+}
+
+// Version for tracking deployments - increment when making changes
+const WRITE_SCENARIO_COMPONENTS_VERSION = '2.5.17-configurable-server-mocks';
+
+function addMockToContent(
+  fileContent: string,
+  importedExport: EnhancedImportedExport,
+  fileAnalyses: DeepReadonly<Analysis[]>,
+  rootAnalysis: Analysis,
+  relativeMocksDir: string,
+  scenarioName: string,
+  importPath?: string,
+) {
+  // First try to find dependency schemas in fileAnalyses (for same-file dependencies)
+  let dependencySchemas = fileAnalyses.find(
+    (a) =>
+      !!a.metadata?.mergedDataStructure?.dependencySchemas?.[
+        importedExport.filePath
+      ]?.[importedExport.name],
+  )?.metadata?.mergedDataStructure?.dependencySchemas;
+
+  // If not found in fileAnalyses, use root analysis's dependency schemas
+  // The root analysis contains merged schemas for all transitive dependencies
+  if (!dependencySchemas) {
+    dependencySchemas =
+      rootAnalysis.metadata?.mergedDataStructure?.dependencySchemas;
+  }
+
+  // Check if we have multiple calls with different variable names
+  // This requires generating separate mock functions for each call site
+  //
+  // IMPORTANT: calls array may contain BOTH base hook calls (e.g., "useFetcher<Type>()")
+  // AND method chain usages (e.g., "useFetcher().functionCallReturnValue.submit(...)").
+  // We only want to count base hook calls for the length comparison with callVariableNames.
+  // A "base call" is one that ends with "()" possibly preceded by a type annotation,
+  // without any subsequent method chains like ".functionCallReturnValue" or ".submit(...)".
+  const baseHookCalls = importedExport.calls?.filter((call) => {
+    // Base hook calls match patterns like:
+    // - "useFetcher()"
+    // - "useFetcher<Type>()"
+    // - "useFetcher<{ complex: Type }>()"
+    // They end with "()" and don't have method chains after the call.
+    // Method chains contain ".functionCallReturnValue" or have property access after "()".
+    return (
+      call.endsWith('()') &&
+      !call.includes('.functionCallReturnValue') &&
+      // Also exclude method chains like "hook().something" or "hook().method()"
+      !call.match(/\(\)\.[a-zA-Z]/)
+    );
+  });
+
+  // Determine if we can generate unique mock functions for multiple variables.
+  // We need:
+  // 1. Multiple variable names (callVariableNames.length > 1)
+  // 2. Base hook calls to match them (baseHookCalls.length > 0)
+  // Note: We use min(baseHookCalls.length, callVariableNames.length) for iteration
+  // to handle cases where data might be slightly out of sync (stale entries).
+  const hasMultipleCallsWithVariables =
+    baseHookCalls &&
+    baseHookCalls.length > 1 &&
+    importedExport.callVariableNames &&
+    importedExport.callVariableNames.length > 1 &&
+    // Only proceed if we have at least as many base calls as variable names,
+    // OR they're close enough (within 1) to handle minor sync issues
+    Math.abs(baseHookCalls.length - importedExport.callVariableNames.length) <=
+      1;
+
+  let mockCode: string | undefined;
+  const variableMockCodes: string[] = [];
+
+  if (hasMultipleCallsWithVariables) {
+    // Generate separate mock functions for each variable-qualified call
+    // Look up canonical keys from dataForMocks and track variable names for function naming
+
+    // Get all call signature keys for this hook from dataForMocks
+    const dataForMocks =
+      rootAnalysis.metadata?.scenariosDataStructure?.dataForMocks;
+    // Match keys that start with the hook name (e.g., "useFetcher" matches "useFetcher<User>()")
+    const callSignatureKeysForHook = dataForMocks
+      ? Object.keys(dataForMocks).filter((key) => {
+          const hookBaseName = importedExport.name.split(/[<(]/)[0];
+          const keyBaseName = key.split(/[<(]/)[0];
+          return keyBaseName === hookBaseName;
+        })
+      : [];
+
+    // Track variable name occurrences for unique function naming
+    const variableNameCounts: Record<string, number> = {};
+
+    // Use the minimum of both array lengths to handle slight mismatches
+    // (e.g., stale data from previous analysis runs)
+    const iterationLimit = Math.min(
+      baseHookCalls!.length,
+      importedExport.callVariableNames!.length,
+    );
+    for (let i = 0; i < iterationLimit; i++) {
+      const variableName = importedExport.callVariableNames![i];
+      if (!variableName) continue;
+
+      // Calculate the occurrence index for this variable name
+      const occurrence = variableNameCounts[variableName] ?? 0;
+      variableNameCounts[variableName] = occurrence + 1;
+
+      // Build indexed variable name for function naming
+      const indexedVariableName =
+        occurrence > 0 ? `${variableName}[${occurrence}]` : variableName;
+
+      // Use safe function name with underscores instead of brackets
+      // e.g., fetcher[1] -> fetcher_1
+      const safeFunctionName = indexedVariableName.replace(/\[(\d+)\]/g, '_$1');
+      // Compute unique mock function name for call site replacement
+      // e.g., useFetcher_entityDiffFetcher, useFetcher_reportFetcher
+      const mockFunctionName = `${importedExport.name}_${safeFunctionName}`;
+
+      // Use the call signature from baseHookCalls[i] as the data key
+      // This matches what's stored in dataForMocks
+      const callSignature = baseHookCalls![i];
+
+      // Generate mock code using the call signature directly
+      // This prevents "symbol already declared" errors when multiple calls exist
+      // Check if this is a package import that won't have scenario copies
+      const isPackageImportForMock = importPath?.startsWith('@');
+      const variableMockCode = constructMockCode(
+        callSignature, // Use call signature format for data lookup
+        dependencySchemas,
+        importedExport.entityType,
+        undefined, // No need for separate canonical key
+        {
+          uniqueFunctionSuffix: safeFunctionName, // Use variable name for unique function naming
+          // For node_modules or package imports, skip spreading from __cyOriginal
+          // since those packages/files don't export *__cyOriginal variants
+          skipOriginalSpread:
+            importedExport.isNodeModule || isPackageImportForMock,
+        },
+      );
+
+      if (variableMockCode) {
+        variableMockCodes.push(variableMockCode);
+
+        // Replace the call site with the variable-specific mock function
+        // e.g., useFetcher<BranchEntityDiffResult>() -> useFetcher_entityDiffFetcher()
+        // e.g., useFetcher() -> useFetcher_reportFetcher()
+        // Escape special regex characters in the call signature
+        const escapedCallSignature = callSignature.replace(
+          /[.*+?^${}()|[\]\\]/g,
+          '\\$&',
+        );
+        // Create regex that matches the call (with optional whitespace variations).
+        // TypeScript formatters commonly break type parameters across lines, e.g.:
+        //   useLoaderData<
+        //     typeof loader
+        //   >()
+        // So we allow optional whitespace around < and > delimiters, not just
+        // where whitespace already exists in the call signature string.
+        const callRegex = new RegExp(
+          escapedCallSignature
+            .replace(/\s+/g, '\\s*')
+            .replace(/</g, '\\s*<\\s*')
+            .replace(/>/g, '\\s*>\\s*'),
+          'gs',
+        );
+        fileContent = fileContent.replace(callRegex, `${mockFunctionName}()`);
+      }
+    }
+  } else {
+    // Check if there's a single variable name that receives this function's return value
+    // Note: callVariableNames tracks the ACTUAL variable assignments (e.g., "const loadError = useLoaderData()")
+    // while calls can contain many entries including method chain accesses (e.g., "useLoaderData().config...")
+    // We only care about how many distinct variables receive the return value.
+    const singleCallVariableName =
+      importedExport.callVariableNames &&
+      importedExport.callVariableNames.length === 1
+        ? importedExport.callVariableNames[0]
+        : undefined;
+
+    if (singleCallVariableName) {
+      // For single variable assignments, use the call signature directly from dataForMocks
+      const dataForMocks =
+        rootAnalysis.metadata?.scenariosDataStructure?.dataForMocks;
+
+      // Find matching call signature key in dataForMocks
+      // IMPORTANT: First try exact match with type parameters (e.g., "useLoaderData<LoaderData>()")
+      // to avoid picking the wrong variant (e.g., "useLoaderData<typeof loader>()" which may
+      // have different properties). Fall back to base name matching only if exact match fails.
+      const hookBaseName = importedExport.name.split(/[<(]/)[0];
+      const expectedKey = importedExport.calls?.[0];
+      let callSignatureKey: string | undefined;
+
+      if (dataForMocks) {
+        const keys = Object.keys(dataForMocks);
+
+        // First try exact match with the expected call signature
+        if (expectedKey && keys.includes(expectedKey)) {
+          callSignatureKey = expectedKey;
+        } else {
+          // Fall back to base name matching
+          callSignatureKey = keys.find((key) => {
+            // Split on ., <, or ( to get the true base name
+            // This handles both "useFlags()" -> "useFlags" and "trpc.useUtils()" -> "trpc"
+            const keyBaseName = key.split(/[.<(]/)[0];
+            return keyBaseName === hookBaseName;
+          });
+        }
+      }
+
+      // Use the call signature if found, otherwise construct it
+      const dataKey =
+        callSignatureKey ??
+        importedExport.calls?.[0] ??
+        `${importedExport.name}()`;
+
+      // IMPORTANT: If the dataKey is a method chain (e.g., "trpc.useUtils()"), we should
+      // use the base name (e.g., "trpc") when calling constructMockCode. This ensures
+      // constructMockCode generates a complete nested mock from the schema without
+      // referencing __cyOriginal variables.
+      const dataKeyBaseName = dataKey.split(/[.<(]/)[0];
+      const isMethodChainDataKey =
+        dataKeyBaseName === importedExport.name &&
+        dataKey !== importedExport.name &&
+        dataKey.includes('.');
+      const mockNameToUse = isMethodChainDataKey
+        ? importedExport.name
+        : dataKey;
+
+      // Keep the original function name since there's only one call
+      // Check if this is a package import that won't have scenario copies
+      const isPackageImportForSingleCall = importPath?.startsWith('@');
+      mockCode = constructMockCode(
+        mockNameToUse,
+        dependencySchemas,
+        importedExport.entityType,
+        undefined,
+        {
+          keepOriginalFunctionName: true,
+          // For node_modules or package imports, skip spreading from __cyOriginal
+          // since those packages/files don't export *__cyOriginal variants
+          skipOriginalSpread:
+            importedExport.isNodeModule || isPackageImportForSingleCall,
+        },
+      );
+      // If constructMockCode didn't generate code, fall back to simple return
+      // IMPORTANT: We inline scenarios().data() inside the function rather than
+      // storing in a const - see comment in constructMockCode.ts for why.
+      if (!mockCode) {
+        mockCode = `// PATCHED: memoize to return stable reference (prevents infinite useEffect re-triggers)
+const _${importedExport.name}Ref = {
+  current: null,
+};
+function ${importedExport.name}(...args) {
+  if (!_${importedExport.name}Ref.current) {
+    _${importedExport.name}Ref.current = scenarios().data()?.["${dataKey}"];
+  }
+  return _${importedExport.name}Ref.current;
+}`;
+      }
+    } else {
+      // Helper to find matching call signature key from dataForMocks
+      // IMPORTANT: First try exact match with type parameters (e.g., "useLoaderData<LoaderData>()")
+      // to avoid picking the wrong variant. Fall back to base name matching only if needed.
+      const hookBaseName = importedExport.name.split(/[<(]/)[0];
+      const expectedKey = importedExport.calls?.[0];
+      const findMatchingKey = (
+        dataForMocks: Record<string, unknown> | undefined,
+      ): string | undefined => {
+        if (!dataForMocks) return undefined;
+        const keys = Object.keys(dataForMocks);
+
+        // First try exact match with the expected call signature
+        if (expectedKey && keys.includes(expectedKey)) {
+          return expectedKey;
+        }
+
+        // Fall back to base name matching
+        return keys.find((key) => {
+          // Split on ., <, or ( to get the true base name
+          // This handles both "useFlags()" -> "useFlags" and "trpc.useUtils()" -> "trpc"
+          const keyBaseName = key.split(/[.<(]/)[0];
+          return keyBaseName === hookBaseName;
+        });
+      };
+
+      // Check rootAnalysis FIRST for matching keys.
+      // The mock DATA is generated from rootAnalysis, so the mock CODE must
+      // also use rootAnalysis keys to ensure the lookup succeeds.
+      let dataKey = findMatchingKey(
+        rootAnalysis.metadata?.scenariosDataStructure?.dataForMocks,
+      );
+
+      // If not found in rootAnalysis, fall back to fileAnalyses
+      if (!dataKey) {
+        for (const analysis of fileAnalyses) {
+          dataKey = findMatchingKey(
+            analysis.metadata?.scenariosDataStructure?.dataForMocks,
+          );
+          if (dataKey) break;
+        }
+      }
+
+      // Use the data key if found, otherwise use call signature or function name.
+      // IMPORTANT: If the dataKey is a method chain (e.g., "trpc.useUtils()"), we should
+      // use the base name (e.g., "trpc") when calling constructMockCode. This ensures
+      // constructMockCode generates a complete nested mock from the schema without
+      // referencing __cyOriginal variables. The __cyOriginal pattern is only needed
+      // for partial mocking where we preserve some original methods, not for complete
+      // method-chain mocks where we provide all implementations.
+      const dataKeyBaseName = dataKey?.split(/[.<(]/)[0];
+      const isMethodChainDataKey =
+        dataKey &&
+        dataKeyBaseName === importedExport.name &&
+        dataKey !== importedExport.name &&
+        dataKey.includes('.');
+      const mockNameToUse = isMethodChainDataKey
+        ? importedExport.name
+        : (dataKey ?? importedExport.calls?.[0] ?? `${importedExport.name}()`);
+
+      mockCode = constructMockCode(
+        mockNameToUse,
+        dependencySchemas,
+        importedExport.entityType,
+        undefined,
+        {
+          keepOriginalFunctionName: true,
+          // For node_modules or package imports, skip spreading from __cyOriginal
+          // since those packages/files don't export *__cyOriginal variants
+          skipOriginalSpread:
+            importedExport.isNodeModule || importPath?.startsWith('@'),
+        },
+      );
+      // If constructMockCode didn't generate code, fall back to simple return
+      // IMPORTANT: We inline scenarios().data() inside the function rather than
+      // storing in a const - see comment in constructMockCode.ts for why.
+      if (!mockCode && dataKey) {
+        mockCode = `// PATCHED: memoize to return stable reference (prevents infinite useEffect re-triggers)
+const _${importedExport.name}Ref = {
+  current: null,
+};
+function ${importedExport.name}(...args) {
+  if (!_${importedExport.name}Ref.current) {
+    _${importedExport.name}Ref.current = scenarios().data()?.["${dataKey}"];
+  }
+  return _${importedExport.name}Ref.current;
+}`;
+      }
+    }
+  }
+
+  // Combine all mock codes
+  const allMockCodes =
+    variableMockCodes.length > 0 ? variableMockCodes.join('\n\n') : mockCode;
+
+  if (!allMockCodes) {
+    return fileContent;
+  }
+
+  if (importPath) {
+    const importedExportNameParts = splitOutsideParenthesesAndArrays(
+      importedExport.name,
+    );
+    const firstPart = importedExportNameParts[0];
+
+    if (importedExport.isDefault && importedExport.isMocked) {
+      const entireImportRegExp = new RegExp(
+        `import\\s+${firstPart}\\s+from\\s+['"]${importPath.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')}['"];?\\s*\\n?`,
+        'm',
+      );
+      fileContent = fileContent.replace(entireImportRegExp, '');
+    } else {
+      // Escape regex special characters in importPath (e.g., brackets in [environmentId])
+      const escapedImportPath = importPath.replace(
+        /[.*+?^${}()|[\]\\]/g,
+        '\\$&',
+      );
+      // Use a simpler, more robust regex pattern that matches the fallback path.
+      // Key improvements:
+      // 1. Uses escapeRegExp(firstPart) to handle special characters in function names
+      // 2. Uses word boundaries (\b) to prevent partial matches
+      // 3. Handles comma BEFORE or AFTER the name: (?:,\s*|\s*,)?
+      // 4. Matches specific import path (escapedImportPath)
+      const importRegExp = new RegExp(
+        `(import\\s*\\{[^}]*?)\\b${escapeRegExp(firstPart)}\\b(?:,\\s*|\\s*,)?((?:[^}]*\\}\\s*from\\s*['"]${escapedImportPath}['"];?))`,
+        'm',
+      );
+
+      // Check if any call signature has multiple parts (e.g., "logger.error(error)")
+      // If so, the mock code will spread from __cyOriginal, so we need to rename the import
+      // EXCEPT:
+      // 1. For node_module imports, the __cyOriginal pattern doesn't work because
+      //    the original package doesn't export *__cyOriginal variants.
+      // 2. For package imports (starting with @), the __cyOriginal pattern doesn't work
+      //    because scenario copies aren't created for package files - they keep the
+      //    original import path which doesn't export *__cyOriginal.
+      const anyCallHasMultipleParts = importedExport.calls?.some((call) => {
+        const callParts = splitOutsideParenthesesAndArrays(call);
+        return callParts.length > 1;
+      });
+
+      // Package imports (starting with @) don't get scenario copies, so __cyOriginal won't exist
+      const isPackageImport = importPath.startsWith('@');
+
+      const shouldRenameToOriginal =
+        !importedExport.isNodeModule &&
+        !isPackageImport &&
+        (importedExportNameParts.length > 1 || anyCallHasMultipleParts);
+
+      if (shouldRenameToOriginal) {
+        fileContent = fileContent.replace(
+          importRegExp,
+          `$1${firstPart}__cyOriginal$2`,
+        );
+      } else {
+        fileContent = fileContent.replace(importRegExp, '$1$2');
+      }
+
+      // Also handle namespace imports (import * as foo from '...')
+      // These need to be renamed to foo__cyOriginal when the mock code spreads from the original.
+      // Note: We match any path (not just escapedImportPath) because the import path may have
+      // been rewritten by transitive import handling before this code runs.
+      if (shouldRenameToOriginal) {
+        const namespaceImportRegExp = new RegExp(
+          `(import\\s+\\*\\s+as\\s+)${escapeRegExp(firstPart)}(\\s+from\\s+['"][^'"]*['"])`,
+          'm',
+        );
+        fileContent = fileContent.replace(
+          namespaceImportRegExp,
+          `$1${firstPart}__cyOriginal$2`,
+        );
+      }
+
+      // Remove empty imports entirely to avoid partial commenting issues with multiline imports
+      // This handles both single-line and multiline empty imports
+      fileContent = fileContent.replace(
+        /import\s*\{\s*\}\s*from\s+['"][^'"]*['"];?\s*\n?/g,
+        '',
+      );
+    }
+  } else {
+    // Fallback: When importPath is undefined (e.g., path alias not in mapping),
+    // we still need to remove the import to avoid "name defined multiple times" errors.
+    // Search for the entity name in any import statement and remove it.
+    const importedExportNameParts = splitOutsideParenthesesAndArrays(
+      importedExport.name,
+    );
+    const firstPart = importedExportNameParts[0];
+
+    // Match the entity name in any named import and remove just that name
+    // This handles imports like: import { useEnvironment, otherThing } from "any/path"
+    // Removes useEnvironment but keeps otherThing
+    const namedImportRegExp = new RegExp(
+      `(import\\s*\\{[^}]*?)\\b${escapeRegExp(firstPart)}\\b(?:,\\s*|\\s*,)?((?:[^}]*\\}\\s*from\\s*['"][^'"]*['"];?))`,
+      'm',
+    );
+
+    // Check if any call signature has multiple parts (e.g., "logger.error(error)")
+    // If so, the mock code will spread from __cyOriginal, so we need to rename the import
+    // EXCEPT: For node_module imports, the __cyOriginal pattern doesn't work because
+    // the original package doesn't export *__cyOriginal variants.
+    const anyCallHasMultipleParts = importedExport.calls?.some((call) => {
+      const callParts = splitOutsideParenthesesAndArrays(call);
+      return callParts.length > 1;
+    });
+
+    const shouldRenameToOriginal =
+      !importedExport.isNodeModule &&
+      (importedExportNameParts.length > 1 || anyCallHasMultipleParts);
+
+    if (shouldRenameToOriginal) {
+      // Rename the import instead of removing (for destructured access patterns)
+      fileContent = fileContent.replace(
+        namedImportRegExp,
+        `$1${firstPart}__cyOriginal$2`,
+      );
+    } else {
+      fileContent = fileContent.replace(namedImportRegExp, '$1$2');
+    }
+
+    // Clean up empty imports that might result from removing the only import
+    fileContent = fileContent.replace(
+      /import\s*\{\s*\}\s*from\s+['"][^'"]*['"];?\s*\n?/g,
+      '',
+    );
+  }
+
+  // Find the last import position using TypeScript AST (more reliable than regex).
+  // This ensures we insert both the scenarios import AND the mock after the original imports,
+  // not at the end of the file.
+  const lastImportEnd = findEndOfImports(fileContent);
+
+  // Build the content to insert: scenarios import (if needed) + mock
+  let insertContent = '';
+
+  if (fileContent.indexOf('import { scenarios } from') === -1) {
+    // Use scenario-specific MockData file to allow multiple scenarios to coexist
+    const mockDataPath = `${relativeMocksDir}/MockData_${safeFileName(scenarioName)}`;
+    insertContent += `\n\nimport { scenarios } from "${mockDataPath}";`;
+  }
+
+  insertContent += `\n\n// Mock constructed using mocksDataStructure from analyses: ${fileAnalyses.map((a) => a.id).join(', ')}\n${allMockCodes}`;
+
+  if (lastImportEnd > 0) {
+    // Insert after the last original import
+    fileContent =
+      fileContent.slice(0, lastImportEnd) +
+      insertContent +
+      fileContent.slice(lastImportEnd);
+  } else {
+    // Fallback: append at end if no imports found
+    fileContent += insertContent;
+  }
+
+  return fileContent;
+}
+
+// Debug utility to capture complex arguments for testing
+function captureArgumentsForTesting(args: WriteScenarioComponentsArgs) {
+  const debugMode =
+    process.env.CODEYAM_DEBUG_WRITE_SCENARIO_COMPONENTS === 'true';
+
+  if (!debugMode) {
+    return;
+  }
+
+  try {
+    const testDataDir = path.join(
+      process.cwd(),
+      'background/src/lib/virtualized/project/__tests__/captured-data',
+    );
+
+    if (!fs.existsSync(testDataDir)) {
+      fs.mkdirSync(testDataDir, { recursive: true });
+    }
+
+    const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
+    const analysisId = args.rootAnalysis?.id || 'unknown';
+    const filename = `writeScenarioComponents-args-${analysisId}-${timestamp}.json`;
+    const filepath = path.join(testDataDir, filename);
+
+    // Extract context data for serialization
+    const contextData = {
+      analyses: {} as any,
+      entities: {} as any,
+    };
+
+    // Get all analyses from context - handle both Map and object formats
+    const allAnalyses = (args.context as any).analyses || new Map();
+    if (allAnalyses instanceof Map) {
+      for (const [filePath, analysisMap] of allAnalyses) {
+        contextData.analyses[filePath] = {};
+        if (analysisMap instanceof Map) {
+          for (const [entityName, analysis] of analysisMap) {
+            contextData.analyses[filePath][entityName] = analysis;
+          }
+        } else {
+          contextData.analyses[filePath] = analysisMap;
+        }
+      }
+    } else {
+      contextData.analyses = allAnalyses;
+    }
+
+    // Get all entities from context - handle both Map and object formats
+    const allEntities = (args.context as any).entities || new Map();
+    if (allEntities instanceof Map) {
+      for (const [filePath, entityMap] of allEntities) {
+        contextData.entities[filePath] = {};
+        if (entityMap instanceof Map) {
+          for (const [entityName, entity] of entityMap) {
+            contextData.entities[filePath][entityName] = entity;
+          }
+        } else {
+          contextData.entities[filePath] = entityMap;
+        }
+      }
+    } else {
+      contextData.entities = allEntities;
+    }
+
+    // Ensure we have the main entity and analysis in context
+    if (!contextData.entities[args.entity.filePath]) {
+      contextData.entities[args.entity.filePath] = {};
+    }
+    contextData.entities[args.entity.filePath][args.entity.name] = args.entity;
+
+    if (!contextData.analyses[args.rootAnalysis.filePath]) {
+      contextData.analyses[args.rootAnalysis.filePath] = {};
+    }
+    contextData.analyses[args.rootAnalysis.filePath][
+      args.rootAnalysis.entityName
+    ] = args.rootAnalysis;
+
+    // Create a serializable version of the arguments
+    const serializedArgs = {
+      project: {
+        id: args.project.id,
+        name: args.project.name,
+        metadata: args.project.metadata,
+        files: args.project.files?.map((f) => ({
+          id: f.id,
+          path: f.path,
+          name: f.name,
+          content: f.content,
+        })),
+      },
+      file: {
+        id: args.file.id,
+        path: args.file.path,
+        name: args.file.name,
+        content: args.file.content,
+      },
+      entity: {
+        name: args.entity.name,
+        filePath: args.entity.filePath,
+        sha: args.entity.sha,
+        entityType: args.entity.entityType,
+        metadata: args.entity.metadata,
+      },
+      rootAnalysis: {
+        id: args.rootAnalysis.id,
+        filePath: args.rootAnalysis.filePath,
+        entityName: args.rootAnalysis.entityName,
+        entitySha: args.rootAnalysis.entitySha,
+        metadata: args.rootAnalysis.metadata,
+      },
+      scenario: {
+        id: args.scenario.id,
+        name: args.scenario.name,
+        description: args.scenario.description,
+        metadata: args.scenario.metadata,
+      },
+      framework: args.framework,
+      mocksDir: args.mocksDir,
+      rootFile: args.rootFile
+        ? {
+            id: args.rootFile.id,
+            path: args.rootFile.path,
+            name: args.rootFile.name,
+            content: args.rootFile.content,
+          }
+        : undefined,
+      namespaceMocks: args.namespaceMocks?.map((m) => ({
+        id: m.id,
+        fileId: m.fileId,
+        filePath: m.filePath,
+        entityName: m.entityName,
+        entitySha: m.entitySha,
+        mockingRequired: m.mockingRequired,
+        nodeModule: m.nodeModule,
+        metadata: m.metadata,
+        analysisId: m.analysisId,
+        localFilePath: m.localFilePath,
+        resolvedFilePath: m.resolvedFilePath,
+        importAlias: m.importAlias,
+        importName: m.importName,
+      })),
+      writtenScenarioComponents: args.writtenScenarioComponents,
+      contextData: contextData,
+    };
+
+    fs.writeFileSync(filepath, JSON.stringify(serializedArgs, null, 2));
+
+    console.log(`CodeYam Debug: Captured arguments to ${filepath}`);
+    console.log(
+      `To use this data in tests, set CODEYAM_DEBUG_WRITE_SCENARIO_COMPONENTS=true`,
+    );
+  } catch (error) {
+    console.error('CodeYam Debug: Failed to capture arguments:', error);
+  }
+}
+
+export default async function writeScenarioComponents({
+  project,
+  file,
+  entity,
+  rootAnalysis,
+  scenario,
+  context,
+  projectAnalyzer,
+  framework,
+  mocksDir,
+  rootFile,
+  namespaceMocks,
+  writtenScenarioComponents = {},
+  fileStore,
+  exportAsNamed,
+}: WriteScenarioComponentsArgs): Promise<{
+  scenarioComponentPaths: string[];
+  writtenScenarioComponents: { [key: string]: string[] };
+}> {
+  // Reset debug timing for this invocation
+  resetDebugTiming();
+  debugLog('START writeScenarioComponents', {
+    filePath: file.path,
+    entityName: entity.name,
+    scenarioName: scenario.name,
+    isRootFile: !rootFile || rootFile === file,
+  });
+
+  // Capture arguments for testing if debug mode is enabled
+  captureArgumentsForTesting({
+    project,
+    file,
+    entity,
+    rootAnalysis,
+    scenario,
+    context,
+    projectAnalyzer,
+    framework,
+    mocksDir,
+    rootFile,
+    namespaceMocks,
+    writtenScenarioComponents,
+  });
+  const indexFile = isIndexPath(file.path);
+  const scenarioComponentPaths: string[] = [];
+
+  // awsLog('CodeYam: Writing scenario components', {
+  //   filePath: file.path,
+  //   entityName: entity.name
+  // });
+
+  if (!rootFile) {
+    rootFile = file;
+  }
+
+  writtenScenarioComponents[file.path] ||= [];
+  writtenScenarioComponents[file.path].push(entity.name);
+
+  const allImportedExports: EnhancedImportedExport[] = [];
+
+  // Get all analyses for this file from context
+  const analysesForFile = context.getAnalysesForFile(file.path) ?? [];
+
+  for (const analysis of analysesForFile) {
+    const entityForAnalysis = context.getEntity(
+      analysis.filePath,
+      analysis.entityName,
+    );
+
+    const nodeModuleImports = Object.keys(
+      entityForAnalysis?.metadata?.nodeModuleImports ?? {},
+    ).flatMap((filePath) =>
+      entityForAnalysis.metadata.nodeModuleImports[filePath].map(
+        (nodeModuleImport) => ({
+          ...nodeModuleImport,
+          filePath,
+          isNodeModule: true,
+          analysisId: analysis.id,
+          resolvedFilePath: filePath,
+        }),
+      ),
+    );
+
+    const importedExports = (
+      entityForAnalysis.metadata?.importedExports ?? []
+    ).map((importedExport) => ({
+      ...importedExport,
+      analysisId: analysis.id,
+    }));
+
+    allImportedExports.push(...importedExports, ...nodeModuleImports);
+  }
+
+  // FALLBACK: If context doesn't have analyses for this file (e.g., when recursing into
+  // a dependency that wasn't directly analyzed), use the entity's own metadata
+  if (
+    analysesForFile.length === 0 &&
+    entity.metadata?.importedExports?.length
+  ) {
+    const entityImportedExports = entity.metadata.importedExports.map(
+      (importedExport) => ({
+        ...importedExport,
+        analysisId: rootAnalysis.id, // Use the root analysis ID as fallback
+      }),
+    );
+
+    const entityNodeModuleImports = Object.keys(
+      entity.metadata?.nodeModuleImports ?? {},
+    ).flatMap((filePath) =>
+      (entity.metadata.nodeModuleImports[filePath] ?? []).map(
+        (nodeModuleImport) => ({
+          ...nodeModuleImport,
+          filePath,
+          isNodeModule: true,
+          analysisId: rootAnalysis.id,
+          resolvedFilePath: filePath,
+        }),
+      ),
+    );
+
+    allImportedExports.push(
+      ...entityImportedExports,
+      ...entityNodeModuleImports,
+    );
+  }
+
+  const relativeImportedExports = allImportedExports.filter(
+    (importedExport, index, self) =>
+      !importedExport.isNodeModule &&
+      index ===
+        self.findIndex(
+          (ie) =>
+            ie.filePath === importedExport.filePath &&
+            ie.name === importedExport.name,
+        ),
+  );
+
+  const nodeModuleImports = allImportedExports.filter(
+    (nodeModuleImport, index, self) =>
+      nodeModuleImport.isNodeModule &&
+      index === self.findIndex((nmi) => nmi.name === nodeModuleImport.name),
+  );
+
+  let fileContent = file.content;
+
+  // Handle .d.ts files: convert type declarations to stub implementations
+  // .d.ts files only have type declarations (e.g., "export declare const logger")
+  // which don't provide runtime exports. We need to generate actual stub implementations.
+  if (file.path.endsWith('.d.ts')) {
+    fileContent = convertDtsToStubs(fileContent, entity.name);
+
+    // After basic .d.ts conversion, enhance the entity's stub with proper mock code
+    // if we have dependency schema data showing its methods/properties.
+    // This prevents "X.method is not a function" errors at runtime.
+    //
+    // NOTE: constructMockCode iterates through ALL file paths in dependencySchemas
+    // to find the entity, so we don't need to check for an exact file path match.
+    // The entity might be stored under a different path (e.g., import alias like
+    // @formbricks/logger vs the actual file path packages/logger/dist/index.d.ts).
+    const dependencySchemas =
+      rootAnalysis.metadata?.mergedDataStructure?.dependencySchemas;
+    if (dependencySchemas) {
+      const mockCode = constructMockCode(
+        entity.name,
+        dependencySchemas,
+        entity.entityType,
+      );
+      if (mockCode) {
+        // Replace the simple `export const entityName = {} as any;` with the full mock code
+        // The mockCode generates `const entityName = { ... }` so we need to add `export`
+        const stubPattern = new RegExp(
+          `export\\s+(const|let|var)\\s+${escapeRegExp(entity.name)}\\s*=\\s*\\{\\s*\\}\\s+as\\s+any;?`,
+        );
+
+        if (fileContent.match(stubPattern)) {
+          // Add export keyword to the mock code since constructMockCode generates `const name = ...`
+          const exportMockCode = mockCode.replace(
+            /^(const|let|var)\s+/,
+            'export $1 ',
+          );
+          fileContent = fileContent.replace(stubPattern, exportMockCode);
+
+          // Add scenarios import at the top for the mock to work
+          const relativeMocksDir = getRelativePath(file.path, mocksDir);
+          const mockDataPath = `${relativeMocksDir}/MockData_${safeFileName(scenario.name)}`;
+          fileContent = `import { scenarios } from "${mockDataPath}";\n\n${fileContent}`;
+        }
+      }
+    }
+  }
+
+  // Extract "use client" or "use server" directive FIRST, before any modifications
+  // This ensures the directive isn't buried by prepended imports
+  const directiveMatch = fileContent.match(
+    /^(\s*["']use (client|server)["'];?\s*\n?)/,
+  );
+  let extractedDirective: string | null = null;
+  if (directiveMatch) {
+    extractedDirective = directiveMatch[1].trim();
+    // Remove the directive from fileContent - we'll add it back at the very end
+    fileContent = fileContent.slice(directiveMatch[0].length);
+  }
+
+  const fileAnalyzer = projectAnalyzer.getFileAnalyzer(file);
+  const importMapping = fileAnalyzer.getRelativeImportMappings();
+
+  let filePath = file.path;
+  if (isFrameworkRoute(file, entity, framework, file === rootFile)) {
+    filePath = getFrameworkRoutePath({
+      file,
+      rootFile,
+      entity,
+      rootAnalysis,
+      scenarioComponentPath: file.path,
+      project,
+      framework,
+      scenario,
+    });
+
+    for (const importFilePath in importMapping) {
+      if (
+        allImportedExports.some(
+          (importedExport) =>
+            importedExport.filePath === importFilePath ||
+            importedExport.filePath === importMapping[importFilePath],
+        )
+      ) {
+        continue;
+      }
+      // const relativePath = getRelativePath(filePath, importFilePath);
+      // const originalImport = importMapping[importFilePath];
+
+      // WHEN IS THIS NEEDED?
+      // fileContent = fileContent.replace(originalImport, relativePath);
+    }
+  }
+
+  const relativeMocksDir = getRelativePath(filePath, mocksDir);
+
+  const fileAnalyses = context.getAnalysesForFile(entity.filePath);
+
+  const sortedImportedExports = relativeImportedExports.sort((a, b) => {
+    if (a.isMocked && !b.isMocked) return -1;
+    if (!a.isMocked && b.isMocked) return 1;
+    return 0;
+  });
+
+  debugLog('Starting main importedExports loop', {
+    count: sortedImportedExports.length,
+    fileContentLength: fileContent.length,
+  });
+
+  let importedExportIndex = 0;
+  const loopStartTime = Date.now();
+  console.log(
+    `[WriteScenario] Starting import loop for ${entity.name}: ${sortedImportedExports.length} imports`,
+  );
+  for (const importedExport of sortedImportedExports) {
+    importedExportIndex++;
+    if (importedExportIndex % 5 === 0 || importedExportIndex === 1) {
+      console.log(
+        `[WriteScenario] ${entity.name} import ${importedExportIndex}/${sortedImportedExports.length}: ${importedExport.name} elapsed=${Date.now() - loopStartTime}ms`,
+      );
+      debugLog(
+        `Processing importedExport ${importedExportIndex}/${sortedImportedExports.length}`,
+        {
+          name: importedExport.name,
+          filePath: importedExport.filePath,
+          isMocked: importedExport.isMocked,
+        },
+      );
+    }
+    // IMPORTANT: The import mapping keys may be either absolute or relative paths
+    // depending on how they were created by the file analyzer. We try multiple formats.
+    // Also need to normalize paths to handle /tmp vs /private/tmp on macOS
+    const normalizePath = (p: string) => {
+      // On macOS, /tmp is a symlink to /private/tmp
+      // Normalize /private/tmp to /tmp for consistency
+      return p.replace(/^\/private\/tmp\//, '/tmp/');
+    };
+
+    const importFilePathForLookup = normalizePath(
+      `${projectAnalyzer.dirPath}/${importedExport.filePath}`,
+    );
+    const resolvedFilePathForLookup = importedExport.resolvedFilePath
+      ? normalizePath(
+          `${projectAnalyzer.dirPath}/${importedExport.resolvedFilePath}`,
+        )
+      : undefined;
+
+    const importedExportEntity =
+      context.getEntity(
+        importedExport.resolvedFilePath ?? importedExport.filePath,
+        importedExport.resolvedName ?? importedExport.name,
+      ) ?? context.getEntity(importedExport.resolvedFilePath, 'default');
+
+    // Check if we actually need the entity for this import
+    const importedExportFilePath =
+      importedExport.resolvedFilePath ?? importedExport.filePath;
+    const isSameFile = importedExportFilePath === file.path;
+
+    // We only need the entity for non-mocked, different-file imports:
+    // - Same-file imports: skip to the isCurrentFile check later (line ~1203)
+    // - Mocked imports: use importedExport.entityType, not importedExportEntity
+    // - Non-mocked different-file: need entity for type check, recursive call, and path rewriting
+    if (!importedExportEntity && !isSameFile && !importedExport.isMocked) {
+      // Check if the file exists in project.files - if not, skip this import.
+      // This happens when a file is excluded from project.files (e.g., Prisma-generated client.ts)
+      // but entities still have importedExports referencing it.
+      const fileExistsInProject = fileStore
+        ? fileStore.getByPath(importedExportFilePath)
+        : project.files.find((f) => f.path === importedExportFilePath);
+
+      if (!fileExistsInProject) {
+        // File not in project.files - likely auto-generated or excluded. Skip it.
+        continue;
+      }
+
+      // All entities should be pre-loaded by loadReadyToBeCaptured's gatherAllReferencedEntities.
+      // If we're here, that function has a bug or an edge case was missed.
+      throw new Error(
+        `Entity not found in context: ${importedExport.resolvedFilePath ?? importedExport.filePath}:${importedExport.resolvedName ?? importedExport.name}\n` +
+          `Looking up for: ${file.path}:${entity.name}\n` +
+          `Context entities for file: ${JSON.stringify(context.getEntityNamesForFile(importedExport.filePath))}\n` +
+          `Context entities for resolved: ${JSON.stringify(context.getEntityNamesForFile(importedExport.resolvedFilePath))}`,
+      );
+    }
+
+    if (importedExportEntity) {
+      (importedExportEntity as Entity).localFilePath = importedExport.filePath;
+    }
+
+    if (!importedExport.isMocked) {
+      // Use resolvedFilePath (actual entity location) for scenario component writing.
+      // This is critical for re-exports: when importing from index.tsx which re-exports
+      // from a nested file, we need to write scenario components for the actual file
+      // where the entity code lives, not the re-export file.
+      if (!isSameFile) {
+        if (
+          !writtenScenarioComponents[importedExportFilePath]?.includes(
+            importedExport.name,
+          )
+        ) {
+          // Use fileStore for O(1) lookup when available
+          let fileNotMocked = fileStore
+            ? fileStore.getByPath(importedExportFilePath)
+            : project.files.find((f) => f.path === importedExportFilePath);
+
+          // Fallback: if file at resolvedFilePath doesn't exist, try filePath
+          // This handles cases where the actual entity file isn't in project.files
+          // (e.g., type re-exports where only the index file is tracked)
+          if (
+            !fileNotMocked &&
+            importedExport.resolvedFilePath &&
+            importedExport.filePath !== importedExport.resolvedFilePath
+          ) {
+            const fallbackPath = importedExport.filePath;
+            fileNotMocked = fileStore
+              ? fileStore.getByPath(fallbackPath)
+              : project.files.find((f) => f.path === fallbackPath);
+          }
+
+          // Skip if file still not found - can't write scenario component without the file
+          if (!fileNotMocked) {
+            continue;
+          }
+
+          // Ensure content is loaded for the file
+          if (
+            fileNotMocked &&
+            fileStore &&
+            !fileStore.isContentLoaded(fileNotMocked.path)
+          ) {
+            fileNotMocked = await fileStore.ensureContent(fileNotMocked.path);
+          }
+
+          // For type/data entities, create a transformed copy WITHOUT recursion.
+          // Data entities don't need mocking - they're just constants/types that need
+          // to be available. But we still need to:
+          // 1. Strip server-only imports (Next.js directive that breaks client components)
+          // 2. Write the transformed file so imports can be rewritten to point to it
+          if (
+            importedExportEntity.entityType === 'type' ||
+            importedExportEntity.entityType === 'data'
+          ) {
+            // For data entities, we write ONE transformed copy per source file, not per entity.
+            // Check if we've already written ANY entity from this file - if so, skip writing.
+            // Use a special marker '__data_file_written__' to track file-level writes.
+            // Also track which SHA was used via '__data_file_sha__:xxx' so subsequent
+            // entities from the same file can reuse it for import rewriting.
+            const dataFileWritten = writtenScenarioComponents[
+              importedExportFilePath
+            ]?.includes('__data_file_written__');
+
+            if (!dataFileWritten) {
+              // Construct the scenario file path for the data file
+              // Use a file-level identifier (first entity's sha) for consistency
+              const dataFileBasePath = safeFolder(
+                fileNotMocked.path.split('/').slice(0, -1).join('/'),
+              );
+              const dataFileExtension = fileNotMocked.name.split('.').pop();
+              const dataFileIsIndex = isIndexPath(fileNotMocked.path);
+              // Use 'data' prefix to distinguish from entity-specific files
+              const dataScenarioPath = `${PROJECT_RELATIVE_PATH}/${dataFileBasePath}/${importedExportEntity.sha}_${dataFileIsIndex ? 'index_' : ''}data_${safeFileName(scenario.name)}.${dataFileExtension}`;
+
+              // Get the file content and apply transformations
+              let dataFileContent = fileNotMocked.content ?? '';
+
+              // Strip server-only imports - these break when imported by client components
+              dataFileContent = stripServerOnlyImport(dataFileContent);
+              dataFileContent = applyServerOnlyMocks(dataFileContent);
+
+              // Write the transformed data entity file
+              await writeFile(dataScenarioPath, dataFileContent);
+              scenarioComponentPaths.push(dataScenarioPath);
+
+              // Mark file as written so we don't duplicate for other entities from same file
+              if (!writtenScenarioComponents[importedExportFilePath]) {
+                writtenScenarioComponents[importedExportFilePath] = [];
+              }
+              writtenScenarioComponents[importedExportFilePath].push(
+                '__data_file_written__',
+              );
+              // Also store the SHA used for this data file so subsequent entities can use it
+              writtenScenarioComponents[importedExportFilePath].push(
+                `__data_file_sha__:${importedExportEntity.sha}`,
+              );
+            }
+
+            // Mark this specific entity as written (for the entity-level check)
+            if (!writtenScenarioComponents[importedExportFilePath]) {
+              writtenScenarioComponents[importedExportFilePath] = [];
+            }
+            writtenScenarioComponents[importedExportFilePath].push(
+              importedExport.name,
+            );
+
+            // Don't recurse - data entities don't need their dependencies processed
+            // The import rewriting will happen later in this same loop iteration
+            // (at lines ~1590-1702) to point imports to this transformed file
+          } else {
+            // For visual/library entities, recurse to process their dependencies
+
+            // When a default export is imported as named (via index re-export), we need
+            // to add a named re-export to the scenario component so the import works.
+            // e.g., if file has `export default X` but parent does `import { X } from '...'`
+            const needsNamedReExport =
+              importedExport.resolvedIsDefault === true &&
+              importedExport.isDefault === false;
+
+            console.log(
+              `[WriteScenario] RECURSE START: ${entity.name} -> ${importedExportEntity.name}`,
+            );
+            const recurseStartTime = Date.now();
+            debugLog(
+              `Recursing into writeScenarioComponents for ${importedExportEntity.name}`,
+              {
+                entityName: importedExportEntity.name,
+                filePath: fileNotMocked.path,
+              },
+            );
+            const {
+              scenarioComponentPaths: newScenarioComponentPaths,
+              writtenScenarioComponents: updatedWrittenScenarioComponents,
+            } = await withTimeout(
+              `recursive writeScenarioComponents for ${importedExportEntity.name}`,
+              writeScenarioComponents({
+                project,
+                file: fileNotMocked,
+                entity: importedExportEntity,
+                rootAnalysis,
+                scenario,
+                context,
+                projectAnalyzer,
+                framework,
+                mocksDir,
+                rootFile,
+                namespaceMocks,
+                writtenScenarioComponents,
+                fileStore,
+                // Pass the import name so we can add `export { default as Name };`
+                exportAsNamed: needsNamedReExport
+                  ? importedExport.name
+                  : undefined,
+              }),
+              180000, // 3 minute timeout for recursive calls (complex components need more time)
+            );
+            console.log(
+              `[WriteScenario] RECURSE END: ${entity.name} -> ${importedExportEntity.name} took ${Date.now() - recurseStartTime}ms`,
+            );
+            debugLog(
+              `Completed recursive writeScenarioComponents for ${importedExportEntity.name}`,
+            );
+            writtenScenarioComponents = updatedWrittenScenarioComponents;
+            scenarioComponentPaths.push(...newScenarioComponentPaths);
+          }
+        }
+      }
+    } else if (
+      importedExport.filePath === file.path &&
+      importedExport.name !== entity.name
+    ) {
+      // For same-file dependencies:
+      // - Strip and replace if we have a mock structure we can generate
+      // - Strip and stub if isMocked is true AND it's a callable entity (visual, library, functionCall)
+      // - Preserve as-is for data entities (like Zod schemas) that need their methods
+      const hasMock = hasMockStructure(importedExport, fileAnalyses);
+      const entityType = importedExport.entityType;
+      // Data and type entities should be preserved - they're not callable and may have methods
+      // that stubbing would break (e.g., Zod schemas with .superRefine())
+      const isDataEntity = entityType === 'data' || entityType === 'type';
+
+      // Heuristic: Zod schemas are often misclassified as 'library' but should be preserved
+      // Detect by: name starts with Z + uppercase letter, AND has Zod method calls
+      const looksLikeZodSchema =
+        entityType === 'library' &&
+        /^Z[A-Z]/.test(importedExport.name) &&
+        importedExport.calls?.some((call: string) =>
+          /\.(superRefine|refine|transform|default|optional|nullable|array|object|string|number|boolean|parse|safeParse)\s*\(/.test(
+            call,
+          ),
+        );
+
+      // Callable entities can be safely stubbed (but not Zod schemas)
+      const isCallable =
+        !isDataEntity && !looksLikeZodSchema && entityType !== undefined;
+
+      // Determine what action to take
+      const shouldStripAndReplace = hasMock;
+      const shouldStripAndStub =
+        !hasMock && importedExport.isMocked && isCallable;
+      const shouldPreserve = !hasMock && importedExport.isMocked && !isCallable;
+
+      // Log warning if entityType is undefined for a mocked same-file dependency
+      // This could indicate an analysis issue where entityType wasn't properly set
+      if (importedExport.isMocked && entityType === undefined) {
+        console.warn(
+          `CodeYam: WARNING - Same-file dependency "${importedExport.name}" is isMocked but has no entityType. ` +
+            `Treating as non-callable (will preserve). File: ${file.path}`,
+        );
+      }
+
+      if (shouldPreserve) {
+        // For data entities (like Zod schemas), don't strip or stub - preserve the original
+        // This ensures schema methods like .superRefine() continue to work
+        // Don't modify fileContent - keep the original code
+      } else if (shouldStripAndReplace || shouldStripAndStub) {
+        // Strip the original code
+        const entityCode = fileAnalyzer.getEntityCode(importedExport.name);
+
+        if (shouldStripAndReplace) {
+          if (entityCode) {
+            fileContent = fileContent.replace(entityCode, '');
+          }
+          // Add mock from mock structure
+          fileContent = addMockToContent(
+            fileContent,
+            importedExport,
+            fileAnalyses,
+            rootAnalysis,
+            relativeMocksDir,
+            scenario.name,
+          );
+        } else if (shouldStripAndStub) {
+          // Generate a simple stub mock that returns scenario data.
+          // This prevents ReferenceError at runtime when the stripped
+          // function is called (e.g., local helper functions like getInitialProps).
+          const functionName = importedExport.name;
+
+          // Add scenarios import if not present
+          if (fileContent.indexOf('import { scenarios } from') === -1) {
+            const mockDataPath = `${relativeMocksDir}/MockData_${safeFileName(scenario.name)}`;
+            const importStatement = `import { scenarios } from "${mockDataPath}";`;
+            fileContent = `${fileContent}\n\n\n${importStatement}`;
+          }
+
+          // Check if the original entity was exported - if so, the stub should be exported too
+          // This ensures other files that import this entity can still access it
+          const wasExported = entityCode?.trimStart().startsWith('export');
+          const exportKeyword = wasExported ? 'export ' : '';
+
+          // Generate a chainable stub mock using a Proxy
+          // This handles cases where the entity is used with method chains like:
+          //   actionClient.use(...).use(...)
+          // The Proxy returns itself for any property access, making it infinitely chainable.
+          // When called as a function, it returns the scenario data.
+          const stubMock = `// Stub mock for local function without mock structure
+// Uses a Proxy to support method chaining (e.g., ${functionName}.use(...))
+const _${functionName}Handler = {
+  get: function(target, prop) {
+    if (prop === 'then') return undefined; // Prevent Promise-like behavior
+    return (...args) => new Proxy(() => scenarios().data()?.["${functionName}()"], _${functionName}Handler);
+  },
+  apply: function(target, thisArg, args) {
+    return scenarios().data()?.["${functionName}()"];
+  }
+};
+${exportKeyword}const ${functionName} = new Proxy(() => scenarios().data()?.["${functionName}()"], _${functionName}Handler);`;
+
+          // Replace the original entity code with the stub mock IN PLACE
+          // This is critical to maintain the original ordering of declarations.
+          // If we append the stub at the end, code that uses this entity before
+          // the stub definition will fail with "Cannot access X before initialization".
+          if (entityCode) {
+            fileContent = fileContent.replace(entityCode, stubMock);
+          } else {
+            // Fallback: append at end if we can't find the original code
+            fileContent += `\n\n${stubMock}`;
+          }
+        }
+      }
+    }
+
+    // Try to find the import mapping using different key formats
+    // The import mapping may use either absolute or relative paths as keys
+    // Use filePath (import source) for looking up the import mapping
+    // because that's how imports are written in source code
+    const mockFilePathRelative = importedExport.filePath;
+
+    // For scenario component paths, prefer resolvedFilePath (where entity actually lives)
+    // because that's where scenario component files are written.
+    // Map nested dist paths to src paths so import paths point to actual source files.
+    const scenarioFilePathRelative = mapNestedDistToSrc(
+      importedExport.resolvedFilePath ?? importedExport.filePath,
+    );
+
+    // Try looking up with different key formats (absolute, relative, resolved absolute)
+    // Filter out undefined keys to avoid misleading undefined lookups
+    const lookupKeys = [
+      importFilePathForLookup,
+      mockFilePathRelative,
+      resolvedFilePathForLookup,
+    ].filter((key) => key !== undefined);
+
+    let mockImportMapping: any = undefined;
+    for (const key of lookupKeys) {
+      if (importMapping[key]) {
+        mockImportMapping = importMapping[key];
+        break;
+      }
+    }
+
+    // Check if importing from the same file (compare relative paths)
+    const isCurrentFile = mockFilePathRelative === file.path;
+
+    // Skip if it's the same file
+    if (isCurrentFile) {
+      continue;
+    }
+
+    // For non-mocked imports, we still need to rewrite paths even without import mapping
+    // (the import mapping is just used to know what string to search for in the import statement)
+
+    if (importedExport.isMocked) {
+      try {
+        fileContent = addMockToContent(
+          fileContent,
+          importedExport,
+          fileAnalyses,
+          rootAnalysis,
+          relativeMocksDir,
+          scenario.name,
+          mockImportMapping,
+        );
+      } catch (error) {
+        console.error('CodeYam Error: Failed to add mock', {
+          filePath: file.path,
+          entityName: entity.name,
+          analysisIds: fileAnalyses.map((a) => a.id),
+          analysisFilePath: fileAnalyses[0].filePath,
+          analysisEntityName: fileAnalyses.map((a) => a.entityName),
+          mockFilePathRelative,
+          importedExport,
+        });
+        throw error;
+      }
+    } else {
+      // For non-mocked imports, rewrite the import path to point to the generated scenario file
+      // Use fileStore for O(1) lookup when available
+      // First try resolvedFilePath (where entity lives), fallback to filePath (import source)
+      let fileNotMocked = fileStore
+        ? fileStore.getByPath(scenarioFilePathRelative)
+        : project.files.find((f) => f.path === scenarioFilePathRelative);
+
+      // Track which path we're actually using for the scenario component
+      let actualScenarioFilePathRelative = scenarioFilePathRelative;
+      if (
+        !fileNotMocked &&
+        importedExport.resolvedFilePath &&
+        importedExport.filePath !== importedExport.resolvedFilePath
+      ) {
+        // Fallback to filePath if file at resolvedFilePath doesn't exist
+        actualScenarioFilePathRelative = importedExport.filePath;
+        fileNotMocked = fileStore
+          ? fileStore.getByPath(actualScenarioFilePathRelative)
+          : project.files.find(
+              (f) => f.path === actualScenarioFilePathRelative,
+            );
+      }
+
+      const fileName = actualScenarioFilePathRelative.split('/').pop();
+      const fileNotMockedIsIndex = isIndexPath(fileNotMocked?.path);
+
+      // For data/type entities, use 'data' instead of entity name since we write
+      // ONE file per source file (not per entity) for data entities
+      const isDataEntity =
+        importedExportEntity?.entityType === 'data' ||
+        importedExportEntity?.entityType === 'type';
+      const scenarioFileName = isDataEntity
+        ? 'data'
+        : safeFileName(importedExportEntity.name);
+
+      // For data entities, look up the SHA that was used to create the data file
+      // (stored when the file was first written). This ensures all entities from
+      // the same source file have their imports rewritten to point to the same file.
+      // Use actualScenarioFilePathRelative as the key since that's what matches
+      // importedExportFilePath used when storing (both are relative paths).
+      let entityShaForPath = importedExportEntity.sha;
+      if (isDataEntity && actualScenarioFilePathRelative) {
+        const storedShaMarker = writtenScenarioComponents[
+          actualScenarioFilePathRelative
+        ]?.find((m) => m.startsWith('__data_file_sha__:'));
+        if (storedShaMarker) {
+          entityShaForPath = storedShaMarker.replace('__data_file_sha__:', '');
+        }
+      }
+
+      const mockFilePath = isFrameworkRoute(
+        fileNotMocked,
+        importedExportEntity,
+        framework,
+        fileNotMocked === rootFile,
+      )
+        ? getFrameworkRoutePath({
+            file: fileNotMocked,
+            rootFile,
+            entity: importedExportEntity,
+            rootAnalysis,
+            scenarioComponentPath: actualScenarioFilePathRelative,
+            project,
+            framework,
+            scenario,
+          })
+            .split('.')
+            .slice(0, -1)
+            .join('.')
+        : actualScenarioFilePathRelative.replace(
+            `${fileName}`,
+            `${entityShaForPath}_${fileNotMockedIsIndex ? 'index_' : ''}${scenarioFileName}_${safeFileName(scenario.name)}`,
+          );
+
+      const path = safeFolder(getRelativePath(filePath, mockFilePath));
+
+      // If we have an import mapping, use it to find the import string to replace
+      // Otherwise, skip rewriting (we can't find the import statement without knowing what to search for)
+      if (mockImportMapping) {
+        const importRegExp = new RegExp(
+          `${escapeRegExp(mockImportMapping)}(['"])`,
+          'g',
+        );
+
+        // Check if there are mocked entities from the same file
+        // If so, we can't replace the entire import path - the mocked entities will be inlined
+        // and we need to keep the original import path for this non-mocked entity
+        const sameFileImports = sortedImportedExports.filter(
+          (ie) => ie.filePath === importedExport.filePath,
+        );
+        const hasMockedSameFileImports = sameFileImports.some(
+          (ie) => ie.isMocked && ie.name !== importedExport.name,
+        );
+
+        // Check if the original import path still exists in the file
+        // If it was already rewritten by another entity from the same module,
+        // we need to add a separate import statement for this entity
+        if (fileContent.match(importRegExp)) {
+          if (hasMockedSameFileImports) {
+            // There are mocked entities from the same file that will be inlined
+            // Don't rewrite the import path - the original import is still needed for this non-mocked entity
+            // The mocked entities will be inlined, so the original import statement will remain
+            // with only the non-mocked entities after the mocked ones are removed
+          } else {
+            // No mocked entities from the same file - safe to replace entire path
+            fileContent = fileContent.replace(importRegExp, `${path}$1`);
+          }
+        } else {
+          // The import path was already rewritten - add a new import for this entity
+          // This handles cases where multiple entities are imported from the same index file
+          // (e.g., import { A, B, C } from '@pkg') and each has its own scenario file
+          const entityImportName = importedExport.name;
+          // Use default import syntax for default exports, named import for named exports
+          const newImport = importedExport.isDefault
+            ? `import ${entityImportName} from '${path}';`
+            : `import { ${entityImportName} } from '${path}';`;
+
+          // First, try to remove this entity from the already-rewritten grouped import
+          // This prevents duplicate/conflicting imports
+          // Use AST-based removal to properly handle type-only imports like `type EntityName`
+          const escapedEntityName = escapeRegExp(entityImportName);
+
+          // For default imports: remove "DefaultName, " from "import DefaultName, { ... }"
+          // This handles the case where a default export is being split out
+          if (importedExport.isDefault) {
+            const defaultImportPattern = new RegExp(
+              `(import\\s+)${escapedEntityName}\\s*,\\s*(\\{)`,
+              'gm',
+            );
+            fileContent = fileContent.replace(defaultImportPattern, '$1$2');
+          }
+
+          // Remove the named import using AST parsing
+          // This properly handles:
+          // - Regular imports: `import { EntityName } from '...'`
+          // - Type-only imports: `import { type EntityName } from '...'`
+          // - Mixed imports: `import { type EntityName, OtherName } from '...'`
+          fileContent = removeNamedImportAst(fileContent, entityImportName);
+
+          // Add the new import at the beginning of fileContent
+          // Note: The header comment (// Scenario:) doesn't exist yet - it's prepended at writeFile time
+          // So prepending here puts the import right after the header in the final output
+          //
+          // IMPORTANT: We cannot use indexOf('// Scenario:') + '\n\n' here because:
+          // 1. The header doesn't exist during processing
+          // 2. indexOf would return -1, and indexOf('\n\n', -1) starts from 0
+          // 3. This could find a \n\n inside a function body, inserting the import there!
+          fileContent = newImport + '\n' + fileContent;
+        }
+      }
+    }
+  }
+
+  // Track post-import-loop timing
+  const postLoopStartTime = Date.now();
+  console.log(`[WriteScenario] POST-LOOP START: ${entity.name}`);
+
+  // Collect universal mocks BEFORE processing nodeModuleImports
+  // This is needed to check if a node module import is handled by a universal mock
+  const universalMocks = project.metadata?.universalMocks ?? [];
+  const nodeModuleUniversalMocks = universalMocks.filter(
+    (mock) => mock.nodeModule && mock.content,
+  );
+
+  // Create a set of import paths that have universal mocks for quick lookup
+  const universalMockPaths = new Set(
+    nodeModuleUniversalMocks.map((mock) => mock.filePath),
+  );
+
+  for (const nodeModuleImport of nodeModuleImports) {
+    if (!nodeModuleImport.isMocked) continue;
+
+    // Skip generating local mock functions for imports that have universal mocks.
+    // Universal mocks provide the exports via rewritten import paths (handled below).
+    // Generating a local mock function would cause "name defined multiple times" errors.
+    if (universalMockPaths.has(nodeModuleImport.filePath)) {
+      continue;
+    }
+
+    fileContent = addMockToContent(
+      fileContent,
+      nodeModuleImport,
+      fileAnalyses,
+      rootAnalysis,
+      relativeMocksDir,
+      scenario.name,
+      importMapping[nodeModuleImport.filePath] ?? nodeModuleImport.filePath,
+    );
+  }
+
+  // Rewrite node_module imports that have universal mocks
+  // Universal mocks create mock files at __codeyamMocks__/{safeFileName}.tsx
+  // We need to rewrite imports like `import { logger } from "@formbricks/logger"`
+  // to `import { logger } from "../__codeyamMocks__/_formbricks_logger"`
+
+  for (const universalMock of nodeModuleUniversalMocks) {
+    const originalPath = universalMock.filePath;
+    // Create the mock file name using the same safeFileName function as writeUniversalMocks
+    const safeMockFileName = safeFileName(originalPath);
+    const mockFileRelativePath = `${relativeMocksDir}/${safeMockFileName}`;
+
+    // Match both single and double quotes, and handle both named and default imports
+    // Pattern 1: import { x, y } from "module"
+    // Pattern 2: import x from "module"
+    // Pattern 3: import * as x from "module"
+    const importRegex = new RegExp(
+      `(import\\s+(?:\\{[^}]*\\}|\\*\\s+as\\s+\\w+|\\w+)\\s+from\\s+)['"]${originalPath.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')}['"]`,
+      'g',
+    );
+
+    fileContent = fileContent.replace(
+      importRegex,
+      `$1'${mockFileRelativePath}'`,
+    );
+  }
+
+  console.log(
+    `[WriteScenario] POST-LOOP ${entity.name}: node+universal mocks took ${Date.now() - postLoopStartTime}ms`,
+  );
+
+  if (
+    rootAnalysis.entitySha === entity.sha &&
+    entity.metadata?.notExported &&
+    entity.name !== 'default'
+  ) {
+    const type = entity.entityType === 'type';
+    const exportStatement = `export${type ? ' type ' : ' '}{ ${entity.name} };`;
+    if (!fileContent.includes(exportStatement)) {
+      fileContent = `${fileContent}\n\n${exportStatement}`;
+    }
+  }
+
+  // When a default export is imported as named via re-export (e.g., index.tsx re-exports
+  // a default export as named), we need to add a named export so the parent's
+  // import statement works: `import { Name } from './path'`
+  // The local variable name is the same as exportAsNamed (e.g., ScenarioEditor)
+  if (exportAsNamed) {
+    const namedExport = `export { ${exportAsNamed} };`;
+    if (!fileContent.includes(namedExport)) {
+      fileContent = `${fileContent}\n\n${namedExport}`;
+    }
+  }
+
+  const basePath = safeFolder(file.path.split('/').slice(0, -1).join('/'));
+  const extension = file.name.split('.').pop();
+  // Include scenario name in path to allow multiple scenarios to coexist
+  let scenarioComponentPath = `${PROJECT_RELATIVE_PATH}/${basePath}/${entity.sha}_${indexFile ? 'index_' : ''}${safeFileName(entity.name)}_${safeFileName(scenario.name)}.${extension}`;
+
+  if (isFrameworkRoute(file, entity, framework, file === rootFile)) {
+    scenarioComponentPath = getFrameworkRoutePath({
+      file,
+      rootFile,
+      entity,
+      rootAnalysis,
+      scenarioComponentPath,
+      project,
+      framework,
+      scenario,
+    });
+  }
+
+  debugLog('Route path computed', { scenarioComponentPath });
+
+  // Strip <html> and <body> tags from root layout files for Next.js
+  // These tags cause hydration errors when the scenario layout is nested under the real root
+  debugLog('Starting stripHtmlBodyTags');
+  fileContent = stripHtmlBodyTags(fileContent, file.path, framework);
+  debugLog('Completed stripHtmlBodyTags');
+
+  // Strip "server-only" imports for Next.js
+  // These cause errors when the scenario component is rendered client-side
+  debugLog('Starting stripServerOnlyImport');
+  fileContent = stripServerOnlyImport(fileContent);
+  debugLog('Starting applyServerOnlyMocks');
+  fileContent = applyServerOnlyMocks(fileContent);
+  debugLog('Completed server-only processing');
+
+  // Rewrite asset imports (CSS, images, fonts, etc.) to correct relative paths
+  // The original file path is relative to PROJECT_RELATIVE_PATH, the new path is scenarioComponentPath
+  debugLog('Starting rewriteAssetImports');
+  fileContent = rewriteAssetImports(
+    fileContent,
+    `${PROJECT_RELATIVE_PATH}/${file.path}`,
+    scenarioComponentPath,
+  );
+  debugLog('Completed rewriteAssetImports');
+
+  // Rewrite relative TypeScript/JavaScript module imports to correct relative paths
+  // This handles cases where the file is moved (e.g., from [environmentId]/ to _environmentId_/)
+  // and relative imports like "./lib/organization" need to be rewritten
+  debugLog('Starting rewriteRelativeModuleImports');
+  fileContent = rewriteRelativeModuleImports(
+    fileContent,
+    `${PROJECT_RELATIVE_PATH}/${file.path}`,
+    scenarioComponentPath,
+  );
+  debugLog('Completed rewriteRelativeModuleImports');
+
+  console.log(
+    `[WriteScenario] POST-LOOP ${entity.name}: transformations took ${Date.now() - postLoopStartTime}ms`,
+  );
+
+  /**
+   * Recursively process a file's imports to create transitive copies with server-only stripped.
+   * This handles chains of arbitrary depth: A -> B -> C -> D where each needs server-only removed.
+   * Uses a visited set to detect and break circular import chains.
+   *
+   * @param content - The file content to process
+   * @param sourceFilePath - The original source file path (for resolving relative imports)
+   * @param targetFilePath - The path where this content will be written (for computing relative imports)
+   * @param visitedPaths - Set of file paths currently being processed (for cycle detection)
+   * @returns The modified content with imports rewritten to point to transitive copies
+   */
+  async function processTransitiveImportsRecursively(
+    content: string,
+    sourceFilePath: string,
+    targetFilePath: string,
+    visitedPaths: Set<string> = new Set(),
+    depth: number = 0,
+    startTime: number = Date.now(),
+  ): Promise<string> {
+    // Global timeout for entire transitive processing
+    const GLOBAL_TIMEOUT_MS = 180000; // 3 minutes max for all transitive processing (complex components need more time)
+    const elapsed = Date.now() - startTime;
+    if (elapsed > GLOBAL_TIMEOUT_MS) {
+      throw new Error(
+        `processTransitiveImportsRecursively exceeded ${GLOBAL_TIMEOUT_MS}ms (elapsed: ${elapsed}ms) at depth=${depth} for ${sourceFilePath}`,
+      );
+    }
+
+    const importPaths = extractInternalImportPaths(content);
+    // Always log to help debug timeout issues
+    console.log(
+      `[TransitiveImports] depth=${depth} file=${path.basename(sourceFilePath)} imports=${importPaths.length} visited=${visitedPaths.size} elapsed=${Date.now() - startTime}ms`,
+    );
+    debugLog(`processTransitiveImportsRecursively depth=${depth}`, {
+      sourceFilePath,
+      importCount: importPaths.length,
+      visitedCount: visitedPaths.size,
+    });
+    let modifiedContent = content;
+
+    // Safety check: limit iterations to prevent infinite loops
+    const MAX_IMPORTS_PER_FILE = 100;
+    if (importPaths.length > MAX_IMPORTS_PER_FILE) {
+      console.warn(
+        `[WriteScenario] WARNING: File ${sourceFilePath} has ${importPaths.length} imports (> ${MAX_IMPORTS_PER_FILE}), limiting processing`,
+      );
+    }
+
+    let importIndex = 0;
+    debugLog(
+      `Starting import loop at depth=${depth}, ${importPaths.length} imports to process`,
+    );
+    const slicedImports = importPaths.slice(0, MAX_IMPORTS_PER_FILE);
+    for (const importPath of slicedImports) {
+      if (!importPath) {
+        continue;
+      }
+      importIndex++;
+      debugLog(
+        `[LOOP] depth=${depth} import ${importIndex}/${Math.min(importPaths.length, MAX_IMPORTS_PER_FILE)}: ${importPath}`,
+      );
+      debugLog(`[LOOP] Calling resolveImportPath...`);
+      const resolvedPath = resolveImportPath(
+        importPath,
+        sourceFilePath,
+        project,
+      );
+      debugLog(
+        `[LOOP] resolveImportPath returned: ${resolvedPath?.slice(0, 80) ?? 'null'}`,
+      );
+      if (!resolvedPath) continue;
+
+      debugLog(`[LOOP] Looking up importFile...`);
+      let importFile = fileStore
+        ? fileStore.getByPath(resolvedPath)
+        : project.files?.find((f) => f.path === resolvedPath);
+      debugLog(
+        `[LOOP] importFile lookup result: ${importFile ? 'found' : 'not found'}`,
+      );
+      if (!importFile) continue;
+
+      // Build the transitive file path (needed for import rewriting even if we skip creating)
+      const basePath = safeFolder(
+        importFile.path.split('/').slice(0, -1).join('/'),
+      );
+      const extension = importFile.name.split('.').pop();
+      const isIndex = isIndexPath(importFile.path);
+      // Limit pathHash length to prevent ENAMETOOLONG errors on macOS (255 char limit)
+      const pathHash = safeFileName(importFile.path, { maxLength: 80 });
+      const scenarioSlug = safeFileName(scenario.name, { maxLength: 60 });
+      const transitiveFilePath = `${PROJECT_RELATIVE_PATH}/${basePath}/${pathHash}_${isIndex ? 'index_' : ''}transitive_${scenarioSlug}.${extension}`;
+
+      // Check if this is a circular import (we're already processing this file)
+      const isCircularImport = visitedPaths.has(resolvedPath);
+
+      // Check if already processed
+      const alreadyProcessed = writtenScenarioComponents[
+        resolvedPath
+      ]?.includes('__transitive_file_written__');
+
+      // Only create the transitive file if not circular and not already processed
+      if (!isCircularImport && !alreadyProcessed) {
+        // Load content if needed
+        if (!importFile.content && fileStore) {
+          importFile = await fileStore.ensureContent(resolvedPath);
+        }
+        if (!importFile?.content) {
+          // Can't create transitive, but still try to rewrite import below
+        } else {
+          // Mark as being processed BEFORE recursing (to detect cycles)
+          visitedPaths.add(resolvedPath);
+
+          // Strip server-only and mock server-only packages, then recursively process imports
+          let transitiveContent = stripServerOnlyImport(importFile.content);
+          transitiveContent = applyServerOnlyMocks(transitiveContent);
+          debugLog(
+            `processTransitiveImportsRecursively depth=${depth} for ${importFile.path}`,
+          );
+          debugLog(
+            `Calling processTransitiveImportsRecursively depth=${depth + 1} for ${importFile.path}`,
+          );
+          transitiveContent = await withTimeout(
+            `processTransitiveImportsRecursively depth=${depth} for ${path.basename(importFile.path)}`,
+            processTransitiveImportsRecursively(
+              transitiveContent,
+              importFile.path,
+              transitiveFilePath,
+              visitedPaths,
+              depth + 1,
+              startTime, // Pass through the original start time
+            ),
+            30000, // 30 second timeout per transitive import
+          );
+          debugLog(
+            `withTimeout returned for depth=${depth}, transitiveContent length=${transitiveContent.length}`,
+          );
+          debugLog(
+            `Completed processTransitiveImportsRecursively depth=${depth} for ${importFile.path}`,
+          );
+
+          debugLog(`Writing transitive file depth=${depth}`, {
+            transitiveFilePath: path.basename(transitiveFilePath),
+            contentLength: transitiveContent.length,
+          });
+          await writeFile(transitiveFilePath, transitiveContent);
+          debugLog(`Wrote transitive file depth=${depth}`);
+          scenarioComponentPaths.push(transitiveFilePath);
+
+          if (!writtenScenarioComponents[resolvedPath]) {
+            writtenScenarioComponents[resolvedPath] = [];
+          }
+          writtenScenarioComponents[resolvedPath].push(
+            '__transitive_file_written__',
+          );
+        }
+      }
+
+      // ALWAYS rewrite the import to point to the transitive copy
+      // (even for circular imports or already-processed files)
+      debugLog(`Rewriting import path depth=${depth}`, {
+        importPath,
+        resolvedPath,
+      });
+      const relativePath = getRelativePath(targetFilePath, transitiveFilePath);
+      const relativePathWithoutExt = relativePath.replace(
+        /\.(ts|tsx|js|jsx)$/,
+        '',
+      );
+      const safeRelativePath = safeFolder(relativePathWithoutExt);
+      const escapedImportPath = importPath.replace(
+        /[.*+?^${}()|[\]\\]/g,
+        '\\$&',
+      );
+      debugLog(`Applying regex depth=${depth}`, {
+        escapedImportPath,
+        contentLength: modifiedContent.length,
+      });
+      // Quick check if the import path even exists in content
+      const simpleCheck = modifiedContent.includes(importPath);
+      debugLog(
+        `Simple check: importPath "${importPath}" exists: ${simpleCheck}`,
+      );
+      if (!simpleCheck) {
+        debugLog(`Skipping regex - import path not found in content`);
+      } else {
+        const regexPattern = `(from\\s*["'])${escapedImportPath}(["'])`;
+        debugLog(`Regex pattern: ${regexPattern.slice(0, 100)}`);
+        const importRegex = new RegExp(regexPattern, 'g');
+        debugLog(`About to call replace...`);
+
+        // Timing for regex replace to detect slow operations
+        const replaceStart = Date.now();
+        modifiedContent = modifiedContent.replace(
+          importRegex,
+          `$1${safeRelativePath}$2`,
+        );
+        const replaceTime = Date.now() - replaceStart;
+        if (replaceTime > 100) {
+          console.warn(
+            `[WriteScenario] SLOW regex replace: ${replaceTime}ms for pattern ${regexPattern.slice(0, 50)} on ${modifiedContent.length} bytes`,
+          );
+        }
+        debugLog(`Regex applied depth=${depth} in ${replaceTime}ms`);
+      }
+      debugLog(`[LOOP END] depth=${depth} import ${importIndex} completed`);
+    }
+
+    debugLog(`[LOOP DONE] Exiting import loop at depth=${depth}`);
+    debugLog(
+      `Returning from processTransitiveImportsRecursively depth=${depth}`,
+    );
+    return modifiedContent;
+  }
+
+  console.log(
+    `[WriteScenario] POST-LOOP ${entity.name}: before remaining imports ${Date.now() - postLoopStartTime}ms`,
+  );
+
+  // Process remaining internal imports that weren't in importedExports
+  // This handles transitive dependencies: when the file content includes code (e.g., from
+  // other functions in the same file) that imports from files with "server-only"
+  debugLog('Extracting remaining import paths');
+  const remainingImportPaths = extractInternalImportPaths(fileContent);
+  debugLog('Found remaining import paths', {
+    count: remainingImportPaths.length,
+  });
+
+  // Get all file paths that are in importedExports - these are handled by main processing
+  const importedExportFilePaths = new Set(
+    allImportedExports.map((ie) => ie.resolvedFilePath || ie.filePath),
+  );
+
+  debugLog('Starting remaining imports loop', {
+    remainingCount: remainingImportPaths.length,
+    importedExportCount: importedExportFilePaths.size,
+  });
+
+  let remainingImportIndex = 0;
+  debugLog(
+    `[REMAINING] Starting remaining imports loop, ${remainingImportPaths.length} imports`,
+  );
+  for (const importPath of remainingImportPaths) {
+    remainingImportIndex++;
+    debugLog(
+      `[REMAINING LOOP] import ${remainingImportIndex}/${remainingImportPaths.length}: ${importPath}`,
+    );
+    // Resolve the import path to a project file path
+    const resolvedFilePath = resolveImportPath(importPath, file.path, project);
+
+    if (!resolvedFilePath) {
+      // Can't resolve - might be a path we don't handle, skip it
+      continue;
+    }
+
+    // Find the file in project.files, using fileStore for O(1) lookup when available
+    // We need to find the file BEFORE checking importedExports to see if it has server-only
+    let targetFile = fileStore
+      ? fileStore.getByPath(resolvedFilePath)
+      : project.files?.find((f) => f.path === resolvedFilePath);
+
+    // Skip if this import is in importedExports - it's handled by the main processing loop
+    // UNLESS the file contains "server-only", in which case we still need a transitive copy
+    // with server-only stripped (even if some exports from the file are mocked).
+    if (importedExportFilePaths.has(resolvedFilePath)) {
+      // Check if the file has server-only before skipping
+      let fileContent = targetFile?.content;
+      if (!fileContent && targetFile && fileStore) {
+        // Load content to check for server-only
+        const loadedFile = await fileStore.ensureContent(resolvedFilePath);
+        fileContent = loadedFile?.content;
+        if (loadedFile) {
+          targetFile = loadedFile;
+        }
+      }
+
+      const hasServerOnly =
+        fileContent && /import\s+["']server-only["']/.test(fileContent);
+
+      if (!hasServerOnly) {
+        continue;
+      }
+      // File has server-only - continue processing to create transitive copy
+    }
+    if (!targetFile) {
+      continue;
+    }
+
+    // Compute the transformed file path (needed for import rewriting even if already processed)
+    const targetFileBasePath = safeFolder(
+      targetFile.path.split('/').slice(0, -1).join('/'),
+    );
+    const targetFileExtension = targetFile.name.split('.').pop();
+    const targetFileIsIndex = isIndexPath(targetFile.path);
+    // Limit path hash length to prevent ENAMETOOLONG errors
+    const filePathHash = safeFileName(targetFile.path, { maxLength: 80 });
+    const targetScenarioSlug = safeFileName(scenario.name, { maxLength: 60 });
+    const transformedFilePath = `${PROJECT_RELATIVE_PATH}/${targetFileBasePath}/${filePathHash}_${targetFileIsIndex ? 'index_' : ''}transitive_${targetScenarioSlug}.${targetFileExtension}`;
+
+    // Check if we've already processed this file as a transitive copy
+    // Note: __data_file_written__ is for entity-specific scenario files with different naming,
+    // but we still need transitive copies for server-only stripping. Data entity files may
+    // only export specific items, so we need a full transitive copy with all exports.
+    const alreadyTransitive = writtenScenarioComponents[
+      resolvedFilePath
+    ]?.includes('__transitive_file_written__');
+
+    if (!alreadyTransitive) {
+      // Ensure content is loaded (LazyFileStore loads content on-demand)
+      if (!targetFile.content && fileStore) {
+        targetFile = await fileStore.ensureContent(resolvedFilePath);
+      }
+      if (!targetFile?.content) {
+        continue;
+      }
+
+      // Get the file content and apply transformations
+      let transformedContent = targetFile.content;
+      transformedContent = stripServerOnlyImport(transformedContent);
+      transformedContent = applyServerOnlyMocks(transformedContent);
+
+      // Recursively process this transitive file's imports
+      // This handles the nested case: service.ts → brevo.ts → constants.ts
+      const nestedImportPaths = extractInternalImportPaths(transformedContent);
+      debugLog(
+        `[NESTED] Processing ${nestedImportPaths.length} nested imports for ${targetFile.path}`,
+      );
+      let nestedIndex = 0;
+      for (const nestedImportPath of nestedImportPaths) {
+        nestedIndex++;
+        debugLog(
+          `[NESTED LOOP] import ${nestedIndex}/${nestedImportPaths.length}: ${nestedImportPath}`,
+        );
+        const nestedResolvedPath = resolveImportPath(
+          nestedImportPath,
+          targetFile.path,
+          project,
+        );
+        if (!nestedResolvedPath) continue;
+
+        // Get file info for building the transformed path
+        let nestedFile = fileStore
+          ? fileStore.getByPath(nestedResolvedPath)
+          : project.files?.find((f) => f.path === nestedResolvedPath);
+        if (!nestedFile) continue;
+
+        // Build the transformed path (needed for import rewriting even if already processed)
+        const nestedBasePath = safeFolder(
+          nestedFile.path.split('/').slice(0, -1).join('/'),
+        );
+        const nestedExtension = nestedFile.name.split('.').pop();
+        const nestedIsIndex = isIndexPath(nestedFile.path);
+        // Limit path hash length to prevent ENAMETOOLONG errors
+        const nestedPathHash = safeFileName(nestedFile.path, { maxLength: 80 });
+        const nestedScenarioSlug = safeFileName(scenario.name, {
+          maxLength: 60,
+        });
+        const nestedTransformedPath = `${PROJECT_RELATIVE_PATH}/${nestedBasePath}/${nestedPathHash}_${nestedIsIndex ? 'index_' : ''}transitive_${nestedScenarioSlug}.${nestedExtension}`;
+
+        // Check if already processed as a transitive file (we can rewrite to point to it)
+        // Note: __data_file_written__ is for entity-specific scenario files with different naming,
+        // but we still need transitive copies for server-only stripping in the import chain
+        const nestedAlreadyTransitive = writtenScenarioComponents[
+          nestedResolvedPath
+        ]?.includes('__transitive_file_written__');
+
+        if (!nestedAlreadyTransitive) {
+          // Ensure content is loaded for nested files
+          if (!nestedFile.content && fileStore) {
+            nestedFile = await fileStore.ensureContent(nestedResolvedPath);
+          }
+          if (!nestedFile?.content) continue;
+
+          // Strip server-only, mock server-only packages, and recursively process imports
+          // This handles chains of any depth: A -> B -> C -> D
+          let nestedContent = stripServerOnlyImport(nestedFile.content);
+          nestedContent = applyServerOnlyMocks(nestedContent);
+          debugLog(
+            `processTransitiveImportsRecursively (nested) for ${nestedFile.path}`,
+          );
+          nestedContent = await withTimeout(
+            `processTransitiveImportsRecursively (nested) for ${path.basename(nestedFile.path)}`,
+            processTransitiveImportsRecursively(
+              nestedContent,
+              nestedFile.path,
+              nestedTransformedPath,
+            ),
+            30000, // 30 second timeout per nested transitive import
+          );
+          debugLog(
+            `Completed processTransitiveImportsRecursively (nested) for ${nestedFile.path}`,
+          );
+
+          await writeFile(nestedTransformedPath, nestedContent);
+          scenarioComponentPaths.push(nestedTransformedPath);
+
+          // Mark as written
+          if (!writtenScenarioComponents[nestedResolvedPath]) {
+            writtenScenarioComponents[nestedResolvedPath] = [];
+          }
+          writtenScenarioComponents[nestedResolvedPath].push(
+            '__transitive_file_written__',
+          );
+        }
+
+        // Rewrite the import to point to the transitive copy
+        const nestedRelativePath = getRelativePath(
+          transformedFilePath,
+          nestedTransformedPath,
+        );
+        // Strip the extension before applying safeFolder - import paths in TypeScript
+        // should NOT include extensions (Next.js resolves .ts/.tsx automatically).
+        // Without this, safeFolder converts ".ts" to "_ts" causing "Module not found" errors.
+        const nestedRelativePathWithoutExt = nestedRelativePath.replace(
+          /\.(ts|tsx|js|jsx)$/,
+          '',
+        );
+        const safeNestedRelativePath = safeFolder(nestedRelativePathWithoutExt);
+        const escapedNestedImportPath = nestedImportPath.replace(
+          /[.*+?^${}()|[\]\\]/g,
+          '\\$&',
+        );
+        const nestedImportRegex = new RegExp(
+          `(from\\s*["'])${escapedNestedImportPath}(["'])`,
+          'g',
+        );
+        transformedContent = transformedContent.replace(
+          nestedImportRegex,
+          `$1${safeNestedRelativePath}$2`,
+        );
+      }
+
+      // Re-check if file was written during recursive processing
+      // (processTransitiveImportsRecursively might have created this file while processing
+      // a nested import that circularly imports back to this file)
+      const writtenDuringRecursion = writtenScenarioComponents[
+        resolvedFilePath
+      ]?.includes('__transitive_file_written__');
+
+      if (!writtenDuringRecursion) {
+        // Write the transformed file
+        await writeFile(transformedFilePath, transformedContent);
+        scenarioComponentPaths.push(transformedFilePath);
+
+        // Mark as written
+        if (!writtenScenarioComponents[resolvedFilePath]) {
+          writtenScenarioComponents[resolvedFilePath] = [];
+        }
+        writtenScenarioComponents[resolvedFilePath].push(
+          '__transitive_file_written__',
+        );
+      }
+    }
+
+    // ALWAYS rewrite the import in fileContent to point to the transformed file
+    // (even if the transitive copy was already created by another entity)
+    const relativePath = getRelativePath(
+      scenarioComponentPath,
+      transformedFilePath,
+    );
+    // Strip the extension before applying safeFolder - import paths in TypeScript
+    // should NOT include extensions (Next.js resolves .ts/.tsx automatically).
+    // Without this, safeFolder converts ".ts" to "_ts" causing "Module not found" errors.
+    const relativePathWithoutExt = relativePath.replace(
+      /\.(ts|tsx|js|jsx)$/,
+      '',
+    );
+    const safeRelativePath = safeFolder(relativePathWithoutExt);
+
+    // Escape special regex characters in the import path
+    const escapedImportPath = importPath.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+    const importRegex = new RegExp(
+      `(from\\s*["'])${escapedImportPath}(["'])`,
+      'g',
+    );
+    fileContent = fileContent.replace(importRegex, `$1${safeRelativePath}$2`);
+  }
+
+  console.log(
+    `[WriteScenario] POST-LOOP ${entity.name}: remaining imports loop took ${Date.now() - postLoopStartTime}ms`,
+  );
+
+  const scenarioComponentComment = `// This file is auto-generated by CodeYam. Do not edit this file manually.
+// This file contains content for a scenario component:
+// Analyses being written: ${JSON.stringify(fileAnalyses?.map((a) => ({ id: a.id, entityName: a.entityName })))}
+// Entity being written: ${entity.filePath} ${entity.metadata?.exportAlias ?? entity.name} - ${entity.sha}
+// File being written: ${file.path}
+// Root analysis:
+// Analysis: ${rootAnalysis.id}
+// Entity: ${rootAnalysis.entitySha} ${rootAnalysis.entityName}
+// Scenario: ${scenario.id} - ${scenario.name}
+`;
+
+  // Final pass: Rename any namespace imports (import * as X from '...') that have
+  // corresponding mock code using ...X__cyOriginal spread pattern.
+  // This handles cases where:
+  // 1. The import path was rewritten by transitive import handling
+  // 2. The import wasn't caught by the earlier renaming logic
+  // We scan for all __cyOriginal references in the mock code and ensure the imports are renamed.
+  const cyOriginalReferences = fileContent.match(/\.\.\.(\w+)__cyOriginal/g);
+  if (cyOriginalReferences) {
+    const uniqueNames = [
+      ...new Set(
+        cyOriginalReferences.map((ref) =>
+          ref.replace('...', '').replace('__cyOriginal', ''),
+        ),
+      ),
+    ];
+    for (const name of uniqueNames) {
+      // Match namespace imports for this name that haven't been renamed yet
+      const namespaceImportRegex = new RegExp(
+        `(import\\s+\\*\\s+as\\s+)${escapeRegExp(name)}(\\s+from\\s+['"][^'"]*['"])`,
+        'g',
+      );
+      fileContent = fileContent.replace(
+        namespaceImportRegex,
+        `$1${name}__cyOriginal$2`,
+      );
+    }
+  }
+
+  // Use the directive that was extracted at the beginning of processing
+  // This ensures it stays at the very top even after imports are prepended
+  // NOTE: We only preserve "use client" directives, NOT "use server" directives.
+  // Server action files get mocked with objects that aren't async functions,
+  // which would violate Next.js's "use server" requirement:
+  // "A 'use server' file can only export async functions, found object."
+  let finalContent: string;
+  if (extractedDirective && extractedDirective.includes('client')) {
+    finalContent = `${extractedDirective}\n\n${scenarioComponentComment}\n\n${fileContent}`;
+  } else {
+    finalContent = `${scenarioComponentComment}\n\n${fileContent}`;
+  }
+
+  debugLog('About to write final scenario file', {
+    scenarioComponentPath,
+    contentLength: finalContent.length,
+  });
+  await writeFile(scenarioComponentPath, finalContent);
+  debugLog('Successfully wrote scenario file');
+  scenarioComponentPaths.push(scenarioComponentPath);
+
+  console.log(
+    `[WriteScenario] POST-LOOP ${entity.name}: COMPLETE total=${Date.now() - postLoopStartTime}ms`,
+  );
+
+  return { scenarioComponentPaths, writtenScenarioComponents };
+}