Function injectRootComponentToApp

src/nodes/code/utils.ts:214–269

(state: GraphState, cache: CodeCache)

 * Reads existing App.tsx, adds import and renders the root component
 */
export async function injectRootComponentToApp(state: GraphState, cache: CodeCache): Promise<void> {
    try {
        // Check if already injected
        if (isAppInjected(cache)) {
            logger.printInfoLog('⏭️  Skipping App.tsx injection (already injected)');
            return;
        }

        logger.printInfoLog('💉 Injecting root component into App.tsx...');

        // Construct App.tsx path
        const appTsxPath = workspaceManager.resolveAppSrc(state.workspace, 'App.tsx');

        // Read existing App.tsx or use default template
        let appContent: string;
        try {
            appContent = fs.readFileSync(appTsxPath, 'utf8');
        } catch {
            // Use default template if App.tsx doesn't exist
            logger.printWarnLog('App.tsx not found, using default template');
            appContent = DEFAULT_APP_CONTENT;
        }

        // Get root component information
        const rootNode = state.protocol!;
        const componentName = rootNode.data.name || 'RootComponent';
        const componentPath = rootNode.data.path || '';

        // Generate prompt
        const prompt = injectRootComponentPrompt({
            appContent,
            componentName,
            componentPath,
        });

        // Call AI model
        const updatedCode = await callModel({
            question: prompt,
        });

        // Extract code (no markdown blocks expected based on prompt requirements)
        const finalCode = updatedCode.includes('```') ? extractCode(updatedCode) : updatedCode.trim();

        // Write updated App.tsx
        const appFolderPath = path.dirname(appTsxPath);
        writeFile(appFolderPath, 'App.tsx', finalCode);

        // Mark as injected and save immediately
        saveAppInjected(cache, state.workspace);

        logger.printSuccessLog(`Successfully injected ${componentName} into App.tsx`);
    } catch (error) {
        logger.printErrorLog(`Failed to inject root component: ${(error as Error).message}`);
        // Don't throw - allow the process to continue even if injection fails
    }
}