From d2f24d0be20080050bfd4af7d21c094184403764 Mon Sep 17 00:00:00 2001 From: LittleSheep Date: Tue, 17 Sep 2024 00:35:42 +0800 Subject: [PATCH] :sparkles: Subscriptions on the air --- pkg/internal/services/posts.go | 21 ++++ pkg/internal/services/subscriptions.go | 146 +++++++++++++++++++++++++ 2 files changed, 167 insertions(+) diff --git a/pkg/internal/services/posts.go b/pkg/internal/services/posts.go index 2b83a2f..c2b75ed 100644 --- a/pkg/internal/services/posts.go +++ b/pkg/internal/services/posts.go @@ -343,6 +343,27 @@ func NewPost(user models.Account, item models.Post) (models.Post, error) { } } + // Notify the subscriptions + if content, ok := item.Body["content"].(string); ok { + var title *string + title, _ = item.Body["title"].(*string) + go func() { + if err := NotifyUserSubscription(user, content, title); err != nil { + log.Error().Err(err).Msg("An error occurred when notifying subscriptions user by user...") + } + for _, tag := range item.Tags { + if err := NotifyTagSubscription(tag, user, content, title); err != nil { + log.Error().Err(err).Msg("An error occurred when notifying subscriptions user by tag...") + } + } + for _, category := range item.Categories { + if err := NotifyCategorySubscription(category, user, content, title); err != nil { + log.Error().Err(err).Msg("An error occurred when notifying subscriptions user by category...") + } + } + }() + } + log.Debug().Dur("elapsed", time.Since(start)).Msg("The post is posted.") return item, nil } diff --git a/pkg/internal/services/subscriptions.go b/pkg/internal/services/subscriptions.go index 77974c6..ac14a83 100644 --- a/pkg/internal/services/subscriptions.go +++ b/pkg/internal/services/subscriptions.go @@ -1,10 +1,15 @@ package services import ( + "context" "errors" "fmt" + "time" + "git.solsynth.dev/hydrogen/dealer/pkg/hyper" + "git.solsynth.dev/hydrogen/dealer/pkg/proto" "git.solsynth.dev/hydrogen/interactive/pkg/internal/database" + "git.solsynth.dev/hydrogen/interactive/pkg/internal/gap" "git.solsynth.dev/hydrogen/interactive/pkg/internal/models" "gorm.io/gorm" ) @@ -134,3 +139,144 @@ func UnsubscribeFromCategory(user models.Account, target models.Category) error err := database.C.Delete(&subscription).Error return err } + +func NotifyUserSubscription(poster models.Account, content string, title *string) error { + var subscriptions []models.Subscription + if err := database.C.Where("account_id = ?", poster.ID).Preload("Follower").Find(&subscriptions).Error; err != nil { + return fmt.Errorf("unable to get subscriptions: %v", err) + } + + nTitle := fmt.Sprintf("New post from %s (%s)", poster.Nick, poster.Name) + nSubtitle := "From your subscription" + + var body string + if len(content) > 80 { + body = content[:80] + } else { + body = content + } + if title != nil { + body = fmt.Sprintf("%s\n%s", *title, body) + } + + userIDs := make([]uint64, 0, len(subscriptions)) + for _, subscription := range subscriptions { + userIDs = append(userIDs, uint64(subscription.Follower.ID)) + } + + pc, err := gap.H.GetServiceGrpcConn(hyper.ServiceTypeAuthProvider) + if err != nil { + return err + } + + ctx, cancel := context.WithTimeout(context.Background(), time.Second*5) + defer cancel() + + _, err = proto.NewNotifierClient(pc).NotifyUserBatch(ctx, &proto.NotifyUserBatchRequest{ + UserId: userIDs, + Notify: &proto.NotifyRequest{ + Topic: "interactive.subscription", + Title: nTitle, + Subtitle: &nSubtitle, + Body: body, + IsRealtime: false, + IsForcePush: true, + }, + }) + + return err +} + +func NotifyTagSubscription(poster models.Tag, og models.Account, content string, title *string) error { + var subscriptions []models.Subscription + if err := database.C.Where("tag_id = ?", poster.ID).Preload("Follower").Find(&subscriptions).Error; err != nil { + return fmt.Errorf("unable to get subscriptions: %v", err) + } + + nTitle := fmt.Sprintf("New post in %s by %s (%s)", poster.Name, og.Nick, og.Name) + nSubtitle := "From your subscription" + + var body string + if len(content) > 80 { + body = content[:80] + } else { + body = content + } + if title != nil { + body = fmt.Sprintf("%s\n%s", *title, body) + } + + userIDs := make([]uint64, 0, len(subscriptions)) + for _, subscription := range subscriptions { + userIDs = append(userIDs, uint64(subscription.Follower.ID)) + } + + pc, err := gap.H.GetServiceGrpcConn(hyper.ServiceTypeAuthProvider) + if err != nil { + return err + } + + ctx, cancel := context.WithTimeout(context.Background(), time.Second*5) + defer cancel() + + _, err = proto.NewNotifierClient(pc).NotifyUserBatch(ctx, &proto.NotifyUserBatchRequest{ + UserId: userIDs, + Notify: &proto.NotifyRequest{ + Topic: "interactive.subscription", + Title: nTitle, + Subtitle: &nSubtitle, + Body: body, + IsRealtime: false, + IsForcePush: true, + }, + }) + + return err +} + +func NotifyCategorySubscription(poster models.Category, og models.Account, content string, title *string) error { + var subscriptions []models.Subscription + if err := database.C.Where("category_id = ?", poster.ID).Preload("Follower").Find(&subscriptions).Error; err != nil { + return fmt.Errorf("unable to get subscriptions: %v", err) + } + + nTitle := fmt.Sprintf("New post in %s by %s (%s)", poster.Name, og.Nick, og.Name) + nSubtitle := "From your subscription" + + var body string + if len(content) > 80 { + body = content[:80] + } else { + body = content + } + if title != nil { + body = fmt.Sprintf("%s\n%s", *title, body) + } + + userIDs := make([]uint64, 0, len(subscriptions)) + for _, subscription := range subscriptions { + userIDs = append(userIDs, uint64(subscription.Follower.ID)) + } + + pc, err := gap.H.GetServiceGrpcConn(hyper.ServiceTypeAuthProvider) + if err != nil { + return err + } + + ctx, cancel := context.WithTimeout(context.Background(), time.Second*5) + defer cancel() + + _, err = proto.NewNotifierClient(pc).NotifyUserBatch(ctx, &proto.NotifyUserBatchRequest{ + UserId: userIDs, + Notify: &proto.NotifyRequest{ + Topic: "interactive.subscription", + Title: nTitle, + Subtitle: &nSubtitle, + Body: body, + IsRealtime: false, + IsForcePush: true, + }, + }) + + return err +}