aboutsummaryrefslogblamecommitdiffhomepage
path: root/server/initializers/migrator.ts
blob: 4b3be6d166fb901ded0be147491b127fb3320e4f (plain) (tree)
1
2
3
4
5
                            
                                   
 
                                           
                                                    












                                                            
                                   
                                                                           

       




                                                             
 


                            
                                             

                                      


                                                                                                  
                                                                                              





                                                                                                  
 

               
 
          



                                                                              


         
 

                                                                              





                                                                           









                                                       
                         


    
                                                                                                
                                                    

                                         
                                                      






                                                                                          
                                        


                                                       
                             
     
 
                                      

                                           
                                                                      
        

    
import * as path from 'path'
import * as Promise from 'bluebird'

import { database as db } from './database'
import { LAST_MIGRATION_VERSION } from './constants'
import { logger, readdirPromise } from '../helpers'

function migrate () {
  const p = db.sequelize.getQueryInterface().showAllTables()
    .then(tables => {
      // No tables, we don't need to migrate anything
      // The installer will do that
      if (tables.length === 0) throw null
    })
    .then(() => {
      return db.Application.loadMigrationVersion()
    })
    .then(actualVersion => {
      if (actualVersion === null) {
        return db.Application.create({ migrationVersion: 0 }).then(() => 0)
      }

      return actualVersion
    })
    .then(actualVersion => {
      // No need migrations, abort
      if (actualVersion >= LAST_MIGRATION_VERSION) throw null

      return actualVersion
    })
    .then(actualVersion => {
      // If there are a new migration scripts
      logger.info('Begin migrations.')

      return getMigrationScripts().then(migrationScripts => ({ actualVersion, migrationScripts }))
    })
    .then(({ actualVersion, migrationScripts }) => {
      return Promise.each(migrationScripts, entity => executeMigration(actualVersion, entity))
    })
    .then(() => {
      logger.info('Migrations finished. New migration version schema: %s', LAST_MIGRATION_VERSION)
    })
    .catch(err => {
      if (err === null) return undefined

      throw err
    })

  return p
}

// ---------------------------------------------------------------------------

export {
  migrate
}

// ---------------------------------------------------------------------------

function getMigrationScripts () {
  return readdirPromise(path.join(__dirname, 'migrations')).then(files => {
    const filesToMigrate: {
      version: string,
      script: string
    }[] = []

    files.forEach(function (file) {
      // Filename is something like 'version-blabla.js'
      const version = file.split('-')[0]
      filesToMigrate.push({
        version,
        script: file
      })
    })

    return filesToMigrate
  })
}

function executeMigration (actualVersion: number, entity: { version: string, script: string }) {
  const versionScript = parseInt(entity.version, 10)

  // Do not execute old migration scripts
  if (versionScript <= actualVersion) return undefined

  // 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))

  return db.sequelize.transaction(t => {
    const options = {
      transaction: t,
      queryInterface: db.sequelize.getQueryInterface(),
      sequelize: db.sequelize
    }

    return migrationScript.up(options)
      .then(() => {
        // Update the new migration version
        return db.Application.updateMigrationVersion(versionScript, t)
      })
  })
}