1 files changed, 134 insertions, 0 deletions

diff --git a/server/initializers/migrator.ts b/server/initializers/migrator.ts
new file mode 100644
index 000000000..cfa3220e0
--- /dev/null
+++ b/server/initializers/migrator.ts
@@ -0,0 +1,134 @@
+import { waterfall, eachSeries } from 'async'
+import fs = require('fs')
+import path = require('path')
+
+const db = require('./database')
+import { LAST_MIGRATION_VERSION } from './constants'
+import { logger } from '../helpers'
+
+function migrate (finalCallback) {
+  waterfall([
+
+    function checkApplicationTableExists (callback) {
+      db.sequelize.getQueryInterface().showAllTables().asCallback(function (err, tables) {
+        if (err) return callback(err)
+
+        // No tables, we don't need to migrate anything
+        // The installer will do that
+        if (tables.length === 0) return finalCallback(null)
+
+        return callback(null)
+      })
+    },
+
+    function loadMigrationVersion (callback) {
+      db.Application.loadMigrationVersion(callback)
+    },
+
+    function createMigrationRowIfNotExists (actualVersion, callback) {
+      if (actualVersion === null) {
+        db.Application.create({
+          migrationVersion: 0
+        }, function (err) {
+          return callback(err, 0)
+        })
+      }
+
+      return callback(null, actualVersion)
+    },
+
+    function abortMigrationIfNotNeeded (actualVersion, callback) {
+      // No need migrations
+      if (actualVersion >= LAST_MIGRATION_VERSION) return finalCallback(null)
+
+      return callback(null, actualVersion)
+    },
+
+    function getMigrations (actualVersion, callback) {
+      // If there are a new migration scripts
+      logger.info('Begin migrations.')
+
+      getMigrationScripts(function (err, migrationScripts) {
+        return callback(err, actualVersion, migrationScripts)
+      })
+    },
+
+    function doMigrations (actualVersion, migrationScripts, callback) {
+      eachSeries(migrationScripts, function (entity, callbackEach) {
+        executeMigration(actualVersion, entity, callbackEach)
+      }, function (err) {
+        if (err) return callback(err)
+
+        logger.info('Migrations finished. New migration version schema: %s', LAST_MIGRATION_VERSION)
+        return callback(null)
+      })
+    }
+  ], finalCallback)
+}
+
+// ---------------------------------------------------------------------------
+
+export {
+  migrate
+}
+
+// ---------------------------------------------------------------------------
+
+function getMigrationScripts (callback) {
+  fs.readdir(path.join(__dirname, 'migrations'), function (err, files) {
+    if (err) return callback(err)
+
+    const filesToMigrate = []
+
+    files.forEach(function (file) {
+      // Filename is something like 'version-blabla.js'
+      const version = file.split('-')[0]
+      filesToMigrate.push({
+        version,
+        script: file
+      })
+    })
+
+    return callback(err, filesToMigrate)
+  })
+}
+
+function executeMigration (actualVersion, entity, callback) {
+  const versionScript = parseInt(entity.version)
+
+  // Do not execute old migration scripts
+  if (versionScript <= actualVersion) return callback(null)
+
+  // Load the migration module and run it
+  const migrationScriptName = entity.script
+  logger.info('Executing %s migration script.', migrationScriptName)
+
+  const migrationScript = require(path.join(__dirname, 'migrations', migrationScriptName))
+
+  db.sequelize.transaction().asCallback(function (err, t) {
+    if (err) return callback(err)
+
+    const options = {
+      transaction: t,
+      queryInterface: db.sequelize.getQueryInterface(),
+      sequelize: db.sequelize,
+      Sequelize: db.Sequelize
+    }
+    migrationScript.up(options, function (err) {
+      if (err) {
+        t.rollback()
+        return callback(err)
+      }
+
+      // Update the new migration version
+      db.Application.updateMigrationVersion(versionScript, t, function (err) {
+        if (err) {
+          t.rollback()
+          return callback(err)
+        }
+
+        t.commit().asCallback(callback)
+      })
+    })
+  })
+}