add backup state

Signed-off-by: Tobias Gurtzick <[email protected]>

Author: wzrdtales

lib/commands/fix.js

@@ -30,8 +30,9 @@ async function prepare (internals, config) {
     }
 
     internals.migrationsDir = migrator.directory;
+    const backupState = internals.argv['backup-state'];
 
-    await migrator.createMigrationsTable({ emptyState: true });
+    await migrator.createMigrationsTable({ emptyState: true, backupState });
     log.verbose('migration table created');
 
     return migrator;

lib/commands/set-default-argv.js

@@ -16,7 +16,8 @@ module.exports = function (internals, isModule) {
     'migrations-dir': internals.cwd + '/migrations',
     'vcseeder-dir': internals.cwd + '/VCSeeder',
     'staticseeder-dir': internals.cwd + '/Seeder',
-    'ignore-completed-migrations': false
+    'ignore-completed-migrations': false,
+    'backup-state': false
   };
 
   if (!isModule) {
@@ -113,6 +114,8 @@ module.exports = function (internals, isModule) {
       .boolean('ignore-completed-migrations')
       .describe('log-level', 'Set the log-level, for example sql|warn')
       .string('log-level')
+      .describe('backup-state', 'For use with fix, to take a backup of the current state before continuing.')
+      .boolean('backup-state')
       .parse();
   } else {
     internals.argv = Object.assign(defaultConfig, internals.cmdOptions);

lib/state.js

@@ -1,6 +1,8 @@
 const MSTATE = '__dbmigrate_state__';
 const SSTATE = '__dbmigrate_schema__';
 const log = require('db-migrate-shared').log;
+const fs = require('fs').promises;
+const path = require('path');
 
 module.exports = {
   lockState: async function (driver, state, internals) {
@@ -29,9 +31,26 @@ module.exports = {
     }
   },
 
-  init: async function (driver, internals, { emptyState }) {
+  init: async function (driver, internals, { emptyState, backupState }) {
     await driver._createKV(internals.migrationState);
-    const schema = emptyState !== true ? await driver._getKV(internals.migrationState, SSTATE) : null;
+    const _schema = await driver._getKV(internals.migrationState, SSTATE);
+
+    if (_schema && backupState) {
+      const newName = `${internals.migrationState}_b_${Math.floor((new Date() - 0) / 1000)}`;
+      log.info(`[state] Created a backup of ${internals.migrationState} by writing to file ${newName}.dbmigrate`);
+
+      await fs.writeFile(path.resolve(`${newName}.dbmigrate`), JSON.stringify(_schema), 'utf8');
+
+      if (emptyState) {
+        await driver.renameTable(internals.migrationState,
+          newName);
+
+        log.info(`[state] Created a backup of ${internals.migrationState} by renaming table to ${newName}`);
+        await driver._createKV(internals.migrationState);
+      }
+    }
+
+    const schema = emptyState !== true ? _schema : null;
     if (schema) {
       internals.schema = JSON.parse(schema.value);
     } else if (!emptyState) {