uptime-kuma/server/notification.js

227 lines
7.8 KiB
JavaScript
Raw Normal View History

2021-07-27 17:47:13 +00:00
const { R } = require("redbean-node");
const { log } = require("../src/util");
const Alerta = require("./notification-providers/alerta");
const AlertNow = require("./notification-providers/alertnow");
const AliyunSms = require("./notification-providers/aliyun-sms");
2021-09-07 14:42:46 +00:00
const Apprise = require("./notification-providers/apprise");
const Bark = require("./notification-providers/bark");
const ClickSendSMS = require("./notification-providers/clicksendsms");
const DingDing = require("./notification-providers/dingding");
2021-09-07 14:42:46 +00:00
const Discord = require("./notification-providers/discord");
const Feishu = require("./notification-providers/feishu");
const GoogleChat = require("./notification-providers/google-chat");
const Gorush = require("./notification-providers/gorush");
2021-09-07 14:42:46 +00:00
const Gotify = require("./notification-providers/gotify");
const HomeAssistant = require("./notification-providers/home-assistant");
2021-09-07 14:42:46 +00:00
const Line = require("./notification-providers/line");
const LineNotify = require("./notification-providers/linenotify");
2021-09-07 14:42:46 +00:00
const LunaSea = require("./notification-providers/lunasea");
2021-10-05 18:03:56 +00:00
const Matrix = require("./notification-providers/matrix");
const Mattermost = require("./notification-providers/mattermost");
const Ntfy = require("./notification-providers/ntfy");
2021-09-07 14:42:46 +00:00
const Octopush = require("./notification-providers/octopush");
const OneBot = require("./notification-providers/onebot");
const PagerDuty = require("./notification-providers/pagerduty");
const PromoSMS = require("./notification-providers/promosms");
2021-09-07 14:42:46 +00:00
const Pushbullet = require("./notification-providers/pushbullet");
const PushDeer = require("./notification-providers/pushdeer");
2021-09-07 14:42:46 +00:00
const Pushover = require("./notification-providers/pushover");
const Pushy = require("./notification-providers/pushy");
const RocketChat = require("./notification-providers/rocket-chat");
const SerwerSMS = require("./notification-providers/serwersms");
2021-09-07 14:42:46 +00:00
const Signal = require("./notification-providers/signal");
const Slack = require("./notification-providers/slack");
const SMTP = require("./notification-providers/smtp");
const Squadcast = require("./notification-providers/squadcast");
const Stackfield = require("./notification-providers/stackfield");
const Teams = require("./notification-providers/teams");
const TechulusPush = require("./notification-providers/techulus-push");
2021-09-07 14:42:46 +00:00
const Telegram = require("./notification-providers/telegram");
const Webhook = require("./notification-providers/webhook");
2021-12-25 17:33:47 +00:00
const WeCom = require("./notification-providers/wecom");
const GoAlert = require("./notification-providers/goalert");
const SMSManager = require("./notification-providers/smsmanager");
const ServerChan = require("./notification-providers/serverchan");
2021-07-09 06:14:03 +00:00
class Notification {
2021-07-18 10:51:58 +00:00
2021-09-07 14:42:46 +00:00
providerList = {};
/** Initialize the notification providers */
2021-09-07 14:42:46 +00:00
static init() {
log.info("notification", "Prepare Notification Providers");
2021-09-07 14:42:46 +00:00
this.providerList = {};
const list = [
new Alerta(),
new AlertNow(),
2021-10-13 03:55:01 +00:00
new AliyunSms(),
new Apprise(),
new Bark(),
new ClickSendSMS(),
2021-10-13 08:13:46 +00:00
new DingDing(),
2021-09-07 14:42:46 +00:00
new Discord(),
new Feishu(),
new GoogleChat(),
new Gorush(),
2021-09-07 14:42:46 +00:00
new Gotify(),
new HomeAssistant(),
2021-09-07 14:42:46 +00:00
new Line(),
new LineNotify(),
2021-09-07 14:42:46 +00:00
new LunaSea(),
2021-10-05 18:03:56 +00:00
new Matrix(),
new Mattermost(),
new Ntfy(),
2021-09-07 14:42:46 +00:00
new Octopush(),
new OneBot(),
new PagerDuty(),
new PromoSMS(),
2021-09-07 14:42:46 +00:00
new Pushbullet(),
new PushDeer(),
2021-09-07 14:42:46 +00:00
new Pushover(),
new Pushy(),
new RocketChat(),
new ServerChan(),
new SerwerSMS(),
2021-09-07 14:42:46 +00:00
new Signal(),
new SMSManager(),
2021-09-07 14:42:46 +00:00
new Slack(),
new SMTP(),
new Squadcast(),
new Stackfield(),
new Teams(),
new TechulusPush(),
2021-09-07 14:42:46 +00:00
new Telegram(),
new Webhook(),
2021-12-25 17:33:47 +00:00
new WeCom(),
2022-08-23 17:52:54 +00:00
new GoAlert(),
2021-09-07 14:42:46 +00:00
];
for (let item of list) {
if (! item.name) {
throw new Error("Notification provider without name");
}
if (this.providerList[item.name]) {
throw new Error("Duplicate notification provider name");
}
this.providerList[item.name] = item;
}
}
2021-07-18 12:49:46 +00:00
/**
* Send a notification
* @param {BeanModel} notification
* @param {string} msg General Message
* @param {Object} monitorJSON Monitor details (For Up/Down only)
* @param {Object} heartbeatJSON Heartbeat details (For Up/Down only)
2021-07-18 12:49:46 +00:00
* @returns {Promise<string>} Successful msg
* @throws Error with fail msg
2021-07-18 12:49:46 +00:00
*/
2021-07-09 11:33:22 +00:00
static async send(notification, msg, monitorJSON = null, heartbeatJSON = null) {
2021-09-07 14:42:46 +00:00
if (this.providerList[notification.type]) {
return this.providerList[notification.type].send(notification, msg, monitorJSON, heartbeatJSON);
2021-07-09 06:14:03 +00:00
} else {
2021-09-07 14:42:46 +00:00
throw new Error("Notification type is not supported");
2021-07-09 06:14:03 +00:00
}
}
/**
* Save a notification
* @param {Object} notification Notification to save
* @param {?number} notificationID ID of notification to update
* @param {number} userID ID of user who adds notification
* @returns {Promise<Bean>}
*/
2021-07-09 06:14:03 +00:00
static async save(notification, notificationID, userID) {
let bean;
2021-07-09 06:14:03 +00:00
if (notificationID) {
bean = await R.findOne("notification", " id = ? AND user_id = ? ", [
notificationID,
userID,
]);
2021-07-09 06:14:03 +00:00
if (! bean) {
throw new Error("notification not found");
2021-07-09 06:14:03 +00:00
}
} else {
bean = R.dispense("notification");
2021-07-09 06:14:03 +00:00
}
bean.name = notification.name;
bean.user_id = userID;
bean.config = JSON.stringify(notification);
2021-09-09 12:22:32 +00:00
bean.is_default = notification.isDefault || false;
await R.store(bean);
if (notification.applyExisting) {
await applyNotificationEveryMonitor(bean.id, userID);
}
return bean;
2021-07-09 06:14:03 +00:00
}
/**
* Delete a notification
* @param {number} notificationID ID of notification to delete
* @param {number} userID ID of user who created notification
* @returns {Promise<void>}
*/
2021-07-09 06:14:03 +00:00
static async delete(notificationID, userID) {
let bean = await R.findOne("notification", " id = ? AND user_id = ? ", [
notificationID,
userID,
]);
2021-07-09 06:14:03 +00:00
if (! bean) {
throw new Error("notification not found");
2021-07-09 06:14:03 +00:00
}
await R.trash(bean);
2021-07-09 06:14:03 +00:00
}
2021-07-09 17:08:08 +00:00
/**
* Check if apprise exists
* @returns {boolean} Does the command apprise exist?
*/
2021-07-18 10:51:58 +00:00
static checkApprise() {
2021-07-27 17:47:13 +00:00
let commandExistsSync = require("command-exists").sync;
let exists = commandExistsSync("apprise");
2021-07-18 10:51:58 +00:00
return exists;
2021-07-10 03:38:00 +00:00
}
2021-07-18 12:49:46 +00:00
}
2021-07-10 03:38:00 +00:00
/**
* Apply the notification to every monitor
* @param {number} notificationID ID of notification to apply
* @param {number} userID ID of user who created notification
* @returns {Promise<void>}
*/
async function applyNotificationEveryMonitor(notificationID, userID) {
let monitors = await R.getAll("SELECT id FROM monitor WHERE user_id = ?", [
userID
]);
for (let i = 0; i < monitors.length; i++) {
let checkNotification = await R.findOne("monitor_notification", " monitor_id = ? AND notification_id = ? ", [
monitors[i].id,
notificationID,
]);
if (! checkNotification) {
let relation = R.dispense("monitor_notification");
relation.monitor_id = monitors[i].id;
relation.notification_id = notificationID;
await R.store(relation);
}
}
}
2021-07-09 06:14:03 +00:00
module.exports = {
Notification,
};