package repository

import (
	"go-humas-be/app/database"
	"go-humas-be/app/database/entity"
	"go-humas-be/app/module/user_levels/request"
	"go-humas-be/utils/paginator"
)

type userLevelsRepository struct {
	DB *database.Database
}

// UserLevelsRepository define interface of IUserLevelsRepository
type UserLevelsRepository interface {
	GetAll(req request.UserLevelsQueryRequest) (userLevelss []*entity.UserLevels, paging paginator.Pagination, err error)
	FindOne(id uint) (userLevels *entity.UserLevels, err error)
	Create(userLevels *entity.UserLevels) (err error)
	Update(id uint, userLevels *entity.UserLevels) (err error)
	Delete(id uint) (err error)
}

func NewUserLevelsRepository(db *database.Database) UserLevelsRepository {
	return &userLevelsRepository{
		DB: db,
	}
}

// implement interface of IUserLevelsRepository
func (_i *userLevelsRepository) GetAll(req request.UserLevelsQueryRequest) (userLevelss []*entity.UserLevels, paging paginator.Pagination, err error) {
	var count int64

	query := _i.DB.DB.Model(&entity.UserLevels{})
	query.Count(&count)

	req.Pagination.Count = count
	req.Pagination = paginator.Paging(req.Pagination)

	err = query.Offset(req.Pagination.Offset).Limit(req.Pagination.Limit).Find(&userLevelss).Error
	if err != nil {
		return
	}

	paging = *req.Pagination

	return
}

func (_i *userLevelsRepository) FindOne(id uint) (userLevels *entity.UserLevels, err error) {
	if err := _i.DB.DB.First(&userLevels, id).Error; err != nil {
		return nil, err
	}

	return userLevels, nil
}

func (_i *userLevelsRepository) Create(userLevels *entity.UserLevels) (err error) {
	return _i.DB.DB.Create(userLevels).Error
}

func (_i *userLevelsRepository) Update(id uint, userLevels *entity.UserLevels) (err error) {
	return _i.DB.DB.Model(&entity.UserLevels{}).
		Where(&entity.UserLevels{ID: id}).
		Updates(userLevels).Error
}

func (_i *userLevelsRepository) Delete(id uint) error {
	return _i.DB.DB.Delete(&entity.UserLevels{}, id).Error
}