smoonb 1.0.11 → 1.0.13

package/src/commands/backup/steps/11-migrations.js

@@ -1,79 +1,35 @@
-const chalk = require('chalk');
 const path = require('path');
-const { execSync } = require('child_process');
-const { extractPasswordFromDbUrl, ensureCleanLink } = require('../../../utils/supabaseLink');
-const { cleanDir, countFiles, copyDirSafe } = require('../../../utils/fsExtra');
-const { t } = require('../../../i18n');
-const ui = require('../../../utils/cliUi');
-
-/**
- * Etapa 10: Migrations Backup (NOVA ETAPA INDEPENDENTE)
- */
-module.exports = async (context) => {
-  const { projectId, accessToken, databaseUrl, backupDir } = context;
+const fs = require('fs').promises;
+const os = require('os');
+const { runSupabase } = require('../../../utils/supabaseCli');
+const { runSql } = require('../../restore/pg');
+const { copyDirectoryRecursive } = require('../../restore/utils');
+module.exports = async ({ databaseUrl, accessToken, backupDir, postgresMajor = 17 }, { query = runSql, run = runSupabase } = {}) => {
+  const root = await fs.mkdtemp(path.join(os.tmpdir(), 'smoonb-migrations-'));
+  let phase = 'migration_history_check_failed';
   try {
-    const getT = global.smoonbI18n?.t || t;
-    // Reset de link ao projeto de ORIGEM
-    const dbPassword = extractPasswordFromDbUrl(databaseUrl);
-    await ensureCleanLink(projectId, accessToken, dbPassword);
-    
-    // Limpar migrations local (opcional, mas recomendado para garantir servidor como fonte da verdade)
-    const migrationsDir = path.join(process.cwd(), 'supabase', 'migrations');
-    await cleanDir(migrationsDir);
-    console.log(chalk.white(`   - ${getT('backup.steps.migrations.cleaning')}`));
-    
-    // Baixar todas as migrations do servidor usando migration fetch
-    console.log(chalk.white(`   - ${getT('backup.steps.migrations.downloading')}`));
-    
-    const env = {
-      ...process.env,
-      SUPABASE_ACCESS_TOKEN: accessToken
-    };
-    
-    try {
-      execSync('supabase migration fetch', {
-        cwd: process.cwd(),
-        stdio: 'pipe',
-        encoding: 'utf8',
-        timeout: 120000,
-        env
-      });
-    } catch (error) {
-      console.log(chalk.yellow(`   ⚠️ ${getT('backup.steps.migrations.fetchError', { message: error.message })}`));
-      console.log(chalk.yellow(`   💡 ${getT('backup.steps.migrations.fetchTip')}`));
-      return { success: false };
+    const history = await query(databaseUrl, "BEGIN READ ONLY; SET LOCAL statement_timeout = '20s'; SELECT to_regclass('supabase_migrations.schema_migrations') IS NOT NULL; COMMIT;", { postgresMajor });
+    if (!['t', 'f'].includes(history.trim())) throw new Error('Unexpected history response');
+    let count = 0;
+    if (history.trim() === 't') {
+      const value = await query(databaseUrl, "BEGIN READ ONLY; SET LOCAL statement_timeout = '20s'; SELECT count(*) FROM supabase_migrations.schema_migrations; COMMIT;", { postgresMajor });
+      if (!/^\d+$/.test(value.trim())) throw new Error('Unexpected migration count');
+      count = Number(value.trim());
     }
-    
-    // Contar arquivos baixados
-    const fileCount = await countFiles(migrationsDir);
-    console.log(chalk.white(`   - ${getT('backup.steps.migrations.downloaded', { count: fileCount })}`));
-    
-    // Copiar migrations para o backup
-    const backupMigrationsDir = path.join(backupDir, 'migrations');
-    const copiedCount = await copyDirSafe(migrationsDir, backupMigrationsDir);
-    const relativePath = path.relative(process.cwd(), backupMigrationsDir);
-    console.log(chalk.white(`   - ${getT('backup.steps.migrations.copying', { path: relativePath, count: copiedCount })}`));
-    
-    if (copiedCount > 0) {
-      console.log(chalk.green(`   ✅ ${getT('backup.steps.migrations.copied', { count: copiedCount })}`));
+    if (count === 0) {
+      await fs.mkdir(path.join(backupDir, 'migrations'), { recursive: true });
+      console.log('Migrations: nenhum histórico registrado no banco; nada a exportar.');
+      return { success: true, file_count: 0, reason: history.trim() === 'f' ? 'history_absent' : 'history_empty' };
     }
-    
-    // Usar flag de limpeza do contexto (já foi perguntado no início)
-    const shouldClean = context?.cleanupFlags?.cleanMigrations || false;
-    
-    if (shouldClean) {
-      await cleanDir(migrationsDir);
-      ui.hint(`   - ${getT('backup.steps.migrations.cleaned')}`);
-    }
-    
-    return {
-      success: true,
-      file_count: copiedCount
-    };
-  } catch (error) {
-    const getT = global.smoonbI18n?.t || t;
-    console.log(chalk.yellow(`   ⚠️ ${getT('backup.steps.migrations.error', { message: error.message })}`));
-    return { success: false };
-  }
+    phase = 'migration_export_failed';
+    await fs.mkdir(path.join(root, 'supabase', 'migrations'), { recursive: true });
+    await fs.writeFile(path.join(root, 'supabase', 'config.toml'), 'project_id = "smoonb-backup"\n');
+    run('supabase', ['migration', 'fetch', '--db-url', databaseUrl, '--workdir', root, '--yes'], { cwd: root, env: { ...process.env, SUPABASE_ACCESS_TOKEN: accessToken }, stdio: 'pipe' });
+    const source = path.join(root, 'supabase', 'migrations');
+    await copyDirectoryRecursive(source, path.join(backupDir, 'migrations'));
+    const fileCount = (await fs.readdir(source)).filter(name => name.endsWith('.sql')).length;
+    if (fileCount !== count) throw new Error('Migration export count mismatch');
+    return { success: true, file_count: fileCount };
+  } catch { return { success: false, reason: phase, action: 'Check database connectivity and access to supabase_migrations.schema_migrations; retry the backup.' }; }
+  finally { await fs.rm(root, { recursive: true, force: true }); }
 };
-