src/Security/DocumentVoter.php line 11

  1. <?php
  3. namespace App\Security;
  5. use App\Entity\Document;
  6. use App\Entity\DocumentPermission;
  7. use App\Entity\User;
  8. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  9. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  11. class DocumentVoter extends Voter {
  12.     const VIEW = 'view';
  13.     const EDIT = 'edit';
  15.     protected function supports(string $attribute, $subject): bool
  16.     {
  17.         // if the attribute isn't one we support, return false
  18.         if (!in_array($attribute, [self::VIEW, self::EDIT])) {
  19.             return false;
  20.         }
  22.         // only vote on "Document" objects
  23.         if(!$subject instanceof Document) {
  24.             return false;
  25.         }
  26.         return true;
  27.     }
  29.     protected function voteOnAttribute(string $attribute, $subject, TokenInterface $token): bool
  30.     {
  31.         $user = $token->getUser();
  32.         if(!$user instanceof User) {
  33.             return false;
  34.         }
  35.         /** @var Document $document */
  36.         $document = $subject;
  38.         switch ($attribute) {
  39.             case self::VIEW:
  40.                 return $this->canView($document, $user);
  41.             case self::EDIT:
  42.                 return $this->canEdit($document, $user);
  43.             default:
  44.                 throw new \LogicException('This code should not be reached');
  45.         }
  46.     }
  48.     private function canView(Document $document, User $user) {
  49.         if($this->canEdit($document, $user))
  50.             return true;
  51.         return false;
  52.     }
  53.     private function canEdit(Document $document, User $user) {
  54.         /** @var DocumentPermission $permission */
  55.         foreach($document->getDocumentPermissions() as $permission) {
  56.             if($user === $permission->getUser())
  57.                 return true;
  58.         }
  59.             return $user === $document->getOwner();
  60.     }
  61. }