@javalabs/prisma-client 1.0.0

package/src/scripts/reset-database.ts

@@ -0,0 +1,264 @@
+import * as pg from "pg";
+import * as dotenv from "dotenv";
+import { Logger } from "@nestjs/common";
+import { execSync } from "child_process";
+import { PrismaClient } from "@prisma/client";
+
+dotenv.config();
+
+export class DatabaseResetTool {
+  private readonly logger = new Logger("DatabaseResetTool");
+  private readonly pool: pg.Pool;
+  private readonly adminPool: pg.Pool;
+  private readonly prisma: PrismaClient;
+
+  constructor(private readonly databaseUrl: string) {
+    // Conexión a la base de datos específica
+    this.pool = new pg.Pool({
+      connectionString: this.databaseUrl,
+    });
+
+    // Conexión a la base de datos postgres (admin)
+    const adminUrl = this.databaseUrl.replace(/\/[^/]+$/, "/postgres");
+    this.adminPool = new pg.Pool({
+      connectionString: adminUrl,
+    });
+
+    // Cliente Prisma
+    this.prisma = new PrismaClient({
+      datasources: {
+        db: {
+          url: this.databaseUrl,
+        },
+      },
+    });
+  }
+
+  async resetDatabase(recreateDb = false) {
+    try {
+      this.logger.log("Starting complete database reset process");
+
+      // Extraer el nombre de la base de datos de la URL
+      const dbName = this.extractDatabaseName(this.databaseUrl);
+      
+      if (!dbName) {
+        throw new Error("Could not extract database name from connection URL");
+      }
+
+      // 1. Eliminar todos los schemas personalizados y sus contenidos
+      await this.dropAllSchemas();
+
+      // 2. Eliminar todas las tablas del schema public
+      await this.dropAllPublicTables();
+
+      // 3. Eliminar todos los tipos personalizados (enums, etc.)
+      await this.dropAllCustomTypes();
+
+      if (recreateDb) {
+        // 4. Terminar todas las conexiones y eliminar la base de datos
+        await this.terminateConnections(dbName);
+        await this.adminPool.query(`DROP DATABASE IF EXISTS "${dbName}"`);
+        
+        // 5. Recrear la base de datos
+        this.logger.log(`Recreating database: ${dbName}`);
+        await this.adminPool.query(`CREATE DATABASE "${dbName}"`);
+        
+        // 6. Ejecutar prisma migrate para recrear el esquema base
+        this.logger.log("Running prisma migrate to recreate base schema");
+        try {
+          execSync(`DATABASE_URL="${this.databaseUrl}" npx prisma migrate deploy`, {
+            stdio: 'inherit'
+          });
+        } catch (migrationError) {
+          this.logger.error(`Error running prisma migrate: ${migrationError.message}`);
+        }
+      }
+
+      this.logger.log("Database has been completely reset");
+      return true;
+    } catch (error) {
+      this.logger.error(
+        `Error resetting database: ${error.message}`,
+        error.stack
+      );
+      return false;
+    } finally {
+      await this.cleanup();
+    }
+  }
+
+  private async dropAllSchemas() {
+    this.logger.log("Dropping all custom schemas");
+    
+    try {
+      // Obtener todos los schemas excepto los del sistema
+      const schemas = await this.getCustomSchemas();
+      this.logger.log(`Found ${schemas.length} custom schemas to drop`);
+      
+      // Desactivar restricciones de clave foránea temporalmente
+      await this.pool.query(`SET session_replication_role = 'replica'`);
+      
+      // Eliminar cada schema y su contenido
+      for (const schema of schemas) {
+        try {
+          await this.pool.query(`DROP SCHEMA IF EXISTS "${schema}" CASCADE`);
+          this.logger.log(`Dropped schema: ${schema}`);
+        } catch (error) {
+          this.logger.error(`Error dropping schema ${schema}: ${error.message}`);
+        }
+      }
+      
+      // Reactivar restricciones de clave foránea
+      await this.pool.query(`SET session_replication_role = 'origin'`);
+    } catch (error) {
+      this.logger.error(`Error dropping schemas: ${error.message}`);
+    }
+  }
+
+  private async dropAllPublicTables() {
+    this.logger.log("Dropping all tables in public schema");
+    
+    try {
+      // Obtener todas las tablas del schema public
+      const tables = await this.getPublicTables();
+      this.logger.log(`Found ${tables.length} tables in public schema to drop`);
+      
+      if (tables.length > 0) {
+        // Desactivar restricciones de clave foránea temporalmente
+        await this.pool.query(`SET session_replication_role = 'replica'`);
+        
+        // Eliminar todas las tablas en una sola operación
+        const tableList = tables.map(table => `"${table}"`).join(", ");
+        if (tableList) {
+          await this.pool.query(`DROP TABLE IF EXISTS ${tableList} CASCADE`);
+          this.logger.log("Dropped all tables in public schema");
+        }
+        
+        // Reactivar restricciones de clave foránea
+        await this.pool.query(`SET session_replication_role = 'origin'`);
+      }
+    } catch (error) {
+      this.logger.error(`Error dropping public tables: ${error.message}`);
+    }
+  }
+
+  private async dropAllCustomTypes() {
+    this.logger.log("Dropping all custom types (enums, domains, etc.)");
+    
+    try {
+      // Obtener todos los tipos personalizados
+      const typesQuery = `
+        SELECT t.typname AS name
+        FROM pg_type t 
+        JOIN pg_catalog.pg_namespace n ON n.oid = t.typnamespace
+        WHERE n.nspname = 'public'
+        AND t.typtype = 'e' -- enum types
+      `;
+      
+      const typesResult = await this.pool.query(typesQuery);
+      const types = typesResult.rows.map(row => row.name);
+      
+      this.logger.log(`Found ${types.length} custom types to drop`);
+      
+      // Eliminar cada tipo
+      for (const type of types) {
+        try {
+          await this.pool.query(`DROP TYPE IF EXISTS "${type}" CASCADE`);
+          this.logger.log(`Dropped type: ${type}`);
+        } catch (error) {
+          this.logger.error(`Error dropping type ${type}: ${error.message}`);
+        }
+      }
+    } catch (error) {
+      this.logger.error(`Error dropping custom types: ${error.message}`);
+    }
+  }
+
+  private async getCustomSchemas(): Promise<string[]> {
+    const result = await this.pool.query(`
+      SELECT schema_name
+      FROM information_schema.schemata
+      WHERE schema_name NOT IN ('public', 'information_schema', 'pg_catalog', 'pg_toast')
+      AND schema_name NOT LIKE 'pg_%'
+    `);
+    
+    return result.rows.map(row => row.schema_name);
+  }
+
+  private async getPublicTables(): Promise<string[]> {
+    const result = await this.pool.query(`
+      SELECT table_name
+      FROM information_schema.tables
+      WHERE table_schema = 'public'
+      AND table_type = 'BASE TABLE'
+    `);
+    
+    return result.rows.map(row => row.table_name);
+  }
+
+  private extractDatabaseName(url: string): string | null {
+    // Extraer el nombre de la base de datos de la URL de conexión
+    const matches = url.match(/\/([^/?]+)($|\?)/);
+    return matches ? matches[1] : null;
+  }
+
+  private async terminateConnections(dbName: string) {
+    this.logger.log(`Terminating all connections to database: ${dbName}`);
+    
+    try {
+      await this.adminPool.query(`
+        SELECT pg_terminate_backend(pg_stat_activity.pid)
+        FROM pg_stat_activity
+        WHERE pg_stat_activity.datname = $1
+        AND pid <> pg_backend_pid()
+      `, [dbName]);
+      
+      this.logger.log("All connections terminated successfully");
+    } catch (error) {
+      this.logger.error(`Error terminating connections: ${error.message}`);
+    }
+  }
+
+  private async cleanup() {
+    this.logger.log("Cleaning up database connections");
+    await this.prisma.$disconnect();
+    await this.pool.end();
+    await this.adminPool.end();
+  }
+}
+
+// Script para ejecutar desde línea de comandos
+if (require.main === module) {
+  const run = async () => {
+    try {
+      // Load environment variables first
+      dotenv.config();
+      
+      // Get database URL from command line args or .env
+      const databaseUrl = process.argv[2] || process.env.DATABASE_URL;
+      
+      if (!databaseUrl) {
+        console.error("Error: DATABASE_URL not found in environment variables or command line arguments");
+        process.exit(1);
+      }
+      
+      // Check for recreate flag
+      const recreateFlag = process.argv.includes('--recreate') || process.argv.includes('-r');
+      
+      const resetTool = new DatabaseResetTool(databaseUrl);
+      const success = await resetTool.resetDatabase(recreateFlag);
+      
+      if (!success) {
+        console.error("Database reset failed");
+        process.exit(1);
+      }
+      
+      process.exit(0);
+    } catch (error) {
+      console.error("Error:", error.message);
+      process.exit(1);
+    }
+  };
+
+  run();
+}

package/src/scripts/retry-failed-migrations.ts

@@ -0,0 +1,416 @@
+import * as fs from "fs";
+import * as path from "path";
+import { PrismaClient } from "@prisma/client";
+import * as dotenv from "dotenv";
+import { Logger } from "@nestjs/common";
+import * as pg from "pg";
+import { MigrationErrorHandler } from "./error-handler";
+
+dotenv.config();
+
+export class FailedMigrationRetry {
+  private readonly logger = new Logger("FailedMigrationRetry");
+  private readonly targetPrisma: PrismaClient;
+  private readonly targetPool: pg.Pool;
+  private readonly errorHandler: MigrationErrorHandler;
+  private schemaCache: Record<string, any> = {};
+
+  constructor(private readonly errorLogPath: string) {
+    // Target database connection
+    this.targetPool = new pg.Pool({
+      connectionString: process.env.DATABASE_URL,
+    });
+
+    // Target Prisma client
+    this.targetPrisma = new PrismaClient({
+      datasources: {
+        db: {
+          url: process.env.DATABASE_URL,
+        },
+      },
+    });
+
+    // Inicializar el manejador de errores
+    this.errorHandler = new MigrationErrorHandler();
+  }
+
+  async retryFailedMigrations() {
+    try {
+      this.logger.log(`Loading error log from: ${this.errorLogPath}`);
+
+      // Cargar el archivo de log de errores
+      if (!fs.existsSync(this.errorLogPath)) {
+        this.logger.error(`Error log file not found: ${this.errorLogPath}`);
+        return;
+      }
+
+      const errorLogContent = fs.readFileSync(this.errorLogPath, "utf8");
+      const errorLog: Record<string, any[]> = JSON.parse(errorLogContent);
+
+      let totalErrors = 0;
+      let retrySuccessCount = 0;
+      let retryFailCount = 0;
+
+      // Procesar cada tenant
+      for (const tenantId in errorLog) {
+        const tenantErrors = errorLog[tenantId];
+        if (tenantErrors.length === 0) continue;
+
+        totalErrors += tenantErrors.length;
+        this.logger.log(
+          `Processing ${tenantErrors.length} errors for tenant: ${tenantId}`
+        );
+
+        // Crear cliente Prisma específico para este tenant
+        const tenantPrisma = new PrismaClient({
+          datasources: {
+            db: {
+              url: `${process.env.DATABASE_URL}?schema=${tenantId}`,
+            },
+          },
+          transactionOptions: {
+            maxWait: 120000,
+            timeout: 120000,
+          },
+        });
+
+        // Procesar cada error
+        for (const errorEntry of tenantErrors) {
+          const { entity, recordId, data } = errorEntry;
+
+          this.logger.log(`Retrying ${entity} with ID ${recordId}`);
+
+          try {
+            // Obtener el esquema de la tabla si no está en caché
+            if (!this.schemaCache[`${tenantId}.${entity}`]) {
+              this.schemaCache[`${tenantId}.${entity}`] =
+                await this.getTableSchema(entity, tenantId);
+            }
+
+            const tableSchema = this.schemaCache[`${tenantId}.${entity}`];
+
+            if (!tableSchema || tableSchema.length === 0) {
+              this.logger.warn(
+                `Schema not found for ${entity} in tenant ${tenantId}. Skipping.`
+              );
+              retryFailCount++;
+              continue;
+            }
+
+            // Preparar los datos para la inserción/actualización
+            const createData: Record<string, any> = {};
+            const updateData: Record<string, any> = {};
+
+            // Obtener la lista de columnas
+            const columnNames = tableSchema.map((col) => col.column_name);
+
+            // Si no tenemos los datos originales, no podemos reintentar
+            if (!data) {
+              this.logger.warn(
+                `No data available for ${entity} with ID ${recordId}. Skipping.`
+              );
+              retryFailCount++;
+              continue;
+            }
+
+            // Procesar cada columna
+            for (const column of tableSchema) {
+              const columnName = column.column_name;
+
+              // Omitir columnas que no están en los datos
+              if (!Object.prototype.hasOwnProperty.call(data, columnName)) {
+                continue;
+              }
+
+              // Manejar tipos especiales
+              if (column.data_type === "numeric" && data[columnName] !== null) {
+                // Convertir a número
+                const numericValue = parseFloat(data[columnName]);
+                if (!isNaN(numericValue)) {
+                  createData[columnName] = numericValue;
+                  updateData[columnName] = numericValue;
+                } else {
+                  createData[columnName] = null;
+                  updateData[columnName] = null;
+                }
+              }
+              // Manejar tipos enum
+              else if (
+                column.data_type === "USER-DEFINED" &&
+                column.udt_name.startsWith("enum_")
+              ) {
+                // Obtener valores válidos para el enum
+                const enumValues = await this.getEnumValues(column.udt_name);
+              
+                // Inicializar _enumTypes si no existe
+                if (!createData._enumTypes) createData._enumTypes = {};
+                if (!updateData._enumTypes) updateData._enumTypes = {};
+              
+                // Siempre guardar el tipo de enum para esta columna
+                createData._enumTypes[columnName] = column.udt_name;
+                updateData._enumTypes[columnName] = column.udt_name;
+              
+                if (data[columnName] === null || data[columnName] === "") {
+                  createData[columnName] = null;
+                  updateData[columnName] = null;
+                }
+                // Si el valor es un string, intentar encontrar una coincidencia en los valores del enum
+                else if (typeof data[columnName] === "string") {
+                  const matchingValue = this.findMatchingEnumValue(
+                    data[columnName],
+                    enumValues
+                  );
+                  if (matchingValue) {
+                    createData[columnName] = matchingValue;
+                    updateData[columnName] = matchingValue;
+                  } else {
+                    // Si no hay coincidencia, usar null
+                    createData[columnName] = null;
+                    updateData[columnName] = null;
+                    this.logger.warn(
+                      `No matching enum value found for ${columnName}="${data[columnName]}" in ${entity}. Using NULL.`
+                    );
+                  }
+                } else {
+                  // Para otros tipos, usar el valor tal cual
+                  const enumValue = data[columnName];
+                  createData[columnName] = data[columnName];
+                  updateData[columnName] = data[columnName];
+                }
+              } else {
+                // Para otros tipos de datos, usar el valor tal cual
+                createData[columnName] = data[columnName];
+                updateData[columnName] = data[columnName];
+              }
+            }
+
+            // Asegurarse de incluir el ID
+            if (columnNames.includes(recordId)) {
+              createData[recordId] = data[recordId];
+            }
+
+            // Ejecutar la operación upsert usando SQL raw en lugar de acceso directo al modelo
+            // Verificar primero si el registro existe
+            const checkExistQuery = `
+              SELECT 1 FROM "${tenantId}"."${entity}" 
+              WHERE "${recordId}" = $1
+              LIMIT 1
+            `;
+
+            const existResult = await this.targetPool.query(checkExistQuery, [
+              data[recordId],
+            ]);
+            const recordExists = existResult.rows.length > 0;
+
+            if (recordExists) {
+              // Construir la consulta UPDATE
+              const updateFields = Object.keys(updateData)
+                .filter((key) => key !== recordId && key !== '_enumTypes') // Excluir el campo ID y _enumTypes
+                .map((key, index) => {
+                  // Aplicar casting explícito para tipos enum
+                  if (updateData._enumTypes && updateData._enumTypes[key]) {
+                    return `"${key}" = $${index + 2}::${updateData._enumTypes[key]}`;
+                  }
+                  return `"${key}" = $${index + 2}`;
+                });
+
+              if (updateFields.length > 0) {
+                const updateQuery = `
+                  UPDATE "${tenantId}"."${entity}"
+                  SET ${updateFields.join(", ")}
+                  WHERE "${recordId}" = $1
+                `;
+
+                const updateValues = [
+                  data[recordId],
+                  ...Object.keys(updateData)
+                    .filter((key) => key !== recordId && key !== '_enumTypes')
+                    .map((key) => updateData[key]),
+                ];
+
+                await this.targetPool.query(updateQuery, updateValues);
+              }
+            } else {
+              // Construir la consulta INSERT
+              const insertFields = Object.keys(createData)
+                .filter(key => key !== '_enumTypes')
+                .map((key) => `"${key}"`);
+              
+              const insertPlaceholders = Object.keys(createData)
+                .filter(key => key !== '_enumTypes')
+                .map((key, index) => {
+                  // Aplicar casting explícito para tipos enum
+                  if (createData._enumTypes && createData._enumTypes[key]) {
+                    return `$${index + 1}::${createData._enumTypes[key]}`;
+                  }
+                  return `$${index + 1}`;
+                });
+
+              const insertQuery = `
+                INSERT INTO "${tenantId}"."${entity}" (${insertFields.join(", ")})
+                VALUES (${insertPlaceholders.join(", ")})
+              `;
+
+              const insertValues = Object.keys(createData)
+                .filter(key => key !== '_enumTypes')
+                .map((key) => createData[key]);
+
+              await this.targetPool.query(insertQuery, insertValues);
+            }
+
+            this.logger.log(
+              `Successfully migrated ${entity} with ID ${recordId}`
+            );
+            retrySuccessCount++;
+          } catch (error) {
+            this.logger.error(
+              `Error retrying ${entity} with ID ${recordId}: ${error.message}`
+            );
+            // Registrar el error en el nuevo log
+            this.errorHandler.logError(tenantId, entity, recordId, error, data);
+            retryFailCount++;
+          }
+        }
+
+        // Cerrar la conexión del cliente tenant
+        await tenantPrisma.$disconnect();
+      }
+
+      // Mostrar resumen
+      this.logger.log(`
+Retry Summary:
+- Total errors: ${totalErrors}
+- Successfully retried: ${retrySuccessCount}
+- Failed retries: ${retryFailCount}
+      `);
+
+      // Generar informe de errores restantes
+      const remainingErrorsReport = this.errorHandler.generateErrorReport();
+      this.logger.log("Remaining Errors Report:\n" + remainingErrorsReport);
+    } catch (error) {
+      this.logger.error(
+        `Error in retry process: ${error.message}`,
+        error.stack
+      );
+    } finally {
+      await this.cleanup();
+    }
+  }
+
+  private async getTableSchema(
+    tableName: string,
+    schemaName: string
+  ): Promise<any[]> {
+    try {
+      const schemaQuery = `
+        SELECT column_name, data_type, column_default, is_nullable, udt_name
+        FROM information_schema.columns
+        WHERE table_name = $1
+        AND table_schema = $2
+        ORDER BY ordinal_position
+      `;
+
+      const result = await this.targetPool.query(schemaQuery, [
+        tableName,
+        schemaName,
+      ]);
+      return result.rows;
+    } catch (error) {
+      this.logger.error(
+        `Error getting schema for table ${tableName} in schema ${schemaName}: ${error.message}`
+      );
+      return [];
+    }
+  }
+
+  private async getEnumValues(enumTypeName: string): Promise<string[]> {
+    try {
+      const enumQuery = `
+        SELECT e.enumlabel
+        FROM pg_enum e
+        JOIN pg_type t ON e.enumtypid = t.oid
+        WHERE t.typname = $1
+        ORDER BY e.enumsortorder
+      `;
+
+      const result = await this.targetPool.query(enumQuery, [enumTypeName]);
+      return result.rows.map((row) => row.enumlabel);
+    } catch (error) {
+      this.logger.error(
+        `Error getting enum values for ${enumTypeName}: ${error.message}`
+      );
+      return [];
+    }
+  }
+
+  private findMatchingEnumValue(
+    value: string,
+    enumValues: string[]
+  ): string | null {
+    // Buscar coincidencia exacta
+    if (enumValues.includes(value)) {
+      return value;
+    }
+
+    // Buscar coincidencia sin distinción de mayúsculas/minúsculas
+    const lowerValue = value.toLowerCase();
+    for (const enumValue of enumValues) {
+      if (enumValue.toLowerCase() === lowerValue) {
+        return enumValue;
+      }
+    }
+
+    // Buscar coincidencia parcial
+    const similarValues = enumValues.filter(
+      (v) =>
+        v.toLowerCase().includes(lowerValue) ||
+        lowerValue.includes(v.toLowerCase())
+    );
+
+    if (similarValues.length === 1) {
+      return similarValues[0];
+    } else if (similarValues.length > 1) {
+      // Si hay múltiples coincidencias, usar la más cercana por longitud
+      return similarValues.reduce((prev, curr) =>
+        Math.abs(curr.length - value.length) <
+        Math.abs(prev.length - value.length)
+          ? curr
+          : prev
+      );
+    }
+
+    return null;
+  }
+
+  private async cleanup() {
+    this.logger.log("Cleaning up database connections");
+    await this.targetPrisma.$disconnect();
+    await this.targetPool.end();
+  }
+}
+
+// Script para ejecutar desde línea de comandos
+if (require.main === module) {
+  const run = async () => {
+    try {
+      // Obtener la ruta del archivo de log de errores
+      const errorLogPath = process.argv[2];
+
+      if (!errorLogPath) {
+        console.error(
+          "Error: No error log file path provided. Please provide the path to the error log file."
+        );
+        process.exit(1);
+      }
+
+      const retryTool = new FailedMigrationRetry(errorLogPath);
+      await retryTool.retryFailedMigrations();
+      process.exit(0);
+    } catch (error) {
+      console.error("Error:", error.message);
+      process.exit(1);
+    }
+  };
+
+  run();
+}