connect.js

/**
 * This file is going to disappear.
 * Only still here for backwards compatibility.
 * */
var recursive = require('final-fs').readdirRecursive;
var fs = require('fs');
var driver = require('./lib/driver');
var path = require('path');
var log = require('db-migrate-shared').log;
const Promise = require('bluebird');

Promise.promisifyAll(driver);

exports.connect = async function (config, PassedClass) {
  var internals = {};
  var prefix = 'migration';
  if (config.config) {
    prefix = config.prefix || prefix;
    internals = config.internals;
    config = config.config;
  }

  const db2 = await driver.connectAsync(config, internals);

  return Promise.fromCallback(callback => {
    driver.connect(config, internals, function (err, db) {
      if (err) {
        callback(err);
        return;
      }

      const realClose = db.close;
      // close both lines with one disconnect action
      db.close = function (cb) {
        db2.close(function () {});
        db.close = realClose;
        db.close(cb);
      };

      var dirPath = path.resolve(
        internals.argv['migrations-dir'] || 'migrations'
      );
      if (internals.migrationMode) {
        if (internals.migrationMode !== 'all') {
          var switched = false;
          var newConf;

          try {
            newConf = require(path.resolve(
              internals.argv['migrations-dir'] || 'migrations',
              internals.migrationMode
            ) + '/config.json');
            log.info(
              'loaded extra config for migration subfolder: "' +
                internals.migrationMode +
                '/config.json"'
            );
            switched = true;
          } catch (e) {}

          if (switched) {
            db.switchDatabase(newConf, function (err) {
              if (err) {
                return callback(err);
              }
              internals.locTitle = internals.migrationMode;
              callback(
                null,
                new PassedClass(
                  db,
                  dirPath,
                  internals.mode !== 'static',
                  internals,
                  prefix,
                  { db2 }
                )
              );
            });
          } else {
            internals.locTitle = internals.migrationMode;
            callback(
              null,
              new PassedClass(
                db,
                dirPath,
                internals.mode !== 'static',
                internals,
                prefix
              )
            );
          }
        } else {
          recursive(
            dirPath,
            false,
            internals.argv['migrations-dir'] || 'migrations'
          ).then(function (files) {
            var oldClose = db.close;

            files = files.filter(function (file) {
              return file !== 'migrations' && fs.statSync(file).isDirectory();
            });

            files.push('');

            db.close = function (cb) {
              migrationFiles(
                files,
                callback,
                config,
                internals,
                PassedClass,
                db,
                oldClose,
                prefix,
                db2,
                cb
              );
            };

            db.close();
          });
        }
      } else {
        callback(
          null,
          new PassedClass(
            db,
            dirPath,
            internals.mode !== 'static',
            internals,
            prefix,
            { db2 }
          )
        );
      }
    });
  });
};

exports.driver = function (config, callback) {
  var internals = {};
  if (config.config) {
    internals = config.internals;
    config = config.config;
  }

  driver.connect(config, internals, callback);
};

function migrationFiles (
  files,
  callback,
  config,
  internals,
  PassedClass,
  db,
  close,
  prefix,
  db2,
  cb
) {
  var file;
  var switched = false;
  var newConf;

  if (files.length === 1) {
    db.close = close;
  }

  file = files.pop();
  log.info('Enter scope "' + (file !== '' ? file : '/') + '"');

  if (file !== '') {
    try {
      fs.statSync(path.resolve(file + '/config.json'));
      newConf = require(path.resolve(file + '/config.json'));
      log.info(
        'loaded extra config for migration subfolder: "' +
          file +
          '/config.json"'
      );
      switched = true;
    } catch (e) {}
  }

  db.switchDatabase(switched ? newConf : config.database, function () {
    internals.matching = file.substr(
      file.indexOf(internals.argv['migrations-dir'] || 'migrations') +
        (internals.argv['migrations-dir'] || 'migrations').length +
        1
    );

    if (internals.matching.length === 0) {
      internals.matching = '';
    }

    internals.locTitle = internals.matching;
    callback(
      null,
      new PassedClass(
        db,
        internals.argv['migrations-dir'],
        internals.mode !== 'static',
        internals,
        prefix,
        { db2 }
      )
    );

    if (typeof cb === 'function') {
      cb();
    }
  });
}