Merge pull request #2 from IrPgFKS0/co-work

Nitrado API Retry and other Docker quality of life changes

util/NitradoAPI.js

@@ -3,64 +3,83 @@ const concat = require('concat-stream');
 const { Readable } = require('stream');
 const FormData = require('form-data');
 const fs = require('fs');
+const maxRetries = 5;
+const retryDelay = 5000; // 5 second
 
 module.exports = {
-  
   DownloadNitradoFile: async (client, filename, outputDir) => {
-    const res = await fetch(`https://api.nitrado.net/services/${client.config.Nitrado.ServerID}/gameservers/file_server/download?file=${filename}`, {
-      headers: {
-        "Authorization": client.config.Nitrado.Auth
+    for (let retries = 0; retries <= maxRetries; retries++) {
+      try {
+        const res = await fetch(`https://api.nitrado.net/services/${client.config.Nitrado.ServerID}/gameservers/file_server/download?file=${filename}`, {
+          headers: {
+            "Authorization": client.config.Nitrado.Auth
+          }
+        }).then(response => 
+          response.json().then(data => data)
+        ).then(res => res);
+      
+        const stream = fs.createWriteStream(outputDir);
+        if (!res.data || !res.data.token) {
+          client.error(`Error downloading File "${filename}":`);
+          client.error(res);
+          return -1;
+        }
+        const { body } = await fetch(res.data.token.url);
+        await finished(Readable.fromWeb(body).pipe(stream));
+        return 0;
+      } catch (error) {
+        if (retries === maxRetries) {
+            throw new Error(`Failed to fetch data after ${maxRetries} retries`);
+        }
       }
-    }).then(response => 
-      response.json().then(data => data)
-    ).then(res => res);
-  
-    const stream = fs.createWriteStream(outputDir);
-    if (!res.data || !res.data.token) {
-      client.error(`Error downloading File "${filename}":`);
-      client.error(res);
-      return -1;
+      await new Promise(resolve => setTimeout(resolve, retryDelay)); // Delay before retrying
     }
-    const { body } = await fetch(res.data.token.url);
-    await finished(Readable.fromWeb(body).pipe(stream));
-    return 0;
   },
 
   HandlePlayerBan: async (client, gamertag, ban) => {
     // get current bans
-    const res = await fetch(`https://api.nitrado.net/services/${client.config.Nitrado.ServerID}/gameservers`, {
-      headers: {
-        "Authorization": client.config.Nitrado.Auth
-      }
-    }).then(response => 
-      response.json().then(data => data)
-    ).then(res => res);
-  
-    let bans = res.data.gameserver.settings.general.bans;
-    if (ban) bans += `\r\n${gamertag}`;
-    else if (!ban) bans = bans.replace(gamertag, '');
-    else client.error("Incorrect Ban Option: HandlePlayerBan");
-
-    const formData = new FormData();
-    formData.append("category", "general");
-    formData.append("key", "bans");
-    formData.append("value", bans);
-    formData.pipe(concat(data => {
-      async function sendList() {
-        await fetch(`https://api.nitrado.net/services/${client.config.Nitrado.ServerID}/gameservers/settings`, {
-          method: "POST",
-          credentials: 'include',
+    for (let retries = 0; retries <= maxRetries; retries++) {
+      try {
+        const res = await fetch(`https://api.nitrado.net/services/${client.config.Nitrado.ServerID}/gameservers`, {
           headers: {
-            ...formData.getHeaders(),
             "Authorization": client.config.Nitrado.Auth
-          },
-          body: data,
-        });
-      }
-      sendList();
-    }));
-  },
+          }
+        }).then(response => 
+          response.json().then(data => data)
+        ).then(res => res);
+      
+        let bans = res.data.gameserver.settings.general.bans;
+        if (ban) bans += `\r\n${gamertag}`;
+        else if (!ban) bans = bans.replace(gamertag, '');
+        else client.error("Incorrect Ban Option: HandlePlayerBan");
 
+        const formData = new FormData();
+        formData.append("category", "general");
+        formData.append("key", "bans");
+        formData.append("value", bans);
+        formData.pipe(concat(data => {
+          async function sendList() {
+            await fetch(`https://api.nitrado.net/services/${client.config.Nitrado.ServerID}/gameservers/settings`, {
+              method: "POST",
+              credentials: 'include',
+              headers: {
+                ...formData.getHeaders(),
+                "Authorization": client.config.Nitrado.Auth
+              },
+              body: data,
+            });
+          }
+          sendList();
+        }));
+      } catch (error) {
+        if (retries === maxRetries) {
+            throw new Error(`Failed to fetch data after ${maxRetries} retries`);
+        }
+      }
+      await new Promise(resolve => setTimeout(resolve, retryDelay)); // Delay before retrying
+    }
+  },
+  
   /*
     Allows more explicit function names outside this file;
     i.e BanPlayer() & UnbanPlayer() that both call to the 
@@ -68,7 +87,7 @@ module.exports = {
     two whole different functions for each.
   */
 
-  BanPlayer:   async (client, gamertag) => module.exports.HandlePlayerBan(client, gamertag, true),
-  UnbanPlayer: async (client, gamertag) => module.exports.HandlePlayerBan(client, gamertag, false),
-
-}
+    BanPlayer:   async (client, gamertag) => module.exports.HandlePlayerBan(client, gamertag, true),
+    UnbanPlayer: async (client, gamertag) => module.exports.HandlePlayerBan(client, gamertag, false),
+  
+  }