From 8682583d416f1b06f963cc7bcb3212b8cf439e0e Mon Sep 17 00:00:00 2001 From: Mitchell3514 Date: Sat, 2 Mar 2024 16:11:57 +0100 Subject: [PATCH] chore(pollux.js): cleanup --- pollux.js | 520 +++++++++++++++++++++++++++++------------------------- 1 file changed, 284 insertions(+), 236 deletions(-) diff --git a/pollux.js b/pollux.js index d4fda6a3..f686d9d2 100644 --- a/pollux.js +++ b/pollux.js @@ -1,44 +1,41 @@ -const FLAVORED_CLIENT = process.env.PRIME_FLAVORED_CLIENT; +/* eslint-disable @typescript-eslint/no-use-before-define */ +/* eslint-disable implicit-arrow-linebreak */ +/* eslint-disable comma-dangle */ +/* eslint-disable camelcase */ +/* eslint-disable import/order */ +/* eslint-disable import/no-unresolved */ +/* eslint-disable import/extensions */ +const FLAVORED_CLIENT = process.env.PRIME_FLAVORED_CLIENT; -const SHARDS_PER_CLUSTER = parseInt(process.env.SHARDS_PER_CLUSTER) || 1; -const CLUSTER_ID = parseInt(process.env.CLUSTER_ID) || 0; -const TOTAL_SHARDS = parseInt(process.env.TOTAL_SHARDS) || 1; +const SHARDS_PER_CLUSTER = parseInt(process.env.SHARDS_PER_CLUSTER) || 1; +const CLUSTER_ID = parseInt(process.env.CLUSTER_ID) || 0; +const TOTAL_SHARDS = parseInt(process.env.TOTAL_SHARDS) || 1; -const isPRIME = process.env.PRIME === "true" || process.env.PRIME === true; +const isPRIME = process.env.PRIME === "true" || process.env.PRIME === true; process.env.UV_THREADPOOL_SIZE = 256; -global.clusterNames = (require("@polestar/constants/clusters"))?.default; +global.clusterNames = require("@polestar/constants/clusters")?.default; require("./instrumentation.js"); - global.Promise = require("bluebird"); Promise.config({ longStackTraces: true }); +const path = require("path"); -const readdirAsync = Promise.promisify(require("fs").readdir); -const { performance } = require("perf_hooks"); -const path = require("path"); - -const ERIS = require("eris"); -const Eris = require("eris-additions")(ERIS); -const axios = require("axios"); -const DBSchema = require("@polestar/database_schema"); -const cmdPreproc = require("./core/structures/CommandPreprocessor"); -const Gearbox = require("./core/utilities/Gearbox"); -const cfg = require("./config.json"); +const ERIS = require("eris"); +const Eris = require("eris-additions")(ERIS); +const axios = require("axios"); +const DBSchema = require("@polestar/database_schema"); +const cmdPreproc = require("./core/structures/CommandPreprocessor"); +const Gearbox = require("./core/utilities/Gearbox"); +const cfg = require("./config.json"); const WebhookDigester = require("./utils/WebhookDigester.js"); -const TopGG = require("@top-gg/sdk"); - -// STARTUP FLAIR -// process.stdout.write("\x1Bc"); +const TopGG = require("@top-gg/sdk"); console.log(require("./resources/asciiPollux.js").ascii()); -// =========================================== - -// || isPRIME ? "prime" : "main"; -const TopGG_api = new TopGG.Api( cfg.topgg ); +const TopGG_api = new TopGG.Api(cfg.topgg); const DummyFlavorDefault = { token: cfg.token, @@ -47,10 +44,12 @@ const DummyFlavorDefault = { name: "dummy_default", }; -const FLAVOR_SWARM_CONFIG = require("./flavored_swarm.config.js"); -const FLAVORED_CLIENT_DATA = FLAVOR_SWARM_CONFIG.find((cli) => cli.name === FLAVORED_CLIENT) || DummyFlavorDefault; +const FLAVOR_SWARM_CONFIG = require("./flavored_swarm.config.js"); +const { readdirSync } = require("fs"); -// global.Sentry = require("@sentry/node"); +const FLAVORED_CLIENT_DATA = + FLAVOR_SWARM_CONFIG.find((cli) => cli.name === FLAVORED_CLIENT) || + DummyFlavorDefault; // Eris Mods-----// require("./core/structures/ReactionCollector.js")(ERIS); @@ -78,63 +77,6 @@ Eris.Channel.prototype.createMessage = function createMsgModded(...args) { return oldSend(...args); }; -/* NOTE: - FUTURE: In case cleanContent is removed -get cleanContent() { - let cleanContent = this.content && this.content.replace(//g, "$1") || ""; - - let authorName = this.author.username; - if(this.channel.guild) { - const member = this.channel.guild.members.get(this.author.id); - if(member && member.nick) { - authorName = member.nick; - } - } - cleanContent = cleanContent.replace(new RegExp(`<@!?${this.author.id}>`, "g"), "@\u200b" + authorName); - - if(this.mentions) { - this.mentions.forEach((mention) => { - if(this.channel.guild) { - const member = this.channel.guild.members.get(mention.id); - if(member && member.nick) { - cleanContent = cleanContent.replace(new RegExp(`<@!?${mention.id}>`, "g"), "@\u200b" + member.nick); - } - } - cleanContent = cleanContent.replace(new RegExp(`<@!?${mention.id}>`, "g"), "@\u200b" + mention.username); - }); - } - - if(this.channel.guild && this.roleMentions) { - for(const roleID of this.roleMentions) { - const role = this.channel.guild.roles.get(roleID); - const roleName = role ? role.name : "deleted-role"; - cleanContent = cleanContent.replace(new RegExp(`<@&${roleID}>`, "g"), "@\u200b" + roleName); - } - } - - this.channelMentions.forEach((id) => { - const channel = this._client.getChannel(id); - if(channel && channel.name && channel.mention) { - cleanContent = cleanContent.replace(channel.mention, "#" + channel.name); - } - }); - - return cleanContent.replace(/@everyone/g, "@\u200beveryone").replace(/@here/g, "@\u200bhere"); - } -*/ - -//------------------------------------------------------- - -/* -Sentry.init({ - dsn: cfg.sentryDSN, - environment: process.env.NODE_ENV, - serverName: `Polaris-[C${CLUSTER_ID}]`, - autoSessionTracking: false, - -}); -*/ - console.table({ SHARDS_PER_CLUSTER, CLUSTER_ID, @@ -142,47 +84,57 @@ console.table({ FLAVORED_CLIENT_DATA, }); -global.PLX = new Eris.CommandClient(FLAVORED_CLIENT_DATA.token, { - maxShards: TOTAL_SHARDS, - firstShardID: (SHARDS_PER_CLUSTER * CLUSTER_ID), - lastShardID: SHARDS_PER_CLUSTER * (CLUSTER_ID + 1) - 1, - defaultImageSize: 512, - restMode: true, - ratelimiterOffset: 128, - rest: { - baseURL: "/api/v9", - latencyThreshold: 5000, - ratelimiterOffset: 800, - }, - defaultImageFormat: "png", - intents: 1927, - disableEvents: { - TYPING_START: true, - TYPING_STOP: true, - GUILD_MEMBER_SPEAKING: true, +global.PLX = new Eris.CommandClient( + FLAVORED_CLIENT_DATA.token, + { + maxShards: TOTAL_SHARDS, + firstShardID: SHARDS_PER_CLUSTER * CLUSTER_ID, + lastShardID: SHARDS_PER_CLUSTER * (CLUSTER_ID + 1) - 1, + defaultImageSize: 512, + restMode: true, + ratelimiterOffset: 128, + rest: { + baseURL: "/api/v9", + latencyThreshold: 5000, + ratelimiterOffset: 800, + }, + defaultImageFormat: "png", + intents: 1927, + disableEvents: { + TYPING_START: true, + TYPING_STOP: true, + GUILD_MEMBER_SPEAKING: true, + }, }, -}, { - defaultHelpCommand: false, - ignoreBots: true, - requestTimeout: 5000, - defaultCommandOptions: cmdPreproc.DEFAULT_CMD_OPTS, - prefix: ["+", "p!", "plx!", "@mention"], -}); + { + defaultHelpCommand: false, + ignoreBots: true, + requestTimeout: 5000, + defaultCommandOptions: cmdPreproc.DEFAULT_CMD_OPTS, + prefix: ["+", "p!", "plx!", "@mention"], + } +); global.MARKET_TOKEN = cfg["pollux-api-token"]; PLX.engine = Eris; PLX.beta = cfg.beta || process.env.NODE_ENV !== "production"; PLX.maintenance = process.env.maintenance; -PLX.isPRIME = (isPRIME === "true" || isPRIME === true); +PLX.isPRIME = isPRIME === "true" || isPRIME === true; PLX._flavordata = FLAVORED_CLIENT_DATA; -PLX.cluster = isPRIME === true - ? { id: 0, name: `Prime: ${FLAVORED_CLIENT_DATA.fname}` } - : { id: CLUSTER_ID, name: clusterNames[CLUSTER_ID] }; +if (isPRIME === true) { + PLX.cluster = { id: 0, name: `Prime: ${FLAVORED_CLIENT_DATA.fname}` }; +} else { + PLX.cluster = { id: CLUSTER_ID, name: clusterNames[CLUSTER_ID] }; +} -console.report = (...args) => console.log(` ${PLX.cluster.name} `.white.bgBlue + " • ".gray + [...args].join(" ")); +console.report = (...args) => { + console.log( + ` ${PLX.cluster.name} `.white.bgBlue + " • ".gray + [...args].join(" ") + ); +}; const debugHook = new WebhookDigester(PLX); Object.assign(global, Gearbox.Global); @@ -202,19 +154,27 @@ PLX.registerOne = cmdPreproc.registerOne; PLX.blackListedUsers = []; PLX.blackListedServers = []; -PLX.updateBlacklists = (DB) => Promise.all([ - DB.users.find({ blacklisted: { $exists: true } }, { id: 1, _id: 0 }).lean().exec(), - DB.servers.find({ blacklisted: { $exists: true } }, { id: 1, _id: 0 }).lean().exec(), -]).then(([users, servers]) => { - PLX.blacklistedUsers = (users || []).map((usr) => usr.id); - PLX.blacklistedServers = (servers || []).map((svr) => svr.id); -}); +PLX.updateBlacklists = async (DB) => { + const result = await Promise.all([ + DB.users + .find({ blacklisted: { $exists: true } }, { id: 1, _id: 0 }) + .lean() + .exec(), + DB.servers + .find({ blacklisted: { $exists: true } }, { id: 1, _id: 0 }) + .lean() + .exec(), + ]).then(([users, servers]) => { + PLX.blacklistedUsers = (users || []).map((usr) => usr.id); + PLX.blacklistedServers = (servers || []).map((svr) => svr.id); + }); + return result; +}; const dbConnectionData = { hook: debugHook, url: PLX.beta ? cfg.dbURL_beta : cfg.dbURL, options: { - useNewUrlParser: true, keepAlive: true, connectTimeoutMS: 8000, @@ -237,38 +197,35 @@ const vanillaConnection = { }, }; +function postConnect() { + console.log("Discord Client Connected".cyan); + initializeEvents(); + console.log("•".cyan, "Events Listening"); +} + DBSchema(dbConnectionData, { redis: { host: "127.0.0.1", port: 6379, }, -}).then((Connection) => { - global.DB = Connection; - - try { - let ProgMgr = require("@polestar/progression"); - ProgMgr.init(PLX); - console.log("PROGRESSION MANAGER LOADED") - //(require("./core/archetypes/Progression.js")).init(); - // (require("./core/archetypes/Achievements.js")).init(); - } catch (err) { - console.error("PROGRESSION MANAGER LOADED FAILED",err); - //process.exit(1); - } +}) + .then((Connection) => { + global.DB = Connection; + + try { + let ProgMgr = require("@polestar/progression"); + ProgMgr.init(PLX); + console.log("PROGRESSION MANAGER LOADED"); + } catch (err) { + console.error("PROGRESSION MANAGER LOADED FAILED", err); + } - //setTimeout(() => { console.log("Discord connection start..."); - // eslint-disable-next-line @typescript-eslint/no-use-before-define PLX.connect().then(postConnect).catch(console.error); - - - - //}, CLUSTER_ID * 1500); - - -}).catch((err) => { - console.error(err); -}); + }) + .catch((err) => { + console.error(err); + }); DBSchema(vanillaConnection, { redis: null }).then((vConnection) => { global.vDB = vConnection; @@ -285,20 +242,19 @@ translateEngineStart(); // BOT EVENT HANDLER //= ======================================// -// const {msgPreproc} = require('./core/subroutines/onEveryMessage'); let ReadyCount = 0; -PLX.on("ready", () => { +PLX.on("ready", () => { console.log(" READY ".bold.bgYellow, "ReadyCount:", ReadyCount); ReadyCount++; + // eslint-disable-next-line no-undef INSTR.gauge("READY_count", ReadyCount); }); PLX.once("ready", async () => { - - if (PLX._flavordata?.name === "main" ){ + if (PLX._flavordata?.name === "main") { await TopGG_api.postStats({ - serverCount: PLX.guilds.size , - shardCount: TOTAL_SHARDS + serverCount: PLX.guilds.size, + shardCount: TOTAL_SHARDS, }); } @@ -308,76 +264,135 @@ PLX.once("ready", async () => { if (payload.t === "INTERACTION_CREATE") { require("./eventHandlers/interactions")(payload); } - if (PLX.logRaw) console.log(`${" RAW WS ".bgCyan} }`, require("util").inspect(payload, 0, 2, 1)); + if (PLX.logRaw) { + console.log( + `${" RAW WS ".bgCyan} }`, + require("util").inspect(payload, 0, 2, 1) + ); + } }); console.log(" READY ".bold.bgCyan); require("./core/subroutines/cronjobs.js").run(); if (PLX.shard) { PLX.user.setStatus("online"); - console.log(`${"● ".green}Shard${1 + PLX.shard.id}/${PLX.shard.count} [ONLINE]`); + console.log( + `${"● ".green}Shard${1 + PLX.shard.id}/${PLX.shard.count} [ONLINE]` + ); } PLX.registerCommands(); - PLX.updateBlacklists(DB).then(() => { - console.log("• ".blue, "Blacklist Loaded!"); - }).catch(console.error); + PLX.updateBlacklists(DB) + .then(() => { + console.log("• ".blue, "Blacklist Loaded!"); + }) + .catch(console.error); }); - function initializeEvents(){ - PLX.eventHandlerFunctions = {}; - readdirAsync("./eventHandlers/").then((files) => { - files.forEach((file) => { - const eventide = file.split(".")[0]; - PLX.on(eventide, (...args) => { - const eventor = require(`./eventHandlers/${file}`); - PLX.eventHandlerFunctions[eventide] = eventor; - return eventor(...args); - }); - }); - }).catch(console.error); - - ///* - PLX.microserverStart = () => { - //return; +function initializeEvents() { + function handleEvent(eventide, file) { + PLX.on(eventide, async (...args) => { + let result; + try { - PLX.microserver = new (require("./core/archetypes/Microserver"))(cfg.crossAuth); - PLX.microserver.microtasks.updateServerCache("all"); - PLX.microserver.microtasks.updateChannels("all"); - } catch (e) { - console.error(" ERROR Microserver Start ".bgRed); - console.error(e); - console.error("--------------------------------------------"); - - // process.exit(1); + const eventor = require(`./eventHandlers/${file}`); + PLX.eventHandlerFunctions[eventide] = eventor; + result = await eventor(...args); + } catch (err) { + console.error(`Error handling event: ${eventide}`, err); + result = undefined; } - }; - //*/ - //PLX.microserverStart = () => null;/* - //debugHook.info(`**INFO:** Cluster connected and all shards reported online! - // Startup Time: ${(((performance.now() - runtime - (CLUSTER_ID * 20000)) / 1000).toFixed(3))}s`); - //*/ + return result; + }); + } + + PLX.eventHandlerFunctions = {}; - require("./core/utilities/debugTools"); + const files = readdirSync("./eventHandlers/"); + for (const file of files) { + const [eventide, suffix] = file.split("."); + if (suffix !== "js") continue; + + try { + handleEvent(eventide, file); + } catch (err) { + console.error(`Error registering event: ${eventide}`, err); + } } + PLX.microserverStart = () => { + try { + PLX.microserver = new (require("./core/archetypes/Microserver"))( + cfg.crossAuth + ); + PLX.microserver.microtasks.updateServerCache("all"); + PLX.microserver.microtasks.updateChannels("all"); + } catch (e) { + console.error(" ERROR Microserver Start ".bgRed); + console.error(e); + console.error("--------------------------------------------"); + } + }; + + require("./core/utilities/debugTools"); +} + PLX.on("debug", (payload, s) => { if (PLX.logDebug) console.log(`${s} -- ${" D E B U G ".bgGray} }`, payload); }); -PLX.on("hello", (trace, shard) => console.error(`${"[Pollux]".blue} ${shard !== undefined ? `Shard ${shard}` : "Hello!"}:`, trace)); -PLX.on("unknown", (pack, shard) => PLX.logDebug && console.error(`${"[Pollux]".bgRed} SHARD ${shard} :: UNKNOWN PACKET`, pack)); -PLX.on("error", (error, shard) => error && console.error(`${"[Pollux]".red} ${shard !== undefined ? `Shard ${shard} error` : "Error"}:`, error)); -PLX.on("warn", (message, shard) => PLX.logDebug && message && console.error(`${"[Pollux]".yellow} ${shard !== undefined ? `Shard ${shard} warning` : "WARNING"}:`, message)); -PLX.on("disconnect", () => console.error(`${"[Pollux]".yellow} Disconnected from Discord`)); -PLX.on("guildUnavailable", (g) => console.error(`${"[Pollux]".yellow} Unavailable Guild Created`, g)); -PLX.on("unavailableGuildCreate", (g) => console.error(`${"[Pollux]".yellow} Guild unavailable [${g.id}]`)); -PLX.on("shardPreReady", (shard) => console.log("•".cyan, "Shard", (`${shard}`).blue, "getting ready...")); -PLX.on("shardReady", (shard) => console.log("•".green, "Shard", (`${shard}`).magenta, "is Ready -")); -PLX.on("shardResume", (shard) => console.error("•".yellow, "Shard", (`${shard}`).magenta, "resumed Activity -")); +PLX.on("hello", (trace, shard) => + console.error( + `${"[Pollux]".blue} ${shard !== undefined ? `Shard ${shard}` : "Hello!"}:`, + trace + ) +); +PLX.on("unknown", (pack, shard) => { + if (PLX.logDebug) { + console.error(`${"[Pollux]".bgRed} SHARD ${shard} :: UNKNOWN PACKET`, pack); + } +}); +PLX.on("error", (error, shard) => { + if (!error) return; + console.error( + `${"[Pollux]".red} ${ + shard !== undefined ? `Shard ${shard} error` : "Error" + }:`, + error + ); +}); +PLX.on("warn", (message, shard) => { + if (!PLX.logDebug || !message) return; + + console.error( + `${"[Pollux]".yellow} ${ + shard !== undefined ? `Shard ${shard} warning` : "WARNING" + }:`, + message + ); +}); + +PLX.on("disconnect", () => { + console.error(`${"[Pollux]".yellow} Disconnected from Discord`); +}); +PLX.on("guildUnavailable", (g) => { + console.error(`${"[Pollux]".yellow} Unavailable Guild Created`, g); +}); +PLX.on("unavailableGuildCreate", (g) => { + console.error(`${"[Pollux]".yellow} Guild unavailable [${g.id}]`); +}); +PLX.on("shardPreReady", (shard) => { + console.log("•".cyan, "Shard", `${shard}`.blue, "getting ready..."); +}); +PLX.on("shardReady", (shard) => { + console.log("•".green, "Shard", `${shard}`.magenta, "is Ready -"); +}); +PLX.on("shardResume", (shard) => { + console.error("•".yellow, "Shard", `${shard}`.magenta, "resumed Activity -"); +}); PLX.on("shardDisconnect", (err, shard) => { - console.warn("•".red, "Shard", (`${shard}`).blue, "Disconnected -"); + console.warn("•".red, "Shard", `${shard}`.blue, "Disconnected -"); console.error(err, " < Error"); }); @@ -390,15 +405,24 @@ PLX.softKill = (msg) => { PLX.restarting = true; PLX.removeListener("messageCreate", PLX.eventHandlerFunctions.messageCreate); - Promise.all(PLX.execQueue).then(async () => { - if (msg) await msg.reply(`${_emoji("yep")} Queue consumed. Rebooting now...`); - PLX.disconnect({ reconnect: false }); - process.exit(0); - }).timeout(30e3).catch(async (error) => { - if (msg) await msg.reply(`${_emoji("nope")} Queue errored or timed out. Hard-rebooting now...`); - console.error(error); - process.exit(1); - }); + Promise.all(PLX.execQueue) + .then(async () => { + if (msg) { + await msg.reply(`${_emoji("yep")} Queue consumed. Rebooting now...`); + } + PLX.disconnect({ reconnect: false }); + process.exit(0); + }) + .timeout(30e3) + .catch(async (error) => { + if (msg) { + await msg.reply( + `${_emoji("nope")} Queue errored or timed out. Hard-rebooting now...` + ); + } + console.error(error); + process.exit(1); + }); }; PLX.hardKill = () => { console.log("Hard killing".red); @@ -412,17 +436,46 @@ PLX.setAvatar = async (url) => { headers: { Accept: "image/*" }, responseType: "arraybuffer", }); - await PLX.editSelf({ avatar: `data:${response.headers["content-type"]};base64,${response.data.toString("base64")}` }); + await PLX.editSelf({ + avatar: `data:${ + response.headers["content-type"] + };base64,${response.data.toString("base64")}`, + }); } catch (err) { console.error(err); } }; +// eslint-disable-next-line import/extensions require("./core/utilities/SelfAPI.js"); -PLX.bean = (guild, user, delete_message_days = 0, reason = "No reason specified") => axios.put(`https://discord.com/api/guilds/${guild}/bans/${user}`, { delete_message_days, reason }, { headers: { Authorization: PLX._token } }); -PLX.unbean = (guild, user, delete_message_days = 0, reason = "No reason specified") => axios.delete(`https://discord.com/api/guilds/${guild}/bans/${user}`, { delete_message_days, reason }, { headers: { Authorization: PLX._token } }); -PLX.reply = (msg, content, ping = false) => { +PLX.bean = async ( + guild, + user, + delete_message_days = 0, + reason = "No reason specified" +) => { + await axios.put( + `https://discord.com/api/guilds/${guild}/bans/${user}`, + { delete_message_days, reason }, + { headers: { Authorization: PLX._token } } + ); +}; + +PLX.unbean = async ( + guild, + user, + delete_message_days = 0, + reason = "No reason specified" +) => { + await axios.delete( + `https://discord.com/api/guilds/${guild}/bans/${user}`, + { delete_message_days, reason }, + { headers: { Authorization: PLX._token } } + ); +}; + +PLX.reply = async (msg, content, ping = false) => { const payload = { allowed_mentions: { replied_user: ping }, message_reference: { @@ -434,47 +487,42 @@ PLX.reply = (msg, content, ping = false) => { if (typeof content === "string") payload.content = content; else Object.assign(payload, content); - return axios.post(`https://discord.com/api/v8/channels/${msg.channel.id}/messages`, payload, { headers: { Authorization: PLX._token } }); + const result = await axios.post( + `https://discord.com/api/v8/channels/${msg.channel.id}/messages`, + payload, + { headers: { Authorization: PLX._token } } + ); + return result; }; -function postConnect() { - console.log("Discord Client Connected".cyan); - initializeEvents(); - console.log("•".cyan,"Events Listening"); - - - - // POST STATS TO LISTS -} - global.errorsHook = cfg.errorsHook; process.on("uncaughtException", (err) => { - // Sentry.captureException(err); console.error(" UNCAUGHT EXCEPTION ".bgRed); console.error(err); - debugHook.error(` + debugHook.error( + ` **Uncaught Exception** \`\`\`js ${err.slice(0, 1900)} \`\`\` - `, { hook: cfg.errorsHook }); - // if(!PLX.beta) PLX.softKill(); - // else PLX.hardKill(); + `, + { hook: cfg.errorsHook } + ); }); process.on("unhandledRejection", (err) => { - // Sentry.captureException(err); console.error(" UNHANDLED REJECTION ".bgYellow); console.error(err); - debugHook.warn(` + debugHook.warn( + ` **Unhandled Rejection** \`\`\`js ${err?.stack?.slice(0, 1900)} \`\`\` - `, { hook: cfg.errorsHook }); - // if(!PLX.beta) PLX.softKill(); - // else PLX.hardKill(); + `, + { hook: cfg.errorsHook } + ); }); PLX.getOrCreateUser = async (user) => {