package repository

import (
	"context"

	"gorm.io/gorm"

	"github.com/gochat/gochat/internal/model"
)

// PushTokenRepo implements GORM repository for PushToken.
// Reference: Chatwoot PushSubscription model + P2B M8 spec
type PushTokenRepo struct {
	db *gorm.DB
}

// NewPushTokenRepo creates a new PushToken repository.
func NewPushTokenRepo(db *gorm.DB) *PushTokenRepo {
	return &PushTokenRepo{db: db}
}

// FindByID retrieves a push token by primary key.
func (r *PushTokenRepo) FindByID(ctx context.Context, id uint) (*model.PushToken, error) {
	var t model.PushToken
	err := r.db.WithContext(ctx).First(&t, id).Error
	if err != nil {
		return nil, err
	}
	return &t, nil
}

// ListByUser retrieves all push tokens for a user.
func (r *PushTokenRepo) ListByUser(ctx context.Context, userID uint) ([]model.PushToken, error) {
	var tokens []model.PushToken
	err := r.db.WithContext(ctx).Where("user_id = ?", userID).Find(&tokens).Error
	return tokens, err
}

// Create inserts a new push token.
func (r *PushTokenRepo) Create(ctx context.Context, token *model.PushToken) error {
	return r.db.WithContext(ctx).Create(token).Error
}

// Delete removes a push token by ID.
func (r *PushTokenRepo) Delete(ctx context.Context, id uint) error {
	return r.db.WithContext(ctx).Delete(&model.PushToken{}, id).Error
}

// DeleteByTokenAndUser removes a push token by its token string and user ID.
func (r *PushTokenRepo) DeleteByTokenAndUser(ctx context.Context, token string, userID uint) error {
	return r.db.WithContext(ctx).
		Where("token = ? AND user_id = ?", token, userID).
		Delete(&model.PushToken{}).Error
}