]>
git.immae.eu Git - github/Chocobozzz/PeerTube.git/blob - server/initializers/migrator.js
3 const waterfall
= require('async/waterfall')
4 const eachSeries
= require('async/eachSeries')
5 const fs
= require('fs')
6 const path
= require('path')
8 const constants
= require('./constants')
9 const db
= require('./database')
10 const logger
= require('../helpers/logger')
16 function migrate (finalCallback
) {
19 function checkApplicationTableExists (callback
) {
20 db
.sequelize
.getQueryInterface().showAllTables().asCallback(function (err
, tables
) {
21 if (err
) return callback(err
)
23 // No tables, we don't need to migrate anything
24 // The installer will do that
25 if (tables
.length
=== 0) return finalCallback(null)
31 function loadMigrationVersion (callback
) {
32 db
.Application
.loadMigrationVersion(callback
)
35 function createMigrationRowIfNotExists (actualVersion
, callback
) {
36 if (actualVersion
=== null) {
37 db
.Application
.create({
40 return callback(err
, 0)
44 return callback(null, actualVersion
)
47 function abortMigrationIfNotNeeded (actualVersion
, callback
) {
49 if (actualVersion
>= constants
.LAST_MIGRATION_VERSION
) return finalCallback(null)
51 return callback(null, actualVersion
)
54 function getMigrations (actualVersion
, callback
) {
55 // If there are a new migration scripts
56 logger
.info('Begin migrations.')
58 getMigrationScripts(function (err
, migrationScripts
) {
59 return callback(err
, actualVersion
, migrationScripts
)
63 function doMigrations (actualVersion
, migrationScripts
, callback
) {
64 eachSeries(migrationScripts
, function (entity
, callbackEach
) {
65 executeMigration(actualVersion
, entity
, callbackEach
)
67 if (err
) return callback(err
)
69 logger
.info('Migrations finished. New migration version schema: %s', constants
.LAST_MIGRATION_VERSION
)
76 // ---------------------------------------------------------------------------
78 module
.exports
= migrator
80 // ---------------------------------------------------------------------------
82 function getMigrationScripts (callback
) {
83 fs
.readdir(path
.join(__dirname
, 'migrations'), function (err
, files
) {
84 if (err
) return callback(err
)
86 const filesToMigrate
= []
88 files
.forEach(function (file
) {
89 // Filename is something like 'version-blabla.js'
90 const version
= file
.split('-')[0]
97 return callback(err
, filesToMigrate
)
101 function executeMigration (actualVersion
, entity
, callback
) {
102 const versionScript
= parseInt(entity
.version
)
104 // Do not execute old migration scripts
105 if (versionScript
<= actualVersion
) return callback(null)
107 // Load the migration module and run it
108 const migrationScriptName
= entity
.script
109 logger
.info('Executing %s migration script.', migrationScriptName
)
111 const migrationScript
= require(path
.join(__dirname
, 'migrations', migrationScriptName
))
113 db
.sequelize
.transaction().asCallback(function (err
, t
) {
114 if (err
) return callback(err
)
118 queryInterface: db
.sequelize
.getQueryInterface(),
119 sequelize: db
.sequelize
,
120 Sequelize: db
.Sequelize
122 migrationScript
.up(options
, function (err
) {
128 // Update the new migration version
129 db
.Application
.updateMigrationVersion(versionScript
, t
, function (err
) {
135 t
.commit().asCallback(callback
)