need learning on reverse, but respect its unlearning

Signed-off-by: Tobias Gurtzick <[email protected]>
db-migrate · Sep 7, 2023 · 458c5e8 · 458c5e8
1 parent 2504712
commit 458c5e8
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git a/lib/learn.js b/lib/learn.js
@@ -2,7 +2,7 @@ const Promise = require('bluebird');
 const Shadow = require('./driver/shadow');
 
 class STD {
- constructor ({ schema, modSchema: mod }, driver) {
+ constructor ({ schema, modSchema: mod, unlearn }, driver) {
  this.checkColumn = function (t, c) {
  if (!this.schema[t]) {
  throw new Error(`There is no ${t} table in schema!`);
@@ -12,6 +12,7 @@ class STD {
  throw new Error(`There is no ${c} column in schema!`);
  }
  };
+ this.unlearn = unlearn;
  this.validations = {};
  this.driver = driver;
  this.indizies = schema.i;
@@ -21,10 +22,14 @@ class STD {
  schema.e = {};
  }
  this.extra = schema.e;
+ if (unlearn) {
+ this.modC = [];
+ } else {
+ this.modC = mod.s;
+ }
  this.modS = mod.c;
  this.modI = mod.i;
  this.modF = mod.f;
- this.modC = mod.s;
  }
 
  dropTable (t) {
@@ -124,7 +129,7 @@ class STD {
  const hasColumns = this.schema[t].columns !== undefined;
  const columns = hasColumns ? this.schema[t].columns : this.schema[t];
 
- if (columns[c].notNull === true) {
+ if (columns[c].notNull === true && !this.unlearn) {
  if (this.validations.columnStrategies !== true) {
  if (
  this.driver._meta &&
@@ -172,7 +177,7 @@ class STD {
 
  this.modS[t] = {};
 
- if (columns[c].notNull === true) {
+ if (columns[c].notNull === true && !this.unlearn) {
  switch (o.columnStrategy) {
  case 'delay':
  this.modS[t][c] = columns[c];
@@ -196,7 +201,7 @@ class STD {
  this.modC.push({ t: 1, a: 'addColumn', c: [t, c, o] });
  break;
  }
- } else {
+ } else if (!this.unlearn) {
  this.modS[t][c] = columns[c];
 
  this.modC.push({ t: 1, a: 'addColumn', c: [t, c, o] });

diff --git a/lib/methods/v2/translatestate.js b/lib/methods/v2/translatestate.js
@@ -2,6 +2,7 @@ const Promise = require('bluebird');
 const State = require('../../state');
 const Chain = require('../../chain');
 const Migrate = require('./migrate');
+const Learn = require('../../learn');
 const log = require('db-migrate-shared').log;
 
 const methods = {
@@ -107,7 +108,9 @@ module.exports = async (context, file, driver, internals) => {
  const mod = internals.modSchema;
 
  const chain = new Chain(context, file, driver, internals);
- // chain.addChain(Learn);
+ internals.unlearn = true;
+
+ chain.addChain(Learn);
  chain.addChain(Migrate);
  await Promise.resolve(mod.s.reverse()).each(args =>
  processEntry(context, file, chain, internals, args)