mirror of
https://github.com/louislam/uptime-kuma.git
synced 2024-11-23 23:04:04 +00:00
Merge branch 'master' into hevans/add-heii-on-call-notification-provider
This commit is contained in:
commit
1db750a2e1
5 changed files with 158 additions and 64 deletions
|
@ -8,6 +8,7 @@ const User = require("../server/model/user");
|
||||||
const { io } = require("socket.io-client");
|
const { io } = require("socket.io-client");
|
||||||
const { localWebSocketURL } = require("../server/config");
|
const { localWebSocketURL } = require("../server/config");
|
||||||
const args = require("args-parser")(process.argv);
|
const args = require("args-parser")(process.argv);
|
||||||
|
|
||||||
const rl = readline.createInterface({
|
const rl = readline.createInterface({
|
||||||
input: process.stdin,
|
input: process.stdin,
|
||||||
output: process.stdout
|
output: process.stdout
|
||||||
|
@ -19,10 +20,10 @@ const main = async () => {
|
||||||
}
|
}
|
||||||
|
|
||||||
console.log("Connecting the database");
|
console.log("Connecting the database");
|
||||||
Database.initDataDir(args);
|
|
||||||
await Database.connect(false, false, true);
|
|
||||||
|
|
||||||
try {
|
try {
|
||||||
|
Database.initDataDir(args);
|
||||||
|
await Database.connect(false, false, true);
|
||||||
// No need to actually reset the password for testing, just make sure no connection problem. It is ok for now.
|
// No need to actually reset the password for testing, just make sure no connection problem. It is ok for now.
|
||||||
if (!process.env.TEST_BACKEND) {
|
if (!process.env.TEST_BACKEND) {
|
||||||
const user = await R.findOne("user");
|
const user = await R.findOne("user");
|
||||||
|
|
|
@ -328,9 +328,9 @@ class Monitor extends BeanModel {
|
||||||
/**
|
/**
|
||||||
* Start monitor
|
* Start monitor
|
||||||
* @param {Server} io Socket server instance
|
* @param {Server} io Socket server instance
|
||||||
* @returns {void}
|
* @returns {Promise<void>}
|
||||||
*/
|
*/
|
||||||
start(io) {
|
async start(io) {
|
||||||
let previousBeat = null;
|
let previousBeat = null;
|
||||||
let retries = 0;
|
let retries = 0;
|
||||||
|
|
||||||
|
@ -1102,9 +1102,9 @@ class Monitor extends BeanModel {
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Stop monitor
|
* Stop monitor
|
||||||
* @returns {void}
|
* @returns {Promise<void>}
|
||||||
*/
|
*/
|
||||||
stop() {
|
async stop() {
|
||||||
clearTimeout(this.heartbeatInterval);
|
clearTimeout(this.heartbeatInterval);
|
||||||
this.isStop = true;
|
this.isStop = true;
|
||||||
|
|
||||||
|
|
|
@ -81,7 +81,8 @@ class MqttMonitorType extends MonitorType {
|
||||||
|
|
||||||
let client = mqtt.connect(mqttUrl, {
|
let client = mqtt.connect(mqttUrl, {
|
||||||
username,
|
username,
|
||||||
password
|
password,
|
||||||
|
clientId: "uptime-kuma_" + Math.random().toString(16).substr(2, 8)
|
||||||
});
|
});
|
||||||
|
|
||||||
client.on("connect", () => {
|
client.on("connect", () => {
|
||||||
|
|
|
@ -1,6 +1,7 @@
|
||||||
const NotificationProvider = require("./notification-provider");
|
const NotificationProvider = require("./notification-provider");
|
||||||
const axios = require("axios");
|
const axios = require("axios");
|
||||||
const { DOWN, UP } = require("../../src/util");
|
const { setting } = require("../util-server");
|
||||||
|
const { DOWN, UP, getMonitorRelativeURL } = require("../../src/util");
|
||||||
|
|
||||||
class Teams extends NotificationProvider {
|
class Teams extends NotificationProvider {
|
||||||
name = "teams";
|
name = "teams";
|
||||||
|
@ -9,89 +10,172 @@ class Teams extends NotificationProvider {
|
||||||
* Generate the message to send
|
* Generate the message to send
|
||||||
* @param {const} status The status constant
|
* @param {const} status The status constant
|
||||||
* @param {string} monitorName Name of monitor
|
* @param {string} monitorName Name of monitor
|
||||||
|
* @param {boolean} withStatusSymbol If the status should be prepended as symbol
|
||||||
* @returns {string} Status message
|
* @returns {string} Status message
|
||||||
*/
|
*/
|
||||||
_statusMessageFactory = (status, monitorName) => {
|
_statusMessageFactory = (status, monitorName, withStatusSymbol) => {
|
||||||
if (status === DOWN) {
|
if (status === DOWN) {
|
||||||
return `🔴 Application [${monitorName}] went down`;
|
return (withStatusSymbol ? "🔴 " : "") + `[${monitorName}] went down`;
|
||||||
} else if (status === UP) {
|
} else if (status === UP) {
|
||||||
return `✅ Application [${monitorName}] is back online`;
|
return (withStatusSymbol ? "✅ " : "") + `[${monitorName}] is back online`;
|
||||||
}
|
}
|
||||||
return "Notification";
|
return "Notification";
|
||||||
};
|
};
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Select theme color to use based on status
|
* Select the style to use based on status
|
||||||
* @param {const} status The status constant
|
* @param {const} status The status constant
|
||||||
* @returns {string} Selected color in hex RGB format
|
* @returns {string} Selected style for adaptive cards
|
||||||
*/
|
*/
|
||||||
_getThemeColor = (status) => {
|
_getStyle = (status) => {
|
||||||
if (status === DOWN) {
|
if (status === DOWN) {
|
||||||
return "ff0000";
|
return "attention";
|
||||||
}
|
}
|
||||||
if (status === UP) {
|
if (status === UP) {
|
||||||
return "00e804";
|
return "good";
|
||||||
}
|
}
|
||||||
return "008cff";
|
return "emphasis";
|
||||||
};
|
};
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Generate payload for notification
|
* Generate payload for notification
|
||||||
* @param {object} args Method arguments
|
* @param {object} args Method arguments
|
||||||
* @param {const} args.status The status of the monitor
|
* @param {object} args.heartbeatJSON Heartbeat details
|
||||||
* @param {string} args.monitorMessage Message to send
|
* @param {string} args.monitorName Name of the monitor affected
|
||||||
* @param {string} args.monitorName Name of monitor affected
|
* @param {string} args.monitorUrl URL of the monitor affected
|
||||||
* @param {string} args.monitorUrl URL of monitor affected
|
* @param {string} args.dashboardUrl URL of the dashboard affected
|
||||||
* @returns {object} Notification payload
|
* @returns {object} Notification payload
|
||||||
*/
|
*/
|
||||||
_notificationPayloadFactory = ({
|
_notificationPayloadFactory = ({
|
||||||
status,
|
heartbeatJSON,
|
||||||
monitorMessage,
|
|
||||||
monitorName,
|
monitorName,
|
||||||
monitorUrl,
|
monitorUrl,
|
||||||
|
dashboardUrl,
|
||||||
}) => {
|
}) => {
|
||||||
const notificationMessage = this._statusMessageFactory(
|
const status = heartbeatJSON?.status;
|
||||||
status,
|
|
||||||
monitorName
|
|
||||||
);
|
|
||||||
|
|
||||||
const facts = [];
|
const facts = [];
|
||||||
|
const actions = [];
|
||||||
|
|
||||||
|
if (dashboardUrl) {
|
||||||
|
actions.push({
|
||||||
|
"type": "Action.OpenUrl",
|
||||||
|
"title": "Visit Uptime Kuma",
|
||||||
|
"url": dashboardUrl
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
if (heartbeatJSON?.msg) {
|
||||||
|
facts.push({
|
||||||
|
title: "Description",
|
||||||
|
value: heartbeatJSON.msg,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
if (monitorName) {
|
if (monitorName) {
|
||||||
facts.push({
|
facts.push({
|
||||||
name: "Monitor",
|
title: "Monitor",
|
||||||
value: monitorName,
|
value: monitorName,
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
if (monitorUrl && monitorUrl !== "https://") {
|
if (monitorUrl && monitorUrl !== "https://") {
|
||||||
facts.push({
|
facts.push({
|
||||||
name: "URL",
|
title: "URL",
|
||||||
value: monitorUrl,
|
// format URL as markdown syntax, to be clickable
|
||||||
|
value: `[${monitorUrl}](${monitorUrl})`,
|
||||||
|
});
|
||||||
|
actions.push({
|
||||||
|
"type": "Action.OpenUrl",
|
||||||
|
"title": "Visit Monitor URL",
|
||||||
|
"url": monitorUrl
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
return {
|
if (heartbeatJSON?.localDateTime) {
|
||||||
"@context": "https://schema.org/extensions",
|
facts.push({
|
||||||
"@type": "MessageCard",
|
title: "Time",
|
||||||
themeColor: this._getThemeColor(status),
|
value: heartbeatJSON.localDateTime + (heartbeatJSON.timezone ? ` (${heartbeatJSON.timezone})` : ""),
|
||||||
summary: notificationMessage,
|
});
|
||||||
sections: [
|
}
|
||||||
|
|
||||||
|
const payload = {
|
||||||
|
"type": "message",
|
||||||
|
// message with status prefix as notification text
|
||||||
|
"summary": this._statusMessageFactory(status, monitorName, true),
|
||||||
|
"attachments": [
|
||||||
{
|
{
|
||||||
activityImage:
|
"contentType": "application/vnd.microsoft.card.adaptive",
|
||||||
"https://raw.githubusercontent.com/louislam/uptime-kuma/master/public/icon.png",
|
"contentUrl": "",
|
||||||
activityTitle: "**Uptime Kuma**",
|
"content": {
|
||||||
|
"type": "AdaptiveCard",
|
||||||
|
"body": [
|
||||||
|
{
|
||||||
|
"type": "Container",
|
||||||
|
"verticalContentAlignment": "Center",
|
||||||
|
"items": [
|
||||||
|
{
|
||||||
|
"type": "ColumnSet",
|
||||||
|
"style": this._getStyle(status),
|
||||||
|
"columns": [
|
||||||
|
{
|
||||||
|
"type": "Column",
|
||||||
|
"width": "auto",
|
||||||
|
"verticalContentAlignment": "Center",
|
||||||
|
"items": [
|
||||||
|
{
|
||||||
|
"type": "Image",
|
||||||
|
"width": "32px",
|
||||||
|
"style": "Person",
|
||||||
|
"url": "https://raw.githubusercontent.com/louislam/uptime-kuma/master/public/icon.png",
|
||||||
|
"altText": "Uptime Kuma Logo"
|
||||||
|
}
|
||||||
|
]
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
activityTitle: notificationMessage,
|
"type": "Column",
|
||||||
|
"width": "stretch",
|
||||||
|
"items": [
|
||||||
|
{
|
||||||
|
"type": "TextBlock",
|
||||||
|
"size": "Medium",
|
||||||
|
"weight": "Bolder",
|
||||||
|
"text": `**${this._statusMessageFactory(status, monitorName, false)}**`,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
activityTitle: "**Description**",
|
"type": "TextBlock",
|
||||||
text: monitorMessage,
|
"size": "Small",
|
||||||
facts,
|
"weight": "Default",
|
||||||
|
"text": "Uptime Kuma Alert",
|
||||||
|
"isSubtle": true,
|
||||||
|
"spacing": "None"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
]
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
"type": "FactSet",
|
||||||
|
"separator": false,
|
||||||
|
"facts": facts
|
||||||
|
}
|
||||||
],
|
],
|
||||||
|
"$schema": "http://adaptivecards.io/schemas/adaptive-card.json",
|
||||||
|
"version": "1.5"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
]
|
||||||
};
|
};
|
||||||
|
|
||||||
|
if (actions) {
|
||||||
|
payload.attachments[0].content.body.push({
|
||||||
|
"type": "ActionSet",
|
||||||
|
"actions": actions,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
return payload;
|
||||||
};
|
};
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -112,7 +196,9 @@ class Teams extends NotificationProvider {
|
||||||
*/
|
*/
|
||||||
_handleGeneralNotification = (webhookUrl, msg) => {
|
_handleGeneralNotification = (webhookUrl, msg) => {
|
||||||
const payload = this._notificationPayloadFactory({
|
const payload = this._notificationPayloadFactory({
|
||||||
monitorMessage: msg
|
heartbeatJSON: {
|
||||||
|
msg: msg
|
||||||
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
return this._sendNotification(webhookUrl, payload);
|
return this._sendNotification(webhookUrl, payload);
|
||||||
|
@ -130,26 +216,32 @@ class Teams extends NotificationProvider {
|
||||||
return okMsg;
|
return okMsg;
|
||||||
}
|
}
|
||||||
|
|
||||||
let url;
|
let monitorUrl;
|
||||||
|
|
||||||
switch (monitorJSON["type"]) {
|
switch (monitorJSON["type"]) {
|
||||||
case "http":
|
case "http":
|
||||||
case "keywork":
|
case "keywork":
|
||||||
url = monitorJSON["url"];
|
monitorUrl = monitorJSON["url"];
|
||||||
break;
|
break;
|
||||||
case "docker":
|
case "docker":
|
||||||
url = monitorJSON["docker_host"];
|
monitorUrl = monitorJSON["docker_host"];
|
||||||
break;
|
break;
|
||||||
default:
|
default:
|
||||||
url = monitorJSON["hostname"];
|
monitorUrl = monitorJSON["hostname"];
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const baseURL = await setting("primaryBaseURL");
|
||||||
|
let dashboardUrl;
|
||||||
|
if (baseURL) {
|
||||||
|
dashboardUrl = baseURL + getMonitorRelativeURL(monitorJSON.id);
|
||||||
|
}
|
||||||
|
|
||||||
const payload = this._notificationPayloadFactory({
|
const payload = this._notificationPayloadFactory({
|
||||||
monitorMessage: heartbeatJSON.msg,
|
heartbeatJSON: heartbeatJSON,
|
||||||
monitorName: monitorJSON.name,
|
monitorName: monitorJSON.name,
|
||||||
monitorUrl: url,
|
monitorUrl: monitorUrl,
|
||||||
status: heartbeatJSON.status,
|
dashboardUrl: dashboardUrl,
|
||||||
});
|
});
|
||||||
|
|
||||||
await this._sendNotification(notification.webhookUrl, payload);
|
await this._sendNotification(notification.webhookUrl, payload);
|
||||||
|
|
|
@ -986,7 +986,7 @@ let needSetup = false;
|
||||||
log.info("manage", `Delete Monitor: ${monitorID} User ID: ${socket.userID}`);
|
log.info("manage", `Delete Monitor: ${monitorID} User ID: ${socket.userID}`);
|
||||||
|
|
||||||
if (monitorID in server.monitorList) {
|
if (monitorID in server.monitorList) {
|
||||||
server.monitorList[monitorID].stop();
|
await server.monitorList[monitorID].stop();
|
||||||
delete server.monitorList[monitorID];
|
delete server.monitorList[monitorID];
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1703,11 +1703,11 @@ async function startMonitor(userID, monitorID) {
|
||||||
]);
|
]);
|
||||||
|
|
||||||
if (monitor.id in server.monitorList) {
|
if (monitor.id in server.monitorList) {
|
||||||
server.monitorList[monitor.id].stop();
|
await server.monitorList[monitor.id].stop();
|
||||||
}
|
}
|
||||||
|
|
||||||
server.monitorList[monitor.id] = monitor;
|
server.monitorList[monitor.id] = monitor;
|
||||||
monitor.start(io);
|
await monitor.start(io);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -1737,7 +1737,7 @@ async function pauseMonitor(userID, monitorID) {
|
||||||
]);
|
]);
|
||||||
|
|
||||||
if (monitorID in server.monitorList) {
|
if (monitorID in server.monitorList) {
|
||||||
server.monitorList[monitorID].stop();
|
await server.monitorList[monitorID].stop();
|
||||||
server.monitorList[monitorID].active = 0;
|
server.monitorList[monitorID].active = 0;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -1754,7 +1754,7 @@ async function startMonitors() {
|
||||||
}
|
}
|
||||||
|
|
||||||
for (let monitor of list) {
|
for (let monitor of list) {
|
||||||
monitor.start(io);
|
await monitor.start(io);
|
||||||
// Give some delays, so all monitors won't make request at the same moment when just start the server.
|
// Give some delays, so all monitors won't make request at the same moment when just start the server.
|
||||||
await sleep(getRandomInt(300, 1000));
|
await sleep(getRandomInt(300, 1000));
|
||||||
}
|
}
|
||||||
|
@ -1775,7 +1775,7 @@ async function shutdownFunction(signal) {
|
||||||
log.info("server", "Stopping all monitors");
|
log.info("server", "Stopping all monitors");
|
||||||
for (let id in server.monitorList) {
|
for (let id in server.monitorList) {
|
||||||
let monitor = server.monitorList[id];
|
let monitor = server.monitorList[id];
|
||||||
monitor.stop();
|
await monitor.stop();
|
||||||
}
|
}
|
||||||
await sleep(2000);
|
await sleep(2000);
|
||||||
await Database.close();
|
await Database.close();
|
||||||
|
|
Loading…
Reference in a new issue