src/Security/TruVoter.php line 19

Open in your IDE?
  1. <?php
  2. /**
  3.  * Tru voter check access tru
  4.  *
  5.  * @package RMCS
  6.  * @author Vlad Shashkov  <vlad.s@zimalab.com>
  7.  * @copyright 2014 - 2019 The Zimalab
  8.  */
  9. declare(strict_types=1);
  11. namespace App\Security;
  13. use App\Entity\Main\{Permission, User};
  14. use Doctrine\Persistence\{ManagerRegistry, ObjectManager};
  15. use Symfony\Component\HttpFoundation\{Request, RequestStack};
  16. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  17. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  19. class TruVoter extends Voter
  20. {
  22.     /**
  23.      * @var array const
  24.      */
  25.     private const FLAGS = [
  26.         'CheckReadTru' => 'ROLE_R',
  27.         'CheckReadWriteTru' => 'ROLE_RW',
  28.         'CheckAdminTru' => 'ROLE_ADMIN',
  29.         'CheckOwnerTRu' => 'ROLE_OWNER'
  30.     ];
  33.     /**
  34.      * @var array const
  35.      */
  36.     public const ROLES = [
  37.         'r' => 'ROLE_R',
  38.         'rw' => 'ROLE_RW',
  39.         'admin'  => 'ROLE_ADMIN',
  40.         'root' => 'ROLE_OWNER'
  41.     ];
  43.     /**
  44.      * @var array
  45.      */
  46.     private const WEIGHTS = [
  47.         'ROLE_R' => 1,
  48.         'ROLE_RW' => 2,
  49.         'ROLE_ADMIN' => 3,
  50.         'ROLE_OWNER' => 4
  51.     ];
  53.     /**
  54.      * @var ObjectManager
  55.      */
  56.     private $em;
  59.     /**
  60.      * @var Request|null
  61.      */
  62.     private $request;
  65.     /**
  66.      * TruVoter constructor.
  67.      *
  68.      * @param ManagerRegistry $registry
  69.      * @param RequestStack $requestStack
  70.      */
  71.     public function __construct(ManagerRegistry $registry, RequestStack $requestStack)
  72.     {
  73.         $this->em = $registry->getManager('default');
  74.         $this->request = $requestStack->getCurrentRequest();
  75.     }
  77.     /**
  78.      * Determines if the attribute and subject are supported by this voter.
  79.      *
  80.      * @param string $attribute An attribute
  81.      * @param mixed $subject The subject to secure, e.g. an object the user wants to access or any other PHP type
  82.      *
  83.      * @return bool True if the attribute and subject are supported, false otherwise
  84.      */
  85.     protected function supports($attribute, $subject):bool
  86.     {
  87.         return in_array($attribute, array_keys(self::FLAGS));
  88.     }
  90.     /**
  91.      * Perform a single access check operation on a given attribute, subject and token.
  92.      * It is safe to assume that $attribute and $subject already passed the "supports()" method check.
  93.      *
  94.      * @param string $attribute
  95.      * @param mixed $subject
  96.      *
  97.      * @param TokenInterface $token
  98.      * @return bool
  99.      */
  100.     protected function voteOnAttribute($attribute, $subject, TokenInterface $token):bool
  101.     {
  102.         $user = $token->getUser();
  104.         if (!$user instanceof User) {
  105.             return false;
  106.         }
  108.         $userRoleWeight = $this->getUserRoleWeight($user);
  109.         $accessWeight = $this->getAccessWeight($attribute);
  111.         return $userRoleWeight >= $accessWeight;
  112.     }
  114.     /**
  115.      * @param User $user
  116.      * @return bool
  117.      */
  118.     private function getUserRoleWeight(User $user): bool
  119.     {
  120.         // Get user role weight
  121.         $roleUserWeight = $this->getRoleWeight(current($user->getRoles()));
  123.         // Get site role weight
  124.         $client = $user->getClient()->getId();
  125.         $idTru = intval($this->request->get('id_tru'));
  127.         $siteRole = $this->em->getRepository(Permission::class)
  128.             ->findPermissionByClient($client, $idTru);
  131.         $siteRoleWeight = $this->getRoleWeight($siteRole);
  133.         // Which role weight we should use for get access ?
  134.         return $roleUserWeight > $siteRoleWeight ?
  135.             $siteRoleWeight :
  136.             $roleUserWeight;
  137.     }
  139.     /**
  140.      * @param $attribute
  141.      * @return false|int|string
  142.      */
  143.     private function getAccessWeight(string $attribute)
  144.     {
  145.         $accessRole = self::FLAGS[$attribute];
  147.         return $this->getRoleWeight($accessRole);
  148.     }
  150.     /**
  151.      * @param $role
  152.      * @return false|int|string
  153.      */
  154.     private function getRoleWeight(string $role)
  155.     {
  156.         return array_search($role, self::WEIGHTS);
  157.     }
  159.     /**
  160.      * @return array|false
  161.      */
  162.     public static function getRoles()
  163.     {
  164.         return array_combine(range(1, count(self::ROLES)), array_values(self::ROLES));
  165.     }
  166. }