package repositories

import (
	"GoMembership/internal/database"
	"GoMembership/internal/models"
)

type DriversLicenceInterface interface {
	FindCategoryByName(categoryName string) (models.LicenceCategory, error)
	FindCategoriesByIDs(ids []uint) ([]models.LicenceCategory, error)
	GetAllCategories() ([]models.LicenceCategory, error)
}

type DriversLicenceRepository struct{}

func (r *DriversLicenceRepository) GetAllCategories() ([]models.LicenceCategory, error) {
	var categories []models.LicenceCategory
	err := database.DB.Find(&categories).Error
	return categories, err
}
func (r *DriversLicenceRepository) FindCategoriesByIDs(ids []uint) ([]models.LicenceCategory, error) {
	var categories []models.LicenceCategory
	err := database.DB.Where("id IN ?", ids).Find(&categories).Error
	return categories, err
}

func (r *DriversLicenceRepository) FindCategoryByName(categoryName string) (models.LicenceCategory, error) {
	var category models.LicenceCategory
	err := database.DB.Where("category = ?", categoryName).First(&category).Error
	return category, err
}