package repo

import (
	"errors"
	"gitee.com/rio-studio/gin-admin-common/pkg/data/model"
	"gorm.io/gorm"
)

type PermissionRetriever interface {
	FindByUrlAndMethod(url, method string) (*model.Permission, error)
	FindById(id int64) (*model.Permission, error)
	GetPermissionsByIds(ids []int64) ([]*model.Permission, error)
}

type PermissionRetrieverRepo struct {
	db *gorm.DB
}

func NewPermissionRetrieverRepo(db *gorm.DB) PermissionRetriever {
	return &PermissionRetrieverRepo{db: db}
}

func (r *PermissionRetrieverRepo) FindByUrlAndMethod(url, method string) (*model.Permission, error) {
	var permission = &(model.Permission{})
	err := r.db.Model(&model.Permission{}).
		Select("*").
		Where("url = ?", url).
		Where("method = ?", method).
		First(permission).Error

	if permission.ID > 0 {
		return permission, nil
	}

	return nil, err
}

func (r *PermissionRetrieverRepo) FindById(id int64) (*model.Permission, error) {
	var permission = &(model.Permission{})
	err := r.db.Model(&model.Permission{}).First(permission, id).Error
	if errors.Is(err, gorm.ErrRecordNotFound) {
		return nil, err
	}

	if permission.ID > 0 {
		return permission, nil
	}

	return nil, err
}

func (r *PermissionRetrieverRepo) GetPermissionsByIds(ids []int64) ([]*model.Permission, error) {
	var entities []*model.Permission
	err := r.db.Model(&model.Permission{}).Where("id in (?)", ids).Find(&entities).Error
	return entities, err
}