SowinskiBraeden/dayz-reforger
1
0 Fork 0
Code Issues Packages Projects Releases Wiki Activity

refactor/support multiple discord guilds with 1 nitrado server each + better handling for NitradoAPI

Browse Source
This commit is contained in:
SowinskiBraeden committed 2024-04-04 20:11:48 -07:00
1 parent 71d03ffa72
commit 55c8a9ee7f
33 files changed
+888 -1028
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

util/NitradoAPI.js
+85 -65
View File
@@ -8,20 +8,18 @@ const retryDelay = 5000; // 5 seconds
// Private functions (only called locally)
const UploadNitradoFile = async (client, remoteDir, remoteFilename, localFileDir) => {
const UploadNitradoFile = async (nitrado_cred, client, remoteDir, remoteFilename, localFileDir) => {
for (let retries = 0; retries <= maxRetries; retries++) {
try {
const res = await fetch(`https://api.nitrado.net/services/${client.config.Nitrado.ServerID}/gameservers/file_server/upload?` + new URLSearchParams({
const res = await fetch(`https://api.nitrado.net/services/${nitrado_cred.ServerID}/gameservers/file_server/upload?` + new URLSearchParams({
path: remoteDir,
file: remoteFilename
}), {
method: "POST",
headers: {
"Authorization": client.config.Nitrado.Auth
"Authorization": nitrado_cred.Auth
},
}).then(response =>
response.json().then(data => data)
).then(res => res);
}).then(response => response.json());
let contents = fs.readFileSync(localFileDir, 'utf8');
@@ -34,22 +32,24 @@ const UploadNitradoFile = async (client, remoteDir, remoteFilename, localFileDir
body: contents,
})
if (!uploadRes.ok) {
const errorText = await uploadRes.text();
client.error(`Failed to upload file to Nitrado (${client.config.Nitrado.ServerID}): status: ${uploadRes.status}, message: ${errorText}: UploadNitradoFile`);
client.error(`Failed to upload file to Nitrado (${nitrado_cred.ServerID}): status: ${uploadRes.status}, message: ${res.statusText}: UploadNitradoFile`);
if (retries === 2) return 1; // Return error status on the second failed status code.
} else {
return uploadRes;
}
} catch (error) {
client.error(`UploadNitradoFile: Error connecting to server (${client.config.Nitrado.ServerID}): ${error.message}`);
if (retries === maxRetries) throw new Error(`UploadNitradoFile: Error connecting to server (${client.config.Nitrado.ServerID}) after ${maxRetries} retries`);
client.error(`UploadNitradoFile: Error connecting to server (${nitrado_cred.ServerID}): ${error.message}`);
if (retries === maxRetries) {
client.error(`UploadNitradoFile: Error connecting to server (${nitrado_cred.ServerID}) after ${maxRetries} retries`);
return 1;
}
}
await new Promise(resolve => setTimeout(resolve, retryDelay)); // Delay before retrying
}
}
const HandlePlayerBan = async (client, gamertag, ban) => {
const data = await module.exports.FetchServerSettings(client, 'HandlePlayerBan'); // Fetch server status
const HandlePlayerBan = async (nitrado_cred, client, gamertag, ban) => {
const data = await module.exports.FetchServerSettings(nitrado_cred, client, 'HandlePlayerBan'); // Fetch server status
if (data && data != 1) {
let bans = data.data.gameserver.settings.general.bans;
@@ -59,26 +59,31 @@ const HandlePlayerBan = async (client, gamertag, ban) => {
let category = 'general';
let key = 'bans';
return await module.exports.PostServerSettings(client, category, key, bans); // returns 1 (failed) or 0 (not failed)
return await module.exports.PostServerSettings(nitrado_cred, client, category, key, bans); // returns 1 (failed) or 0 (not failed)
}
}
const GetRemoteDir = async (client, dir="") => {
const GetRemoteDir = async (nitrado_cred, client, dir="") => {
const dirParam = client.exists(dir) ? `?dir=${dir}` : "";
for (let retries = 0; retries <= maxRetries; retries++) {
try {
const res = await fetch(`https://api.nitrado.net/services/${client.config.Nitrado.ServerID}/gameservers/file_server/list${dirParam}`, {
const res = await fetch(`https://api.nitrado.net/services/${nitrado_cred.ServerID}/gameservers/file_server/list${dirParam}`, {
headers: {
"Authorization": client.config.Nitrado.Auth
"Authorization": nitrado_cred.Auth
}
}).then(response =>
response.json().then(data => data)
).then(res => res);
if (res.status === "error") return 1;
return res.data.entries;
} catch (error) {
client.error(`GetRemoteDir: Error connecting to server (${client.config.Nitrado.ServerID}): ${error.message}`);
if (retries == maxRetries) throw new Error(`GetRemoteDir: Error connecting to server (${client.config.Nitrado.ServerID}) after ${maxRetries} retries`)
client.error(`GetRemoteDir: Error connecting to server (${nitrado_cred.ServerID}): ${error}`);
if (retries == maxRetries) {
client.error(`GetRemoteDir: Error connecting to server (${nitrado_cred.ServerID}) after ${maxRetries} retries`);
return 1;
}
}
await new Promise(resolve => setTimeout(resolve, retryDelay)); // Delay before retrying
}
@@ -88,12 +93,12 @@ const GetRemoteDir = async (client, dir="") => {
module.exports = {
DownloadNitradoFile: async(client, filename, outputDir) => {
DownloadNitradoFile: async(nitrado_cred, 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}`, {
const res = await fetch(`https://api.nitrado.net/services/${nitrado_cred.ServerID}/gameservers/file_server/download?file=${filename}`, {
headers: {
"Authorization": client.config.Nitrado.Auth
"Authorization": nitrado_cred.Auth
}
}).then(response =>
response.json().then(data => data)
@@ -101,16 +106,18 @@ module.exports = {
const stream = fs.createWriteStream(outputDir);
if (!res.data || !res.data.token) {
const errorText = await res.text();
client.error(`Error downloading File "${filename}": message: ${errorText}: DownloadNitradoFile`);
client.error(`Error downloading File "${filename}": message: ${res.message}: DownloadNitradoFile`);
return 1;
}
const { body } = await fetch(res.data.token.url);
await finished(Readable.fromWeb(body).pipe(stream));
return 0;
} catch (error) {
client.error(`DownloadNitradoFile: Error connecting to server (${client.config.Nitrado.ServerID}): ${error.message}`);
if (retries === maxRetries) throw new Error(`DownloadNitradoFile: Error connecting to server (${client.config.Nitrado.ServerID}) after ${maxRetries} retries`);
client.error(`DownloadNitradoFile: Error connecting to server (${nitrado_cred.ServerID}): ${error.message}`);
if (retries === maxRetries) {
client.error(`DownloadNitradoFile: Error connecting to server (${nitrado_cred.ServerID}) after ${maxRetries} retries`);
return 1;
}
}
await new Promise(resolve => setTimeout(resolve, retryDelay)); // Delay before retrying
}
@@ -122,67 +129,70 @@ module.exports = {
rather than write two whole different functions for each.
*/
BanPlayer: async (client, gamertag) => await HandlePlayerBan(client, gamertag, true),
UnbanPlayer: async (client, gamertag) => await HandlePlayerBan(client, gamertag, false),
BanPlayer: async (nitrado_cred, client, gamertag) => await HandlePlayerBan(nitrado_cred, client, gamertag, true),
UnbanPlayer: async (nitrado_cred, client, gamertag) => await HandlePlayerBan(nitrado_cred, client, gamertag, false),
RestartServer: async (client, restart_message, message) => {
RestartServer: async (nitrado_cred, client, restart_message, message) => {
const params = {
restart_message: restart_message,
message: message
};
// client.log('Restarting server...');
for (let retries = 0; retries < maxRetries; retries++) {
try {
const res = await fetch(`https://api.nitrado.net/services/${client.config.Nitrado.ServerID}/gameservers/restart`, {
const res = await fetch(`https://api.nitrado.net/services/${nitrado_cred.ServerID}/gameservers/restart`, {
method: "POST",
headers: {
"Authorization": client.config.Nitrado.Auth,
"Authorization": nitrado_cred.Auth,
},
body: JSON.stringify(params)
});
if (!res.ok) {
const errorText = await res.text();
client.error(`Failed to restart Nitrado server (${client.config.Nitrado.ServerID}): status: ${res.status}, message: ${errorText}: RestartServer`);
client.error(`Failed to restart Nitrado server (${nitrado_cred.ServerID}): status: ${res.status}, message: ${res.statusText}: RestartServer`);
return 1; // Return error status on failed status code.
} else {
return 0;
}
} catch (error) {
client.error(`RestartServer: Error connecting to server (${client.config.Nitrado.ServerID}): ${error.message}`);
if (retries === maxRetries) throw new Error(`RestartServer: Error connecting to server (${client.config.Nitrado.ServerID}) after ${maxRetries} retries`);
client.error(`RestartServer: Error connecting to server (${nitrado_cred.ServerID}): ${error.message}`);
if (retries === maxRetries) {
client.error(`RestartServer: Error connecting to server (${nitrado_cred.ServerID}) after ${maxRetries} retries`);
return 1;
}
}
await new Promise(resolve => setTimeout(resolve, retryDelay)); // Delay before retrying
}
},
FetchServerSettings: async (client, fetcher) => {
FetchServerSettings: async (nitrado_cred, client, fetcher) => {
for (let retries = 0; retries <= maxRetries; retries++) {
try {
// get current status
const res = await fetch(`https://api.nitrado.net/services/${client.config.Nitrado.ServerID}/gameservers`, {
const res = await fetch(`https://api.nitrado.net/services/${nitrado_cred.ServerID}/gameservers`, {
headers: {
"Authorization": client.config.Nitrado.Auth
"Authorization": nitrado_cred.Auth
}
});
if (!res.ok) {
const errorText = await res.text();
client.error(`Failed to get Nitrado server stats (${client.config.Nitrado.ServerID}): status: ${res.status}, message: ${errorText}: ${fetcher} via FetchServerSettings`);
client.error(`Failed to get Nitrado server stats (${nitrado_cred.ServerID}): status: ${res.status}, message: ${res.statusText}: ${fetcher} via FetchServerSettings`);
if (retries === 2) return 1; // Return error status on the second failed status code.
} else {
const data = await res.json();
return data;
}
} catch (error) {
client.error(`${fetcher} via FetchServerSettings: Error connecting to server (${client.config.Nitrado.ServerID}): ${error.message}`);
if (retries === maxRetries) throw new Error(`${fetcher} via FetchServerSettings: Error connecting to server (${client.config.Nitrado.ServerID}) after ${maxRetries} retries`);
client.error(`${fetcher} via FetchServerSettings: Error connecting to server (${nitrado_cred.ServerID}): ${error.message}`);
if (retries === maxRetries) {
client.error(`${fetcher} via FetchServerSettings: Error connecting to server (${nitrado_cred.ServerID}) after ${maxRetries} retries`);
return 1;
}
}
await new Promise(resolve => setTimeout(resolve, retryDelay)); // Delay before retrying
}
},
PostServerSettings: async (client, category, key, value) => {
PostServerSettings: async (nitrado_cred, client, category, key, value) => {
for (let retries = 0; retries <= maxRetries; retries++) {
try {
const formData = new FormData();
@@ -191,18 +201,17 @@ module.exports = {
formData.append("value", value);
formData.pipe(concat(data => {
async function postData() {
const res = await fetch(`https://api.nitrado.net/services/${client.config.Nitrado.ServerID}/gameservers/settings`, {
const res = await fetch(`https://api.nitrado.net/services/${nitrado_cred.ServerID}/gameservers/settings`, {
method: "POST",
credentials: 'include',
headers: {
...formData.getHeaders(),
"Authorization": client.config.Nitrado.Auth
"Authorization": nitrado_cred.Auth
},
body: data,
});
if (!res.ok) {
const errorText = await res.text();
client.error(`Failed to get post Nitrado server settings (${client.config.Nitrado.ServerID}): status: ${res.status}, message: ${errorText}: PostServerSettings`);
client.error(`Failed to get post Nitrado server settings (${nitrado_cred.ServerID}): status: ${res.status}, message: ${res.statusText}: PostServerSettings`);
if (retries === 2) return 1; // Return error status on the second failed status code.
} else {
const data = await res.json();
@@ -213,39 +222,46 @@ module.exports = {
}));
return 0;
} catch (error) {
client.error(`PostServerSettings: Error connecting to server (${client.config.Nitrado.ServerID}): ${error.message}`);
if (retries === maxRetries) throw new Error(`PostServerSettings: Error connecting to server (${client.config.Nitrado.ServerID}) after ${maxRetries} retries`);
client.error(`PostServerSettings: Error connecting to server (${nitrado_cred.ServerID}): ${error.message}`);
if (retries === maxRetries) {
client.error(`PostServerSettings: Error connecting to server (${nitrado_cred.ServerID}) after ${maxRetries} retries`);
return 1;
}
}
await new Promise(resolve => setTimeout(resolve, retryDelay)); // Delay before retrying
}
},
CheckServerStatus: async (client) => {
const data = await module.exports.FetchServerSettings(client, 'CheckServerStatus'); // Fetch server status
CheckServerStatus: async (nitrado_cred, client) => {
const data = await module.exports.FetchServerSettings(nitrado_cred, client, 'CheckServerStatus'); // Fetch server status
if (data && data != 1) {
if (data && data.data.gameserver.status === 'stopped') {
client.log(`Restart of Nitrado server ${client.config.Nitrado.ServerID} has been invoked by the bot, the periodic check showed status of "${data.data.gameserver.status}".`);
client.log(`Restart of Nitrado server ${nitrado_cred.ServerID} has been invoked by the bot, the periodic check showed status of "${data.data.gameserver.status}".`);
// Write optional "restart_message" to set in the Nitrado server logs and send a notice "message" to your server community.
restart_message = 'Server being restarted by periodic bot check.';
message = 'The server was restarted by periodic bot check!';
module.exports.RestartServer(client, restart_message, message);
module.exports.RestartServer(nitrado_cred, client, restart_message, message);
}
}
},
DisableBaseDamage: async (client, preference) => {
DisableBaseDamage: async (nitrado_cred, client, preference) => {
const pref = preference ? '1' : '0';
const posted = await module.exports.PostServerSettings(client, "config", "disableBaseDamage", pref);
const posted = await module.exports.PostServerSettings(nitrado_cred, client, "config", "disableBaseDamage", pref);
if (posted == 1) return 1;
const basePath = await GetRemoteDir(client).then(dirs => dirs.filter(dir => dir.type == 'dir')[0].path)
const missionPath = await GetRemoteDir(client, basePath).then(dirs => dirs[0].path)
const remoteDirs = await GetRemoteDir(nitrado_cred, client);
if (remoteDirs == 1) return 1;
const basePath = remoteDirs.filter(dir => dir.type == 'dir')[0].path
const remoteDirsFromBase = await GetRemoteDir(nitrado_cred, client, basePath);
if (remoteDirsFromBase == 1) return 1;
const missionPath = remoteDirsFromBase[0].path;
const cfggameplayPath = `${missionPath}/cfggameplay.json`;
const jsonDir = `./logs/cfggameplay.json`;
await module.exports.DownloadNitradoFile(client, cfggameplayPath, jsonDir);
await module.exports.DownloadNitradoFile(nitrado_cred, client, cfggameplayPath, jsonDir);
let gameplay = JSON.parse(fs.readFileSync(jsonDir));
gameplay.GeneralData.disableBaseDamage = preference;
@@ -253,23 +269,27 @@ module.exports = {
// write JSON to file
fs.writeFileSync(jsonDir, JSON.stringify(gameplay, null, 2));
const uploaded = await UploadNitradoFile(client, missionPath, 'cfggameplay.json', jsonDir);
const uploaded = await UploadNitradoFile(nitrado_cred, client, missionPath, 'cfggameplay.json', jsonDir);
if (uploaded == 1) return 1;
return 0;
},
DisableContainerDamage: async (client, preference) => {
DisableContainerDamage: async (nitrado_cred, client, preference) => {
const pref = preference ? '1' : '0';
const posted = await module.exports.PostServerSettings(client, "config", "disableContainerDamage", pref);
const posted = await module.exports.PostServerSettings(nitrado_cred, client, "config", "disableContainerDamage", pref);
if (posted == 1) return 1;
const basePath = await GetRemoteDir(client).then(dirs => dirs.filter(dir => dir.type == 'dir')[0].path)
const missionPath = await GetRemoteDir(client, basePath).then(dirs => dirs[0].path)
const remoteDirs = await GetRemoteDir(nitrado_cred, client);
if (remoteDirs == 1) return 1;
const basePath = remoteDirs.filter(dir => dir.type == 'dir')[0].path
const remoteDirsFromBase = await GetRemoteDir(nitrado_cred, client, basePath);
if (remoteDirsFromBase == 1) return 1;
const missionPath = remoteDirsFromBase[0].path;
const cfggameplayPath = `${missionPath}/cfggameplay.json`;
const jsonDir = `./logs/cfggameplay.json`;
await module.exports.DownloadNitradoFile(client, cfggameplayPath, jsonDir);
await module.exports.DownloadNitradoFile(nitrado_cred, client, cfggameplayPath, jsonDir);
let gameplay = JSON.parse(fs.readFileSync(jsonDir));
gameplay.GeneralData.disableContainerDamage = preference;
@@ -277,7 +297,7 @@ module.exports = {
// write JSON to file
fs.writeFileSync(jsonDir, JSON.stringify(gameplay, null, 2));
const uploaded = await UploadNitradoFile(client, missionPath, 'cfggameplay.json', jsonDir);
const uploaded = await UploadNitradoFile(nitrado_cred, client, missionPath, 'cfggameplay.json', jsonDir);
if (uploaded == 1) return 1;
return 0;