Part-DB-server/src/Security/Voter/StructureVoter.php
2020-01-04 20:24:09 +01:00

101 lines
3.7 KiB
PHP

<?php
/**
* This file is part of Part-DB (https://github.com/Part-DB/Part-DB-symfony).
*
* Copyright (C) 2019 Jan Böhmer (https://github.com/jbtronics)
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
*/
namespace App\Security\Voter;
use App\Entity\Attachments\AttachmentType;
use App\Entity\Devices\Device;
use App\Entity\Parts\Category;
use App\Entity\Parts\Footprint;
use App\Entity\Parts\Manufacturer;
use App\Entity\Parts\MeasurementUnit;
use App\Entity\Parts\Storelocation;
use App\Entity\Parts\Supplier;
use App\Entity\PriceInformations\Currency;
use App\Entity\UserSystem\User;
class StructureVoter extends ExtendedVoter
{
/**
* Determines if the attribute and subject are supported by this voter.
*
* @param string $attribute An attribute
* @param mixed $subject The subject to secure, e.g. an object the user wants to access or any other PHP type
*
* @return bool True if the attribute and subject are supported, false otherwise
*/
protected function supports($attribute, $subject)
{
if (\is_object($subject)) {
$permission_name = $this->instanceToPermissionName($subject);
//If permission name is null, then the subject is not supported
return (null !== $permission_name) && $this->resolver->isValidOperation($permission_name, $attribute);
}
}
/**
* Maps a instance type to the permission name.
*
* @param $subject mixed The subject for which the permission name should be generated
*
* @return string|null the name of the permission for the subject's type or null, if the subject is not supported
*/
protected function instanceToPermissionName($subject): ?string
{
$class_name = \get_class($subject);
switch ($class_name) {
case AttachmentType::class:
return 'attachment_types';
case Category::class:
return 'categories';
case Device::class:
return 'devices';
case Footprint::class:
return 'footprints';
case Manufacturer::class:
return 'manufacturers';
case Storelocation::class:
return 'storelocations';
case Supplier::class:
return 'suppliers';
case Currency::class:
return 'currencies';
case MeasurementUnit::class:
return 'measurement_units';
}
//When the class is not supported by this class return null
return null;
}
/**
* Similar to voteOnAttribute, but checking for the anonymous user is already done.
* The current user (or the anonymous user) is passed by $user.
*
* @param $attribute
* @param $subject
*/
protected function voteOnUser($attribute, $subject, User $user): bool
{
$permission_name = $this->instanceToPermissionName($subject);
//Just resolve the permission
return $this->resolver->inherit($user, $permission_name, $attribute) ?? false;
}
}