const { finished } = require('stream/promises'); 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) => { 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`); } } await new Promise(resolve => setTimeout(resolve, retryDelay)); // Delay before retrying } }, HandlePlayerBan: async (client, gamertag, ban) => { // get current bans for (let retries = 0; retries <= maxRetries; retries++) { try { 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', 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 parent function HandlePlayerBan() rather than write 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) }