不要怂,就是干,撸起袖子干!

migrator.js 2.55 KB
var fs        = require("fs")
  , path      = require("path")
  , moment    = require("moment")
  , Utils     = require("./utils")
  , DataTypes = require("./data-types")

module.exports = (function() {
  var Migrator = function(sequelize, options) {
    this.sequelize = sequelize
    this.options   = Utils._.extend({
      path: __dirname + '/../migrations',
      from: null,
      to: null
    }, options || {})
  }

  Migrator.prototype.migrate = function() {
    var self = this

    return new Utils.CustomEventEmitter(function(emitter) {
      getLastMigrationId.call(self).success(function(migrationId) {
        migrationId = migrationId || '19700101000000'
console.log(migrationId)
        var nonMigratedFiles = getMigrationsFilesSince.call(self, migrationId)
        console.log(nonMigratedFiles)
        emitter.emit('success')
      })
    }).run()
  }

  // private

  var getLastMigrationId = function() {
    var self = this

    return new Utils.CustomEventEmitter(function(emitter) {
      var SequelizeMeta       = self.sequelize.modelManager.getModel('SequelizeMeta')

      var findLastMigrationId = function() {
        SequelizeMeta.find({ order: 'id DESC' }).success(function(sequelizeMeta) {
          emitter.emit('success', sequelizeMeta ? sequelizeMeta.lastMigrationId : null)
        })
      }

      if(SequelizeMeta) {
        findLastMigrationId()
      } else {
        SequelizeMeta = self.sequelize.define('SequelizeMeta', {
          lastMigrationId: DataTypes.STRING
        })

        SequelizeMeta.sync().success(function() {
          findLastMigrationId()
        }).error(function(err) {
          emitter.emit('failure', err)
        })
      }
    }).run()
  }

  var getFormattedDateString = function(s) {
    var result = null

    try {
      result = s.match(/(\d{4})(\d{2})(\d{2})(\d{2})(\d{2})(\d{2})/).slice(1, 6).join('-')
    } catch(e) {
      throw new Error(s + ' is no valid migration timestamp format! Use YYYYMMDDHHmmss!')
    }

    return result
  }

  var stringToDate = function(s) {
    var formattedString = getFormattedDateString(s)
      , date            = moment(formattedString, "YYYYMMDDHHmmss")

    return date
  }

  var getMigrationsFilesSince = function(sinceString)  {
    var result = []
      , since  = stringToDate(sinceString)

    var undoneMigrationFiles = fs.readdirSync(this.options.path).filter(function(file) {
      var fileDateString = file.split("-")[0]
        , fileDate       = stringToDate(fileDateString)

      return fileDate.diff(since) > 0
    })

    return undoneMigrationFiles
  }

  return Migrator
})()