@friggframework/core 2.0.0-next.45 → 2.0.0-next.46

package/database/utils/prisma-runner.js

@@ -0,0 +1,400 @@
+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. Lambda layer path (where the schema actually exists in deployed Lambda)
+    // 2. Local node_modules (where @friggframework/core is installed - production scenario)
+    // 3. Parent node_modules (workspace/monorepo setup)
+    const possiblePaths = [
+        // Lambda layer path - this is where the schema actually exists in deployed Lambda
+        `/opt/nodejs/node_modules/generated/prisma-${dbType}/schema.prisma`,
+        // 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;
+
+        // In Lambda, also check the layer path (/opt/nodejs/node_modules)
+        const lambdaLayerClientPath = `/opt/nodejs/node_modules/generated/prisma-${dbType}/client.js`;
+
+        const clientExists = fs.existsSync(generatedClientPath) || (isLambdaEnvironment && fs.existsSync(lambdaLayerClientPath));
+
+        if (clientExists) {
+            const foundPath = fs.existsSync(generatedClientPath) ? generatedClientPath : lambdaLayerClientPath;
+            if (verbose) {
+                console.log(chalk.gray(`✓ Prisma client already generated at: ${foundPath}`));
+            }
+            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 prismaBin = getPrismaBinaryPath();
+
+        // Use direct path instead of npx to avoid WASM file resolution issues
+        const isDirectBinary = prismaBin !== 'npx prisma';
+        const command = isDirectBinary
+            ? `${prismaBin} migrate status --schema=${schemaPath}`
+            : `npx prisma migrate status --schema=${schemaPath}`;
+
+        const output = execSync(
+            command,
+            {
+                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
+        };
+    }
+}
+
+/**
+ * Gets the path to the Prisma CLI entry point
+ * 
+ * IMPORTANT: We invoke prisma/build/index.js directly instead of .bin/prisma
+ * because .bin/prisma uses __dirname to find WASM files, and when the symlink
+ * is resolved during Lambda packaging, __dirname points to .bin/ instead of
+ * prisma/build/, causing WASM files to not be found.
+ * 
+ * @returns {string} Command to run Prisma CLI (e.g., 'node /path/to/index.js' or 'npx prisma')
+ */
+function getPrismaBinaryPath() {
+    const fs = require('fs');
+
+    // Check function's bundled Prisma (Lambda) - use actual CLI location
+    const functionPrisma = '/var/task/node_modules/prisma/build/index.js';
+    if (fs.existsSync(functionPrisma)) {
+        return `node ${functionPrisma}`;
+    }
+
+    // Check Lambda layer path - use actual CLI location
+    const layerPrisma = '/opt/nodejs/node_modules/prisma/build/index.js';
+    if (fs.existsSync(layerPrisma)) {
+        return `node ${layerPrisma}`;
+    }
+
+    // Check local node_modules - use actual CLI location
+    const localPrisma = path.join(process.cwd(), 'node_modules', 'prisma', 'build', 'index.js');
+    if (fs.existsSync(localPrisma)) {
+        return `node ${localPrisma}`;
+    }
+
+    // Fallback to npx (local dev)
+    return 'npx prisma';
+}
+
+/**
+ * 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}`));
+            }
+
+            // Execute the command (prismaBin might be 'node /path/to/index.js' or 'npx prisma')
+            const [executable, ...executableArgs] = prismaBin.split(' ');
+            const fullArgs = [...executableArgs, ...args];
+
+            const proc = spawn(executable, fullArgs, {
+                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
+};

package/database/utils/prisma-schema-parser.js

@@ -0,0 +1,182 @@
+/**
+ * Prisma Schema Parser for MongoDB Collections
+ *
+ * Dynamically parses the Prisma schema file to extract MongoDB collection names.
+ * This ensures collection names stay in sync with the schema without hardcoding.
+ *
+ * Handles:
+ * - @@map() directives (custom collection names)
+ * - Models without @@map() (uses model name)
+ * - Comments and whitespace
+ * - Multiple schema file locations
+ */
+
+const fs = require('fs');
+const path = require('path');
+
+/**
+ * Parse Prisma schema file to extract collection names
+ *
+ * Reads the schema.prisma file and extracts all model definitions,
+ * returning the actual MongoDB collection names (from @@map directives).
+ *
+ * @param {string} schemaPath - Path to schema.prisma file
+ * @returns {Promise<string[]>} Array of collection names
+ *
+ * @example
+ * ```js
+ * const collections = await parseCollectionsFromSchema('./prisma/schema.prisma');
+ * // Returns: ['User', 'Token', 'Credential', ...]
+ * ```
+ */
+async function parseCollectionsFromSchema(schemaPath) {
+    try {
+        const schemaContent = await fs.promises.readFile(schemaPath, 'utf-8');
+        return extractCollectionNames(schemaContent);
+    } catch (error) {
+        throw new Error(
+            `Failed to parse Prisma schema at ${schemaPath}: ${error.message}`
+        );
+    }
+}
+
+/**
+ * Synchronous version of parseCollectionsFromSchema
+ *
+ * @param {string} schemaPath - Path to schema.prisma file
+ * @returns {string[]} Array of collection names
+ */
+function parseCollectionsFromSchemaSync(schemaPath) {
+    try {
+        const schemaContent = fs.readFileSync(schemaPath, 'utf-8');
+        return extractCollectionNames(schemaContent);
+    } catch (error) {
+        throw new Error(
+            `Failed to parse Prisma schema at ${schemaPath}: ${error.message}`
+        );
+    }
+}
+
+/**
+ * Extract collection names from Prisma schema content
+ *
+ * Parses the schema content to find:
+ * 1. All model definitions
+ * 2. Their @@map() directives (if present)
+ * 3. Falls back to model name if no @@map()
+ *
+ * @param {string} schemaContent - Content of schema.prisma file
+ * @returns {string[]} Array of collection names
+ * @private
+ */
+function extractCollectionNames(schemaContent) {
+    const collections = [];
+
+    // Match model blocks: "model ModelName { ... }"
+    // Using non-greedy match to handle multiple models
+    const modelRegex = /model\s+(\w+)\s*\{([^}]+)\}/g;
+
+    let match;
+    while ((match = modelRegex.exec(schemaContent)) !== null) {
+        const modelName = match[1];
+        const modelBody = match[2];
+
+        // Look for @@map("CollectionName") directive
+        const mapMatch = modelBody.match(/@@map\s*\(\s*["'](\w+)["']\s*\)/);
+
+        if (mapMatch) {
+            // Use mapped collection name
+            collections.push(mapMatch[1]);
+        } else {
+            // Use model name as collection name (Prisma default)
+            collections.push(modelName);
+        }
+    }
+
+    return collections;
+}
+
+/**
+ * Find Prisma MongoDB schema file
+ *
+ * Searches for the schema.prisma file in common locations:
+ * 1. prisma-mongodb/schema.prisma (Frigg convention)
+ * 2. prisma/schema.prisma (Prisma default)
+ * 3. schema.prisma (root)
+ *
+ * @param {string} startDir - Directory to start searching from
+ * @returns {string|null} Path to schema file, or null if not found
+ */
+function findMongoDBSchemaFile(startDir = __dirname) {
+    // Start from database directory and work up
+    const baseDir = path.resolve(startDir, '../..');
+
+    const searchPaths = [
+        path.join(baseDir, 'prisma-mongodb', 'schema.prisma'),
+        path.join(baseDir, 'prisma', 'schema.prisma'),
+        path.join(baseDir, 'schema.prisma'),
+    ];
+
+    for (const schemaPath of searchPaths) {
+        if (fs.existsSync(schemaPath)) {
+            return schemaPath;
+        }
+    }
+
+    return null;
+}
+
+/**
+ * Get MongoDB collection names from Prisma schema
+ *
+ * Convenience function that finds and parses the schema automatically.
+ *
+ * @returns {Promise<string[]>} Array of collection names
+ * @throws {Error} If schema file not found or parsing fails
+ *
+ * @example
+ * ```js
+ * const collections = await getCollectionsFromSchema();
+ * await ensureCollectionsExist(collections);
+ * ```
+ */
+async function getCollectionsFromSchema() {
+    const schemaPath = findMongoDBSchemaFile();
+
+    if (!schemaPath) {
+        throw new Error(
+            'Could not find Prisma MongoDB schema file. ' +
+            'Searched: prisma-mongodb/schema.prisma, prisma/schema.prisma, schema.prisma'
+        );
+    }
+
+    return await parseCollectionsFromSchema(schemaPath);
+}
+
+/**
+ * Synchronous version of getCollectionsFromSchema
+ *
+ * @returns {string[]} Array of collection names
+ * @throws {Error} If schema file not found or parsing fails
+ */
+function getCollectionsFromSchemaSync() {
+    const schemaPath = findMongoDBSchemaFile();
+
+    if (!schemaPath) {
+        throw new Error(
+            'Could not find Prisma MongoDB schema file. ' +
+            'Searched: prisma-mongodb/schema.prisma, prisma/schema.prisma, schema.prisma'
+        );
+    }
+
+    return parseCollectionsFromSchemaSync(schemaPath);
+}
+
+module.exports = {
+    parseCollectionsFromSchema,
+    parseCollectionsFromSchemaSync,
+    extractCollectionNames,
+    findMongoDBSchemaFile,
+    getCollectionsFromSchema,
+    getCollectionsFromSchemaSync,
+};

package/encrypt/Cryptor.js

@@ -17,7 +17,7 @@
  */
 
 const crypto = require('crypto');
-const AWS = require('aws-sdk');
+const { KMSClient, GenerateDataKeyCommand, DecryptCommand } = require('@aws-sdk/client-kms');
 const aes = require('./aes');
 
 class Cryptor {
@@ -27,16 +27,15 @@ class Cryptor {
 
     async generateDataKey() {
         if (this.shouldUseAws) {
-            const kmsClient = new AWS.KMS();
-            const dataKey = await kmsClient
-                .generateDataKey({
-                    KeyId: process.env.KMS_KEY_ARN,
-                    KeySpec: 'AES_256',
-                })
-                .promise();
+            const kmsClient = new KMSClient({});
+            const command = new GenerateDataKeyCommand({
+                KeyId: process.env.KMS_KEY_ARN,
+                KeySpec: 'AES_256',
+            });
+            const dataKey = await kmsClient.send(command);
 
             const keyId = Buffer.from(dataKey.KeyId).toString('base64');
-            const encryptedKey = dataKey.CiphertextBlob.toString('base64');
+            const encryptedKey = Buffer.from(dataKey.CiphertextBlob).toString('base64');
             const plaintext = dataKey.Plaintext;
             return { keyId, encryptedKey, plaintext };
         }
@@ -70,13 +69,12 @@ class Cryptor {
 
     async decryptDataKey(keyId, encryptedKey) {
         if (this.shouldUseAws) {
-            const kmsClient = new AWS.KMS();
-            const dataKey = await kmsClient
-                .decrypt({
-                    KeyId: keyId,
-                    CiphertextBlob: encryptedKey,
-                })
-                .promise();
+            const kmsClient = new KMSClient({});
+            const command = new DecryptCommand({
+                KeyId: keyId,
+                CiphertextBlob: encryptedKey,
+            });
+            const dataKey = await kmsClient.send(command);
 
             return dataKey.Plaintext;
         }

package/generated/prisma-mongodb/client.d.ts

@@ -0,0 +1 @@
+export * from "./index"

package/generated/prisma-mongodb/client.js

@@ -0,0 +1,4 @@
+
+/* !!! This is code generated by Prisma. Do not edit directly. !!!
+/* eslint-disable */
+module.exports = { ...require('.') }

package/generated/prisma-mongodb/default.d.ts

@@ -0,0 +1 @@
+export * from "./index"

package/generated/prisma-mongodb/default.js

@@ -0,0 +1,4 @@
+
+/* !!! This is code generated by Prisma. Do not edit directly. !!!
+/* eslint-disable */
+module.exports = { ...require('#main-entry-point') }

package/generated/prisma-mongodb/edge.d.ts

@@ -0,0 +1 @@
+export * from "./default"