Part-DB-server/src/Security/Voter/StructureVoter.php

<?php
/**
 * This file is part of Part-DB (https://github.com/Part-DB/Part-DB-symfony).
 *
 * Copyright (C) 2019 - 2022 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 Affero General Public License as published
 * by the Free Software Foundation, either version 3 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

declare(strict_types=1);

namespace App\Security\Voter;

use App\Entity\Attachments\AttachmentType;
use App\Entity\ProjectSystem\Project;
use App\Entity\Parts\Category;
use App\Entity\Parts\Footprint;
use App\Entity\Parts\Manufacturer;
use App\Entity\Parts\MeasurementUnit;
use App\Entity\Parts\StorageLocation;
use App\Entity\Parts\Supplier;
use App\Entity\PriceInformations\Currency;
use App\Services\UserSystem\VoterHelper;
use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
use Symfony\Component\Security\Core\Authorization\Voter\Vote;
use Symfony\Component\Security\Core\Authorization\Voter\Voter;

use function is_object;

/**
 * @phpstan-extends Voter<non-empty-string, AttachmentType|Category|Project|Footprint|Manufacturer|StorageLocation|Supplier|Currency|MeasurementUnit|class-string>
 */
final class StructureVoter extends Voter
{
    protected const OBJ_PERM_MAP = [
        AttachmentType::class => 'attachment_types',
        Category::class => 'categories',
        Project::class => 'projects',
        Footprint::class => 'footprints',
        Manufacturer::class => 'manufacturers',
        StorageLocation::class => 'storelocations',
        Supplier::class => 'suppliers',
        Currency::class => 'currencies',
        MeasurementUnit::class => 'measurement_units',
    ];

    public function __construct(private readonly VoterHelper $helper)
    {
    }

    /**
     * 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(string $attribute, mixed $subject): bool
    {
        if (is_object($subject) || is_string($subject)) {
            $permission_name = $this->instanceToPermissionName($subject);
            //If permission name is null, then the subject is not supported
            return (null !== $permission_name) && $this->helper->isValidOperation($permission_name, $attribute);
        }

        return false;
    }

    public function supportsType(string $subjectType): bool
    {
        return $subjectType === 'string' || $this->instanceToPermissionName($subjectType) !== null;
    }

    /**
     * Maps an instance type to the permission name.
     *
     * @param  object|string  $subject 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(object|string $subject): ?string
    {
        $class_name = is_string($subject) ? $subject : $subject::class;

        //If it is existing in index, we can skip the loop
        if (isset(static::OBJ_PERM_MAP[$class_name])) {
            return static::OBJ_PERM_MAP[$class_name];
        }

        foreach (static::OBJ_PERM_MAP as $class => $ret) {
            if (is_a($class_name, $class, true)) {
                return $ret;
            }
        }

        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  string  $attribute
     */
    protected function voteOnAttribute(string $attribute, $subject, TokenInterface $token, ?Vote $vote = null): bool
    {
        $permission_name = $this->instanceToPermissionName($subject);
        //Just resolve the permission
        return $this->helper->isGranted($token, $permission_name, $attribute, $vote);
    }
}