mirror/Checkmate
1
0
Fork 0
mirror of https://github.com/bluewave-labs/Checkmate.git synced 2026-02-05 00:28:30 -06:00
Code Issues Packages Projects Releases Wiki Activity

Configured a config object in the notification schema.

Browse Source
This commit is contained in:
Skorpios
2025-02-09 12:48:22 -08:00
parent 01e73fe5b8
commit d8379f7e88
2 changed files with 36 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
Server/controllers/notificationController.js
View File

@@ -6,15 +6,15 @@ class NotificationController {
}
async triggerNotification(req, res) {
const { monitorId, type, webhookUrl, botToken, chatId } = req.body;
const { monitorId, type, config } = req.body;
if (!monitorId || !type) {
return res.status(400).json({
success: false,
msg: "monitorId and type are required"
});
}
try {
const networkResponse = {
monitor: { _id: monitorId, name: "Test Monitor", url: "http://www.google.com" },
@@ -22,33 +22,37 @@ class NotificationController {
statusChanged: true,
prevStatus: true,
};
if (type === "telegram") {
if (!botToken || !chatId) {
if (!config?.botToken || !config?.chatId) {
return res.status(400).json({
success: false,
msg: "botToken and chatId are required for Telegram notifications"
});
}
await this.notificationService.sendWebhookNotification(networkResponse, null, type, botToken, chatId);
await this.notificationService.sendWebhookNotification(
networkResponse, null, type, config.botToken, config.chatId
);
} else if (type === "discord" || type === "slack") {
if (!webhookUrl) {
if (!config?.webhookUrl) {
return res.status(400).json({
success: false,
msg: `webhookUrl is required for ${type} notifications`
});
}
await this.notificationService.sendWebhookNotification(networkResponse, webhookUrl, type);
await this.notificationService.sendWebhookNotification(
networkResponse, config.webhookUrl, type
);
} else if (type === "email") {
if (!req.body.address) {
if (!config?.address) {
return res.status(400).json({
success: false,
msg: "address is required for email notifications"
});
}
await this.notificationService.sendEmail(networkResponse, req.body.address);
await this.notificationService.sendEmail(networkResponse, config.address);
}
res.json({ success: true, msg: "Notification sent successfully" });
} catch (error) {
logger.error({
@@ -60,6 +64,7 @@ class NotificationController {
res.status(500).json({ success: false, msg: "Failed to send notification" });
}
}
}
export default NotificationController;

30
Server/db/models/Notification.js
View File

@@ -8,17 +8,14 @@ const NotificationSchema = mongoose.Schema(
},
type: {
type: String,
enum: ["email", "sms", "discord", "slack", "telegram"],
enum: ["email", "sms"],
},
webhookUrl: {
type: String,
},
botToken: {
type: String,
},
chatId: {
type: String,
},
config: {
webhookUrl: { type: String }, // For Discord & Slack
botToken: { type: String }, // For Telegram
chatId: { type: String }, // For Telegram
},
address: {
type: String,
},
@@ -59,6 +56,19 @@ const NotificationSchema = mongoose.Schema(
}
);
NotificationSchema.pre("save", function (next) {
if (this.type === "telegram" && (!this.config.botToken || !this.config.chatId)) {
return next(new Error("botToken and chatId are required for Telegram notifications"));
}
if ((this.type === "discord" || this.type === "slack") && !this.config.webhookUrl) {
return next(new Error(`webhookUrl is required for ${this.type} notifications`));
}
if (this.type === "email" && !this.config.address) {
return next(new Error("address is required for email notifications"));
}
next();
});
NotificationSchema.pre("save", function (next) {
if (!this.cpuAlertThreshold || this.isModified("alertThreshold")) {
this.cpuAlertThreshold = this.alertThreshold;