summaryrefslogtreecommitdiff
path: root/nodejs/database/db.ts
blob: 2fba09bf6169f4034a6e1338b49c5314fcbdb437 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
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;