src/Security/Voters/DocumentVoter.php line 14

Open in your IDE?
  1. <?php
  4. namespace App\Security\Voters;
  7. use App\Entity\CPSUser;
  8. use App\Entity\Document;
  9. use App\Entity\User;
  10. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  11. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  12. use Symfony\Component\Security\Core\Security;
  14. class DocumentVoter extends Voter
  15. {
  16.   const EDIT = 'edit';
  17.   const VIEW = 'view';
  20.   private $security;
  22.   public function __construct(Security $security)
  23.   {
  24.     $this->security = $security;
  25.   }
  27.   protected function supports($attribute, $subject)
  28.   {
  29.     // if the attribute isn't one we support, return false
  30.     if (!in_array($attribute, [self::EDIT, self::VIEW])) {
  31.       return false;
  32.     }
  34.     // only vote on `Document` objects
  35.     if ($subject && !$subject instanceof Document) {
  36.       return false;
  37.     }
  39.     return true;
  40.   }
  42.   protected function voteOnAttribute($attribute, $subject, TokenInterface $token)
  43.   {
  44.     $user = $token->getUser();
  46.     if (!$user instanceof User) {
  47.       // the user must be logged in; if not, deny access
  48.       return false;
  49.     }
  51.     // you know $subject is a Document object, thanks to `supports()`
  52.     /** @var Document $document */
  53.     $document = $subject;
  55.     switch ($attribute) {
  56.       case self::EDIT:
  57.         return $this->canEdit($document, $user);
  58.       case self::VIEW:
  59.         return $this->canView($document, $user);
  61.     }
  63.     throw new \LogicException('This code should not be reached!');
  64.   }
  66.   private function canView(Document $document, User $user)
  67.   {
  68.     // if they can edit, they can view
  69.     if ($this->canEdit($document, $user)) {
  70.       return true;
  71.     }
  73.     /** @var CPSUser $user */
  74.     return in_array($user->getCodiceFiscale(), array_merge([$document->getOwner()->getCodiceFiscale()], $document->getRelatedCFs()));
  76.   }
  78.   private function canEdit(Document $document, User $user)
  79.   {
  80.     if ($this->security->isGranted('ROLE_ADMIN') || $this->security->isGranted('ROLE_OPERATORE')) {
  81.       return true;
  82.     }
  83.     return false;
  84.   }
  85. }