From c7b29fa19510daf4e766b689c7bc105742f35119 Mon Sep 17 00:00:00 2001 From: Mo Tarbin Date: Wed, 17 Jul 2024 01:09:47 -0400 Subject: Add CircleGroup notifications to chore notifications --- internal/notifier/service/planner.go | 48 ++++++++++++++++++++++++++++++++++ internal/notifier/telegram/telegram.go | 31 +++++++++++++++++----- 2 files changed, 72 insertions(+), 7 deletions(-) (limited to 'internal/notifier') diff --git a/internal/notifier/service/planner.go b/internal/notifier/service/planner.go index 05d600d..2f7f207 100644 --- a/internal/notifier/service/planner.go +++ b/internal/notifier/service/planner.go @@ -62,6 +62,9 @@ func (n *NotificationPlanner) GenerateNotifications(c context.Context, chore *ch if mt.Nagging { notifications = append(notifications, generateOverdueNotifications(chore, assignees)...) } + if mt.CircleGroup { + notifications = append(notifications, generateCircleGroupNotifications(chore, mt)...) + } n.nRepo.BatchInsertNotifications(notifications) return true @@ -150,3 +153,48 @@ func generateOverdueNotifications(chore *chModel.Chore, users []*cModel.UserCirc return notifications } + +func generateCircleGroupNotifications(chore *chModel.Chore, mt *chModel.NotificationMetadata) []*nModel.Notification { + var notifications []*nModel.Notification + if !mt.CircleGroup || mt.CircleGroupID == nil || *mt.CircleGroupID == 0 { + return notifications + } + if mt.DueDate { + notifications = append(notifications, &nModel.Notification{ + ChoreID: chore.ID, + IsSent: false, + ScheduledFor: *chore.NextDueDate, + CreatedAt: time.Now().UTC(), + TypeID: 1, + TargetID: fmt.Sprint(*mt.CircleGroupID), + Text: fmt.Sprintf("📅 Reminder: *%s* is due today.", chore.Name), + }) + } + if mt.PreDue { + notifications = append(notifications, &nModel.Notification{ + ChoreID: chore.ID, + IsSent: false, + ScheduledFor: *chore.NextDueDate, + CreatedAt: time.Now().UTC().Add(-time.Hour * 3), + TypeID: 3, + TargetID: fmt.Sprint(*mt.CircleGroupID), + Text: fmt.Sprintf("📢 Heads up! *%s* is due soon (on %s).", chore.Name, chore.NextDueDate.Format("January 2nd")), + }) + } + if mt.Nagging { + for _, hours := range []int{24, 48, 72} { + scheduleTime := chore.NextDueDate.Add(time.Hour * time.Duration(hours)) + notifications = append(notifications, &nModel.Notification{ + ChoreID: chore.ID, + IsSent: false, + ScheduledFor: scheduleTime, + CreatedAt: time.Now().UTC(), + TypeID: 2, + TargetID: fmt.Sprint(*mt.CircleGroupID), + Text: fmt.Sprintf("🚨 *%s* is now %d hours overdue. Please complete it as soon as possible.", chore.Name, hours), + }) + } + } + + return notifications +} diff --git a/internal/notifier/telegram/telegram.go b/internal/notifier/telegram/telegram.go index 54c0905..3f064b6 100644 --- a/internal/notifier/telegram/telegram.go +++ b/internal/notifier/telegram/telegram.go @@ -2,6 +2,7 @@ package telegram import ( "context" + "encoding/json" "fmt" "strconv" @@ -49,22 +50,38 @@ func (tn *TelegramNotifier) SendChoreReminder(c context.Context, chore *chModel. } } -func (tn *TelegramNotifier) SendChoreCompletion(c context.Context, chore *chModel.Chore, users []*uModel.User) { +func (tn *TelegramNotifier) SendChoreCompletion(c context.Context, chore *chModel.Chore, user *uModel.User) { log := logging.FromContext(c) - for _, user := range users { - if user.ChatID == 0 { - continue + var mt *chModel.NotificationMetadata + if err := json.Unmarshal([]byte(*chore.NotificationMetadata), &mt); err != nil { + log.Error("Error unmarshalling notification metadata", err) + } + + targets := []int64{} + if user.ChatID != 0 { + targets = append(targets, user.ChatID) + } + if mt.CircleGroup && mt.CircleGroupID != nil { + // attempt to parse it: + + if *mt.CircleGroupID != 0 { + targets = append(targets, *mt.CircleGroupID) } - text := fmt.Sprintf("🎉 *%s* is completed! is off the list, %s! 🌟 ", chore.Name, user.DisplayName) - msg := tgbotapi.NewMessage(user.ChatID, text) + + } + + text := fmt.Sprintf("🎉 *%s* is completed! is off the list, %s! 🌟 ", chore.Name, user.DisplayName) + for _, target := range targets { + msg := tgbotapi.NewMessage(target, text) + msg.ParseMode = "Markdown" _, err := tn.bot.Send(msg) if err != nil { log.Error("Error sending message to user: ", err) log.Debug("Error sending message, chore: ", chore.Name, " user: ", user.DisplayName, " chatID: ", user.ChatID, " user id: ", user.ID) } - } + } func (tn *TelegramNotifier) SendChoreOverdue(c context.Context, chore *chModel.Chore, users []*uModel.User) { -- cgit