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
}