convert notifications into modules

server/notification-providers/discord.js

@@ -0,0 +1,105 @@
+const NotificationProvider = require("./notification-provider");
+const axios = require("axios");
+const { DOWN, UP } = require("../../src/util");
+
+class Discord extends NotificationProvider {
+
+    name = "discord";
+
+    async send(notification, msg, monitorJSON = null, heartbeatJSON = null) {
+        let okMsg = "Sent Successfully. ";
+
+        try {
+            const discordDisplayName = notification.discordUsername || "Uptime Kuma";
+
+            // If heartbeatJSON is null, assume we're testing.
+            if (heartbeatJSON == null) {
+                let discordtestdata = {
+                    username: discordDisplayName,
+                    content: msg,
+                }
+                await axios.post(notification.discordWebhookUrl, discordtestdata)
+                return okMsg;
+            }
+
+            let url;
+
+            if (monitorJSON["type"] === "port") {
+                url = monitorJSON["hostname"];
+                if (monitorJSON["port"]) {
+                    url += ":" + monitorJSON["port"];
+                }
+
+            } else {
+                url = monitorJSON["url"];
+            }
+
+            // If heartbeatJSON is not null, we go into the normal alerting loop.
+            if (heartbeatJSON["status"] == DOWN) {
+                let discorddowndata = {
+                    username: discordDisplayName,
+                    embeds: [{
+                        title: "❌ Your service " + monitorJSON["name"] + " went down. ❌",
+                        color: 16711680,
+                        timestamp: heartbeatJSON["time"],
+                        fields: [
+                            {
+                                name: "Service Name",
+                                value: monitorJSON["name"],
+                            },
+                            {
+                                name: "Service URL",
+                                value: url,
+                            },
+                            {
+                                name: "Time (UTC)",
+                                value: heartbeatJSON["time"],
+                            },
+                            {
+                                name: "Error",
+                                value: heartbeatJSON["msg"],
+                            },
+                        ],
+                    }],
+                }
+                await axios.post(notification.discordWebhookUrl, discorddowndata)
+                return okMsg;
+
+            } else if (heartbeatJSON["status"] == UP) {
+                let discordupdata = {
+                    username: discordDisplayName,
+                    embeds: [{
+                        title: "✅ Your service " + monitorJSON["name"] + " is up! ✅",
+                        color: 65280,
+                        timestamp: heartbeatJSON["time"],
+                        fields: [
+                            {
+                                name: "Service Name",
+                                value: monitorJSON["name"],
+                            },
+                            {
+                                name: "Service URL",
+                                value: url.startsWith("http") ? "[Visit Service](" + url + ")" : url,
+                            },
+                            {
+                                name: "Time (UTC)",
+                                value: heartbeatJSON["time"],
+                            },
+                            {
+                                name: "Ping",
+                                value: heartbeatJSON["ping"] + "ms",
+                            },
+                        ],
+                    }],
+                }
+                await axios.post(notification.discordWebhookUrl, discordupdata)
+                return okMsg;
+            }
+        } catch (error) {
+            this.throwGeneralAxiosError(error)
+        }
+    }
+
+}
+
+module.exports = Discord;