81acbd8eba
fix Issue with Scheduler Support NotificationPlatformWebhook support Discord as notification target
69 lines
2 KiB
Go
69 lines
2 KiB
Go
package notifier
|
|
|
|
import (
|
|
"context"
|
|
|
|
"donetick.com/core/internal/events"
|
|
nModel "donetick.com/core/internal/notifier/model"
|
|
"donetick.com/core/internal/notifier/service/discord"
|
|
pushover "donetick.com/core/internal/notifier/service/pushover"
|
|
telegram "donetick.com/core/internal/notifier/service/telegram"
|
|
|
|
"donetick.com/core/logging"
|
|
)
|
|
|
|
type Notifier struct {
|
|
Telegram *telegram.TelegramNotifier
|
|
Pushover *pushover.Pushover
|
|
discord *discord.DiscordNotifier
|
|
eventsProducer *events.EventsProducer
|
|
}
|
|
|
|
func NewNotifier(t *telegram.TelegramNotifier, p *pushover.Pushover, ep *events.EventsProducer, d *discord.DiscordNotifier) *Notifier {
|
|
return &Notifier{
|
|
Telegram: t,
|
|
Pushover: p,
|
|
eventsProducer: ep,
|
|
discord: d,
|
|
}
|
|
}
|
|
|
|
func (n *Notifier) SendNotification(c context.Context, notification *nModel.NotificationDetails) error {
|
|
log := logging.FromContext(c)
|
|
var err error
|
|
switch notification.TypeID {
|
|
case nModel.NotificationPlatformTelegram:
|
|
if n.Telegram == nil {
|
|
log.Error("Telegram bot is not initialized, Skipping sending message")
|
|
return nil
|
|
}
|
|
err = n.Telegram.SendNotification(c, notification)
|
|
case nModel.NotificationPlatformPushover:
|
|
if n.Pushover == nil {
|
|
log.Error("Pushover is not initialized, Skipping sending message")
|
|
return nil
|
|
}
|
|
err = n.Pushover.SendNotification(c, notification)
|
|
case nModel.NotificationPlatformDiscord:
|
|
if n.discord == nil {
|
|
log.Error("Discord is not initialized, Skipping sending message")
|
|
return nil
|
|
}
|
|
err = n.discord.SendNotification(c, notification)
|
|
|
|
case nModel.NotificationPlatformWebhook:
|
|
// TODO: Implement webhook notification
|
|
// currently we have eventProducer to send events always as a webhook
|
|
// if NotificationPlatform is selected. this a case to catch
|
|
// when we only want to send a webhook
|
|
|
|
default:
|
|
log.Error("Unknown notification type", "type", notification.TypeID)
|
|
return nil
|
|
}
|
|
if err != nil {
|
|
log.Error("Failed to send notification", "err", err)
|
|
}
|
|
|
|
return nil
|
|
}
|