mpp-server-dev2/src/Database.js

const fs = require('fs');
const { promisify } = require('util');
const createKeccakHash = require('keccak');
const ColorEncoder = require('./ColorEncoder');
const UserModel = require('./UserModel');
const mongoose = require('mongoose');
const level = require('level');
const { db } = require('./UserModel');
const Logger = require('./Logger');

var logger = new Logger("Database");

mongoose.connect(process.env.MONGO_URL, {
    useNewUrlParser: true,
    useUnifiedTopology: true
}, err => {
    if (err) {
        console.error(err);
        return;
    }
    logger.log("Connected");
});

class Database {
    static userdb;
    static roomdb;

    static async load() {
        this.userdb = mongoose.connection;
        this.roomdb = level('src/db/rooms.db');
        // const writeFile = promisify(fs.writeFile);
        // const readdir = promisify(fs.readdir);

        // let files = await readdir("src/db/");
        // if (!files.includes("users.json")) {
        //     await writeFile('src/db/users.json', JSON.stringify(this.default_db, null, 2))
        //     this.userdb = new Map(Object.entries(require("./db/users.json")));
        // } else {
        //     this.userdb = new Map(Object.entries(require("./db/users.json")));
        // }
    }

    static async getUserData(cl, server) {
        if (!this.userdb) {
            await this.load();
        }
        
        let _id = createKeccakHash('keccak256').update((cl.server._id_Private_Key + cl.ip)).digest('hex').substr(0, 24);
        let user = await UserModel.findOne({ _id: _id }).exec();
        
        if (user == null) {
            user = this.createUser(_id);
        }

        return user;
    }

    static async createUser(_id) {
        let user = new UserModel({
            name: "Anonymous",
            _id: _id,
            color: "#" + ColorEncoder.intToRGB(ColorEncoder.hashCode(_id)),
            flags: {

            }
        });
        user.save();
        return user;
    }

    static async updateUser(_id, data) {
        UserModel.findOneAndUpdate({_id: _id}, data, {new: true}, (err, doc) => {
            if (err) {
                logger.error(err);
                return err;
            }
            
            if (doc == null) {
                logger.warn('Could not find user to save.');
                return;
            }
        });
    }

    static async wipe() {
        await UserModel.find({}, (err, docs) => {
            docs.forEach(doc => {
                doc.remove();
            });
        }).exec();
    }

    static strMapToObj(strMap) {
        return [...strMap.entries()].reduce((obj, [key, value]) => (obj[key] = value, obj), {});
    }

    static getRoomSettings(_id, cb) {
        let key = "room~"+_id;
        
        roomSettings
        
        this.roomdb.get(key, (err, value) => {
            if (err || !value || value == "") {
                cb(err, value);
                return;
            }
            cb(undefined, value);
        });
    }

    static setRoomSettings(_id, roomSettings, chat) {
        let roomData = new RoomDataModel(roomSettings, chat);
        let key = "room~"+_id;
        this.roomdb.put(key, JSON.stringify(roomData));
    }

    static getRoomSettings(_id, cb) {
        let key = "room~"+_id;
        this.roomdb.get(key, (err, value) => {
            if (err) {
                cb(err);
                return;
            }
            let settings = JSON.parse(value);
            cb(err, settings);
        });
    }

    static deleteRoomSettings(_id) {
        this.roomdb.del("room~"+_id);
    }
}

class RoomDataModel {
    constructor (settings, chat) {
        this.settings = settings;
        this.chat = chat;
    }
}

module.exports = Database;