Function run

scripts/restore-drill.js:101–200

()

}

async function run() {
  const args = parseArgs(process.argv.slice(2));
  const sourceDatabaseUrl = process.env.POSTGRES_URL;
  if (!sourceDatabaseUrl) {
    throw new Error('Missing required environment variable: POSTGRES_URL');
  }

  const keepDrillDatabase = args['keep-db'] === true || args['keep-db'] === 'true';
  const retentionDays = Number.parseInt(args['retention-days'] || process.env.BACKUP_RETENTION_DAYS || '14', 10);
  const backupDir = path.resolve(args['backup-dir'] || process.env.BACKUP_DIR || path.join(process.cwd(), 'backups'));

  ensureCommand('pg_dump');
  ensureCommand('pg_restore');

  await mkdir(backupDir, { recursive: true });

  const stamp = createTimestamp();
  const backupPath = path.join(backupDir, `restore-drill-${stamp}.dump`);
  const drillDatabaseName = `titanbot_restore_drill_${stamp}`;
  const maintenanceUrl = buildDatabaseUrlWithName(sourceDatabaseUrl, 'postgres');
  const drillDatabaseUrl = buildDatabaseUrlWithName(sourceDatabaseUrl, drillDatabaseName);

  const maintenancePool = new Pool({ connectionString: maintenanceUrl });

  logger.info('Starting restore drill', {
    event: 'restore_drill.start',
    drillDatabaseName
  });

  try {
    runCommand('pg_dump', [
      '--format=custom',
      '--no-owner',
      '--no-privileges',
      '--file',
      backupPath,
      sourceDatabaseUrl
    ]);

    const createClient = await maintenancePool.connect();
    try {
      await createClient.query(`CREATE DATABASE "${drillDatabaseName}" TEMPLATE template0`);
    } finally {
      createClient.release();
    }

    runCommand('pg_restore', [
      '--clean',
      '--if-exists',
      '--no-owner',
      '--no-privileges',
      '--dbname',
      drillDatabaseUrl,
      backupPath
    ]);

    const verifyPool = new Pool({ connectionString: drillDatabaseUrl });
    try {