package infrastructure

import (
	"context"

	"mengstack/internal/modules/notification/domain"

	"gorm.io/gorm"
)

type notificationRepo struct {
	db *gorm.DB
}

func NewNotificationRepository(db *gorm.DB) domain.NotificationRepository {
	return &notificationRepo{db: db}
}

func (r *notificationRepo) Create(ctx context.Context, n *domain.Notification) error {
	return r.db.WithContext(ctx).Create(n).Error
}

func (r *notificationRepo) FindByID(ctx context.Context, tenantID string, id uint) (*domain.Notification, error) {
	var n domain.Notification
	err := r.db.WithContext(ctx).Where("id = ? AND tenant_id = ?", id, tenantID).First(&n).Error
	if err != nil {
		return nil, err
	}
	return &n, nil
}

func (r *notificationRepo) FindByUser(ctx context.Context, tenantID string, userID uint, page, pageSize int) ([]domain.Notification, int64, error) {
	var notifications []domain.Notification
	var total int64

	query := r.db.WithContext(ctx).Where("tenant_id = ? AND user_id = ?", tenantID, userID)
	query.Model(&domain.Notification{}).Count(&total)

	offset := (page - 1) * pageSize
	err := query.Order("created_at DESC").Offset(offset).Limit(pageSize).Find(&notifications).Error
	return notifications, total, err
}

func (r *notificationRepo) MarkAsRead(ctx context.Context, tenantID string, id uint) error {
	return r.db.WithContext(ctx).
		Where("id = ? AND tenant_id = ?", id, tenantID).
		Model(&domain.Notification{}).
		Update("is_read", true).Error
}

func (r *notificationRepo) MarkAllAsRead(ctx context.Context, tenantID string, userID uint) error {
	return r.db.WithContext(ctx).
		Where("tenant_id = ? AND user_id = ? AND is_read = ?", tenantID, userID, false).
		Model(&domain.Notification{}).
		Update("is_read", true).Error
}

func (r *notificationRepo) CountUnread(ctx context.Context, tenantID string, userID uint) (int64, error) {
	var count int64
	err := r.db.WithContext(ctx).
		Where("tenant_id = ? AND user_id = ? AND is_read = ?", tenantID, userID, false).
		Model(&domain.Notification{}).
		Count(&count).Error
	return count, err
}

func (r *notificationRepo) Delete(ctx context.Context, tenantID string, id uint) error {
	return r.db.WithContext(ctx).
		Where("id = ? AND tenant_id = ?", id, tenantID).
		Delete(&domain.Notification{}).Error
}