feat: delete downloaded images files automatically after a minute and minor changes in mem monitoring

Author: mrepol742

.env.example

@@ -3,7 +3,8 @@ PROJECT_CANIS_ALIAS=Canis
 # automatically reboot canis, if a potential memory leak has been detected,
 # or the memory usage was just way too high
 PROJECT_AUTO_RESTART=false
-PROJECT_MAX_MEMORY=1024 # 1GB
+PROJECT_THRESHOLD_MEMORY=1024
+PROJECT_MAX_MEMORY=2048 # 2GB
 
 # whether to enable debug mode
 # in debug mode, additional logging is enabled

src/components/utils/download.ts

@@ -3,6 +3,7 @@ import path from "path";
 import crypto from "crypto";
 import { MessageMedia } from "whatsapp-web.js";
 import axios from "../axios";
+import log from "./log";
 
 export async function download(
   url: string,
@@ -20,6 +21,20 @@ export async function download(
 
   try {
     await fs.writeFile(tempPath, response.data, "base64");
+
+    // Schedule deletion after 1 minute
+    setTimeout(async () => {
+      try {
+        await fs.unlink(tempPath);
+        log.info("Download", `Temporary file deleted: ${tempPath}`);
+      } catch (err) {
+        log.error(
+          "Download",
+          `Failed to delete temp file: ${(err as Error).message}`,
+        );
+      }
+    }, 60 * 1000);
+
     return MessageMedia.fromFilePath(tempPath);
   } catch (err) {
     throw new Error(

src/components/utils/memMonitor.ts

@@ -25,7 +25,7 @@ export default class MemoryMonitor {
 
   constructor({
     interval = 60000,
-    thresholdMB = 500,
+    thresholdMB = 1024,
   }: MemoryMonitorOptions = {}) {
     this.interval = interval;
     this.thresholdMB = thresholdMB;

src/index.ts

@@ -11,7 +11,14 @@ import "./components/process";
 import "./components/server";
 import MemoryMonitor from "./components/utils/memMonitor";
 
-const monitor = new MemoryMonitor({ interval: 30000 });
+const monitor = new MemoryMonitor({
+  interval: 60000,
+  thresholdMB: parseInt(
+    process.env.PROJECT_THRESHOLD_MEMORY || "1024",
+    10
+  )
+});
+
 monitor.start();
 checkRequirements();