153 lines
		
	
	
		
			4.6 KiB
		
	
	
	
		
			PHP
		
	
	
	
			
		
		
	
	
			153 lines
		
	
	
		
			4.6 KiB
		
	
	
	
		
			PHP
		
	
	
	
<?php namespace BookStack\Repos;
 | 
						|
 | 
						|
 | 
						|
use BookStack\Exceptions\PermissionsException;
 | 
						|
use BookStack\RolePermission;
 | 
						|
use BookStack\Role;
 | 
						|
use BookStack\Services\PermissionService;
 | 
						|
use Setting;
 | 
						|
 | 
						|
class PermissionsRepo
 | 
						|
{
 | 
						|
 | 
						|
    protected $permission;
 | 
						|
    protected $role;
 | 
						|
    protected $permissionService;
 | 
						|
 | 
						|
    protected $systemRoles = ['admin', 'public'];
 | 
						|
 | 
						|
    /**
 | 
						|
     * PermissionsRepo constructor.
 | 
						|
     * @param RolePermission $permission
 | 
						|
     * @param Role $role
 | 
						|
     * @param PermissionService $permissionService
 | 
						|
     */
 | 
						|
    public function __construct(RolePermission $permission, Role $role, PermissionService $permissionService)
 | 
						|
    {
 | 
						|
        $this->permission = $permission;
 | 
						|
        $this->role = $role;
 | 
						|
        $this->permissionService = $permissionService;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Get all the user roles from the system.
 | 
						|
     * @return \Illuminate\Database\Eloquent\Collection|static[]
 | 
						|
     */
 | 
						|
    public function getAllRoles()
 | 
						|
    {
 | 
						|
        return $this->role->all();
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Get all the roles except for the provided one.
 | 
						|
     * @param Role $role
 | 
						|
     * @return mixed
 | 
						|
     */
 | 
						|
    public function getAllRolesExcept(Role $role)
 | 
						|
    {
 | 
						|
        return $this->role->where('id', '!=', $role->id)->get();
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Get a role via its ID.
 | 
						|
     * @param $id
 | 
						|
     * @return mixed
 | 
						|
     */
 | 
						|
    public function getRoleById($id)
 | 
						|
    {
 | 
						|
        return $this->role->findOrFail($id);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Save a new role into the system.
 | 
						|
     * @param array $roleData
 | 
						|
     * @return Role
 | 
						|
     */
 | 
						|
    public function saveNewRole($roleData)
 | 
						|
    {
 | 
						|
        $role = $this->role->newInstance($roleData);
 | 
						|
        $role->name = str_replace(' ', '-', strtolower($roleData['display_name']));
 | 
						|
        // Prevent duplicate names
 | 
						|
        while ($this->role->where('name', '=', $role->name)->count() > 0) {
 | 
						|
            $role->name .= strtolower(str_random(2));
 | 
						|
        }
 | 
						|
        $role->save();
 | 
						|
 | 
						|
        $permissions = isset($roleData['permissions']) ? array_keys($roleData['permissions']) : [];
 | 
						|
        $this->assignRolePermissions($role, $permissions);
 | 
						|
        $this->permissionService->buildJointPermissionForRole($role);
 | 
						|
        return $role;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Updates an existing role.
 | 
						|
     * Ensure Admin role always has all permissions.
 | 
						|
     * @param $roleId
 | 
						|
     * @param $roleData
 | 
						|
     * @throws PermissionsException
 | 
						|
     */
 | 
						|
    public function updateRole($roleId, $roleData)
 | 
						|
    {
 | 
						|
        $role = $this->role->findOrFail($roleId);
 | 
						|
 | 
						|
        $permissions = isset($roleData['permissions']) ? array_keys($roleData['permissions']) : [];
 | 
						|
        $this->assignRolePermissions($role, $permissions);
 | 
						|
 | 
						|
        if ($role->name === 'admin') {
 | 
						|
            $permissions = $this->permission->all()->pluck('id')->toArray();
 | 
						|
            $role->permissions()->sync($permissions);
 | 
						|
        }
 | 
						|
 | 
						|
        $role->fill($roleData);
 | 
						|
        $role->save();
 | 
						|
        $this->permissionService->buildJointPermissionForRole($role);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Assign an list of permission names to an role.
 | 
						|
     * @param Role $role
 | 
						|
     * @param array $permissionNameArray
 | 
						|
     */
 | 
						|
    public function assignRolePermissions(Role $role, $permissionNameArray = [])
 | 
						|
    {
 | 
						|
        $permissions = [];
 | 
						|
        $permissionNameArray = array_values($permissionNameArray);
 | 
						|
        if ($permissionNameArray && count($permissionNameArray) > 0) {
 | 
						|
            $permissions = $this->permission->whereIn('name', $permissionNameArray)->pluck('id')->toArray();
 | 
						|
        }
 | 
						|
        $role->permissions()->sync($permissions);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Delete a role from the system.
 | 
						|
     * Check it's not an admin role or set as default before deleting.
 | 
						|
     * If an migration Role ID is specified the users assign to the current role
 | 
						|
     * will be added to the role of the specified id.
 | 
						|
     * @param $roleId
 | 
						|
     * @param $migrateRoleId
 | 
						|
     * @throws PermissionsException
 | 
						|
     */
 | 
						|
    public function deleteRole($roleId, $migrateRoleId)
 | 
						|
    {
 | 
						|
        $role = $this->role->findOrFail($roleId);
 | 
						|
 | 
						|
        // Prevent deleting admin role or default registration role.
 | 
						|
        if ($role->system_name && in_array($role->system_name, $this->systemRoles)) {
 | 
						|
            throw new PermissionsException(trans('errors.role_system_cannot_be_deleted'));
 | 
						|
        } else if ($role->id == setting('registration-role')) {
 | 
						|
            throw new PermissionsException(trans('errors.role_registration_default_cannot_delete'));
 | 
						|
        }
 | 
						|
 | 
						|
        if ($migrateRoleId) {
 | 
						|
            $newRole = $this->role->find($migrateRoleId);
 | 
						|
            if ($newRole) {
 | 
						|
                $users = $role->users->pluck('id')->toArray();
 | 
						|
                $newRole->users()->sync($users);
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        $this->permissionService->deleteJointPermissionsForRole($role);
 | 
						|
        $role->delete();
 | 
						|
    }
 | 
						|
 | 
						|
} |