Files
PteroStats/index.js

248 lines
11 KiB
JavaScript

const { Client, GatewayIntentBits, EmbedBuilder, time, ActivityType, ActionRowBuilder, ButtonBuilder, ButtonStyle } = require("discord.js");
const cachePath = require('node:path').join(__dirname, "./cache.json");
const fs = require("node:fs");
const cliColor = require("cli-color");
const package = require("./package.json")
console.log(`
_${cliColor.blueBright.bold(`${cliColor.underline("Ptero")}dact${cliColor.underline("yl & P")}eli${cliColor.underline("can")}`)}___ ______ ______
/\\ ___\\ /\\__ _\\ /\\ __ \\ /\\__ _\\ /\\ ___\\
\\ \\___ \\ \\/_ \\ \\/ \\ \\ \\_\\ \\ \\/_/\\ \\/ \\ \\___ \\
\\/\\_____\\ \\ \\_\\ \\ \\_\\ \\_\\ \\ \\_\\ \\/\\_____\\
\\/_____/ \\/_/ \\/_/\\/_/ \\/_/ \\/_____/${cliColor.yellowBright.bold(`${package.version}-dev`)}`);
console.log(
` \nCopyright © 2022 - ${new Date().getFullYear()} HirziDevs & Contributors` +
" \n \nDiscord: https://discord.znproject.my.id" +
" \n Source: https://github.com/HirziDevs/PteroStats" +
" \nLicense: https://github.com/Hirzidevs/PteroStats/blob/main/LICENSE" +
` \n \n${package.description}\n `
);
if (!fs.existsSync(".env")) return require("./handlers/installer.js")()
require("dotenv").config()
const config = require("./handlers/config.js");
const convertUnits = require("./handlers/convertUnits.js");
const getStats = require("./handlers/getStats.js");
console.log(cliColor.cyanBright("[PteroStats] ") + cliColor.green("Starting app..."));
console.log(cliColor.cyanBright("[PteroStats] ") + cliColor.redBright("You are using a development build! Some features may not work as intended."));
const client = new Client({
intents: [GatewayIntentBits.Guilds]
});
async function startGetStatus() {
try {
const results = await getStats();
createMessage({
panel: true,
nodes: results.nodes,
servers: results.servers,
users: results.users,
});
} catch {
console.log(cliColor.cyanBright("[PteroStats] ") + cliColor.redBright("Panel is currently offline."));
fs.readFile(cachePath, (err, data) => {
if (err) {
createMessage({ cache: false, panel: false });
return console.log(cliColor.cyanBright("[PteroStats] ") + cliColor.redBright("Last cache was not found!"));
}
try {
const results = JSON.parse(data);
createMessage({
cache: true,
panel: false,
nodes: results.nodes,
servers: results.servers,
users: results.users,
});
} catch {
createMessage({ cache: false, panel: false });
console.log(cliColor.cyanBright("[PteroStats] ") + cliColor.redBright("Something went wrong with cache data..."));
}
});
}
}
client.once("ready", () => {
console.log(cliColor.cyanBright("[PteroStats] ") + cliColor.green(`${cliColor.blueBright(client.user.tag)} is online!`));
if (config.presence.enable) {
if (config.presence.text && config.presence.type) {
switch (config.presence.type.toLowerCase()) {
case "playing":
config.presence.type = ActivityType.Playing;
break;
case "listening":
config.presence.type = ActivityType.Listening;
break;
case "competing":
config.presence.type = ActivityType.Competing;
break;
default:
config.presence.type = ActivityType.Watching;
}
client.user.setActivity(config.presence.text, {
type: config.presence.type,
});
}
if (config.presence.status) {
if (!["idle", "online", "dnd", "invisible"].includes(
config.presence.status.toLowerCase()
))
config.presence.status = "online";
client.user.setStatus(config.presence.status);
}
}
startGetStatus();
});
async function createMessage({ cache, panel, nodes, servers, users }) {
let embed = new EmbedBuilder()
.setAuthor({
name: config.embed.nodes.author.name || null,
iconURL: config.embed.nodes.author.icon || null
})
.setTitle(config.embed.nodes.title || null)
.setColor(config.embed.nodes.color || null)
.setURL(config.embed.nodes.url || null)
.setThumbnail(config.embed.nodes.thumbnail || null)
let embeds = [embed];
if (config.nodes_settings.details) {
nodes.forEach((node, index) => {
if (index % 25 === 0 && index !== 0) {
embed = new EmbedBuilder().setColor(config.embed.nodes.color);
if (embeds.length < 9) embeds.push(embed);
}
embed.addFields({
name: `${node.attributes.name} - ${node.status ? config.status.online : config.status.offline}`,
value:
"```\n" +
`Memory: ${convertUnits(node.attributes.allocated_resources.memory, node.attributes.memory, config.nodes_settings.unit)}\n` +
`Disk : ${convertUnits(node.attributes.allocated_resources.disk, node.attributes.disk, config.nodes_settings.unit)}` +
(node.attributes?.allocated_resources?.cpu ? `\nCPU : ${node.attributes?.allocated_resources?.cpu || 0}%` : "") +
"```"
});
});
} else {
embeds[0].setDescription(nodes.map(node => `**${node.attributes.name}** - ${node.status ? config.status.online : config.status.offline}`).join("\n"));
}
let panelEmbed = new EmbedBuilder()
.setAuthor({
name: config.embed.panel.author.name || null,
iconURL: config.embed.panel.author.icon || null
})
.setColor(config.embed.panel.color || null)
.setTitle(config.embed.panel.title || null)
.setURL(config.embed.panel.url || null)
.setTimestamp(config.embed.panel.timestamp ? new Date() : null)
.setFooter({
text: config.embed.panel.footer.text || null,
iconURL: config.embed.panel.footer.icon || null
})
.setThumbnail(config.embed.panel.thumbnail || null)
.setImage(config.embed.panel.image || null)
.setDescription(
config.embed.panel.description
.replace(
"{{time}}",
time(new Date(Date.now() + (config.refresh * 1000) + 1000), "R")
) || null
)
.addFields({
name: `Panel - ${panel ? config.status.online : config.status.offline}`,
value:
"```\n" +
`Nodes : ${nodes.length}\n` +
(config.panel_settings.servers ? `Servers: ${servers || "Unknown"}\n` : "") +
(config.panel_settings.users ? `Users : ${users || "Unknown"}\n` : "") +
"```"
});
if (config.panel_settings.status) embeds.unshift(panelEmbed);
embeds[embeds.length - 1]
.setTimestamp(config.embed.nodes.timestamp ? new Date() : null)
.setFooter({
text: config.embed.nodes.footer.text || null,
iconURL: config.embed.nodes.footer.icon || null
})
.setImage(config.embed.nodes.image || null)
if (!cache && !panel) {
embeds[embeds.length - 1].setDescription(
embeds[embeds.length - 1].data.description ? embeds[embeds.length - 1].data.description + "\n\nThere are no nodes to be display!" : "There are no nodes to be display!"
);
}
const components = []
if (config.button.enable && config.button.row1?.length > 0) {
components.push(
new ActionRowBuilder().addComponents(
config.button.row1.map(button =>
new ButtonBuilder()
.setLabel(button.label)
.setURL(button.url)
.setStyle(ButtonStyle.Link)
)
)
)
}
try {
const channel = await client.channels.fetch(process.env?.DiscordChannel);
const messages = await channel.messages.fetch({ limit: 10 });
const botMessage = messages.find(msg => msg.author.id === client.user.id);
console.log(cliColor.cyanBright("[PteroStats] ") + cliColor.green(`Panel stats successfully posted to the ${cliColor.blueBright(channel.name)} channel!`));
setTimeout(() => startGetStatus(), config.refresh * 1000);
if (botMessage) {
await botMessage.edit({ content: config.message.content || null, embeds, components });
} else {
await channel.send({ content: config.message.content || null, embeds, components });
}
} catch (error) {
DiscordErrorHandler(error);
}
}
function DiscordErrorHandler(error) {
if (error.rawError?.code === 429) {
console.log(cliColor.cyanBright("[PteroStats] ") + cliColor.redBright("Error 429 | Your IP has been rate limited by either Discord or your website. If it's a rate limit with Discord, you must wait. If it's a issue with your website, consider whitelisting your server IP."));
} else if (error.rawError?.code === 403) {
console.log(cliColor.cyanBright("[PteroStats] ") + cliColor.redBright("FORBIDDEN | The channel ID you provided is incorrect. Please double check you have the right ID. If you're not sure, read our documentation: https://github.com/HirziDevs/PteroStats#getting-channel-id"));
} else if (error.code === "ENOTFOUND") {
console.log(cliColor.cyanBright("[PteroStats] ") + cliColor.redBright("ENOTFOUND | DNS Error. Ensure your network connection and DNS server are functioning correctly."));
} else if (error.rawError?.code === 50001) {
console.log(cliColor.cyanBright("[PteroStats] ") + cliColor.redBright("Discord Error | Your discord bot doesn't have access to see/send message/edit message in the channel!"));
} else if (error.rawError?.errors && Object?.values(error.rawError.errors)[0]._errors[0].code === "MAX_EMBED_SIZE_EXCEEDED") {
console.log(cliColor.cyanBright("[PteroStats] ") + cliColor.redBright("Discord Error | Embed message limit exceeded! Please limit or decrease the nodes that need to be shown in the config!"));
} else if (error.rawError?.errors && Object?.values(error.rawError.errors)[0]._errors[0].code) {
console.log(Object.values(error.rawError.errors)[0]._errors[0].message);
} else {
console.error(cliColor.cyanBright("[PteroStats] ") + cliColor.redBright("Discord Error"), error);
}
process.exit();
}
try {
client.login(process.env?.DiscordBotToken);
} catch {
console.log(cliColor.cyanBright("[PteroStats] ") + cliColor.redBright("Discord Error | Invalid Discord Bot Token! Make sure you have the correct token in the config!"));
process.exit();
}