package repository

import (
	"context"

	"apskel-pos-be/internal/entities"

	"github.com/google/uuid"
	"gorm.io/gorm"
)

type NotificationDeliveryRepository interface {
	Create(ctx context.Context, delivery *entities.NotificationDelivery) error
	BulkCreate(ctx context.Context, deliveries []*entities.NotificationDelivery) error
	GetByID(ctx context.Context, id uuid.UUID) (*entities.NotificationDelivery, error)
	Update(ctx context.Context, delivery *entities.NotificationDelivery) error
	ListByReceiverID(ctx context.Context, receiverID uuid.UUID) ([]*entities.NotificationDelivery, error)
}

type NotificationDeliveryRepositoryImpl struct {
	db *gorm.DB
}

func NewNotificationDeliveryRepository(db *gorm.DB) *NotificationDeliveryRepositoryImpl {
	return &NotificationDeliveryRepositoryImpl{db: db}
}

func (r *NotificationDeliveryRepositoryImpl) Create(ctx context.Context, delivery *entities.NotificationDelivery) error {
	return r.db.WithContext(ctx).Create(delivery).Error
}

func (r *NotificationDeliveryRepositoryImpl) BulkCreate(ctx context.Context, deliveries []*entities.NotificationDelivery) error {
	if len(deliveries) == 0 {
		return nil
	}
	return r.db.WithContext(ctx).Create(&deliveries).Error
}

func (r *NotificationDeliveryRepositoryImpl) GetByID(ctx context.Context, id uuid.UUID) (*entities.NotificationDelivery, error) {
	var delivery entities.NotificationDelivery
	err := r.db.WithContext(ctx).First(&delivery, "id = ?", id).Error
	if err != nil {
		return nil, err
	}
	return &delivery, nil
}

func (r *NotificationDeliveryRepositoryImpl) Update(ctx context.Context, delivery *entities.NotificationDelivery) error {
	return r.db.WithContext(ctx).Save(delivery).Error
}

func (r *NotificationDeliveryRepositoryImpl) ListByReceiverID(ctx context.Context, receiverID uuid.UUID) ([]*entities.NotificationDelivery, error) {
	var deliveries []*entities.NotificationDelivery
	err := r.db.WithContext(ctx).
		Where("notification_receiver_id = ?", receiverID).
		Order("created_at DESC").
		Find(&deliveries).Error
	return deliveries, err
}