Add Ping Command

2026-09-13 18:59:19 +00:00 · 2022-01-25 19:55:26 +07:00
parent a142ff43f0
commit a3d4d7a923
2 changed files with 129 additions and 4 deletions
@@ -0,0 +1,128 @@
+import { CommandInteraction, Message, Interaction } from "discord.js";
+import { makeSuccessEmbed, makeProcessingEmbed, sendMessageOrInteractionResponse } from "../utils/DiscordMessage";
+import DiscordProvider from "../providers/Discord";
+import NodePing from "ping";
+import fs from "fs";
+import path from "path";
+import os from "os";
+import Logger from "../libs/Logger";
+import Environment from "../providers/Environment";
+
+enum MeasureType {
+    Ping = "ping",
+    DiscordHTTPPing = "discordhttp",
+    DiscordWebsocket = "discordwebsocket"
+}
+
+let measureList = [
+    {
+        type: MeasureType.Ping, 
+        title: "☁️ Internet | ",
+        host: "1.1.1.1"
+    },
+    {
+        type: MeasureType.DiscordWebsocket, 
+        title: "🚀 Discord Websocket | "
+    },
+    {
+        type: MeasureType.DiscordHTTPPing, 
+        title: "🏓 Discord HTTP | "
+    }
+];
+
+if(fs.existsSync(path.join(process.cwd(), 'configs/Ping.json'))) {
+    try {
+        const rawData = fs.readFileSync(path.join(process.cwd(), 'configs/Ping.json'));
+        const jsonData = JSON.parse(rawData.toString());
+        measureList = jsonData;
+    } catch (err) {
+        Logger.error("Unable to load custom Ping config: " + err);
+    }
+}
+
+const EMBEDS = {
+    PING_INFO: (data: Message | Interaction, description: string) => {
+        return makeSuccessEmbed({
+            icon: '🌎',
+            title: `Network performance`,
+            description: description,
+            user: (data instanceof Interaction) ? data.user : data.author
+        });
+    }
+}
+
+export default class Ping {
+
+    async onCommand(command: string, args: any, message: Message) {
+        if(command.toLowerCase() !== 'ping') return;
+        await this.process(message, args);
+    }
+
+    async interactionCreate(interaction: CommandInteraction) { 
+        if(interaction.isCommand()) {
+            if(typeof interaction.commandName === 'undefined') return;
+            if((interaction.commandName).toLowerCase() !== 'ping') return;
+            await this.process(interaction, interaction.options);
+        }
+    }
+
+    async process(data: Interaction | Message, args: any) {
+        const isSlashCommand = data instanceof CommandInteraction && data.isCommand();
+        const isMessage = data instanceof Message;
+
+        if(!isSlashCommand && !isMessage) return;
+
+        let placeholder;
+
+        const loadingEmbed = makeProcessingEmbed({
+            icon: isMessage ? undefined : '⌛',
+            title: `Measuring network performance`,
+            user: (data instanceof Interaction) ? data.user : data.author
+        });
+
+        let beforeEditDate = Date.now();
+        placeholder = await sendMessageOrInteractionResponse(data, { embeds:[loadingEmbed]} );
+        let afterEditDate = Date.now();
+
+
+        let desString = [];
+
+        // TODO: Optimize speed of this
+
+        for(let entry of measureList) {
+
+            let stringCurrent = `${entry.title}`;
+
+            if(entry.type === MeasureType.Ping) {
+                const res = await NodePing.promise.probe(entry.host!, { min_reply: 4 });
+    
+                if(!res.alive) {
+                    stringCurrent += "Failed";
+                    return desString.push(stringCurrent);
+                }
+    
+                if(res.avg === 'unknown' || res.min === 'unknown' || res.max === 'unknown') {
+                    stringCurrent += "Failed";
+                    return desString.push(stringCurrent);
+                }
+    
+                stringCurrent += `${parseFloat(res.avg).toFixed(1)}ms (${parseFloat(res.min).toFixed(1)}ms - ${parseFloat(res.max).toFixed(1)}ms)`;
+                desString.push(stringCurrent);
+            } else if (entry.type === MeasureType.DiscordWebsocket) {
+                stringCurrent += `${Math.round(DiscordProvider.client.ws.ping).toFixed(1)}ms`;
+                desString.push(stringCurrent);
+            } else if (entry.type === MeasureType.DiscordHTTPPing) {
+                stringCurrent += `${Math.abs(afterEditDate - beforeEditDate).toFixed(1)}ms`;
+                desString.push(stringCurrent);
+            } 
+            
+        }
+
+        desString.push("");
+        desString.push("💻 Running on " + `${os.hostname()}${Environment.get().NODE_ENV === "development" ? ' / Development Environment' : ''}`);
+
+        if(isSlashCommand) return await data.editReply({ embeds: [EMBEDS.PING_INFO(data, desString.join('\n'))] });
+        else if(typeof placeholder !== "undefined" && isMessage && placeholder instanceof Message) return await placeholder.edit({ embeds: [EMBEDS.PING_INFO(data, desString.join('\n'))] });
+    }
+
+}