node-baseline/index.js

A simple database migration and versioning tool

#! /usr/bin/env node

var fs = require('fs');
var pathutil = require('path');
var options = require('./lib/options');
var log = require('./lib/utils/logger');
var getMergedDbConfig = require('./lib/utils/getMergedDbConfig');
var readFromFile = require('./lib/config').readFromFile;
var stat = require('./lib/utils/stat');
var init = require('./lib/commands/init');
var up = require('./lib/commands/up');
var backup = require('./lib/commands/backup');
var restore = require('./lib/commands/restore');
var colors = require('colors/safe');
var factory = require('./lib/factory');

/**
 * the main entry point for the program.
 * @param {Object} config configuration information loaded
 *  from config file specified via --config or .baselinerc.
 */
function main(config) {
  if (!config.databases) {
    log.error('Config error: at least one database should be configured. ');
    return;
  }

  // merge database configurations
  var map = {};
  for (var i = 0; i < config.databases.length; i++) {
    var dbConfig = getMergedDbConfig(config.databases[i], config);
    config.databases[i] = dbConfig;

    try {
      factory.getProvider(dbConfig.dialect);
    } catch (e) {
      log.error('Error: %s', e.message);
      return;
    }

    // check the required information of database config.
    if (!dbConfig.name) {
      log.error('Config error: missing name of database #%d. ', i);
      return;
    }

    if (map[dbConfig.name.toLowerCase()]) {
      log.error('Config error: database \'%s\' already exists. ', dbConfig.name);
      return;
    }

    var required = ['host', 'user', 'password', 'dialect'];
    for (var k = 0; k < required.length; k++) {
      if (!dbConfig[required[k]]) {
        log.error(
          'Config error: missing %s of database \'%s\'. ',
          colors.underline(required[k]),
          dbConfig.name
        );
        return;
      }
    }

    map[dbConfig.name.toLowerCase()] = dbConfig;
  }

  if (options.database) {
    config.databases = config.databases.filter(
      x => x.name === options.database
    );
  }

  if (/^init$/i.test(options.command)) {
    init(config, options.force).catch(e => {
      log.error(e.stack);
    })
  } else if (/^up$/i.test(options.command)) {
    up(config).catch(e => {
      log.error(e.stack);
    });
  } else if (/^backup/i.test(options.command)) {
    backup(config, options.output, options.outputFile).catch(e => {
      log.error(e.stack);
    });
  } else if (/^restore/i.test(options.command)) {
    if (!options.database) {
      log.error('"--database" options missing');
      process.exit(1);
    }
    if (!options.input) {
      log.error('"--input" options missing');
      process.exit(1);
    }
    restore(
      config,
      options.database,
      options.input,
      options.dropDatabase
    ).catch(e => {
      log.error(e.stack);
    });
  }
}

if (!/^(init|up|backup|restore)$/i.test(options.command)) {
  log.error('Error: unknown command \'%s\'', options.command);
  return;
}

// if the config option is not specified, try to load the
// .baselinerc from the current working directory.
if (!options.config) {
  var configFile = options.production ?
    '.baselinerc.production' :
    '.baselinerc';
  options.config = pathutil.join(process.cwd(), configFile);

  // check if the .baselinerc config file exists
  stat(options.config).then(() => {
    // yes, try to load config from it.
    readFromFile(options.config).then(result => {
      main(result);
    }).catch(e => {
      log.error(
        `Error load config from ${configFile}: %s`,
        e.message
      );
    });
  }).catch(e => {
    log.warn(
      'Error: configuration file is not specified. You can specify\n' +
      'the config file via --config option or .baselinerc in the\n' +
      'directory where you call baseline. Use --help for more information. '
    );
  });
} else {
  readFromFile(options.config).then(result => {
    main(result);
  }).catch(e => {
    log.error('Error load config file: %s', e.message);
  });
}