142 lines
3.6 KiB
Go
142 lines
3.6 KiB
Go
package services
|
|
|
|
import (
|
|
"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"
|
|
"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,
|
|
AccountID: user.ID,
|
|
}); err != nil {
|
|
subscriber = models.NotificationSubscriber{
|
|
UserAgent: ua,
|
|
Provider: provider,
|
|
DeviceID: id,
|
|
DeviceToken: tk,
|
|
AccountID: user.ID,
|
|
}
|
|
} else {
|
|
prev = subscriber
|
|
}
|
|
|
|
subscriber.UserAgent = ua
|
|
subscriber.Provider = provider
|
|
subscriber.DeviceToken = tk
|
|
|
|
var err error
|
|
if !reflect.DeepEqual(subscriber, prev) {
|
|
err = database.C.Save(&subscriber).Error
|
|
}
|
|
|
|
return subscriber, err
|
|
}
|
|
|
|
func NewNotification(notification models.Notification) error {
|
|
if err := database.C.Save(¬ification).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
go func() {
|
|
err := PushNotification(notification)
|
|
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] {
|
|
_ = conn.WriteMessage(1, models.UnifiedCommand{
|
|
Action: "notifications.new",
|
|
Payload: notification,
|
|
}.Marshal())
|
|
}
|
|
|
|
// Skip push notify
|
|
if GetStatusDisturbable(notification.RecipientID) != nil {
|
|
return nil
|
|
}
|
|
|
|
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 {
|
|
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).
|
|
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{
|
|
ApnsID: subscriber.DeviceID,
|
|
DeviceToken: subscriber.DeviceToken,
|
|
Topic: viper.GetString("apns_topic"),
|
|
Payload: data,
|
|
}
|
|
|
|
if resp, err := ExtAPNS.Push(payload); err != nil {
|
|
log.Warn().Err(err).Msg("An error occurred when notify subscriber via APNs...")
|
|
} else {
|
|
log.Debug().
|
|
Str("reason", resp.Reason).
|
|
Int("status", resp.StatusCode).
|
|
Int("subscriber", int(subscriber.ID)).
|
|
Msg("Notified subscriber via APNs.")
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|