src/Backoffice/Security/BackofficeFeatureVoter.php line 12

Open in your IDE?
  1. <?php
  3. namespace App\Backoffice\Security;
  5. use App\Backoffice\Enum\BackofficeFeatureAccessEnum;
  6. use AppBundle\Entity\User;
  7. use Doctrine\ORM\EntityManagerInterface;
  8. use Symfony\Component\Security\Core\Authentication\Token\SwitchUserToken;
  9. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  10. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  12. class BackofficeFeatureVoter extends Voter
  13. {
  14.     private EntityManagerInterface $em;
  16.     public function __construct(EntityManagerInterface $em)
  17.     {
  18.         $this->em = $em;
  19.     }
  21.     protected function supports($attribute, $subject): bool
  22.     {
  23.         return \in_array($attribute, BackofficeFeatureAccessEnum::getChoices(), true);
  24.     }
  26.     protected function voteOnAttribute($attribute, $subject, TokenInterface $token): bool
  27.     {
  28.         $user = $token->getUser();
  30.         if ($token instanceof SwitchUserToken) {
  31.             $user = $token->getOriginalToken()->getUser();
  32.         }
  34.         if (!$user instanceof User) {
  35.             return false;
  36.         }
  38.         // User hasn't data during a switch
  39.         $user = $this->em->getRepository(User::class)->find($user->getId());
  41.         if ($user === null || $user->getBackofficeProfile() === null) {
  42.             return false;
  43.         }
  45.         return $user->getBackofficeProfile()->hasRole($attribute);
  46.     }
  47. }