src/Security/Voter/ScanVoter.php line 14

Open in your IDE?
  1. <?php
  3. namespace App\Security\Voter;
  5. use App\Entity\Scan;
  6. use App\Entity\User;
  7. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  8. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  9. use Symfony\Component\Security\Core\Security;
  11. /**
  12.  * Class ScanVoter
  13.  */
  14. class ScanVoter extends Voter
  15. {
  16.     const VIEW = 'view';
  17.     const EDIT = 'edit';
  19.     /**
  20.      * @var Security
  21.      */
  22.     private $security;
  24.     /**
  25.      * OfferVoter constructor.
  26.      * @param Security $security
  27.      */
  28.     public function __construct(Security $security)
  29.     {
  30.         $this->security = $security;
  31.     }
  33.     /**
  34.      * @param string $attribute
  35.      * @param mixed $subject
  36.      * @return bool
  37.      */
  38.     protected function supports($attribute, $subject): bool
  39.     {
  40.         // if the attribute isn't one we support, return false
  41.         if (!in_array($attribute, [self::VIEW, self::EDIT])) {
  42.             return false;
  43.         }
  45.         if (!$subject instanceof Scan) {
  46.             return false;
  47.         }
  49.         return true;
  50.     }
  52.     protected function voteOnAttribute($attribute, $subject, TokenInterface $token): bool
  53.     {
  54.         $user = $token->getUser();
  55.         $granted = false;
  57.         if ($user instanceof User) {
  58.             /** @var Scan $order */
  59.             $scan = $subject;
  61.             switch ($attribute) {
  62.                 case self::VIEW:
  63.                     $granted = $this->canView($scan, $user);
  64.                     break;
  65.                 case self::EDIT:
  66.                     $granted = $this->canEdit($scan, $user);
  67.                     break;
  68.             }
  69.         }
  71.         return $granted;
  72.     }
  74.     private function canView(Scan $scan, $user): bool
  75.     {
  76.         if ($this->security->isGranted('ROLE_ADMIN')) {
  77.             return true;
  78.         }
  80.         if ($this->canEdit($scan, $user)) {
  81.             return true;
  82.         }
  84.         return false;
  85.     }
  87.     private function canEdit(Scan $scan, User $user): bool
  88.     {
  89.         $owners = $scan->getProject()->getOwners();
  90.         /** @var User $owner */
  91.         foreach ($owners as $owner) {
  92.             if ($user->getId() === $owner->getId()) {
  93.                 return true;
  94.             }
  95.         }
  97.         $members = $scan->getProject()->getMembers();
  99.         /** @var User $member */
  100.         foreach ($members as $member) {
  101.             if ($user->getId() === $member->getId()) {
  102.                 return true;
  103.             }
  104.         }
  106.         return false;
  107.     }
  108. }