91 lines
2.3 KiB
Go
91 lines
2.3 KiB
Go
package services
|
|
|
|
import (
|
|
"code.smartsheep.studio/hydrogen/passport/pkg/database"
|
|
"code.smartsheep.studio/hydrogen/passport/pkg/external"
|
|
"code.smartsheep.studio/hydrogen/passport/pkg/models"
|
|
"context"
|
|
"firebase.google.com/go/messaging"
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
func AddNotifySubscriber(user models.Account, provider, device, ua string) (models.NotificationSubscriber, error) {
|
|
subscriber := models.NotificationSubscriber{
|
|
UserAgent: ua,
|
|
Provider: provider,
|
|
DeviceID: device,
|
|
AccountID: user.ID,
|
|
}
|
|
|
|
err := database.C.Save(&subscriber).Error
|
|
|
|
return subscriber, err
|
|
}
|
|
|
|
func NewNotification(
|
|
user models.ThirdClient,
|
|
target models.Account,
|
|
subject, content string,
|
|
links []models.NotificationLink,
|
|
important bool,
|
|
) error {
|
|
notification := models.Notification{
|
|
Subject: subject,
|
|
Content: content,
|
|
Links: links,
|
|
IsImportant: important,
|
|
ReadAt: nil,
|
|
SenderID: &user.ID,
|
|
RecipientID: target.ID,
|
|
}
|
|
|
|
if err := database.C.Save(¬ification).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
var subscribers []models.NotificationSubscriber
|
|
if err := database.C.Where(&models.NotificationSubscriber{
|
|
AccountID: target.ID,
|
|
}).Find(&subscribers).Error; err != nil {
|
|
// I don't know why cannot get subscribers list, but whatever, the notifications has created
|
|
log.Error().Err(err).Msg("Unexpected error occurred during the notification.")
|
|
return nil
|
|
}
|
|
|
|
for _, subscriber := range subscribers {
|
|
switch subscriber.Provider {
|
|
case models.NotifySubscriberFirebase:
|
|
if external.Fire == nil {
|
|
// Didn't configure for firebase support
|
|
break
|
|
}
|
|
|
|
ctx := context.Background()
|
|
client, err := external.Fire.Messaging(ctx)
|
|
if err != nil {
|
|
log.Warn().Err(err).Msg("An error occurred when getting firebase FCM client...")
|
|
break
|
|
}
|
|
|
|
message := &messaging.Message{
|
|
Notification: &messaging.Notification{
|
|
Title: notification.Subject,
|
|
Body: notification.Content,
|
|
},
|
|
Token: subscriber.DeviceID,
|
|
}
|
|
|
|
if response, err := client.Send(ctx, message); err != nil {
|
|
log.Warn().Err(err).Msg("An error occurred when notify subscriber though firebase FCM...")
|
|
} else {
|
|
log.Debug().
|
|
Str("response", response).
|
|
Int("subscriber", int(subscriber.ID)).
|
|
Msg("Notified to subscriber though firebase FCM.")
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|