49 lines
1.8 KiB
TypeScript
49 lines
1.8 KiB
TypeScript
import fs from 'fs';
|
|
import path from 'path';
|
|
import Database from 'better-sqlite3';
|
|
|
|
export type DbData = {data: string};
|
|
|
|
export const db = new Database(
|
|
process.env.NODE_ENV === 'test' ? 'test.db' : 'spacetraders.db',
|
|
process.env.NODE_ENV === 'development' ? { verbose: console.log } : undefined
|
|
);
|
|
db.pragma('foreign_keys = ON');
|
|
db.pragma('journal_mode = WAL');
|
|
|
|
function init(): void {
|
|
const filenames = fs.readdirSync('./database/');
|
|
const allMigrations = filenames.filter(e => e.match(/\.sql$/)).map(e => path.join('./database', e));
|
|
db.transaction(function migrate() {
|
|
let version;
|
|
try {
|
|
const res = db.prepare('SELECT version FROM schema_version').get() as {version: number};
|
|
version = res.version;
|
|
} catch {
|
|
version = 0;
|
|
}
|
|
if (version === allMigrations.length) return;
|
|
while (version < allMigrations.length) {
|
|
db.exec(fs.readFileSync(allMigrations[version], 'utf8'));
|
|
version++;
|
|
}
|
|
db.exec(`DELETE FROM schema_version; INSERT INTO schema_version (version) VALUES (${version});`);
|
|
})();
|
|
}
|
|
|
|
export function reset(): void {
|
|
const indices = db.prepare(`SELECT name FROM sqlite_master WHERE type = 'index';`).all() as Array<{name: string}>;
|
|
const tables = db.prepare(`SELECT name FROM sqlite_master WHERE type = 'table';`).all() as Array<{name: string}>;
|
|
const triggers = db.prepare(`SELECT name FROM sqlite_master WHERE type = 'trigger';`).all() as Array<{name: string}>;
|
|
const views = db.prepare(`SELECT name FROM sqlite_master WHERE type = 'view';`).all() as Array<{name: string}>;
|
|
indices.forEach(elt => db.exec(`DROP INDEX ${elt.name};`));
|
|
tables.forEach(elt => db.exec(`DROP TABLE ${elt.name};`));
|
|
triggers.forEach(elt => db.exec(`DROP TRIGGER ${elt.name};`));
|
|
views.forEach(elt => db.exec(`DROP VIEW ${elt.name};`));
|
|
db.exec(`VACUUM;`);
|
|
init();
|
|
}
|
|
|
|
init();
|
|
|
|
export default db;
|