init

server/db/db.js

@@ -0,0 +1,76 @@
+import chalk from 'chalk'
+import path from 'path';
+import fs from 'fs/promises';
+import { pathToFileURL } from 'url';
+
+export default class Database {
+    #nodes;
+    #edges;
+    #labels = {}
+
+    constructor() {
+        this.getData()
+    }
+
+    async getData() {
+        const dbData = await fs.readFile(path.join(process.cwd(), 'db/db.json'), 'utf8');
+        let dbJson;
+        try {
+            dbJson = JSON.parse(dbData);
+        } catch {
+            dbJson = []
+        }
+        this.#nodes = dbJson["nodes"];
+        this.#edges = dbJson["edges"];
+    
+        console.log(chalk.yellow("DB established."))
+        Object.preventExtensions(this);
+    }
+
+    // superKey = "nodes" || "edges"
+    async writeData(superKey, key, value) {
+        const dbData = await fs.readFile(path.join(process.cwd(), 'db/db.json'), 'utf8');
+        let dbJson;
+        try {
+            dbJson = JSON.parse(dbData);
+        } catch {
+            dbJson = []
+        }
+
+        dbJson[superKey][key] = value;
+
+        await fs.writeFile(path.join(process.cwd(), 'db/db.json'), JSON.stringify(dbJson, null, 2), 'utf8')
+    }
+
+    async getLabelModels() {
+        const labelHandlers = {};
+        const labelDir = path.join(process.cwd(), 'src/model/labels');
+        const files = await fs.readdir(labelDir);
+    
+        for (const file of files) {
+            if (!file.endsWith('.js')) continue;
+    
+            const label = path.basename(file, '.js');
+            const modulePath = path.join(labelDir, file);
+            const module = await import(pathToFileURL(modulePath).href);
+            labelHandlers[label] = module.default;
+    
+            if (!this.#labels[label]) {
+                this.#labels[label] = [];
+            }
+        }
+        return labelHandlers
+    }
+
+    generateUserID() {
+        let id = this.#labels["User"].length + 1;
+        while (this.get.user(`user-${id}`)) {
+            id++;
+        }
+        return `user-${id}`; // O(1) most of the time
+    }
+
+    async getAll() {
+        return { nodes: this.#nodes }
+    }
+}