src/Security/HotelVoter.php line 12

Open in your IDE?
  1. <?php
  2. // src/Security/HotelVoter.php
  3. namespace App\Security;
  5. use App\Entity\Hotel;
  6. use App\Entity\User;
  8. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  9. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  10. use Symfony\Component\Security\Core\Security;
  12. class HotelVoter extends Voter
  13. {
  14.     const VIEW = 'view';
  15.     const EDIT = 'edit';
  16.     
  17.     private $security;
  19.     public function __construct(Security $security)
  20.     {
  21.         $this->security = $security;
  22.     }    
  24.     protected function supports(string $attribute, $subject): bool
  25.     {
  26.         // if the attribute isn't one we support, return false
  27.         if (!in_array($attribute, [self::VIEW, self::EDIT])) {
  28.             return false;
  29.         }
  31.         // only vote on `Hotel` objects
  32.         if (!$subject instanceof Hotel) {
  33.             return false;
  34.         }
  36.         return true;
  37.     }
  39.     protected function voteOnAttribute(string $attribute, $subject, TokenInterface $token): bool
  40.     {
  41.         $user = $token->getUser();
  43.         if (!$user instanceof User) {
  44.             // the user must be logged in; if not, deny access
  45.             return false;
  46.         }
  48.         // you know $subject is a Hotel object, thanks to `supports()`
  49.         /** @var Hotel $hotel */
  50.         $hotel = $subject;
  52.         switch ($attribute) {
  53.             case self::VIEW:
  54.                 return $this->canView($hotel, $user);
  55.             case self::EDIT:
  56.                 return $this->canEdit();
  57.         }
  59.         throw new \LogicException('This code should not be reached!');
  60.     }
  62.     private function canView(Hotel $hotel, User $user): bool
  63.     {
  64.         $hotel_users = $hotel->getHotelUsers();
  65.         foreach($hotel_users as $hu){
  66.             if($hu->getUser() == $user){
  67.                 return true;
  68.             }
  69.         }
  71.         // the Hotel object could have, for example, a method `isPrivate()`
  72.         return false;
  73.     }
  75.     private function canEdit(): bool
  76.     {
  77.         if ($this->security->isGranted('ROLE_ADMIN')) {
  78.             return true;
  79.         }
  80.         
  81.         return false;
  82.     }
  83. }