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 }
    }
}