src/Security/Voters/FolderVoter.php line 13

Open in your IDE?
  1. <?php
  4. namespace App\Security\Voters;
  7. use App\Entity\Folder;
  8. use App\Entity\User;
  9. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  10. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  11. use Symfony\Component\Security\Core\Security;
  13. class FolderVoter extends Voter
  14. {
  15.   const EDIT = 'edit';
  16.   const VIEW = 'view';
  19.   private $security;
  21.   public function __construct(Security $security)
  22.   {
  23.     $this->security = $security;
  24.   }
  26.   protected function supports($attribute, $subject): bool
  27.   {
  28.     // if the attribute isn't one we support, return false
  29.     if (!in_array($attribute, [self::EDIT, self::VIEW])) {
  30.       return false;
  31.     }
  33.     // only vote on `Folder` objects
  34.     if ($subject && !$subject instanceof Folder) {
  35.       return false;
  36.     }
  38.     return true;
  39.   }
  41.   protected function voteOnAttribute($attribute, $subject, TokenInterface $token): bool
  42.   {
  43.     $user = $token->getUser();
  45.     if (!$user instanceof User) {
  46.       // the user must be logged in; if not, deny access
  47.       return false;
  48.     }
  50.     // you know $subject is a Folder object, thanks to `supports()`
  51.     /** @var Folder $folder */
  52.     $folder = $subject;
  54.     switch ($attribute) {
  55.       case self::EDIT:
  56.         return $this->canEdit($folder, $user);
  57.       case self::VIEW:
  58.         return $this->canView($folder, $user);
  60.     }
  62.     throw new \LogicException('This code should not be reached!');
  63.   }
  65.   private function canView(Folder $folder, User $user): bool
  66.   {
  67.     // if they can edit, they can view
  68.     if ($this->canEdit($folder, $user)) {
  69.       return true;
  70.     }
  72.     return $user->getId() == $folder->getOwnerId();
  73.   }
  75.   private function canEdit(Folder $folder, User $user): bool
  76.   {
  77.     if ($this->security->isGranted('ROLE_ADMIN') || $this->security->isGranted('ROLE_OPERATORE')) {
  78.       return true;
  79.     }
  80.     return false;
  81.   }
  82. }