[slack] implemented slack bot mode, updating of slack messages

This commit is contained in:
Daan Meijer 2024-10-08 15:02:09 +02:00
parent eca90a2b00
commit a95da7dc54
2 changed files with 190 additions and 42 deletions

View file

@ -1,7 +1,8 @@
const NotificationProvider = require("./notification-provider"); const NotificationProvider = require("./notification-provider");
const axios = require("axios"); const axios = require("axios");
const { setSettings, setting } = require("../util-server"); const { getMonitorRelativeURL, UP, DOWN } = require("../../src/util");
const { getMonitorRelativeURL, UP } = require("../../src/util"); const { Settings } = require("../settings");
const { log } = require("../../src/util");
class Slack extends NotificationProvider { class Slack extends NotificationProvider {
name = "slack"; name = "slack";
@ -14,13 +15,11 @@ class Slack extends NotificationProvider {
* @returns {Promise<void>} * @returns {Promise<void>}
*/ */
static async deprecateURL(url) { static async deprecateURL(url) {
let currentPrimaryBaseURL = await setting("primaryBaseURL"); let currentPrimaryBaseURL = await Settings.get("primaryBaseURL");
if (!currentPrimaryBaseURL) { if (!currentPrimaryBaseURL) {
console.log("Move the url to be the primary base URL"); console.log("Move the url to be the primary base URL");
await setSettings("general", { await Settings.set("primaryBaseURL", url, "general");
primaryBaseURL: url,
});
} else { } else {
console.log("Already there, no need to move the primary base URL"); console.log("Already there, no need to move the primary base URL");
} }
@ -115,30 +114,29 @@ class Slack extends NotificationProvider {
} }
/** /**
* @inheritdoc * Builds the message object to send to Slack
* @param {object} heartbeatJSON The heartbeat bean
* @param {object} monitorJSON The monitor bean
* @param {object} notification The notification config
* @param {string} title The message title
* @param {string} msg The textual message
* @returns {Promise<object>} The message object
*/ */
async send(notification, msg, monitorJSON = null, heartbeatJSON = null) { static async buildMessage(heartbeatJSON, monitorJSON, notification, title, msg) {
const okMsg = "Sent Successfully.";
if (notification.slackchannelnotify) { // check if the notification provider is being tested
msg += " <!channel>";
}
try {
if (heartbeatJSON == null) { if (heartbeatJSON == null) {
let data = { return {
"text": msg, "text": msg,
"channel": notification.slackchannel, "channel": notification.slackchannel,
"username": notification.slackusername, "username": notification.slackusername,
"icon_emoji": notification.slackiconemo, "icon_emoji": notification.slackiconemo,
"attachments": [],
}; };
await axios.post(notification.slackwebhookURL, data);
return okMsg;
} }
const baseURL = await setting("primaryBaseURL"); const baseURL = await Settings.get("primaryBaseURL");
const title = "Uptime Kuma Alert";
let data = { let data = {
"channel": notification.slackchannel, "channel": notification.slackchannel,
"username": notification.slackusername, "username": notification.slackusername,
@ -157,11 +155,147 @@ class Slack extends NotificationProvider {
data.text = `${title}\n${msg}`; data.text = `${title}\n${msg}`;
} }
return data;
}
static ENDPOINTS = {
postMessage: "https://slack.com/api/chat.postMessage",
getPermalink: "https://slack.com/api/chat.getPermalink",
update: "https://slack.com/api/chat.update",
};
// Keeps track of open alerts in order to update/close them
static openAlerts = {};
/**
* Delivers the message object to slack, through the chosen method
* @param {object} options The slack configuration
* @param {object} heartbeatJSON The heartbeat bean
* @param {object} message The message object to send to Slack
* @returns {Promise<T|AxiosResponse<any>>} The response from axios
*/
static async deliverMessage(options, heartbeatJSON, message) {
let response = null;
switch (options.mode) {
case "app":
response = Slack.deliverMessageViaAppApi(options, heartbeatJSON, message);
break;
case "webhook":
default:
response = axios.post(options.slackwebhookURL, message);
}
return response;
}
/**
* Track an open alert for a specific monitor
* @param {string} monitorId The monitor id
* @param {object} data The object representing the message
*/
static trackAlert(monitorId, data) {
Slack.openAlerts[monitorId] = Slack.openAlerts[monitorId] || [];
Slack.openAlerts[monitorId].push(data);
log.debug("notification.slack", `Monitor ${monitorId} now has ${Slack.openAlerts[monitorId].length} open alerts`);
}
/**
* Clears the open alerts for a specific monitor
* @param {string} monitorId The monitor id
*/
static clearAlerts(monitorId) {
Slack.openAlerts[monitorId] = [];
}
/**
* Returns the alert(s) for the ongoing incident for a specific monitor
* @param {string} monitorId The monitor id
* @returns {Array<object>} all open alerts
*/
static getAlerts(monitorId) {
return Slack.openAlerts[monitorId] || [];
}
/**
* Delivers the message through the Slack App API
* @param {object} options Slack configuration
* @param {object} heartbeatJSON The heartbeat bean
* @param {object} message The message object to send
* @returns {Promise<object>} The axios response
*/
static async deliverMessageViaAppApi(options, heartbeatJSON, message) {
let response = null;
const token = options.botToken;
const monitorId = heartbeatJSON?.monitorId;
const axiosConfig = {
headers: {
"Authorization": "Bearer " + token,
}
};
const existingAlerts = Slack.getAlerts(monitorId);
if (existingAlerts.length > 0 && heartbeatJSON?.status === UP) {
log.info("slack", `Updating ${existingAlerts.length} message(s)`);
//Update the messages in parallel
const responses = await Promise.all(existingAlerts.map(( { channel, ts } ) => {
message.channel = channel;
message.ts = ts;
return axios.post(Slack.ENDPOINTS.update, message, axiosConfig);
}));
//get the last response
response = responses.pop();
} else {
response = await axios.post(Slack.ENDPOINTS.postMessage, message, axiosConfig);
}
if (response.data.ok) {
if (heartbeatJSON?.status === DOWN) {
Slack.trackAlert(monitorId, response.data);
} else if (heartbeatJSON?.status === UP) {
Slack.clearAlerts(monitorId);
}
}
return response;
}
/**
* @inheritdoc
*/
async send(notification, msg, monitorJSON = null, heartbeatJSON = null) {
const okMsg = "Sent Successfully.";
if (notification.slackchannelnotify) {
msg += " <!channel>";
}
try {
const title = "Uptime Kuma Alert";
const message = await Slack.buildMessage(heartbeatJSON, monitorJSON, notification, title, msg);
//not sure what this does, I think it can be safely removed
if (notification.slackbutton) { if (notification.slackbutton) {
await Slack.deprecateURL(notification.slackbutton); await Slack.deprecateURL(notification.slackbutton);
} }
await axios.post(notification.slackwebhookURL, data); await Slack.deliverMessage(notification, heartbeatJSON, message);
return okMsg; return okMsg;
} catch (error) { } catch (error) {
this.throwGeneralAxiosError(error); this.throwGeneralAxiosError(error);

View file

@ -1,11 +1,25 @@
<template> <template>
<div class="mb-3"> <div class="mb-3">
<label for="slack-mode">API mode</label>
<select id="slack-mode" v-model="$parent.notification.mode" class="form-control">
<option value="webhook">Webhook</option>
<option value="app">App</option>
</select>
<fieldset v-if="$parent.notification.mode == 'webhook'">
<label for="slack-webhook-url" class="form-label">{{ $t("Webhook URL") }}<span style="color: red;"><sup>*</sup></span></label> <label for="slack-webhook-url" class="form-label">{{ $t("Webhook URL") }}<span style="color: red;"><sup>*</sup></span></label>
<input id="slack-webhook-url" v-model="$parent.notification.slackwebhookURL" type="text" class="form-control" required> <input id="slack-webhook-url" v-model="$parent.notification.slackwebhookURL" type="text" class="form-control" required>
<label for="slack-username" class="form-label">{{ $t("Username") }}</label> <label for="slack-username" class="form-label">{{ $t("Username") }}</label>
<input id="slack-username" v-model="$parent.notification.slackusername" type="text" class="form-control"> <input id="slack-username" v-model="$parent.notification.slackusername" type="text" class="form-control">
<label for="slack-iconemo" class="form-label">{{ $t("Icon Emoji") }}</label> <label for="slack-iconemo" class="form-label">{{ $t("Icon Emoji") }}</label>
<input id="slack-iconemo" v-model="$parent.notification.slackiconemo" type="text" class="form-control"> <input id="slack-iconemo" v-model="$parent.notification.slackiconemo" type="text" class="form-control">
</fieldset>
<fieldset v-if="$parent.notification.mode == 'app'">
<label for="slack-token" class="form-label">{{ $t("Token") }}</label>
<input id="slack-token" v-model="$parent.notification.botToken" type="text" class="form-control">
</fieldset>
<label for="slack-channel" class="form-label">{{ $t("Channel Name") }}</label> <label for="slack-channel" class="form-label">{{ $t("Channel Name") }}</label>
<input id="slack-channel-name" v-model="$parent.notification.slackchannel" type="text" class="form-control"> <input id="slack-channel-name" v-model="$parent.notification.slackchannel" type="text" class="form-control">