tgpholly/EUS
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity

Add api endpoint for stats #6

Merged
tgpholly merged 3 commits from add-stats-api into master 2020-02-20 08:03:10 +00:00
Conversation 1 Commits 3 Files changed 1 +69 -30
1 changed files with 69 additions and 30 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files
Showing only changes of commit 2fe9efd36a - Show all commits

99
EUS.js
View file

@ -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) {
/*