ffffng/server/db/database.js
baldo 2e27e63f94 Fix: Start using sqlite module that actually uses promises.
For now export legacy sqlite3 module until other code is refactored.
2019-03-29 22:26:54 +01:00

90 lines
2.5 KiB
JavaScript

'use strict';
const util = require('util');
const fs = require('graceful-fs');
const glob = util.promisify(require('glob'));
const path = require('path');
const config = require('../config').config;
const Logger = require('../logger');
async function applyPatch(db, file) {
Logger.tag('database', 'migration').info('Checking if patch need to be applied: %s', file);
const contents = await util.promisify(fs.readFile)(file);
const version = path.basename(file, '.sql');
const row = await db.get('SELECT * FROM schema_version WHERE version = ?', version);
if (row) {
// patch is already applied. skip!
Logger.tag('database', 'migration').info('Patch already applied, skipping: %s', file);
return
}
const sql = 'BEGIN TRANSACTION;\n' +
contents.toString() + '\n' +
'INSERT INTO schema_version (version) VALUES (\'' + version + '\');\n' +
'END TRANSACTION;';
await db.exec(sql);
Logger.tag('database', 'migration').info('Patch successfully applied: %s', file);
}
async function applyMigrations(db) {
Logger.tag('database', 'migration').info('Migrating database...');
const sql = 'BEGIN TRANSACTION; CREATE TABLE IF NOT EXISTS schema_version (\n' +
' version VARCHAR(255) PRIMARY KEY ASC,\n' +
' applied_at DATETIME DEFAULT CURRENT_TIMESTAMP NOT NULL\n' +
'); END TRANSACTION;';
await db.exec(sql);
const files = await glob(__dirname + '/patches/*.sql');
for (const file of files) {
await applyPatch(db, file)
}
}
async function init() {
const sqlite = require('sqlite');
const SQLite3 = require('sqlite3');
const file = config.server.databaseFile;
Logger.tag('database').info('Setting up database: %s', file);
let db;
try {
db = await sqlite.open(file);
}
catch (error) {
Logger.tag('database').error('Error initialzing database:', error);
throw error;
}
db.on('profile', (sql, time) => Logger.tag('database').profile('[%sms]\t%s', time, sql));
try {
await applyMigrations(db);
}
catch (error) {
Logger.tag('database').error('Error migrating database:', error);
throw error;
}
let legacyDB;
try {
legacyDB = new SQLite3.Database(file);
}
catch (error) {
Logger.tag('database').error('Error initialzing legacy database lib:', error);
throw error;
}
module.exports.db = legacyDB;
}
module.exports = {
init
};