1 import { waterfall, eachSeries } from 'async'
2 import fs = require('fs')
3 import path = require('path')
4 import * as Sequelize from 'sequelize'
6 import { database as db } from './database'
7 import { LAST_MIGRATION_VERSION } from './constants'
8 import { logger } from '../helpers'
9 import { ApplicationInstance } from '../models'
11 function migrate (finalCallback) {
14 function checkApplicationTableExists (callback) {
15 db.sequelize.getQueryInterface().showAllTables().asCallback(function (err, tables) {
16 if (err) return callback(err)
18 // No tables, we don't need to migrate anything
19 // The installer will do that
20 if (tables.length === 0) return finalCallback(null)
26 function loadMigrationVersion (callback) {
27 db.Application.loadMigrationVersion(callback)
30 function createMigrationRowIfNotExists (actualVersion, callback) {
31 if (actualVersion === null) {
32 db.Application.create({
35 return callback(err, 0)
39 return callback(null, actualVersion)
42 function abortMigrationIfNotNeeded (actualVersion, callback) {
44 if (actualVersion >= LAST_MIGRATION_VERSION) return finalCallback(null)
46 return callback(null, actualVersion)
49 function getMigrations (actualVersion, callback) {
50 // If there are a new migration scripts
51 logger.info('Begin migrations.')
53 getMigrationScripts(function (err, migrationScripts) {
54 return callback(err, actualVersion, migrationScripts)
58 function doMigrations (actualVersion, migrationScripts, callback) {
59 eachSeries(migrationScripts, function (entity, callbackEach) {
60 executeMigration(actualVersion, entity, callbackEach)
62 if (err) return callback(err)
64 logger.info('Migrations finished. New migration version schema: %s', LAST_MIGRATION_VERSION)
71 // ---------------------------------------------------------------------------
77 // ---------------------------------------------------------------------------
79 function getMigrationScripts (callback) {
80 fs.readdir(path.join(__dirname, 'migrations'), function (err, files) {
81 if (err) return callback(err)
83 const filesToMigrate = []
85 files.forEach(function (file) {
86 // Filename is something like 'version-blabla.js'
87 const version = file.split('-')[0]
94 return callback(err, filesToMigrate)
98 function executeMigration (actualVersion, entity, callback) {
99 const versionScript = parseInt(entity.version, 10)
101 // Do not execute old migration scripts
102 if (versionScript <= actualVersion) return callback(null)
104 // Load the migration module and run it
105 const migrationScriptName = entity.script
106 logger.info('Executing %s migration script.', migrationScriptName)
108 const migrationScript = require(path.join(__dirname, 'migrations', migrationScriptName))
110 db.sequelize.transaction().asCallback(function (err, t) {
111 if (err) return callback(err)
115 queryInterface: db.sequelize.getQueryInterface(),
116 sequelize: db.sequelize,
119 migrationScript.up(options, function (err) {
125 // Update the new migration version
126 db.Application.updateMigrationVersion(versionScript, t, function (err) {
132 t.commit().asCallback(callback)