const { finished } = require('stream/promises'); const concat = require('concat-stream'); const { Readable } = require('stream'); const FormData = require('form-data'); const fs = require('fs'); 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 } }).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; }, 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") throw "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(); })); }, /* 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), }