@mtkn/mega-agent 0.1.1 → 0.3.0

package/scripts/prisma-sync.mjs

@@ -6,19 +6,27 @@
  * Reads the mega-agent reference schema and appends/updates
  * enum and model blocks into the host application's schema.
  *
+ * Supports ID type customization:
+ *   --id-type        Primary key strategy: cuid (default), uuid, autoincrement
+ *   --user-id-type   User ID field type: int (default), string
+ *
  * Usage:
- *   node scripts/prisma-sync.mjs --target prisma/schema.prisma
  *   npx mega-agent-prisma-sync --target prisma/schema.prisma
+ *   npx mega-agent-prisma-sync --target prisma/schema.prisma --id-type uuid --user-id-type string
  *
  * Options:
- *   --source  Path to mega-agent schema (default: auto-detect)
- *   --target  Path to host schema (required)
- *   --dry-run Print result without writing
+ *   --source         Path to mega-agent schema (default: auto-detect)
+ *   --target         Path to host schema (required)
+ *   --id-type        Primary key strategy: cuid | uuid | autoincrement (default: cuid)
+ *   --user-id-type   User ID field type: int | string (default: int)
+ *   --no-interactive Skip interactive prompts, use defaults or flags
+ *   --dry-run        Print result without writing
  */
 
 import { readFileSync, writeFileSync, existsSync } from 'node:fs';
-import { resolve, dirname, join } from 'node:path';
+import { resolve, dirname } from 'node:path';
 import { fileURLToPath } from 'node:url';
+import { createInterface } from 'node:readline/promises';
 
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = dirname(__filename);
@@ -26,17 +34,40 @@ const __dirname = dirname(__filename);
 const MARKER_START = '// ═══ @7style/mega-agent MODELS START ═══';
 const MARKER_END = '// ═══ @7style/mega-agent MODELS END ═══';
 
+const VALID_ID_TYPES = ['cuid', 'uuid', 'autoincrement'];
+const VALID_USER_ID_TYPES = ['int', 'string'];
+
+// Foreign key fields that reference mega-agent model IDs (need type change for autoincrement)
+const INTERNAL_FK_FIELDS = {
+  AiChatMessage: ['chatId'],
+  AiWorkflowExecution: ['workflowId'],
+  AiReport: ['artifactId'],
+};
+
 // ── CLI args ──────────────────────────────────────────────────
 
 function parseArgs() {
   const args = process.argv.slice(2);
-  const parsed = { source: null, target: null, dryRun: false };
+  const parsed = {
+    source: null,
+    target: null,
+    idType: null,
+    userIdType: null,
+    noInteractive: false,
+    dryRun: false,
+  };
 
   for (let i = 0; i < args.length; i++) {
     if (args[i] === '--source' && args[i + 1]) {
       parsed.source = args[++i];
     } else if (args[i] === '--target' && args[i + 1]) {
       parsed.target = args[++i];
+    } else if (args[i] === '--id-type' && args[i + 1]) {
+      parsed.idType = args[++i];
+    } else if (args[i] === '--user-id-type' && args[i + 1]) {
+      parsed.userIdType = args[++i];
+    } else if (args[i] === '--no-interactive') {
+      parsed.noInteractive = true;
     } else if (args[i] === '--dry-run') {
       parsed.dryRun = true;
     } else if (args[i] === '--help' || args[i] === '-h') {
@@ -45,12 +76,29 @@ function parseArgs() {
 
 Usage:
   npx mega-agent-prisma-sync --target prisma/schema.prisma
+  npx mega-agent-prisma-sync --target prisma/schema.prisma --id-type uuid --user-id-type string
 
 Options:
-  --source  Path to mega-agent reference schema (auto-detected by default)
-  --target  Path to host schema.prisma (required)
-  --dry-run Print result without writing to file
-  --help    Show this help message
+  --source         Path to mega-agent reference schema (auto-detected by default)
+  --target         Path to host schema.prisma (required)
+  --id-type        Primary key strategy: cuid | uuid | autoincrement (default: cuid)
+  --user-id-type   User ID field type: int | string (default: int)
+  --no-interactive Skip interactive prompts, use defaults or previous config
+  --dry-run        Print result without writing to file
+  --help           Show this help message
+
+Examples:
+  # Interactive mode (first time setup)
+  npx mega-agent-prisma-sync --target prisma/schema.prisma
+
+  # UUID primary keys + String user IDs (for UUID-based User models)
+  npx mega-agent-prisma-sync --target prisma/schema.prisma --id-type uuid --user-id-type string
+
+  # Autoincrement primary keys (integer IDs)
+  npx mega-agent-prisma-sync --target prisma/schema.prisma --id-type autoincrement
+
+  # CI/CD: skip prompts, use previous config or defaults
+  npx mega-agent-prisma-sync --target prisma/schema.prisma --no-interactive
 `);
       process.exit(0);
     }
@@ -59,6 +107,61 @@ Options:
   return parsed;
 }
 
+// ── Config detection from existing markers ────────────────────
+
+/**
+ * Detect previous ID config from existing marker comments in target schema.
+ * Returns { idType, userIdType } or null if not found.
+ */
+function detectPreviousConfig(targetSchema) {
+  const configMatch = targetSchema.match(
+    /\/\/ Config: id-type=(\w+), user-id-type=(\w+)/
+  );
+  if (configMatch) {
+    return {
+      idType: configMatch[1],
+      userIdType: configMatch[2],
+    };
+  }
+  return null;
+}
+
+// ── Interactive prompts ───────────────────────────────────────
+
+async function promptForConfig(defaults = {}) {
+  const rl = createInterface({
+    input: process.stdin,
+    output: process.stdout,
+  });
+
+  console.log('\n📋 mega-agent Prisma Schema Configuration\n');
+
+  const idDefault = defaults.idType || 'cuid';
+  const idAnswer = await rl.question(
+    `  Primary key strategy for mega-agent models? (cuid/uuid/autoincrement) [${idDefault}]: `
+  );
+  const idType = idAnswer.trim().toLowerCase() || idDefault;
+
+  const userIdDefault = defaults.userIdType || 'int';
+  const userIdAnswer = await rl.question(
+    `  User ID type in your app? (int/string) [${userIdDefault}]: `
+  );
+  const userIdType = userIdAnswer.trim().toLowerCase() || userIdDefault;
+
+  rl.close();
+
+  if (!VALID_ID_TYPES.includes(idType)) {
+    console.error(`\nError: Invalid id-type "${idType}". Must be one of: ${VALID_ID_TYPES.join(', ')}`);
+    process.exit(1);
+  }
+  if (!VALID_USER_ID_TYPES.includes(userIdType)) {
+    console.error(`\nError: Invalid user-id-type "${userIdType}". Must be one of: ${VALID_USER_ID_TYPES.join(', ')}`);
+    process.exit(1);
+  }
+
+  return { idType, userIdType };
+}
+
 // ── Schema parsing ────────────────────────────────────────────
 
 /**
@@ -104,16 +207,78 @@ function extractBlocks(schema) {
   return blocks;
 }
 
+// ── Schema transformation ─────────────────────────────────────
+
+/**
+ * Transform a model block based on ID type configuration.
+ * Enum blocks are returned unchanged.
+ */
+function transformBlock(block, idType, userIdType) {
+  if (block.type === 'enum') return block;
+
+  const lines = block.content.split('\n');
+  const fkFields = INTERNAL_FK_FIELDS[block.name] || [];
+
+  const transformedLines = lines.map((line) => {
+    const trimmed = line.trim();
+
+    // 1. Transform primary key @id @default(cuid()) lines
+    if (trimmed.includes('@id') && trimmed.includes('@default(cuid())')) {
+      if (idType === 'uuid') {
+        return line.replace('@default(cuid())', '@default(uuid())');
+      }
+      if (idType === 'autoincrement') {
+        return line
+          .replace(/(\s+)String(\s+)@id/, '$1Int   $2@id')
+          .replace('@default(cuid())', '@default(autoincrement())');
+      }
+    }
+
+    // 2. Transform userId fields
+    if (userIdType === 'string' && /^\s*userId\s+Int\s/.test(line)) {
+      return line.replace(/userId(\s+)Int/, 'userId$1String');
+    }
+
+    // 3. Transform internal FK fields when idType is autoincrement
+    if (idType === 'autoincrement') {
+      for (const fkField of fkFields) {
+        // Match: fieldName String (with optional ?)
+        const fkPattern = new RegExp(`^(\\s*${fkField}\\s+)String(\\??)(\\s)`);
+        if (fkPattern.test(line)) {
+          return line.replace(fkPattern, `$1Int$2   $3`);
+        }
+      }
+    }
+
+    return line;
+  });
+
+  return {
+    ...block,
+    content: transformedLines.join('\n'),
+  };
+}
+
 // ── Main ──────────────────────────────────────────────────────
 
-function main() {
-  const { source, target, dryRun } = parseArgs();
+async function main() {
+  const { source, target, idType: flagIdType, userIdType: flagUserIdType, noInteractive, dryRun } = parseArgs();
 
   if (!target) {
     console.error('Error: --target is required. Example: --target prisma/schema.prisma');
     process.exit(1);
   }
 
+  // Validate flags if provided
+  if (flagIdType && !VALID_ID_TYPES.includes(flagIdType)) {
+    console.error(`Error: Invalid --id-type "${flagIdType}". Must be one of: ${VALID_ID_TYPES.join(', ')}`);
+    process.exit(1);
+  }
+  if (flagUserIdType && !VALID_USER_ID_TYPES.includes(flagUserIdType)) {
+    console.error(`Error: Invalid --user-id-type "${flagUserIdType}". Must be one of: ${VALID_USER_ID_TYPES.join(', ')}`);
+    process.exit(1);
+  }
+
   // Resolve source path
   const sourcePath = source
     ? resolve(source)
@@ -138,17 +303,55 @@ function main() {
   const sourceSchema = readFileSync(sourcePath, 'utf-8');
   const targetSchema = readFileSync(targetPath, 'utf-8');
 
+  // ── Determine ID config ───────────────────────────────────
+  let idType, userIdType;
+
+  if (flagIdType && flagUserIdType) {
+    // Both flags provided → use them directly
+    idType = flagIdType;
+    userIdType = flagUserIdType;
+  } else {
+    // Try to detect from existing markers
+    const previousConfig = detectPreviousConfig(targetSchema);
+
+    if (flagIdType || flagUserIdType) {
+      // Partial flags: use flag + previous/default for the other
+      idType = flagIdType || previousConfig?.idType || 'cuid';
+      userIdType = flagUserIdType || previousConfig?.userIdType || 'int';
+    } else if (noInteractive) {
+      // No interactive: use previous config or defaults
+      idType = previousConfig?.idType || 'cuid';
+      userIdType = previousConfig?.userIdType || 'int';
+      if (previousConfig) {
+        console.log(`\nUsing previous config: id-type=${idType}, user-id-type=${userIdType}`);
+      } else {
+        console.log(`\nUsing defaults: id-type=${idType}, user-id-type=${userIdType}`);
+      }
+    } else {
+      // Interactive mode
+      const config = await promptForConfig(previousConfig || {});
+      idType = config.idType;
+      userIdType = config.userIdType;
+    }
+  }
+
+  console.log(`\nConfig: id-type=${idType}, user-id-type=${userIdType}`);
+
   // Extract blocks from source
   const blocks = extractBlocks(sourceSchema);
   console.log(`Found ${blocks.length} blocks (${blocks.filter(b => b.type === 'enum').length} enums, ${blocks.filter(b => b.type === 'model').length} models)`);
 
+  // Transform blocks based on config
+  const transformedBlocks = blocks.map((b) => transformBlock(b, idType, userIdType));
+
   // Build the managed section
   const managedContent = [
     MARKER_START,
     '// Auto-generated by @7style/mega-agent prisma-sync. Do not edit manually.',
     `// Last sync: ${new Date().toISOString()}`,
+    `// Config: id-type=${idType}, user-id-type=${userIdType}`,
     '',
-    ...blocks.map(b => b.content),
+    ...transformedBlocks.map(b => b.content),
     '',
     MARKER_END,
   ].join('\n');