62 lines
2.1 KiB
Go
62 lines
2.1 KiB
Go
package repository
|
|
|
|
// NotificationSettingRepo provides database operations for NotificationSetting.
|
|
// Reference: Chatwoot notification_settings — find_by account_id + user_id (unique)
|
|
|
|
import (
|
|
"github.com/gochat/gochat/internal/model"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// NotificationSettingRepo handles CRUD for notification settings.
|
|
type NotificationSettingRepo struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewNotificationSettingRepo creates a new repo instance.
|
|
func NewNotificationSettingRepo(db *gorm.DB) *NotificationSettingRepo {
|
|
return &NotificationSettingRepo{db: db}
|
|
}
|
|
|
|
// FindByAccountAndUser returns the notification setting for a specific account/user pair.
|
|
// Reference: Chatwoot `@user.notification_settings.find_by(account_id: Current.account.id)`
|
|
func (r *NotificationSettingRepo) FindByAccountAndUser(accountID, userID uint) (*model.NotificationSetting, error) {
|
|
var ns model.NotificationSetting
|
|
err := r.db.Where("account_id = ? AND user_id = ?", accountID, userID).First(&ns).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &ns, nil
|
|
}
|
|
|
|
// Create creates a new notification setting.
|
|
func (r *NotificationSettingRepo) Create(ns *model.NotificationSetting) (*model.NotificationSetting, error) {
|
|
if err := r.db.Create(ns).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return ns, nil
|
|
}
|
|
|
|
// Update updates an existing notification setting.
|
|
func (r *NotificationSettingRepo) Update(ns *model.NotificationSetting) (*model.NotificationSetting, error) {
|
|
if err := r.db.Save(ns).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return ns, nil
|
|
}
|
|
|
|
// UpsertByAccountAndUser creates or updates the notification setting for a user/account pair.
|
|
// Chatwoot always uses find_or_initialize — we do the same via GORM FirstOrCreate + Save.
|
|
func (r *NotificationSettingRepo) UpsertByAccountAndUser(accountID, userID uint, emailFlags, pushFlags int) (*model.NotificationSetting, error) {
|
|
var ns model.NotificationSetting
|
|
err := r.db.Where("account_id = ? AND user_id = ?", accountID, userID).FirstOrCreate(&ns).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ns.EmailFlags = emailFlags
|
|
ns.PushFlags = pushFlags
|
|
if err := r.db.Save(&ns).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &ns, nil
|
|
} |