uptime-kuma/server/notification.js

139 lines
3.8 KiB
JavaScript
Raw Normal View History

2021-07-09 02:14:03 -04:00
const axios = require("axios");
const {R} = require("redbean-node");
2021-07-09 07:33:22 -04:00
const FormData = require('form-data');
2021-07-09 13:08:08 -04:00
const nodemailer = require("nodemailer");
2021-07-09 23:38:00 -04:00
const Discord = require('discord.js');
2021-07-09 02:14:03 -04:00
class Notification {
2021-07-09 07:33:22 -04:00
static async send(notification, msg, monitorJSON = null, heartbeatJSON = null) {
2021-07-09 02:14:03 -04:00
if (notification.type === "telegram") {
2021-07-09 07:33:22 -04:00
try {
await axios.get(`https://api.telegram.org/bot${notification.telegramBotToken}/sendMessage`, {
params: {
chat_id: notification.telegramChatID,
text: msg,
}
})
return true;
} catch (error) {
console.log(error)
return false;
}
} else if (notification.type === "webhook") {
try {
let data = {
heartbeat: heartbeatJSON,
monitor: monitorJSON,
msg,
};
let finalData;
let config = {};
if (notification.webhookContentType === "form-data") {
finalData = new FormData();
finalData.append('data', JSON.stringify(data));
config = {
headers: finalData.getHeaders()
}
} else {
finalData = data;
2021-07-09 02:14:03 -04:00
}
2021-07-09 07:33:22 -04:00
let res = await axios.post(notification.webhookURL, finalData, config)
return true;
} catch (error) {
console.log(error)
return false;
}
2021-07-09 13:08:08 -04:00
} else if (notification.type === "smtp") {
return await Notification.smtp(notification, msg)
2021-07-09 23:38:00 -04:00
} else if (notification.type === "discord") {
return await Notification.discord(notification, msg)
2021-07-09 02:14:03 -04:00
} else {
throw new Error("Notification type is not supported")
}
}
static async save(notification, notificationID, userID) {
let bean
if (notificationID) {
bean = await R.findOne("notification", " id = ? AND user_id = ? ", [
notificationID,
userID,
])
if (! bean) {
throw new Error("notification not found")
}
} else {
bean = R.dispense("notification")
}
bean.name = notification.name;
bean.user_id = userID;
bean.config = JSON.stringify(notification)
await R.store(bean)
}
static async delete(notificationID, userID) {
let bean = await R.findOne("notification", " id = ? AND user_id = ? ", [
notificationID,
userID,
])
if (! bean) {
throw new Error("notification not found")
}
await R.trash(bean)
}
2021-07-09 13:08:08 -04:00
static async smtp(notification, msg) {
let transporter = nodemailer.createTransport({
host: notification.smtpHost,
port: notification.smtpPort,
secure: notification.smtpSecure,
auth: {
user: notification.smtpUsername,
pass: notification.smtpPassword,
},
});
// send mail with defined transport object
let info = await transporter.sendMail({
from: `"Uptime Kuma" <${notification.smtpFrom}>`,
to: notification.smtpTo,
subject: msg,
text: msg,
});
return true;
}
2021-07-09 23:38:00 -04:00
static async discord(notification, msg) {
const client = new Discord.Client();
await client.login(notification.discordToken)
const channel = await client.channels.fetch(notification.discordChannelID);
await channel.send(msg);
client.destroy()
return true;
}
2021-07-09 02:14:03 -04:00
}
module.exports = {
Notification,
}