86 lines
2.3 KiB
Go
86 lines
2.3 KiB
Go
|
package models
|
||
|
|
|
||
|
|
import (
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"code.nonshy.com/nonshy/website/pkg/log"
|
||
|
|
)
|
||
|
|
|
||
|
|
// PushNotification table for Web Push subscriptions.
|
||
|
|
type PushNotification struct {
|
||
|
|
ID uint64 `gorm:"primaryKey"`
|
||
|
|
UserID uint64 `gorm:"index"`
|
||
|
|
Subscription string `gorm:"index"`
|
||
|
|
CreatedAt time.Time
|
||
|
|
UpdatedAt time.Time
|
||
|
|
}
|
||
|
|
|
||
|
|
// RegisterPushNotification stores a registration for the user.
|
||
|
|
func RegisterPushNotification(user *User, subscription string) (*PushNotification, error) {
|
||
|
|
// Check for an existing registration.
|
||
|
|
pn, err := GetPushNotificationFromSubscription(user, subscription)
|
||
|
|
if err == nil {
|
||
|
|
return pn, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Create it.
|
||
|
|
pn = &PushNotification{
|
||
|
|
UserID: user.ID,
|
||
|
|
Subscription: subscription,
|
||
|
|
}
|
||
|
|
result := DB.Create(pn)
|
||
|
|
return pn, result.Error
|
||
|
|
}
|
||
|
|
|
||
|
|
// GetPushNotificationFromSubscription checks for an existing subscription.
|
||
|
|
func GetPushNotificationFromSubscription(user *User, subscription string) (*PushNotification, error) {
|
||
|
|
var (
|
||
|
|
pn *PushNotification
|
||
|
|
result = DB.Model(&PushNotification{}).Where(
|
||
|
|
"user_id = ? AND subscription = ?",
|
||
|
|
user.ID, subscription,
|
||
|
|
).First(&pn)
|
||
|
|
)
|
||
|
|
return pn, result.Error
|
||
|
|
}
|
||
|
|
|
||
|
|
// CountPushNotificationSubscriptions returns how many subscriptions the user has for push.
|
||
|
|
func CountPushNotificationSubscriptions(user *User) int64 {
|
||
|
|
var count int64
|
||
|
|
result := DB.Where(
|
||
|
|
"user_id = ?",
|
||
|
|
user.ID,
|
||
|
|
).Model(&PushNotification{}).Count(&count)
|
||
|
|
if result.Error != nil {
|
||
|
|
log.Error("CountPushNotificationSubscriptions(%d): %s", user.ID, result.Error)
|
||
|
|
}
|
||
|
|
return count
|
||
|
|
}
|
||
|
|
|
||
|
|
// GetPushNotificationSubscriptions returns all subscriptions for a user.
|
||
|
|
func GetPushNotificationSubscriptions(user *User) ([]*PushNotification, error) {
|
||
|
|
var (
|
||
|
|
pn = []*PushNotification{}
|
||
|
|
result = DB.Model(&PushNotification{}).Where("user_id = ?", user.ID).Scan(&pn)
|
||
|
|
)
|
||
|
|
return pn, result.Error
|
||
|
|
}
|
||
|
|
|
||
|
|
// DeletePushNotifications scrubs data for deleting a user.
|
||
|
|
func DeletePushNotificationSubscriptions(user *User) error {
|
||
|
|
result := DB.Where(
|
||
|
|
"user_id = ?",
|
||
|
|
user.ID,
|
||
|
|
).Delete(&PushNotification{})
|
||
|
|
return result.Error
|
||
|
|
}
|
||
|
|
|
||
|
|
// DeletePushNotification removes a single subscription from the database.
|
||
|
|
func DeletePushNotification(user *User, subscription string) error {
|
||
|
|
result := DB.Where(
|
||
|
|
"user_id = ? AND subscription = ?",
|
||
|
|
user.ID, subscription,
|
||
|
|
).Delete(&PushNotification{})
|
||
|
|
return result.Error
|
||
|
|
}
|