Passport/pkg/internal/services/notifications.go

142 lines
3.6 KiB
Go
Raw Normal View History

package services
import (
2024-06-06 14:48:43 +00:00
"context"
"firebase.google.com/go/messaging"
"git.solsynth.dev/hydrogen/passport/pkg/internal/database"
"git.solsynth.dev/hydrogen/passport/pkg/internal/models"
"github.com/rs/zerolog/log"
"github.com/sideshow/apns2"
payload2 "github.com/sideshow/apns2/payload"
2024-06-07 12:50:27 +00:00
"github.com/spf13/viper"
"reflect"
)
func AddNotifySubscriber(user models.Account, provider, id, tk, ua string) (models.NotificationSubscriber, error) {
var prev models.NotificationSubscriber
var subscriber models.NotificationSubscriber
if err := database.C.Where(&models.NotificationSubscriber{
DeviceID: id,
2024-02-07 15:15:16 +00:00
AccountID: user.ID,
}); err != nil {
subscriber = models.NotificationSubscriber{
UserAgent: ua,
Provider: provider,
DeviceID: id,
DeviceToken: tk,
AccountID: user.ID,
}
} else {
prev = subscriber
2024-02-07 15:15:16 +00:00
}
subscriber.UserAgent = ua
subscriber.Provider = provider
subscriber.DeviceToken = tk
var err error
if !reflect.DeepEqual(subscriber, prev) {
err = database.C.Save(&subscriber).Error
}
2024-02-07 15:15:16 +00:00
return subscriber, err
}
func NewNotification(notification models.Notification) error {
if err := database.C.Save(&notification).Error; err != nil {
return err
}
go func() {
err := PushNotification(notification)
2024-03-31 08:03:59 +00:00
if err != nil {
log.Error().Err(err).Msg("Unexpected error occurred during the notification.")
}
}()
return nil
}
func PushNotification(notification models.Notification) error {
for conn := range wsConn[notification.RecipientID] {
2024-05-09 15:35:13 +00:00
_ = conn.WriteMessage(1, models.UnifiedCommand{
Action: "notifications.new",
2024-05-13 14:31:19 +00:00
Payload: notification,
2024-05-09 15:35:13 +00:00
}.Marshal())
2024-03-31 08:03:59 +00:00
}
2024-06-26 12:05:28 +00:00
// Skip push notify
if GetStatusDisturbable(notification.RecipientID) != nil {
return nil
}
2024-06-06 14:48:43 +00:00
var subscribers []models.NotificationSubscriber
if err := database.C.Where(&models.NotificationSubscriber{
AccountID: notification.RecipientID,
}).Find(&subscribers).Error; err != nil {
return err
}
for _, subscriber := range subscribers {
switch subscriber.Provider {
2024-06-06 14:48:43 +00:00
case models.NotifySubscriberFirebase:
if ExtFire != nil {
ctx := context.Background()
client, err := ExtFire.Messaging(ctx)
if err != nil {
log.Warn().Err(err).Msg("An error occurred when creating FCM client...")
break
}
message := &messaging.Message{
Notification: &messaging.Notification{
Title: notification.Subject,
Body: notification.Content,
},
Token: subscriber.DeviceToken,
}
if response, err := client.Send(ctx, message); err != nil {
log.Warn().Err(err).Msg("An error occurred when notify subscriber via FCM...")
} else {
log.Debug().
Str("response", response).
Int("subscriber", int(subscriber.ID)).
Msg("Notified subscriber via FCM.")
}
}
case models.NotifySubscriberAPNs:
if ExtAPNS != nil {
data, err := payload2.
NewPayload().
AlertTitle(notification.Subject).
AlertBody(notification.Content).
2024-06-08 10:18:28 +00:00
Sound("default").
Category(notification.Type).
MarshalJSON()
if err != nil {
log.Warn().Err(err).Msg("An error occurred when preparing to notify subscriber via APNs...")
}
payload := &apns2.Notification{
2024-06-07 12:50:27 +00:00
ApnsID: subscriber.DeviceID,
DeviceToken: subscriber.DeviceToken,
2024-06-07 12:50:27 +00:00
Topic: viper.GetString("apns_topic"),
Payload: data,
}
2024-06-07 12:24:32 +00:00
if resp, err := ExtAPNS.Push(payload); err != nil {
log.Warn().Err(err).Msg("An error occurred when notify subscriber via APNs...")
2024-06-07 12:24:32 +00:00
} else {
log.Debug().
Str("reason", resp.Reason).
Int("status", resp.StatusCode).
Int("subscriber", int(subscriber.ID)).
Msg("Notified subscriber via APNs.")
}
}
}
}
return nil
}