package repository

import (
	"gofaster/internal/model"

	"gorm.io/gorm"
)

type RoleRepo struct {
	BaseRepo
}

func NewRoleRepo(db *gorm.DB) *RoleRepo {
	return &RoleRepo{BaseRepo{db: db}}
}

func (r *RoleRepo) Create(role *model.Role) error {
	return r.db.Create(role).Error
}

func (r *RoleRepo) GetByID(id uint) (*model.Role, error) {
	var role model.Role
	err := r.db.Preload("Permissions").First(&role, id).Error
	return &role, err
}

func (r *RoleRepo) GetByName(name string) (*model.Role, error) {
	var role model.Role
	err := r.db.Preload("Permissions").Where("name = ?", name).First(&role).Error
	return &role, err
}

func (r *RoleRepo) Update(role *model.Role) error {
	return r.db.Save(role).Error
}

func (r *RoleRepo) Delete(id uint) error {
	return r.db.Delete(&model.Role{}, id).Error
}

func (r *RoleRepo) List(page, pageSize int) ([]model.Role, int64, error) {
	var roles []model.Role
	var count int64

	err := r.db.Model(&model.Role{}).Count(&count).Error
	if err != nil {
		return nil, 0, err
	}

	offset := (page - 1) * pageSize
	err = r.db.Preload("Permissions").Offset(offset).Limit(pageSize).Find(&roles).Error
	return roles, count, err
}

func (r *RoleRepo) AssignPermissions(roleID uint, permissionIDs []uint) error {
	var permissions []model.Permission
	if err := r.db.Find(&permissions, permissionIDs).Error; err != nil {
		return err
	}
	return r.db.Model(&model.Role{BaseModel: model.BaseModel{ID: roleID}}).Association("Permissions").Replace(permissions)
}