PeerTube/server/core/initializers/migrator.ts

107 lines
3.2 KiB
TypeScript
Raw Normal View History

import { readdir } from 'fs/promises'
2021-08-27 14:32:44 +02:00
import { join } from 'path'
import { QueryTypes } from 'sequelize'
import { currentDir } from '@peertube/peertube-node-utils'
import { logger } from '../helpers/logger.js'
import { LAST_MIGRATION_VERSION } from './constants.js'
import { sequelizeTypescript } from './database.js'
async function migrate () {
2017-12-12 17:53:50 +01:00
const tables = await sequelizeTypescript.getQueryInterface().showAllTables()
// No tables, we don't need to migrate anything
// The installer will do that
if (tables.length === 0) return
2017-02-18 11:56:28 +01:00
let actualVersion: number | null = null
2017-12-13 17:46:23 +01:00
2019-04-18 11:28:17 +02:00
const query = 'SELECT "migrationVersion" FROM "application"'
const options = {
type: QueryTypes.SELECT as QueryTypes.SELECT
}
const rows = await sequelizeTypescript.query<{ migrationVersion: number }>(query, options)
2020-01-31 16:56:52 +01:00
if (rows?.[0]?.migrationVersion) {
actualVersion = rows[0].migrationVersion
2017-12-13 17:46:23 +01:00
}
if (actualVersion === null) {
2017-12-13 17:46:23 +01:00
await sequelizeTypescript.query('INSERT INTO "application" ("migrationVersion") VALUES (0)')
actualVersion = 0
}
// No need migrations, abort
if (actualVersion >= LAST_MIGRATION_VERSION) return
// If there are a new migration scripts
logger.info('Begin migrations.')
const migrationScripts = await getMigrationScripts()
for (const migrationScript of migrationScripts) {
2017-11-27 09:47:21 +01:00
try {
await executeMigration(actualVersion, migrationScript)
} catch (err) {
2018-03-26 15:54:13 +02:00
logger.error('Cannot execute migration %s.', migrationScript.version, { err })
2018-01-10 17:18:12 +01:00
process.exit(-1)
2017-11-27 09:47:21 +01:00
}
}
logger.info('Migrations finished. New migration version schema: %s', LAST_MIGRATION_VERSION)
}
// ---------------------------------------------------------------------------
2017-05-15 22:22:03 +02:00
export {
migrate
}
2016-12-25 09:44:57 +01:00
// ---------------------------------------------------------------------------
async function getMigrationScripts () {
const files = await readdir(join(currentDir(import.meta.url), 'migrations'))
const filesToMigrate: {
2020-01-31 16:56:52 +01:00
version: string
script: string
}[] = []
files
2021-12-17 14:20:43 +01:00
.filter(file => file.endsWith('.js'))
.forEach(file => {
// Filename is something like 'version-blabla.js'
const version = file.split('-')[0]
filesToMigrate.push({
version,
script: file
2016-12-25 09:44:57 +01:00
})
})
2016-12-25 09:44:57 +01:00
return filesToMigrate
2016-12-25 09:44:57 +01:00
}
async function executeMigration (actualVersion: number, entity: { version: string, script: string }) {
2017-05-22 20:58:25 +02:00
const versionScript = parseInt(entity.version, 10)
2016-12-25 09:44:57 +01:00
// Do not execute old migration scripts
if (versionScript <= actualVersion) return undefined
2016-12-25 09:44:57 +01:00
// Load the migration module and run it
const migrationScriptName = entity.script
logger.info('Executing %s migration script.', migrationScriptName)
const migrationScript = await import(join(currentDir(import.meta.url), 'migrations', migrationScriptName))
2016-12-25 09:44:57 +01:00
2018-01-10 17:18:12 +01:00
return sequelizeTypescript.transaction(async t => {
2017-02-16 19:19:56 +01:00
const options = {
transaction: t,
2017-12-12 17:53:50 +01:00
queryInterface: sequelizeTypescript.getQueryInterface(),
sequelize: sequelizeTypescript
2017-02-16 19:19:56 +01:00
}
2016-12-25 09:44:57 +01:00
await migrationScript.up(options)
// Update the new migration version
2017-12-13 17:46:23 +01:00
await sequelizeTypescript.query('UPDATE "application" SET "migrationVersion" = ' + versionScript, { transaction: t })
2016-12-25 09:44:57 +01:00
})
}