package repository

import (
	"context"
	"errors"

	"gorm.io/gorm"

	"gitlab.com/mbugroup/lti-api.git/internal/common/repository"
	entity "gitlab.com/mbugroup/lti-api.git/internal/entities"
)

type KandangGroupRepository interface {
	repository.BaseRepository[entity.KandangGroup]
	LocationExists(ctx context.Context, locationId uint) (bool, error)
	PicExists(ctx context.Context, picId uint) (bool, error)
	NameExists(ctx context.Context, name string, excludeID *uint) (bool, error)
	HasDailyChecklistRelation(ctx context.Context, kandangGroupId uint) (bool, error)
}

type KandangGroupRepositoryImpl struct {
	*repository.BaseRepositoryImpl[entity.KandangGroup]
	db *gorm.DB
}

func NewKandangGroupRepository(db *gorm.DB) KandangGroupRepository {
	return &KandangGroupRepositoryImpl{
		BaseRepositoryImpl: repository.NewBaseRepository[entity.KandangGroup](db),
		db:                 db,
	}
}

func (r *KandangGroupRepositoryImpl) LocationExists(ctx context.Context, locationId uint) (bool, error) {
	return repository.Exists[entity.Location](ctx, r.db, locationId)
}

func (r *KandangGroupRepositoryImpl) PicExists(ctx context.Context, picId uint) (bool, error) {
	return repository.Exists[entity.User](ctx, r.db, picId)
}

func (r *KandangGroupRepositoryImpl) NameExists(ctx context.Context, name string, excludeID *uint) (bool, error) {
	return repository.ExistsByName[entity.KandangGroup](ctx, r.db, name, excludeID)
}

func (r *KandangGroupRepositoryImpl) HasDailyChecklistRelation(ctx context.Context, kandangGroupId uint) (bool, error) {
	var marker int
	err := r.db.WithContext(ctx).
		Model(&entity.DailyChecklist{}).
		Select("1").
		Where("kandang_id = ?", kandangGroupId).
		Limit(1).
		Take(&marker).Error
	if errors.Is(err, gorm.ErrRecordNotFound) {
		return false, nil
	}
	if err != nil {
		return false, err
	}
	return true, nil
}