additional state travel but without stepping

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

Author: wzrdtales

lib/executors/versioned/v2.js

@@ -5,7 +5,8 @@ const State = require('../../state');
 const log = require('db-migrate-shared').log;
 const Learn = require('../../learn');
 const Chain = require('../../chain');
-const StateTravel = require('../../methods/v2/statetravel');
+const StateTravel = require('../../methods/v2/statetravel')();
+const StateNoStepTravel = require('../../methods/v2/statetravel')(false);
 const Migrate = require('../../methods/v2/migrate');
 const TranslateState = require('../../methods/v2/translatestate');
 const AddConventions = require('../../methods/v2/conventions');
@@ -33,8 +34,9 @@ const execUnit = {
     if (!_file._meta.noDefaultColumn) {
       chain.addChain(AddConventions);
     }
-    chain.addChain(Learn);
     chain.addChain(StateTravel);
+    chain.addChain(Learn);
+    chain.addChain(StateNoStepTravel);
     chain.addChain(Migrate);
 
     await State.startMigration(context._pdriver, file, context.internals);

lib/methods/v2/statetravel.js

@@ -18,14 +18,16 @@ const DEFAULT = {
 };
 
 class StateTravel {
-  constructor (internals, file, driver, pdriver) {
+  constructor (internals, file, driver, pdriver, stepOnly = true) {
     this.file = file;
     this.internals = internals;
     this.driver = driver;
     this.pdriver = pdriver;
     this._counter = 0;
     this._default = async function _default () {
-      await State.step(this.pdriver, ++this._counter, this.internals);
+      if (!stepOnly) {
+        await State.step(this.pdriver, ++this._counter, this.internals);
+      }
 
       return State.update(
         this.pdriver,
@@ -49,24 +51,25 @@ const noTravelError = prop => {
   };
 };
 
-module.exports = {
-  getInterface: (context, file, driver, internals, pdriver) => {
-    if (context.learnable) {
-      const st = new StateTravel(internals, file, context, pdriver);
+module.exports = function (step = true) {
+  return {
+    getInterface: (context, file, driver, internals, pdriver) => {
+      if (context.learnable) {
+        const st = new StateTravel(internals, file, context, pdriver, step);
+        return Shadow.overshadow(
+          driver,
+          Object.assign(st, context.statechanger),
+          noTravelError
+        );
+      }
+
       return Shadow.overshadow(
         driver,
-        Object.assign(st, context.statechanger),
+        new StateTravel(internals, file, context, pdriver, step),
         noTravelError
       );
     }
 
-    return Shadow.overshadow(
-      driver,
-      new StateTravel(internals, file, context, pdriver),
-      noTravelError
-    );
-  }
-
   /* hasModification: true,
 
   modify: (context, driver, internals) => {
@@ -84,4 +87,5 @@ module.exports = {
 
     return { driver: _driver };
   } */
+  };
 };