summaryrefslogtreecommitdiff
path: root/nodejs/database
diff options
context:
space:
mode:
Diffstat (limited to 'nodejs/database')
-rw-r--r--nodejs/database/000_init.sql8
-rw-r--r--nodejs/database/001_systems.sql6
-rw-r--r--nodejs/database/002_ships.sql6
-rw-r--r--nodejs/database/003_surveys.sql6
-rw-r--r--nodejs/database/config.js23
-rw-r--r--nodejs/database/db.js33
-rw-r--r--nodejs/database/ships.js82
-rw-r--r--nodejs/database/surveys.js19
-rw-r--r--nodejs/database/systems.js71
9 files changed, 254 insertions, 0 deletions
diff --git a/nodejs/database/000_init.sql b/nodejs/database/000_init.sql
new file mode 100644
index 0000000..ecda95a
--- /dev/null
+++ b/nodejs/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/nodejs/database/001_systems.sql b/nodejs/database/001_systems.sql
new file mode 100644
index 0000000..9ad76b1
--- /dev/null
+++ b/nodejs/database/001_systems.sql
@@ -0,0 +1,6 @@
+CREATE TABLE systems (
+ id INTEGER PRIMARY KEY,
+ data JSON NOT NULL,
+ updated DATE DEFAULT NULL
+);
+CREATE UNIQUE INDEX systems_data_symbol on systems (json_extract(data, '$.symbol'));
diff --git a/nodejs/database/002_ships.sql b/nodejs/database/002_ships.sql
new file mode 100644
index 0000000..629f739
--- /dev/null
+++ b/nodejs/database/002_ships.sql
@@ -0,0 +1,6 @@
+CREATE TABLE ships (
+ id INTEGER PRIMARY KEY,
+ data JSON NOT NULL,
+ updated DATE DEFAULT NULL
+);
+CREATE UNIQUE INDEX ships_data_symbol on ships (json_extract(data, '$.symbol'));
diff --git a/nodejs/database/003_surveys.sql b/nodejs/database/003_surveys.sql
new file mode 100644
index 0000000..0548796
--- /dev/null
+++ b/nodejs/database/003_surveys.sql
@@ -0,0 +1,6 @@
+CREATE TABLE surveys (
+ id INTEGER PRIMARY KEY,
+ data JSON NOT NULL
+);
+CREATE INDEX surveys_data_symbol on surveys (json_extract(data, '$.symbol'));
+CREATE INDEX surveys_data_expiration on surveys (json_extract(data, '$.expiration'));
diff --git a/nodejs/database/config.js b/nodejs/database/config.js
new file mode 100644
index 0000000..7a50f68
--- /dev/null
+++ b/nodejs/database/config.js
@@ -0,0 +1,23 @@
+import db from './db.js';
+
+const getTokenStatement = db.prepare(`SELECT value->>'token' as token from config where key = 'register_data';`);
+const registerAgentStatement = db.prepare(`INSERT INTO config(key, value) VALUES ('register_data', json(?));`);
+
+export function getToken() {
+ try {
+ return getTokenStatement.get().token;
+ } catch (err) {
+ console.log(err);
+ return null;
+ }
+}
+
+export function registerAgent(data) {
+ try {
+ registerAgentStatement.run(JSON.stringify(data));
+ return true;
+ } catch (err) {
+ console.log(err);
+ return false;
+ }
+}
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;
diff --git a/nodejs/database/ships.js b/nodejs/database/ships.js
new file mode 100644
index 0000000..0220be6
--- /dev/null
+++ b/nodejs/database/ships.js
@@ -0,0 +1,82 @@
+import db from './db.js';
+
+const getShipStatement = db.prepare(`SELECT data FROM ships WHERE data->>'symbol' = ?;`);
+const setShipStatement = db.prepare(`INSERT INTO ships(data, updated) VALUES (json(?), ?);`);
+const setShipCargoStatement = db.prepare(`UPDATE ships SET data = (SELECT json_set(data, '$.cargo', json(:cargo)) FROM ships WHERE data->>'symbol' = :symbol), updated = :date WHERE data->>'symbol' = :symbol;`);
+const setShipFuelStatement = db.prepare(`UPDATE ships SET data = (SELECT json_set(data, '$.fuel', json(:fuel)) FROM ships WHERE data->>'symbol' = :symbol), updated = :date WHERE data->>'symbol' = :symbol;`);
+const setShipNavStatement = db.prepare(`UPDATE ships SET data = (SELECT json_set(data, '$.nav', json(:nav)) FROM ships WHERE data->>'symbol' = :symbol), updated = :date WHERE data->>'symbol' = :symbol;`);
+const updateShipStatement = db.prepare(`UPDATE ships SET data = json(:data), updated = :date WHERE data->>'symbol' = :symbol;`);
+
+export function getShip(symbol) {
+ try {
+ const data = getShipStatement.get(symbol);
+ if (data === undefined) {
+ return null;
+ }
+ return JSON.parse(data.data);
+ } catch (err) {
+ console.log(err);
+ return null;
+ }
+}
+
+export function setShip(data) {
+ if (getShip(data.symbol) === null) {
+ try {
+ return setShipStatement.run(JSON.stringify(data), new Date().toISOString()).lastInsertRowid;
+ } catch (err) {
+ console.log(err);
+ return null;
+ }
+ } else {
+ try {
+ return updateShipStatement.run({
+ data: JSON.stringify(data),
+ date: new Date().toISOString(),
+ symbol: data.symbol,
+ }).changes;
+ } catch (err) {
+ console.log(err);
+ return null;
+ }
+ }
+}
+
+export function setShipCargo(symbol, cargo) {
+ try {
+ setShipCargoStatement.run({
+ cargo: JSON.stringify(cargo),
+ date: new Date().toISOString(),
+ symbol: symbol,
+ }).changes;
+ } catch (err) {
+ console.log(err);
+ return null;
+ }
+}
+
+export function setShipFuel(symbol, fuel) {
+ try {
+ setShipFuelStatement.run({
+ date: new Date().toISOString(),
+ fuel: JSON.stringify(fuel),
+ symbol: symbol,
+ }).changes;
+ } catch (err) {
+ console.log(err);
+ return null;
+ }
+}
+
+export function setShipNav(symbol, nav) {
+ try {
+ setShipNavStatement.run({
+ date: new Date().toISOString(),
+ nav: JSON.stringify(nav),
+ symbol: symbol,
+ }).changes;
+ } catch (err) {
+ console.log(err);
+ return null;
+ }
+}
diff --git a/nodejs/database/surveys.js b/nodejs/database/surveys.js
new file mode 100644
index 0000000..60c85b2
--- /dev/null
+++ b/nodejs/database/surveys.js
@@ -0,0 +1,19 @@
+import db from './db.js';
+
+const deleteExpiredSurveysStatement = db.prepare(`DELETE FROM surveys WHERE data->>'expiration' < ?;`);
+const getSurveysStatement = db.prepare(`SELECT data FROM surveys WHERE data->>'symbol' = ?;`);
+const setSurveysStatement = db.prepare(`INSERT INTO surveys(data) VALUES (json(?));`);
+
+export function deleteExpired() {
+ return deleteExpiredSurveysStatement.run(new Date().toISOString()).changes;
+}
+
+export function get(symbol) {
+ deleteExpired();
+ return getSurveysStatement.all(symbol);
+}
+
+export function set(survey) {
+ deleteExpired();
+ return setSurveysStatement.run(JSON.stringify(survey));
+}
diff --git a/nodejs/database/systems.js b/nodejs/database/systems.js
new file mode 100644
index 0000000..0d637df
--- /dev/null
+++ b/nodejs/database/systems.js
@@ -0,0 +1,71 @@
+import db from './db.js';
+
+const getSystemStatement = db.prepare(`SELECT data FROM systems WHERE data->>'symbol' = ?;`);
+const getSystemUpdatedStatement = db.prepare(`SELECT updated FROM systems WHERE data->>'symbol' = ?;`);
+const initStatement = db.prepare(`INSERT INTO config(key, value) VALUES ('systems_initialized', TRUE);`);
+const isInitStatement = db.prepare(`SELECT value FROM config WHERE key = 'systems_initialized'`);
+const setSystemStatement = db.prepare(`INSERT INTO systems(data) VALUES (json(?));`);
+const setSystemWaypointsStatement = db.prepare(`UPDATE systems SET data = (SELECT json_set(data, '$.waypoints', json(:waypoints)) FROM systems WHERE data->>'symbol' = :symbol), updated = :date WHERE data->>'symbol' = :symbol;`);
+
+export function init() {
+ try {
+ return initStatement.run().lastInsertRowid;
+ } catch (err) {
+ return null;
+ }
+}
+
+export function isInit() {
+ try {
+ return isInitStatement.get().value === '1';
+ } catch (err) {
+ return false;
+ }
+}
+
+export function getSystem(symbol) {
+ try {
+ const data = getSystemStatement.get(symbol);
+ if (data === undefined) {
+ return null;
+ }
+ return JSON.parse(data.data);
+ } catch (err) {
+ console.log(err);
+ return null;
+ }
+}
+
+export function getSystemUpdated(symbol) {
+ try {
+ const updated = getSystemUpdatedStatement.get(symbol);
+ if (updated === undefined) {
+ return null;
+ }
+ return updated.updated;
+ } catch (err) {
+ console.log(err);
+ return null;
+ }
+}
+
+export function setSystem(data) {
+ try {
+ return setSystemStatement.run(JSON.stringify(data)).lastInsertRowid;
+ } catch (err) {
+ return null;
+ }
+}
+
+export function setSystemWaypoints(symbol, waypoints) {
+ try {
+ return setSystemWaypointsStatement.run({
+ date: new Date().toISOString(),
+ symbol: symbol,
+ waypoints: JSON.stringify(waypoints),
+ }).changes;
+ } catch (err) {
+ console.log(err);
+ return null;
+ }
+}