fix failing

PenguinMod · Feb 26, 2024 · f52e874 · f52e874
1 parent c88c58f
commit f52e874
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 5 deletions.
diff --git a/api/db/UMTests.js b/api/db/UMTests.js
@@ -1,11 +1,13 @@
 const UserManager = require('./UserManager');
+const prompt = require('prompt-sync')();
 const colors = require('colors');
 
 
-const understands = process.argv.includes("-u") || 
+let understands = process.argv.includes("-u") || 
                     process.argv.includes("--understands");
 
 async function tests() {
+
     const manager = new UserManager();
     await manager.init();
 
@@ -393,6 +395,16 @@ async function tests() {
 
 
 (async () => {
+    if (!understands) {
+        let pmt = prompt("This will reset the database, are you sure? (Y/n) ");
+        if (typeof pmt !== "string" ||
+            pmt === "n"
+        ) {
+            process.exit(0);
+        }
+        understands = true;
+    }
+
     let result = await tests();
     if (result) {
         console.log("[ PASS ]".green, "All tests passed");

diff --git a/api/db/UserManager.js b/api/db/UserManager.js
@@ -55,14 +55,21 @@ class UserManager {
      */
     async reset(understands = false) {
         if (!understands) {
-            if (prompt("This deletes ALL DATA. Are you sure? (Y/n) ") === "n")
-            return;
+            let unde = prompt("This deletes ALL DATA. Are you sure? (Y/n) ")
+            if (
+                typeof unde !== "string"
+            ) {
+                return;
+            }
+            console.log(typeof unde)
         }
         await this.users.deleteMany({});
         await this.reports.deleteMany({});
         await this.projects.deleteMany({});
-        fs.rmSync(path.join(__dirname, "projects/files"), { recursive: true, force: true });
-        fs.rmSync(path.join(__dirname, "projects/images"), { recursive: true, force: true });
+        if (fs.existsSync(path.join(__dirname, "projects/files")) && fs.existsSync(path.join(__dirname, "projects/images"))) {
+            fs.rmSync(path.join(__dirname, "projects/files"), { recursive: true, force: true });
+            fs.rmSync(path.join(__dirname, "projects/images"), { recursive: true, force: true });
+        }
         fs.mkdirSync(path.join(__dirname, "projects/files"));
         fs.mkdirSync(path.join(__dirname, "projects/images"));
     }