uptime-kuma/server/notification-providers/discord.js

124 lines
4.7 KiB
JavaScript
Raw Normal View History

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;
}
let address;
2021-09-07 07:42:46 -07:00
switch (monitorJSON["type"]) {
case "ping":
address = monitorJSON["hostname"];
break;
case "port":
case "dns":
case "gamedig":
case "steam":
address = monitorJSON["hostname"];
if (monitorJSON["port"]) {
address += ":" + monitorJSON["port"];
}
break;
default:
address = monitorJSON["url"];
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"],
},
{
name: monitorJSON["type"] === "push" ? "Service Type" : "Service URL",
value: monitorJSON["type"] === "push" ? "Heartbeat" : address,
2021-09-07 07:42:46 -07:00
},
{
name: `Time (${heartbeatJSON["timezone"]})`,
value: heartbeatJSON["localDateTime"],
2021-09-07 07:42:46 -07:00
},
{
name: "Error",
value: heartbeatJSON["msg"] == null ? "N/A" : heartbeatJSON["msg"],
2021-09-07 07:42:46 -07:00
},
],
}],
2022-04-13 09:30:32 -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"],
},
{
name: monitorJSON["type"] === "push" ? "Service Type" : "Service URL",
value: monitorJSON["type"] === "push" ? "Heartbeat" : address,
2021-09-07 07:42:46 -07:00
},
{
name: `Time (${heartbeatJSON["timezone"]})`,
value: heartbeatJSON["localDateTime"],
2021-09-07 07:42:46 -07:00
},
{
name: "Ping",
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
};
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;