@friggframework/core 2.0.0--canary.463.62579dd.0 → 2.0.0--canary.461.84ff4f5.0

package/database/utils/prisma-runner.js

@@ -0,0 +1,377 @@
+const { execSync, spawn } = require('child_process');
+const path = require('path');
+const fs = require('fs');
+const chalk = require('chalk');
+
+/**
+ * Prisma Command Runner Utility
+ * Handles execution of Prisma CLI commands for database setup
+ */
+
+/**
+ * Gets the path to the Prisma schema file for the database type
+ * @param {'mongodb'|'postgresql'} dbType - Database type
+ * @param {string} projectRoot - Project root directory
+ * @returns {string} Absolute path to schema file
+ * @throws {Error} If schema file doesn't exist
+ */
+function getPrismaSchemaPath(dbType, projectRoot = process.cwd()) {
+    // Try multiple locations for the schema file
+    // Priority order:
+    // 1. Local node_modules (where @friggframework/core is installed - production scenario)
+    // 2. Parent node_modules (workspace/monorepo setup)
+    const possiblePaths = [
+        // Check where Frigg is installed via npm (production scenario)
+        path.join(projectRoot, 'node_modules', '@friggframework', 'core', `prisma-${dbType}`, 'schema.prisma'),
+        path.join(projectRoot, '..', 'node_modules', '@friggframework', 'core', `prisma-${dbType}`, 'schema.prisma')
+    ];
+
+    for (const schemaPath of possiblePaths) {
+        if (fs.existsSync(schemaPath)) {
+            return schemaPath;
+        }
+    }
+
+    // If not found in any location, throw error
+    throw new Error(
+        `Prisma schema not found at:\n${possiblePaths.join('\n')}\n\n` +
+        'Ensure @friggframework/core is installed.'
+    );
+}
+
+/**
+ * Runs prisma generate for the specified database type
+ * @param {'mongodb'|'postgresql'} dbType - Database type
+ * @param {boolean} verbose - Enable verbose output
+ * @returns {Promise<Object>} { success: boolean, output?: string, error?: string }
+ */
+async function runPrismaGenerate(dbType, verbose = false) {
+    try {
+        const schemaPath = getPrismaSchemaPath(dbType);
+
+        // Check if Prisma client already exists (e.g., in Lambda or pre-generated)
+        const generatedClientPath = path.join(path.dirname(path.dirname(schemaPath)), 'generated', `prisma-${dbType}`, 'client.js');
+        const isLambdaEnvironment = !!process.env.AWS_LAMBDA_FUNCTION_NAME || !!process.env.LAMBDA_TASK_ROOT;
+
+        if (fs.existsSync(generatedClientPath)) {
+            if (verbose) {
+                console.log(chalk.gray(`✓ Prisma client already generated at: ${generatedClientPath}`));
+            }
+            if (isLambdaEnvironment) {
+                if (verbose) {
+                    console.log(chalk.gray('Skipping generation in Lambda environment (using pre-generated client)'));
+                }
+                return {
+                    success: true,
+                    output: 'Using pre-generated Prisma client (Lambda environment)'
+                };
+            }
+        }
+
+        if (verbose) {
+            console.log(chalk.gray(`Running: npx prisma generate --schema=${schemaPath}`));
+        }
+
+        const output = execSync(
+            `npx prisma generate --schema=${schemaPath}`,
+            {
+                encoding: 'utf8',
+                stdio: verbose ? 'inherit' : 'pipe',
+                env: {
+                    ...process.env,
+                    // Suppress Prisma telemetry prompts
+                    PRISMA_HIDE_UPDATE_MESSAGE: '1'
+                }
+            }
+        );
+
+        return {
+            success: true,
+            output: verbose ? 'Generated successfully' : output
+        };
+
+    } catch (error) {
+        return {
+            success: false,
+            error: error.message,
+            output: error.stdout?.toString() || error.stderr?.toString()
+        };
+    }
+}
+
+/**
+ * Checks database migration status
+ * @param {'mongodb'|'postgresql'} dbType - Database type
+ * @returns {Promise<Object>} { upToDate: boolean, pendingMigrations?: number, error?: string }
+ */
+async function checkDatabaseState(dbType) {
+    try {
+        // Only applicable for PostgreSQL (MongoDB uses db push)
+        if (dbType !== 'postgresql') {
+            return { upToDate: true };
+        }
+
+        const schemaPath = getPrismaSchemaPath(dbType);
+
+        const output = execSync(
+            `npx prisma migrate status --schema=${schemaPath}`,
+            {
+                encoding: 'utf8',
+                stdio: 'pipe',
+                env: {
+                    ...process.env,
+                    PRISMA_HIDE_UPDATE_MESSAGE: '1'
+                }
+            }
+        );
+
+        if (output.includes('Database schema is up to date')) {
+            return { upToDate: true };
+        }
+
+        // Parse pending migrations count
+        const pendingMatch = output.match(/(\d+) migration/);
+        const pendingMigrations = pendingMatch ? parseInt(pendingMatch[1]) : 0;
+
+        return {
+            upToDate: false,
+            pendingMigrations
+        };
+
+    } catch (error) {
+        // If migrate status fails, database might not be initialized
+        return {
+            upToDate: false,
+            error: error.message
+        };
+    }
+}
+
+/**
+ * Get Prisma binary path for Lambda environment
+ * Checks multiple locations in priority order:
+ * 1. Function's bundled Prisma (/var/task/node_modules/.bin/prisma) - for standalone functions
+ * 2. Layer's Prisma (/opt/nodejs/node_modules/.bin/prisma) - for functions using Prisma layer with CLI
+ * 3. Fallback to npx for local development
+ */
+function getPrismaBinaryPath() {
+    const fs = require('fs');
+    const isLambdaEnvironment = !!process.env.AWS_LAMBDA_FUNCTION_NAME || !!process.env.LAMBDA_TASK_ROOT;
+
+    if (!isLambdaEnvironment) {
+        return 'npx';
+    }
+
+    // Check function's own node_modules first (standalone dbMigrate)
+    const functionPrisma = '/var/task/node_modules/.bin/prisma';
+    if (fs.existsSync(functionPrisma)) {
+        return functionPrisma;
+    }
+
+    // Fall back to layer path (functions using Prisma layer with CLI)
+    const layerPrisma = '/opt/nodejs/node_modules/.bin/prisma';
+    if (fs.existsSync(layerPrisma)) {
+        return layerPrisma;
+    }
+
+    // Should not reach here in Lambda, but provide fallback
+    console.warn('⚠️  Prisma binary not found in expected Lambda paths, using npx');
+    return 'npx';
+}
+
+/**
+ * Runs Prisma migrate for PostgreSQL
+ * @param {'dev'|'deploy'} command - Migration command (dev or deploy)
+ * @param {boolean} verbose - Enable verbose output
+ * @returns {Promise<Object>} { success: boolean, output?: string, error?: string }
+ */
+async function runPrismaMigrate(command = 'dev', verbose = false) {
+    return new Promise((resolve) => {
+        try {
+            const schemaPath = getPrismaSchemaPath('postgresql');
+
+            // Get Prisma binary path (checks multiple locations)
+            const isLambdaEnvironment = !!process.env.AWS_LAMBDA_FUNCTION_NAME || !!process.env.LAMBDA_TASK_ROOT;
+            const prismaBin = getPrismaBinaryPath();
+
+            // Determine args based on whether we're using direct binary or npx
+            // Direct binary (e.g., /var/task/node_modules/.bin/prisma): ['migrate', command, ...]
+            // npx (local dev or fallback): ['prisma', 'migrate', command, ...]
+            const isDirectBinary = prismaBin !== 'npx';
+            const args = isDirectBinary
+                ? ['migrate', command, '--schema', schemaPath]
+                : ['prisma', 'migrate', command, '--schema', schemaPath];
+
+            if (verbose) {
+                const displayCmd = isDirectBinary
+                    ? `${prismaBin} ${args.join(' ')}`
+                    : `npx ${args.join(' ')}`;
+                console.log(chalk.gray(`Running: ${displayCmd}`));
+            }
+
+            const proc = spawn(prismaBin, args, {
+                stdio: 'inherit',
+                env: {
+                    ...process.env,
+                    PRISMA_HIDE_UPDATE_MESSAGE: '1'
+                }
+            });
+
+            proc.on('error', (error) => {
+                resolve({
+                    success: false,
+                    error: error.message
+                });
+            });
+
+            proc.on('close', (code) => {
+                if (code === 0) {
+                    resolve({
+                        success: true,
+                        output: 'Migration completed successfully'
+                    });
+                } else {
+                    resolve({
+                        success: false,
+                        error: `Migration process exited with code ${code}`
+                    });
+                }
+            });
+
+        } catch (error) {
+            resolve({
+                success: false,
+                error: error.message
+            });
+        }
+    });
+}
+
+/**
+ * Runs Prisma db push for MongoDB
+ * @param {boolean} verbose - Enable verbose output
+ * @param {boolean} nonInteractive - Run in non-interactive mode (accepts data loss, for Lambda/CI)
+ * @returns {Promise<Object>} { success: boolean, output?: string, error?: string }
+ */
+async function runPrismaDbPush(verbose = false, nonInteractive = false) {
+    return new Promise((resolve) => {
+        try {
+            const schemaPath = getPrismaSchemaPath('mongodb');
+
+            const args = [
+                'prisma',
+                'db',
+                'push',
+                '--schema',
+                schemaPath,
+                '--skip-generate' // We generate separately
+            ];
+
+            // Add non-interactive flag for Lambda/CI environments
+            if (nonInteractive) {
+                args.push('--accept-data-loss');
+            }
+
+            if (verbose) {
+                console.log(chalk.gray(`Running: npx ${args.join(' ')}`));
+            }
+
+            if (nonInteractive) {
+                console.log(chalk.yellow('⚠️  Non-interactive mode: Data loss will be automatically accepted'));
+            } else {
+                console.log(chalk.yellow('⚠️  Interactive mode: You may be prompted if schema changes cause data loss'));
+            }
+
+            const proc = spawn('npx', args, {
+                stdio: nonInteractive ? 'pipe' : 'inherit', // Use pipe for non-interactive to capture output
+                env: {
+                    ...process.env,
+                    PRISMA_HIDE_UPDATE_MESSAGE: '1'
+                }
+            });
+
+            let stdout = '';
+            let stderr = '';
+
+            // Capture output in non-interactive mode
+            if (nonInteractive) {
+                if (proc.stdout) {
+                    proc.stdout.on('data', (data) => {
+                        stdout += data.toString();
+                        if (verbose) {
+                            process.stdout.write(data);
+                        }
+                    });
+                }
+                if (proc.stderr) {
+                    proc.stderr.on('data', (data) => {
+                        stderr += data.toString();
+                        if (verbose) {
+                            process.stderr.write(data);
+                        }
+                    });
+                }
+            }
+
+            proc.on('error', (error) => {
+                resolve({
+                    success: false,
+                    error: error.message
+                });
+            });
+
+            proc.on('close', (code) => {
+                if (code === 0) {
+                    resolve({
+                        success: true,
+                        output: nonInteractive ? stdout || 'Database push completed successfully' : 'Database push completed successfully'
+                    });
+                } else {
+                    resolve({
+                        success: false,
+                        error: `Database push process exited with code ${code}`,
+                        output: stderr || stdout
+                    });
+                }
+            });
+
+        } catch (error) {
+            resolve({
+                success: false,
+                error: error.message
+            });
+        }
+    });
+}
+
+/**
+ * Determines migration command based on STAGE environment variable
+ * @param {string} stage - Stage from CLI option or environment
+ * @returns {'dev'|'deploy'}
+ */
+function getMigrationCommand(stage) {
+    // Always use 'deploy' in Lambda environment (it's non-interactive and doesn't create migrations)
+    const isLambdaEnvironment = !!process.env.AWS_LAMBDA_FUNCTION_NAME || !!process.env.LAMBDA_TASK_ROOT;
+    if (isLambdaEnvironment) {
+        return 'deploy';
+    }
+
+    const normalizedStage = (stage || process.env.STAGE || 'development').toLowerCase();
+
+    const developmentStages = ['dev', 'local', 'test', 'development'];
+
+    if (developmentStages.includes(normalizedStage)) {
+        return 'dev';
+    }
+
+    return 'deploy';
+}
+
+module.exports = {
+    getPrismaSchemaPath,
+    runPrismaGenerate,
+    checkDatabaseState,
+    runPrismaMigrate,
+    runPrismaDbPush,
+    getMigrationCommand
+};