From 1713387ffde9112b4f28654e37d472772e4c9abc Mon Sep 17 00:00:00 2001
From: Essem <smswessem@gmail.com>
Date: Tue, 19 Jul 2022 23:27:43 -0500
Subject: [PATCH] Store directory size cache inside eris-fleet central store

---
 app.js                      |  3 ++-
 events/interactionCreate.js |  2 +-
 events/messageCreate.js     |  2 +-
 utils/tempimages.js         | 18 ++++++++++--------
 4 files changed, 14 insertions(+), 11 deletions(-)

diff --git a/app.js b/app.js
index 3668a44..30172e5 100644
--- a/app.js
+++ b/app.js
@@ -175,8 +175,9 @@ if (isMaster) {
       return promises.stat(`${process.env.TEMPDIR}/${file}`).then((stats) => stats.size);
     });
     const size = await Promise.all(dirstat);
-    process.env.DIRSIZECACHE = size.reduce((a, b)=>{
+    const reduced = size.reduce((a, b) => {
       return a + b;
     }, 0);
+    Admiral.centralStore.set("dirSizeCache", reduced);
   }
 }
diff --git a/events/interactionCreate.js b/events/interactionCreate.js
index 9216081..4c5cdcd 100644
--- a/events/interactionCreate.js
+++ b/events/interactionCreate.js
@@ -40,7 +40,7 @@ export default async (client, cluster, worker, ipc, interaction) => {
       }
       if (result.file.length > fileSize) {
         if (process.env.TEMPDIR && process.env.TEMPDIR !== "") {
-          await upload(client, result, interaction, true);
+          await upload(client, ipc, result, interaction, true);
         } else {
           await interaction[replyMethod]("The resulting image was more than 8MB in size, so I can't upload it.");
         }
diff --git a/events/messageCreate.js b/events/messageCreate.js
index 8e96e30..187598f 100644
--- a/events/messageCreate.js
+++ b/events/messageCreate.js
@@ -129,7 +129,7 @@ export default async (client, cluster, worker, ipc, message) => {
       }
       if (result.file.length > fileSize) {
         if (process.env.TEMPDIR && process.env.TEMPDIR !== "") {
-          await upload(client, result, message);
+          await upload(client, ipc, result, message);
         } else {
           await client.createMessage(message.channel.id, "The resulting image was more than 8MB in size, so I can't upload it.");
         }
diff --git a/utils/tempimages.js b/utils/tempimages.js
index 22ca516..e047fe7 100644
--- a/utils/tempimages.js
+++ b/utils/tempimages.js
@@ -1,7 +1,7 @@
 import * as logger from "../utils/logger.js";
 import { readdir, lstat, rm, writeFile } from "fs/promises";
 
-export async function upload(client, result, context, interaction = false) {
+export async function upload(client, ipc, result, context, interaction = false) {
   const filename = `${Math.random().toString(36).substring(2, 15)}.${result.name.split(".")[1]}`;
   await writeFile(`${process.env.TEMPDIR}/${filename}`, result.file);
   const imageURL = `${process.env.TMP_DOMAIN || "https://tmp.projectlounge.pw"}/${filename}`;
@@ -34,13 +34,14 @@ export async function upload(client, result, context, interaction = false) {
     }));
   }
   if (process.env.THRESHOLD) {
-    process.env.DIRSIZECACHE = parseInt(process.env.DIRSIZECACHE) + result.file.length;
-    await removeOldImages();
+    const size = ipc.centralStore.get("dirSizeCache") + result.file.length;
+    ipc.centralStore.set("dirSizeCache", size);
+    await removeOldImages(ipc, size);
   }
 }
 
-export async function removeOldImages() {
-  if (process.env.DIRSIZECACHE > process.env.THRESHOLD) {
+export async function removeOldImages(ipc, size) {
+  if (size > process.env.THRESHOLD) {
     const files = (await readdir(process.env.TEMPDIR)).map((file) => {
       return lstat(`${process.env.TEMPDIR}/${file}`).then((stats) => {
         if (stats.isSymbolicLink()) return;
@@ -52,14 +53,15 @@ export async function removeOldImages() {
       });
     }).filter(Boolean);
     const resolvedFiles = await Promise.all(files);
-    process.env.DIRSIZECACHE = resolvedFiles.reduce((a, b)=>{
+    let newSize = resolvedFiles.reduce((a, b)=>{
       return a + b.size;
     }, 0);
     const oldestFiles = resolvedFiles.sort((a, b) => a.ctime - b.ctime);
-    while (process.env.DIRSIZECACHE > process.env.THRESHOLD) {
+    while (newSize > process.env.THRESHOLD) {
       if (!oldestFiles[0]) break;
       await rm(`${process.env.TEMPDIR}/${oldestFiles[0].name}`);
-      process.env.DIRSIZECACHE = parseInt(process.env.DIRSIZECACHE) - oldestFiles[0].size;
+      newSize -= oldestFiles[0].size;
+      ipc.centralStore.set("dirSizeCache", newSize);
       logger.log(`Removed oldest image file: ${oldestFiles[0].name}`);
       oldestFiles.shift();
     }