From 36cc33f9e96a38ecea98ac8d26275b4828347d80 Mon Sep 17 00:00:00 2001
From: Julien Dessaux
Date: Sat, 1 Jul 2023 23:13:13 +0200
Subject: Moved the nodejs agent to its own subfolder to make room for my
 haskell agent

---
 nodejs/database/db.js | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 nodejs/database/db.js

(limited to 'nodejs/database/db.js')

diff --git a/nodejs/database/db.js b/nodejs/database/db.js
new file mode 100644
index 0000000..4855233
--- /dev/null
+++ b/nodejs/database/db.js
@@ -0,0 +1,33 @@
+import fs from 'fs';
+import Database from 'better-sqlite3';
+
+const allMigrations = [
+	'database/000_init.sql',
+	'database/001_systems.sql',
+	'database/002_ships.sql',
+	'database/003_surveys.sql',
+];
+
+const db = new Database(
+	process.env.NODE_ENV === 'test' ? 'test.db' : 'spacetraders.db',
+	process.env.NODE_ENV === 'development' ? { verbose: console.log } : null
+);
+db.pragma('foreign_keys = ON');
+db.pragma('journal_mode = WAL');
+
+db.transaction(function migrate() {
+	let version;
+	try {
+		version = db.prepare('SELECT version FROM schema_version').get().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 default db;
-- 
cgit v1.2.3