db-migrate/lib/commands/create-migration.js

Database migration framework for node.js

'use strict';

var _assert = require('./helper/assert');
var log = require('db-migrate-shared').log;
var mkdirp = require('mkdirp');
var fs = require('fs');
var yargs = require('yargs');
var util = require('util');

function createMigrationDir (dir, callback) {
  fs.stat(dir, function (err) {
    if (err) {
      mkdirp(dir, callback);
    } else {
      callback();
    }
  });
}

function executeCreateMigration (internals, config, callback) {
  var migrationsDir = internals.argv['migrations-dir'];

  internals.runTimestamp = new Date();

  if (internals.migrationMode && internals.migrationMode !== 'all') {
    migrationsDir =
      internals.argv['migrations-dir'] + '/' + internals.migrationMode;
  }

  var folder, path;

  if (internals.argv._.length === 0) {
    log.error("'migrationName' is required.");
    if (!internals.isModule) {
      yargs.showHelp();
    }

    if (typeof callback !== 'function') {
      process.exit(1);
    } else {
      return callback(new Error("'migrationName' is required."));
    }
  }

  createMigrationDir(migrationsDir, function (err) {
    var index = require('../../connect');
    var Migration = require('../migration.js');

    if (err) {
      log.error('Failed to create migration directory at ', migrationsDir, err);
      if (typeof callback !== 'function') {
        process.exit(1);
      } else {
        return callback(new Error('Failed to create migration directory.'));
      }
    }

    internals.argv.title = internals.argv._.shift();
    folder = internals.argv.title.split('/');

    internals.argv.title = folder[folder.length - 2] || folder[0];
    path = migrationsDir;

    if (folder.length > 1) {
      path += '/';

      for (var i = 0; i < folder.length - 1; ++i) {
        path += folder[i] + '/';
      }
    }

    var templateType = Migration.TemplateType.DEFAULT_JS;
    if (
      shouldCreateSqlFiles(internals, config) &&
      shouldCreateCoffeeFile(internals, config)
    ) {
      templateType = Migration.TemplateType.COFFEE_SQL_FILE_LOADER;
    } else if (
      shouldCreateSqlFiles(internals, config) &&
      shouldIgnoreOnInitFiles(internals, config)
    ) {
      templateType = Migration.TemplateType.SQL_FILE_LOADER_IGNORE_ON_INIT;
    } else if (shouldCreateSqlFiles(internals, config)) {
      templateType = Migration.TemplateType.SQL_FILE_LOADER;
    } else if (shouldCreateCoffeeFile(internals, config)) {
      templateType = Migration.TemplateType.DEFAULT_COFFEE;
    }
    var migration = new Migration(
      internals.argv.title +
        (shouldCreateCoffeeFile(internals, config) ? '.coffee' : '.js'),
      path,
      internals.runTimestamp,
      templateType
    );
    index.createMigration(migration, function (err, migration) {
      if (_assert(err, callback)) {
        log.info(util.format('Created migration at %s', migration.path));
        if (shouldCreateSqlFiles(internals, config)) {
          createSqlFiles(internals, config, callback);
        } else {
          if (typeof callback === 'function') {
            return callback();
          }
        }
      }
    });
  });
}

function shouldCreateSqlFiles (internals, config) {
  return internals.argv['sql-file'] || config['sql-file'];
}

function shouldIgnoreOnInitFiles (internals, config) {
  return internals.argv['ignore-on-init'] || config['ignore-on-init'];
}

function shouldCreateCoffeeFile (internals, config) {
  return internals.argv['coffee-file'] || config['coffee-file'];
}

function createSqlFiles (internals, config, callback) {
  var migrationsDir = internals.argv['migrations-dir'];

  if (internals.migrationMode && internals.migrationMode !== 'all') {
    migrationsDir =
      internals.argv['migrations-dir'] + '/' + internals.migrationMode;
  }

  var sqlDir = migrationsDir + '/sqls';
  createMigrationDir(sqlDir, function (err) {
    var index = require('../../connect');
    var Migration = require('../migration.js');

    if (err) {
      log.error('Failed to create migration directory at ', sqlDir, err);

      if (typeof callback !== 'function') {
        process.exit(1);
      } else {
        return callback(err);
      }
    }

    var templateTypeDefaultSQL = Migration.TemplateType.DEFAULT_SQL;
    var migrationUpSQL = new Migration(
      internals.argv.title + '-up.sql',
      sqlDir,
      internals.runTimestamp,
      templateTypeDefaultSQL
    );
    index.createMigration(migrationUpSQL, function (err, migration) {
      if (_assert(err, callback)) {
        log.info(
          util.format('Created migration up sql file at %s', migration.path)
        );

        var migrationDownSQL = new Migration(
          internals.argv.title + '-down.sql',
          sqlDir,
          internals.runTimestamp,
          templateTypeDefaultSQL
        );
        index.createMigration(migrationDownSQL, function (err, migration) {
          if (_assert(err, callback)) {
            log.info(
              util.format(
                'Created migration down sql file at %s',
                migration.path
              )
            );
            if (typeof callback === 'function') callback();
          }
        });
      }
    });
  });
}

module.exports = executeCreateMigration;