]>
Commit | Line | Data |
---|---|---|
65fcc311 | 1 | import { waterfall, eachSeries } from 'async' |
4d4e5cd4 C |
2 | import * as fs from 'fs' |
3 | import * as path from 'path' | |
e02643f3 | 4 | import * as Sequelize from 'sequelize' |
00d6b0dd | 5 | |
e02643f3 | 6 | import { database as db } from './database' |
65fcc311 C |
7 | import { LAST_MIGRATION_VERSION } from './constants' |
8 | import { logger } from '../helpers' | |
e02643f3 | 9 | import { ApplicationInstance } from '../models' |
00d6b0dd | 10 | |
69818c93 | 11 | function migrate (finalCallback: (err: Error) => void) { |
5804c0db C |
12 | waterfall([ |
13 | ||
14 | function checkApplicationTableExists (callback) { | |
15 | db.sequelize.getQueryInterface().showAllTables().asCallback(function (err, tables) { | |
16 | if (err) return callback(err) | |
17 | ||
18 | // No tables, we don't need to migrate anything | |
19 | // The installer will do that | |
20 | if (tables.length === 0) return finalCallback(null) | |
21 | ||
22 | return callback(null) | |
23 | }) | |
24 | }, | |
25 | ||
26 | function loadMigrationVersion (callback) { | |
27 | db.Application.loadMigrationVersion(callback) | |
28 | }, | |
00d6b0dd | 29 | |
ed708d62 C |
30 | function createMigrationRowIfNotExists (actualVersion, callback) { |
31 | if (actualVersion === null) { | |
32 | db.Application.create({ | |
33 | migrationVersion: 0 | |
34 | }, function (err) { | |
1f5bf9ef | 35 | return callback(err, 0) |
ed708d62 C |
36 | }) |
37 | } | |
38 | ||
39 | return callback(null, actualVersion) | |
40 | }, | |
41 | ||
5804c0db C |
42 | function abortMigrationIfNotNeeded (actualVersion, callback) { |
43 | // No need migrations | |
65fcc311 | 44 | if (actualVersion >= LAST_MIGRATION_VERSION) return finalCallback(null) |
5804c0db C |
45 | |
46 | return callback(null, actualVersion) | |
47 | }, | |
48 | ||
49 | function getMigrations (actualVersion, callback) { | |
50 | // If there are a new migration scripts | |
00d6b0dd C |
51 | logger.info('Begin migrations.') |
52 | ||
b769007f | 53 | getMigrationScripts(function (err, migrationScripts) { |
5804c0db C |
54 | return callback(err, actualVersion, migrationScripts) |
55 | }) | |
56 | }, | |
00d6b0dd | 57 | |
5804c0db | 58 | function doMigrations (actualVersion, migrationScripts, callback) { |
69818c93 | 59 | eachSeries(migrationScripts, function (entity: any, callbackEach) { |
5804c0db C |
60 | executeMigration(actualVersion, entity, callbackEach) |
61 | }, function (err) { | |
62 | if (err) return callback(err) | |
00d6b0dd | 63 | |
65fcc311 | 64 | logger.info('Migrations finished. New migration version schema: %s', LAST_MIGRATION_VERSION) |
5804c0db | 65 | return callback(null) |
00d6b0dd | 66 | }) |
00d6b0dd | 67 | } |
5804c0db | 68 | ], finalCallback) |
00d6b0dd C |
69 | } |
70 | ||
71 | // --------------------------------------------------------------------------- | |
72 | ||
65fcc311 C |
73 | export { |
74 | migrate | |
75 | } | |
00d6b0dd | 76 | |
b769007f C |
77 | // --------------------------------------------------------------------------- |
78 | ||
69818c93 C |
79 | type GetMigrationScriptsCallback = (err: Error, filesToMigrate?: { version: string, script: string }[]) => void |
80 | function getMigrationScripts (callback: GetMigrationScriptsCallback) { | |
b769007f C |
81 | fs.readdir(path.join(__dirname, 'migrations'), function (err, files) { |
82 | if (err) return callback(err) | |
83 | ||
84 | const filesToMigrate = [] | |
85 | ||
86 | files.forEach(function (file) { | |
87 | // Filename is something like 'version-blabla.js' | |
88 | const version = file.split('-')[0] | |
89 | filesToMigrate.push({ | |
90 | version, | |
91 | script: file | |
92 | }) | |
93 | }) | |
94 | ||
95 | return callback(err, filesToMigrate) | |
96 | }) | |
97 | } | |
98 | ||
69818c93 | 99 | function executeMigration (actualVersion: number, entity: { version: string, script: string }, callback: (err: Error) => void) { |
e02643f3 | 100 | const versionScript = parseInt(entity.version, 10) |
b769007f C |
101 | |
102 | // Do not execute old migration scripts | |
103 | if (versionScript <= actualVersion) return callback(null) | |
104 | ||
105 | // Load the migration module and run it | |
106 | const migrationScriptName = entity.script | |
107 | logger.info('Executing %s migration script.', migrationScriptName) | |
108 | ||
109 | const migrationScript = require(path.join(__dirname, 'migrations', migrationScriptName)) | |
110 | ||
111 | db.sequelize.transaction().asCallback(function (err, t) { | |
112 | if (err) return callback(err) | |
113 | ||
4793c343 C |
114 | const options = { |
115 | transaction: t, | |
116 | queryInterface: db.sequelize.getQueryInterface(), | |
5804c0db | 117 | sequelize: db.sequelize, |
e02643f3 | 118 | Sequelize: Sequelize |
4793c343 C |
119 | } |
120 | migrationScript.up(options, function (err) { | |
b769007f C |
121 | if (err) { |
122 | t.rollback() | |
123 | return callback(err) | |
124 | } | |
125 | ||
126 | // Update the new migration version | |
127 | db.Application.updateMigrationVersion(versionScript, t, function (err) { | |
128 | if (err) { | |
129 | t.rollback() | |
130 | return callback(err) | |
131 | } | |
132 | ||
dea32aac | 133 | t.commit().asCallback(callback) |
b769007f C |
134 | }) |
135 | }) | |
136 | }) | |
137 | } |