From 2fe9efd36a65d8eb097012204348b8fe07208413 Mon Sep 17 00:00:00 2001
From: tgpethan <ethan@pyra-studios.net>
Date: Thu, 20 Feb 2020 07:42:46 +0000
Subject: [PATCH] oops

---
 EUS.js | 99 ++++++++++++++++++++++++++++++++++++++++------------------
 1 file changed, 69 insertions(+), 30 deletions(-)

diff --git a/EUS.js b/EUS.js
index 0502843..b319e68 100644
--- a/EUS.js
+++ b/EUS.js
@@ -3,6 +3,7 @@ config = require("../config/config.json"),
 chalk = require("chalk"),
 busboy = require("connect-busboy"),
 randomstring = require("randomstring"),
+getSize = require('get-folder-size'),
 emoji = require("../misc/emoji_list.json");
 
 // Defines the function of this module
@@ -75,41 +76,79 @@ module.exports = {
             res - Response from server
         */
         
+        let isAPI = false;
+
         if (req.query["stat"] == "get") return res.end('{ "status":1, "version":"'+global.internals.version+'" }');
 
-        // Register the time at the start of the request
-        d = new Date();
-        startTime = d.getTime();
-        // Get the requested image
-        let urs = ""+req.url; urs = urs.split("/")[1];
-        // Get the file type of the image from image_json and make sure it exists
-        fs.access(__dirname + BASE_PATH + "/i/"+urs+image_json[urs], error => {
-            if (error) {
-                // Doesn't exist, handle request normaly
-                if (req.url === "/") { urs = "/index.html" } else { urs = req.url };
-                fs.access(__dirname + BASE_PATH + "/files"+urs, error => {
-                    if (error) {
-                        // Doesn't exist, send a 404 to the client.
-                        res.status(404).end("404!");
-                        d = new Date();
-                        endTime = d.getTime();
-                        global.modules.consoleHelper.printInfo(emoji.cross, `${req.method}: ${chalk.red("[404]")} ${req.url} ${endTime - startTime}ms`);
-                    } else {
-                        // File does exist, send it back to the client.
-                        res.sendFile(__dirname + BASE_PATH + "/files"+req.url);
-                        d = new Date();
-                        endTime = d.getTime();
-                        global.modules.consoleHelper.printInfo(emoji.heavy_check, `${req.method}: ${chalk.green("[200]")} ${req.url} ${endTime - startTime}ms`);
+        if (req.url.split("?")[0] == "/api/get-stats") {
+            isAPI = true;
+            const filesaa = req.query["f"],
+            spaceaa = req.query["s"];
+            let jsonaa = {};
+            if (filesaa == 1) {
+                jsonaa["files"] = {};
+                for (var i2 = 0; i2 < eusConfig.acceptedTypes.length; i2++) {
+                    jsonaa["files"][`${eusConfig.acceptedTypes[i2]}`.replace(".", "")] = 0;
+                }
+                fs.readdir(__dirname + BASE_PATH + "/i", (err, files) => {
+                    if (err) throw err;
+                    for (var i = 0; i < files.length; i++) {
+                        for (var i1 = 0; i1 < eusConfig.acceptedTypes.length; i1++) {
+                            const jsudfg = files[i].split(".");
+                            if (`.${jsudfg[jsudfg.length-1]}` == eusConfig.acceptedTypes[i1]) {
+                                jsonaa["files"][eusConfig.acceptedTypes[i1].replace(".", "")]++;
+                            }
+                        }
                     }
+                    if (spaceaa != 1) return res.end(JSON.stringify(jsonaa));
                 });
-            } else {
-                // Image does exist, send it back.
-                res.sendFile(__dirname + BASE_PATH + "/i/"+urs+image_json[urs]);
-                d = new Date();
-                endTime = d.getTime();
-                global.modules.consoleHelper.printInfo(emoji.heavy_check, `${req.method}: ${chalk.green("[200]")} ${req.url} ${endTime - startTime}ms`);
             }
-        });
+            if (spaceaa == 1) {
+                jsonaa["used"] = "";
+                getSize(__dirname + BASE_PATH + "/i", (err, size) => {
+                    if (err) throw err;
+                    const sizeOfFolder = (size / 1024 / 1024 / 1024).toFixed(2);
+                    jsonaa["used"] = `${sizeOfFolder} GB`;
+                    return res.end(JSON.stringify(jsonaa));
+                });
+            }
+        }
+
+        if (!isAPI) {
+            // Register the time at the start of the request
+            d = new Date();
+            startTime = d.getTime();
+            // Get the requested image
+            let urs = ""+req.url; urs = urs.split("/")[1];
+            // Get the file type of the image from image_json and make sure it exists
+            fs.access(__dirname + BASE_PATH + "/i/"+urs+image_json[urs], error => {
+                if (error) {
+                    // Doesn't exist, handle request normaly
+                    if (req.url === "/") { urs = "/index.html" } else { urs = req.url };
+                    fs.access(__dirname + BASE_PATH + "/files"+urs, error => {
+                        if (error) {
+                            // Doesn't exist, send a 404 to the client.
+                            res.status(404).end("404!");
+                            d = new Date();
+                            endTime = d.getTime();
+                            global.modules.consoleHelper.printInfo(emoji.cross, `${req.method}: ${chalk.red("[404]")} ${req.url} ${endTime - startTime}ms`);
+                        } else {
+                            // File does exist, send it back to the client.
+                            res.sendFile(__dirname + BASE_PATH + "/files"+req.url);
+                            d = new Date();
+                            endTime = d.getTime();
+                            global.modules.consoleHelper.printInfo(emoji.heavy_check, `${req.method}: ${chalk.green("[200]")} ${req.url} ${endTime - startTime}ms`);
+                        }
+                    });
+                } else {
+                    // Image does exist, send it back.
+                    res.sendFile(__dirname + BASE_PATH + "/i/"+urs+image_json[urs]);
+                    d = new Date();
+                    endTime = d.getTime();
+                    global.modules.consoleHelper.printInfo(emoji.heavy_check, `${req.method}: ${chalk.green("[200]")} ${req.url} ${endTime - startTime}ms`);
+                }
+            });
+        }
     },
     post:function(req, res) {
         /*