Move Datastore class to src/client/classes folder

2025-07-01 19:43:39 +00:00 · 2021-09-10 23:25:48 +03:00
parent b9e1945719
commit 57f63c3838
2 changed files with 5 additions and 5 deletions
--- a/src/client/classes/Datastore.js
+++ b/src/client/classes/Datastore.js
@ -0,0 +1,136 @@
+const fs = require('fs');
+const path = require('path');
+
+module.exports = class {
+    constructor() {
+        this.map = new Map();
+        this.loadData();
+    }
+
+    loadData() {
+        const dir = path.join(path.dirname(__dirname), '../../datastore');
+        if (!fs.existsSync(dir)) {
+            fs.mkdirSync(dir);
+        }
+        //console.log("");
+        const dataFiles = fs.readdirSync(path.join(path.dirname(__dirname), '../../datastore')).filter(f => f.endsWith('.json'));
+        for (const file of dataFiles) {
+            try {
+                const json = require(`../../../datastore/${file}`);
+                this.map.set(json.guild.id, json);
+                //console.log('[LOADED] ' + file + " (" + json.guild.id + ")");
+                //console.log(JSON.stringify(json, null, 4));
+            } catch (error) {
+                //console.log('[ERROR] Loading ' + file + ' failed');
+            }
+        }
+        //console.log("");
+    }
+
+    calculateGlobal(client){
+        let guilds = this.map.keys();
+        let stations = client.stations;
+        var statistics = {};
+
+        if(!client.stations) return;
+
+        let calculation = guilds.next();
+
+        while (!calculation.done) {
+            let currentGuild = this.getEntry(calculation.value);
+            if(calculation.value != 'global'){
+                if(stations){
+                    Object.keys(stations).forEach(function(station) {
+                        if(currentGuild.statistics[stations[station].name] && currentGuild.statistics[stations[station].name].time && parseInt(currentGuild.statistics[stations[station].name].time) != 0  && currentGuild.statistics[stations[station].name].used && parseInt(currentGuild.statistics[stations[station].name].used) != 0){
+                            if(!statistics[stations[station].name]){
+                                statistics[stations[station].name] = {};
+                                statistics[stations[station].name].time = 0;
+                                statistics[stations[station].name].used = 0;
+                            }
+
+                            statistics[stations[station].name].time = parseInt(statistics[stations[station].name].time)+parseInt(currentGuild.statistics[stations[station].name].time);
+                            statistics[stations[station].name].used = parseInt(statistics[stations[station].name].used)+parseInt(currentGuild.statistics[stations[station].name].used);
+                        }
+                    });
+                }
+            }
+            calculation = guilds.next();
+        }
+
+        let newData = {};
+        newData.guild = {};
+        newData.guild.id = "global";
+        newData.guild.name = "global";
+        newData.statistics = statistics;
+        this.updateEntry(newData.guild, newData);
+    }
+
+
+    checkEntry(id){
+        if(!this.map.has(id)){
+            this.createEntry(id);
+            //this.showEntry(this.getEntry(id));
+        } else {
+            //this.showEntry(this.getEntry(id));
+        }
+    }
+
+    createEntry(id){
+        let newData = {};
+        newData.guild = {};
+        newData.guild.id = id;
+        newData.statistics = {};
+        newData.state = {};
+        this.map.set(id, newData);
+        this.saveEntry(id, newData);
+    }
+
+    getEntry(id){
+        return this.map.get(id);
+    }
+
+    updateEntry(guild, newData) {
+        newData.guild.name = guild.name;
+
+        let date = new Date();
+        newData.updated = date.toISOString().substring(0, 10)
+
+        this.map.set(guild.id, newData);
+        this.saveEntry(guild.id, newData);
+        //this.showEntry(this.getEntry(guild.id));
+    }
+
+    showEntry(data){
+        console.log(data);
+    }
+
+    createTestFile () {
+        let newData = {
+            "guild": {
+                "id": "test",
+                "name": "Test"
+            },
+            "statistics": {
+                "test": {
+                    "time": 0,
+                    "used": 0
+                }
+            },
+            "state": {
+
+            }
+        }
+
+        this.updateEntry(newData.guild, newData);
+    }
+
+    saveEntry(file, data) {
+        data = JSON.stringify(data, null, 4);
+
+        fs.writeFile(path.join(path.dirname(__dirname), '../../datastore') + "/" + file + ".json", data, 'utf8', function(err) {
+            if (err) {
+                //console.log(err);
+            }
+        });
+    }
+};