const axios = require("axios"); const {R} = require("redbean-node"); const FormData = require('form-data'); const nodemailer = require("nodemailer"); const Discord = require('discord.js'); class Notification { static async send(notification, msg, monitorJSON = null, heartbeatJSON = null) { if (notification.type === "telegram") { 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; } let res = await axios.post(notification.webhookURL, finalData, config) return true; } catch (error) { console.log(error) return false; } } else if (notification.type === "smtp") { return await Notification.smtp(notification, msg) } else if (notification.type === "discord") { return await Notification.discord(notification, msg) } 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) } 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; } 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; } } module.exports = { Notification, }