1 files changed, 23 insertions, 0 deletions
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;
+	}
+}