2021-09-07 07:42:46 -07:00
|
|
|
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) {
|
2021-10-05 12:40:59 -07:00
|
|
|
let okMsg = "Sent Successfully.";
|
2021-09-07 07:42:46 -07:00
|
|
|
|
|
|
|
try {
|
|
|
|
const discordDisplayName = notification.discordUsername || "Uptime Kuma";
|
|
|
|
|
|
|
|
// If heartbeatJSON is null, assume we're testing.
|
|
|
|
if (heartbeatJSON == null) {
|
|
|
|
let discordtestdata = {
|
|
|
|
username: discordDisplayName,
|
|
|
|
content: msg,
|
2022-04-13 09:30:32 -07:00
|
|
|
};
|
|
|
|
await axios.post(notification.discordWebhookUrl, discordtestdata);
|
2021-09-07 07:42:46 -07:00
|
|
|
return okMsg;
|
|
|
|
}
|
|
|
|
|
2022-05-13 23:36:40 -07:00
|
|
|
let address;
|
2021-09-07 07:42:46 -07:00
|
|
|
|
2022-04-10 04:45:07 -07:00
|
|
|
switch (monitorJSON["type"]) {
|
|
|
|
case "ping":
|
2022-05-13 23:36:40 -07:00
|
|
|
address = monitorJSON["hostname"];
|
|
|
|
break;
|
2022-04-11 16:52:07 -07:00
|
|
|
case "port":
|
2022-05-13 23:36:40 -07:00
|
|
|
case "dns":
|
2022-04-11 16:52:07 -07:00
|
|
|
case "steam":
|
2022-05-13 23:36:40 -07:00
|
|
|
address = monitorJSON["hostname"];
|
2022-04-10 04:45:07 -07:00
|
|
|
if (monitorJSON["port"]) {
|
2022-05-13 23:36:40 -07:00
|
|
|
address += ":" + monitorJSON["port"];
|
2022-04-10 04:45:07 -07:00
|
|
|
}
|
|
|
|
break;
|
|
|
|
default:
|
2022-05-13 23:36:40 -07:00
|
|
|
address = monitorJSON["url"];
|
2022-04-10 04:45:07 -07:00
|
|
|
break;
|
2021-09-07 07:42:46 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
// If heartbeatJSON is not null, we go into the normal alerting loop.
|
2022-04-17 00:43:03 -07:00
|
|
|
if (heartbeatJSON["status"] === DOWN) {
|
2021-09-07 07:42:46 -07:00
|
|
|
let discorddowndata = {
|
|
|
|
username: discordDisplayName,
|
|
|
|
embeds: [{
|
|
|
|
title: "❌ Your service " + monitorJSON["name"] + " went down. ❌",
|
|
|
|
color: 16711680,
|
|
|
|
timestamp: heartbeatJSON["time"],
|
|
|
|
fields: [
|
|
|
|
{
|
|
|
|
name: "Service Name",
|
|
|
|
value: monitorJSON["name"],
|
|
|
|
},
|
|
|
|
{
|
2022-05-24 14:14:27 -07:00
|
|
|
name: monitorJSON["type"] === "push" ? "Service Type" : "Service URL",
|
|
|
|
value: monitorJSON["type"] === "push" ? "Heartbeat" : address,
|
2021-09-07 07:42:46 -07:00
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "Time (UTC)",
|
|
|
|
value: heartbeatJSON["time"],
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "Error",
|
2022-12-12 08:58:11 -08:00
|
|
|
value: heartbeatJSON["msg"] == null ? "N/A" : heartbeatJSON["msg"],
|
2021-09-07 07:42:46 -07:00
|
|
|
},
|
|
|
|
],
|
|
|
|
}],
|
2022-04-13 09:30:32 -07:00
|
|
|
};
|
2021-09-12 08:54:12 -07:00
|
|
|
|
|
|
|
if (notification.discordPrefixMessage) {
|
|
|
|
discorddowndata.content = notification.discordPrefixMessage;
|
|
|
|
}
|
|
|
|
|
2022-04-13 09:30:32 -07:00
|
|
|
await axios.post(notification.discordWebhookUrl, discorddowndata);
|
2021-09-07 07:42:46 -07:00
|
|
|
return okMsg;
|
|
|
|
|
2022-04-17 00:43:03 -07:00
|
|
|
} else if (heartbeatJSON["status"] === UP) {
|
2021-09-07 07:42:46 -07:00
|
|
|
let discordupdata = {
|
|
|
|
username: discordDisplayName,
|
|
|
|
embeds: [{
|
|
|
|
title: "✅ Your service " + monitorJSON["name"] + " is up! ✅",
|
|
|
|
color: 65280,
|
|
|
|
timestamp: heartbeatJSON["time"],
|
|
|
|
fields: [
|
|
|
|
{
|
|
|
|
name: "Service Name",
|
|
|
|
value: monitorJSON["name"],
|
|
|
|
},
|
|
|
|
{
|
2022-05-24 14:14:27 -07:00
|
|
|
name: monitorJSON["type"] === "push" ? "Service Type" : "Service URL",
|
|
|
|
value: monitorJSON["type"] === "push" ? "Heartbeat" : address.startsWith("http") ? "[Visit Service](" + address + ")" : address,
|
2021-09-07 07:42:46 -07:00
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "Time (UTC)",
|
|
|
|
value: heartbeatJSON["time"],
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "Ping",
|
2022-05-24 14:14:27 -07:00
|
|
|
value: heartbeatJSON["ping"] == null ? "N/A" : heartbeatJSON["ping"] + " ms",
|
2021-09-07 07:42:46 -07:00
|
|
|
},
|
|
|
|
],
|
|
|
|
}],
|
2022-04-13 09:30:32 -07:00
|
|
|
};
|
2021-09-12 08:54:12 -07:00
|
|
|
|
|
|
|
if (notification.discordPrefixMessage) {
|
|
|
|
discordupdata.content = notification.discordPrefixMessage;
|
|
|
|
}
|
|
|
|
|
2022-04-13 09:30:32 -07:00
|
|
|
await axios.post(notification.discordWebhookUrl, discordupdata);
|
2021-09-07 07:42:46 -07:00
|
|
|
return okMsg;
|
|
|
|
}
|
|
|
|
} catch (error) {
|
2022-04-13 09:30:32 -07:00
|
|
|
this.throwGeneralAxiosError(error);
|
2021-09-07 07:42:46 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
module.exports = Discord;
|