summaryrefslogtreecommitdiff
path: root/database
diff options
context:
space:
mode:
Diffstat (limited to 'database')
-rw-r--r--database/000_init.sql8
-rw-r--r--database/config.js23
-rw-r--r--database/db.js29
3 files changed, 60 insertions, 0 deletions
diff --git a/database/000_init.sql b/database/000_init.sql
new file mode 100644
index 0000000..ecda95a
--- /dev/null
+++ b/database/000_init.sql
@@ -0,0 +1,8 @@
+CREATE TABLE schema_version (
+ version INTEGER NOT NULL
+);
+CREATE TABLE config (
+ id INTEGER PRIMARY KEY,
+ key TEXT NOT NULL UNIQUE,
+ value TEXT NOT NULL
+);
diff --git a/database/config.js b/database/config.js
new file mode 100644
index 0000000..6db856e
--- /dev/null
+++ b/database/config.js
@@ -0,0 +1,23 @@
+import db from './db.js';
+
+const getTokenStatement = db.prepare(`SELECT value from config where key = 'token';`);
+const registerAgentStatement = db.prepare(`INSERT INTO config(key, value) VALUES ('symbol', ?), ('faction', ?), ('token', ?);`);
+
+export function getToken() {
+ try {
+ return getTokenStatement.get().value;
+ } catch (err) {
+ console.log(err);
+ return null;
+ }
+}
+
+export function registerAgent(symbol, faction, token) {
+ try {
+ registerAgentStatement.run(symbol, faction, token);
+ return true;
+ } catch (err) {
+ console.log(err);
+ return false;
+ }
+}
diff --git a/database/db.js b/database/db.js
new file mode 100644
index 0000000..db71080
--- /dev/null
+++ b/database/db.js
@@ -0,0 +1,29 @@
+import fs from 'fs';
+import Database from 'better-sqlite3';
+
+const allMigrations = [
+ 'database/000_init.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.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;