src/Security/SuperheroOffice/ShareholderVoter.php line 15

Open in your IDE?
  1. <?php
  3. namespace App\Security\SuperheroOffice;
  5. use App\Entity\SuperheroOffice\Shareholder;
  6. use App\Handler\SuperheroOffice\UserOfficeHandler;
  7. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  8. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  10. /**
  11.  * Class ShareholderVoter
  12.  *
  13.  * @package App\Security\SuperheroOffice
  14.  */
  15. class ShareholderVoter extends Voter
  16. {
  17.     const EDIT   = 'edit';
  18.     const VIEW   = 'view';
  19.     const DELETE = 'delete';
  21.     /**
  22.      * base voter actions
  23.      */
  24.     private const ATTRIBUTES = [
  25.         self::EDIT,
  26.         self::VIEW,
  27.         self::DELETE
  28.     ];
  30.     /**
  31.      * @var UserOfficeHandler
  32.      */
  33.     private $userOfficeHandler;
  35.     /**
  36.      * ShareholderVoter constructor.
  37.      *
  38.      * @param UserOfficeHandler $userOfficeHandler
  39.      */
  40.     public function __construct(UserOfficeHandler $userOfficeHandler)
  41.     {
  42.         $this->userOfficeHandler = $userOfficeHandler;
  43.     }
  45.     /**
  46.      * @param string $attribute
  47.      * @param mixed  $subject
  48.      *
  49.      * @return bool
  50.      */
  51.     protected function supports($attribute, $subject)
  52.     {
  53.         return $subject instanceof Shareholder
  54.             && in_array($attribute, self::ATTRIBUTES);
  55.     }
  57.     /**
  58.      * @param                $attribute
  59.      * @param Shareholder    $shareholder
  60.      * @param TokenInterface $token
  61.      *
  62.      * @return bool
  63.      */
  64.     protected function voteOnAttribute(
  65.         $attribute,
  66.         $shareholder,
  67.         TokenInterface $token
  68.     ): bool {
  69.         switch ($attribute) {
  70.             case self::EDIT:
  71.             case self::DELETE:
  72.             case self::VIEW:
  73.                 return $this
  74.                     ->userOfficeHandler
  75.                     ->getUserOffice()
  76.                     ->getShareholders()
  77.                     ->contains($shareholder);
  78.                 break;
  79.         }
  81.         throw new \LogicException('Invalid attribute: ' . $attribute);
  82.     }
  83. }