package repository

import (
	"context"
	"github.com/go-nunu/nunu-layout-advanced/internal/model"
)

type GlobalLimitRepository interface {
	GetGlobalLimit(ctx context.Context, id int64) (*model.GlobalLimit, error)
	AddGlobalLimit(ctx context.Context, req *model.GlobalLimit) error
	UpdateGlobalLimitByHostId(ctx context.Context, req *model.GlobalLimit) error
	DeleteGlobalLimitByHostId(ctx context.Context, hostId int64) error
	IsGlobalLimitExistByHostId(ctx context.Context, hostId int64) (bool, error)
	GetGlobalLimitByHostId(ctx context.Context, hostId int64) (*model.GlobalLimit, error)
}

func NewGlobalLimitRepository(
	repository *Repository,
) GlobalLimitRepository {
	return &globalLimitRepository{
		Repository: repository,
	}
}

type globalLimitRepository struct {
	*Repository
}

func (r *globalLimitRepository) GetGlobalLimit(ctx context.Context, id int64) (*model.GlobalLimit, error) {
	var globalLimit model.GlobalLimit

	return &globalLimit, nil
}

func (r *globalLimitRepository) AddGlobalLimit(ctx context.Context, req *model.GlobalLimit) error {
	if err := r.DB(ctx).Create(&req).Error; err != nil {
		return err
	}
	return nil
}

func (r *globalLimitRepository) UpdateGlobalLimitByHostId(ctx context.Context, req *model.GlobalLimit) error {
	if err := r.DB(ctx).Where("host_id = ?", req.HostId).Updates(&req).Error; err != nil {
		return err
	}
	return nil
}

func (r *globalLimitRepository) DeleteGlobalLimitByHostId(ctx context.Context, hostId int64) error {
	if err := r.DB(ctx).Where("host_id = ?", hostId).Delete(&model.GlobalLimit{}).Error; err != nil {
		return err
	}
	return nil
}

func (r *globalLimitRepository) IsGlobalLimitExistByHostId(ctx context.Context, hostId int64) (bool, error) {
	var count int64
	err := r.DB(ctx).Model(&model.GlobalLimit{}).Where("host_id = ?", hostId).Count(&count).Error

	if err != nil {
		return false, err
	}

	return count > 0, nil
}

func (r *globalLimitRepository) GetGlobalLimitByHostId(ctx context.Context, hostId int64) (*model.GlobalLimit, error) {
	var globalLimit model.GlobalLimit
	if err := r.DB(ctx).Where("host_id = ?", hostId).First(&globalLimit).Error; err != nil {
		return nil, err
	}
	return &globalLimit, nil

}