@mtkn/mega-agent 0.1.0

package/scripts/prisma-sync.mjs

@@ -0,0 +1,187 @@
+#!/usr/bin/env node
+
+/**
+ * @7style/mega-agent — Prisma Schema Sync
+ *
+ * Reads the mega-agent reference schema and appends/updates
+ * enum and model blocks into the host application's schema.
+ *
+ * Usage:
+ *   node scripts/prisma-sync.mjs --target prisma/schema.prisma
+ *   npx mega-agent-prisma-sync --target prisma/schema.prisma
+ *
+ * Options:
+ *   --source  Path to mega-agent schema (default: auto-detect)
+ *   --target  Path to host schema (required)
+ *   --dry-run Print result without writing
+ */
+
+import { readFileSync, writeFileSync, existsSync } from 'node:fs';
+import { resolve, dirname, join } from 'node:path';
+import { fileURLToPath } from 'node:url';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+
+const MARKER_START = '// ═══ @7style/mega-agent MODELS START ═══';
+const MARKER_END = '// ═══ @7style/mega-agent MODELS END ═══';
+
+// ── CLI args ──────────────────────────────────────────────────
+
+function parseArgs() {
+  const args = process.argv.slice(2);
+  const parsed = { source: null, target: null, 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] === '--dry-run') {
+      parsed.dryRun = true;
+    } else if (args[i] === '--help' || args[i] === '-h') {
+      console.log(`
+@7style/mega-agent — Prisma Schema Sync
+
+Usage:
+  npx mega-agent-prisma-sync --target prisma/schema.prisma
+
+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
+`);
+      process.exit(0);
+    }
+  }
+
+  return parsed;
+}
+
+// ── Schema parsing ────────────────────────────────────────────
+
+/**
+ * Extract enum and model blocks from a Prisma schema string.
+ * Skips comment-only blocks before the first enum/model.
+ */
+function extractBlocks(schema) {
+  const lines = schema.split('\n');
+  const blocks = [];
+  let current = null;
+  let depth = 0;
+
+  for (const line of lines) {
+    const trimmed = line.trim();
+
+    // Start of enum or model
+    const match = trimmed.match(/^(enum|model)\s+(\w+)\s*\{/);
+    if (match && !current) {
+      current = { type: match[1], name: match[2], lines: [line] };
+      depth = 1;
+      continue;
+    }
+
+    if (current) {
+      current.lines.push(line);
+
+      if (trimmed.includes('{')) depth++;
+      if (trimmed.includes('}')) depth--;
+
+      // Closing brace at depth 0 → block complete
+      if (depth <= 0) {
+        blocks.push({
+          type: current.type,
+          name: current.name,
+          content: current.lines.join('\n'),
+        });
+        current = null;
+        depth = 0;
+      }
+    }
+  }
+
+  return blocks;
+}
+
+// ── Main ──────────────────────────────────────────────────────
+
+function main() {
+  const { source, target, dryRun } = parseArgs();
+
+  if (!target) {
+    console.error('Error: --target is required. Example: --target prisma/schema.prisma');
+    process.exit(1);
+  }
+
+  // Resolve source path
+  const sourcePath = source
+    ? resolve(source)
+    : resolve(__dirname, '..', 'prisma', 'schema.prisma');
+
+  const targetPath = resolve(target);
+
+  if (!existsSync(sourcePath)) {
+    console.error(`Error: Source schema not found at ${sourcePath}`);
+    process.exit(1);
+  }
+
+  if (!existsSync(targetPath)) {
+    console.error(`Error: Target schema not found at ${targetPath}`);
+    process.exit(1);
+  }
+
+  console.log(`Source: ${sourcePath}`);
+  console.log(`Target: ${targetPath}`);
+
+  // Read schemas
+  const sourceSchema = readFileSync(sourcePath, 'utf-8');
+  const targetSchema = readFileSync(targetPath, 'utf-8');
+
+  // 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)`);
+
+  // 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()}`,
+    '',
+    ...blocks.map(b => b.content),
+    '',
+    MARKER_END,
+  ].join('\n');
+
+  let result;
+
+  // Check if markers already exist in target
+  const startIdx = targetSchema.indexOf(MARKER_START);
+  const endIdx = targetSchema.indexOf(MARKER_END);
+
+  if (startIdx !== -1 && endIdx !== -1) {
+    // Replace existing managed section
+    const before = targetSchema.substring(0, startIdx);
+    const after = targetSchema.substring(endIdx + MARKER_END.length);
+    result = before + managedContent + after;
+    console.log('Replaced existing mega-agent section in target schema.');
+  } else {
+    // Append to end of file
+    result = targetSchema.trimEnd() + '\n\n' + managedContent + '\n';
+    console.log('Appended mega-agent section to target schema.');
+  }
+
+  if (dryRun) {
+    console.log('\n--- DRY RUN (no changes written) ---\n');
+    console.log(managedContent);
+  } else {
+    writeFileSync(targetPath, result, 'utf-8');
+    console.log(`\nDone! Target schema updated at ${targetPath}`);
+    console.log('\nNext steps:');
+    console.log('  1. Add User relations to mega-agent models (see reference schema comments)');
+    console.log('  2. Run: npx prisma migrate dev --name add_mega_agent_models');
+    console.log('  3. Run: npx prisma generate');
+  }
+}
+
+main();