@javalabs/prisma-client 1.0.4 → 1.0.6

package/src/scripts/run-migration.ts

@@ -5,12 +5,35 @@ import { promisify } from "util";
 import { SchemaDiscrepancyFixer } from "./fix-schema-discrepancies";
 import { EnumSynchronizer } from "./sync-enum-types";
 import { PrismaClient } from "@prisma/client";
+import { MigrationConfig } from "./data-migration/types";
+import * as fs from "fs";
+import * as path from "path";
 
 dotenv.config();
 
 const execAsync = promisify(exec);
 const program = new Command();
 
+// Función para cargar la configuración de migración
+async function loadMigrationConfig(
+  configPath?: string
+): Promise<MigrationConfig> {
+  const defaultPath = path.join(
+    __dirname,
+    "data-migration",
+    "migration-config.json"
+  );
+  const filePath = configPath || defaultPath;
+
+  try {
+    const configContent = await fs.promises.readFile(filePath, "utf8");
+    return JSON.parse(configContent);
+  } catch (error) {
+    console.error(`Error loading migration config from ${filePath}:`, error);
+    throw error;
+  }
+}
+
 // Añadir esta función después de las importaciones
 async function testDatabaseConnection(
   url: string,
@@ -36,85 +59,6 @@ async function testDatabaseConnection(
 }
 
 // Modificar la acción del comando migrate para incluir la prueba de conexión
-async function initializeDatabase(options: any) {
-  try {
-    const { Pool } = await import("pg");
-    const pool = new Pool({ connectionString: process.env.DATABASE_URL });
-
-    // Check if database is already initialized
-    const tablesExist = await checkIfTablesExist(pool);
-
-    if (!tablesExist) {
-      // Only create tables if they don't exist
-      console.log("Database empty, creating initial schema...");
-      await execAsync("npx prisma db push --skip-generate");
-    } else {
-      console.log("Database schema already exists, skipping schema creation");
-      // Create baseline if it doesn't exist
-      await execAsync(
-        "npx prisma migrate reset --force --skip-generate --skip-seed"
-      );
-    }
-
-    if (!options.skipEnumSync) {
-      console.log("Synchronizing enum types...");
-      const enumSynchronizer = new EnumSynchronizer(
-        process.env.SOURCE_DATABASE_URL,
-        process.env.DATABASE_URL
-      );
-      await enumSynchronizer.synchronizeEnums();
-    }
-
-    // Proceed with data migration if not skipped
-    if (!options.skipDataMigration) {
-      console.log("Starting data migration...");
-      const { DataMigrationTool } = await import(
-        "./data-migration/migration-tool"
-      );
-      const migrationTool = new DataMigrationTool({
-        sourcePool: new Pool({
-          connectionString: process.env.SOURCE_DATABASE_URL,
-        }),
-        targetPool: pool,
-        sourcePrisma: new PrismaClient({
-          datasources: { db: { url: process.env.SOURCE_DATABASE_URL } },
-        }),
-        targetPrisma: new PrismaClient({
-          datasources: { db: { url: process.env.DATABASE_URL } },
-        }),
-      });
-
-      await migrationTool.migrateData({
-        publicOnly: options.mode === "public-only",
-      });
-    }
-
-    await pool.end();
-    return true;
-  } catch (error) {
-    console.error("Error initializing database:", error);
-    return false;
-  }
-}
-
-// Helper function to check if tables exist
-async function checkIfTablesExist(pool: any): Promise<boolean> {
-  try {
-    const result = await pool.query(`
-      SELECT COUNT(*) 
-      FROM information_schema.tables 
-      WHERE table_schema = 'public'
-      AND table_type = 'BASE TABLE'
-    `);
-
-    return parseInt(result.rows[0].count) > 0;
-  } catch (error) {
-    console.error("Error checking tables:", error);
-    return false;
-  }
-}
-
-// Modify the existing action handler
 program
   .command("migrate")
   .description("Run the data migration process")
@@ -134,6 +78,7 @@ program
   .option("--auto-fix-schema", "Automatically fix schema discrepancies", false)
   .option("--force", "Force migration even if validation fails", false)
   .option("-y, --yes", "Automatically answer yes to all prompts", false)
+  .option("--config-path <path>", "Path to migration config file")
   .action(async (options) => {
     // Set environment variables for the migration
     if (options.source) {
@@ -182,9 +127,6 @@ program
     }
 
     try {
-      // Initialize database if empty
-      await initializeDatabase(options);
-
       // Step 0: Pre-migration validation
       if (!options.skipValidation) {
         console.log("Step 0: Running pre-migration validation...");
@@ -198,9 +140,7 @@ program
           console.log(
             `Pre-migration validation found ${validationResult.issueCount} issues.`
           );
-          console.log(
-            `Check the full report at: ${validationResult.reportPath}`
-          );
+          console.log(JSON.stringify(validationResult, null, 2));
 
           // Si la opción --yes o --force está activada, continuar automáticamente
           if (options.yes) {
@@ -253,6 +193,24 @@ program
         } else {
           const { Pool } = await import("pg");
           const pool = new Pool({ connectionString: process.env.DATABASE_URL });
+
+          // Helper function to check if tables exist
+          async function checkIfTablesExist(pool: any): Promise<boolean> {
+            try {
+              const result = await pool.query(`
+                SELECT COUNT(*) 
+                FROM information_schema.tables 
+                WHERE table_schema = 'public'
+                AND table_type = 'BASE TABLE'
+              `);
+
+              return parseInt(result.rows[0].count) > 0;
+            } catch (error) {
+              console.error("Error checking tables:", error);
+              return false;
+            }
+          }
+
           const tablesExist = await checkIfTablesExist(pool);
           if (!tablesExist) {
             await execAsync("npx prisma migrate deploy");
@@ -276,7 +234,6 @@ program
       }
 
       // Step 3: Migrate data with transformation
-      // In the migrate command action
       if (!options.skipDataMigration) {
         console.log("Step 3: Migrating data with transformation...");
         const { DataMigrationTool } = await import(
@@ -285,6 +242,9 @@ program
         const { PrismaClient } = await import("@prisma/client");
         const { Pool } = await import("pg");
 
+        // Cargar la configuración de migración
+        const migrationConfig = await loadMigrationConfig(options.configPath);
+
         // Create connections with increased pool size and timeout
         const connections = {
           sourcePool: new Pool({
@@ -313,20 +273,44 @@ program
             "SET session_replication_role = 'replica';"
           );
 
-          const migrationTool = new DataMigrationTool(connections);
-
+          // Si es public-only, modificar la configuración para solo incluir tablas públicas
           if (options.mode === "public-only") {
-            await migrationTool.migrateData({
-              publicOnly: true,
+            migrationConfig.tables = Object.entries(migrationConfig.tables)
+              .filter(([_, config]) => config.type === "public")
+              .reduce(
+                (acc, [key, value]) => ({
+                  ...acc,
+                  [key]: value,
+                }),
+                {}
+              );
+          }
+
+          const migrationTool = new DataMigrationTool(
+            connections,
+            {
+              ...migrationConfig,
+              tables: Object.entries(migrationConfig.tables).reduce(
+                (acc, [key, value]) => ({
+                  ...acc,
+                  [key]: {
+                    ...value,
+                    sourceTable: value.sourceTable || key,
+                    targetTable: value.targetTable || key,
+                  },
+                }),
+                {}
+              ),
+            },
+            {
+              publicOnly: options.mode === "public-only",
               targetSchema: "public",
               sourceSchema: "public",
-            });
-          } else {
-            await migrationTool.migrateData({
-              forceSingleTenant: false,
-              multiTenant: true,
-            });
-          }
+              multiTenant: options.mode !== "public-only",
+            }
+          );
+
+          await migrationTool.migrate();
         } finally {
           // Re-enable foreign key constraints
           await connections.targetPool.query(
@@ -362,7 +346,9 @@ program
           "./post-migration-validator"
         );
         const validator = new PostMigrationValidator();
-        const result = await validator.validate();
+        const result = await validator.validate({
+          publicOnly: options.mode === "public-only",
+        });
 
         if (result.success) {
           console.log("Post-migration validation successful!");
@@ -370,7 +356,7 @@ program
           console.log(
             `Post-migration validation found ${result.issueCount} issues.`
           );
-          console.log(`Check the full report at: ${result.reportPath}`);
+          console.log(JSON.stringify(result, null, 2));
           console.log(
             "You may need to run 'retry-failed' command to fix these issues."
           );
@@ -537,7 +523,7 @@ program
         console.log(
           `Pre-migration validation found ${result.issueCount} issues.`
         );
-        console.log(`Check the full report at: ${result.reportPath}`);
+        console.log(JSON.stringify(result, null, 2));
 
         // Si la opción --yes está activada, continuar automáticamente
         if (options.yes) {
@@ -643,7 +629,7 @@ program
         console.log(
           `Post-migration validation found ${result.issueCount} issues.`
         );
-        console.log(`Check the full report at: ${result.reportPath}`);
+        console.log(JSON.stringify(result, null, 2));
       }
     } catch (error) {
       console.error("Error during post-migration validation:", error);