src/Security/EntityVoter.php line 17

Open in your IDE?
  1. <?php
  3. namespace App\Security;
  5. use App\Entity\Menu;
  6. use App\Entity\Permission;
  7. use App\Entity\Profile;
  8. use App\Entity\User;
  9. use Doctrine\ORM\EntityManagerInterface;
  10. use Sogec\BOBundle\Routing\CRUDAction;
  11. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  12. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  14. /**
  15.  * Class EntityVoter.
  16.  */
  17. class EntityVoter extends Voter
  18. {
  19.     /**
  20.      * @var EntityManagerInterface
  21.      */
  22.     private $manager;
  24.     /**
  25.      * EntityVoter constructor.
  26.      *
  27.      * @param EntityManagerInterface $manager Entity manager.
  28.      */
  29.     public function __construct(EntityManagerInterface $manager)
  30.     {
  31.         $this->manager = $manager;
  32.     }
  34.     /**
  35.      * {@inheritdoc}
  36.      */
  37.     protected function supports($action, $route)
  38.     {
  39.         // if the attribute isn't one we support, return false
  40.         if (!in_array($action, array(CRUDAction::SHOW, CRUDAction::EDIT, CRUDAction::DELETE, CRUDAction::NEW))) {
  41.             return false;
  42.         }
  44.         return true;
  45.     }
  47.     /**
  48.      * @param string         $action
  49.      * @param mixed          $route
  50.      * @param TokenInterface $token
  51.      *
  52.      * @return bool
  53.      */
  54.     protected function voteOnAttribute($action, $route, TokenInterface $token)
  55.     {
  56.         $user = $token->getUser();
  58.         if (!$user instanceof User) {
  59.             // the user must be logged in; if not, deny access
  60.             return false;
  61.         }
  63.         return $this->canActionByProfile($user, $route, $action);
  64.     }
  66.     /**
  67.      * @param User   $user
  68.      * @param string $route
  69.      * @param string $action
  70.      *
  71.      * @return bool
  72.      */
  73.     private function canActionByProfile(User $user, $route, $action)
  74.     {
  75.         $profile = $user->getProfile();
  76.         $permission = $this->getPermissionByOption($profile, $route);
  77.         if (!$permission) {
  78.             return false;
  79.         }
  80.         if ('ROLE_SUPER_ADMIN' === $user->getRoles()[0]) {
  81.             return true;
  82.         }
  84.         return 1 === (int) $permission->getActions()[$action] ? true : false;
  85.     }
  87.     /**
  88.      * @param Profile $profile
  89.      * @param string  $route
  90.      *
  91.      * @return Permission|bool|null|object
  92.      */
  93.     private function getPermissionByOption(Profile $profile, $route)
  94.     {
  95.         $menu = $this->manager->getRepository(Menu::class)
  96.             ->findOneBy(['route' => $route]);
  98.         $permission = $this->manager->getRepository(Permission::class)
  99.             ->findOneBy(['menu' => $menu, 'profile' => $profile]);
  101.         return $permission ? $permission : false;
  102.     }
  103. }