src/Voter/InvoiceVoter.php line 13

Open in your IDE?
  1. <?php
  3. namespace App\Voter;
  5. use App\Entity\Invoice;
  6. use App\Entity\User as UserEntity;
  7. use App\Model\InvoiceFactory;
  8. use App\Model\User\User;
  9. use Exception;
  10. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  11. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  13. class InvoiceVoter extends Voter
  14. {
  15.     public const REMOVE = 'remove';
  17.     public const SHOW = 'show';
  19.     private $user;
  21.     private $invoiceFactory;
  23.     public function __construct(User $user, InvoiceFactory $invoiceFactory)
  24.     {
  25.         $this->user           = $user;
  26.         $this->invoiceFactory = $invoiceFactory;
  27.     }
  29.     /**
  30.      * @param string $attribute
  31.      * @param mixed $subject
  32.      */
  33.     protected function supports($attribute, $subject): bool
  34.     {
  36.         // if the attribute isn't one we support, return false
  37.         if (! in_array($attribute, [
  38.             self::REMOVE,
  39.             self::SHOW
  40.         ])) {
  41.             return false;
  42.         }
  44.         // only vote on Supplier objects inside this voter
  45.         if (! $subject instanceof Invoice) {
  46.             return false;
  47.         }
  49.         return true;
  50.     }
  52.     /**
  53.      * @param string $attribute
  54.      * @param mixed $entity
  55.      * @return bool|void
  56.      * @throws Exception
  57.      */
  58.     protected function voteOnAttribute($attribute, $entity, TokenInterface $token): bool
  59.     {
  62.         $userEntity = $token->getUser();
  63.         if (! $userEntity instanceof UserEntity) {
  64.             // the user must be logged in; if not, deny access
  65.             return false;
  66.         }
  68.         $this->user->setEntity($userEntity);
  70.         $invoice = $this->invoiceFactory->createModel($entity);
  73.         switch ($attribute) {
  74.             case self::SHOW:
  75.                 return $this->user->canSeeInvoiceEntity($entity);
  76.             case self::REMOVE:
  77.                 return $invoice->canBeRemovedBy($this->user);
  78.         }
  80.         return false;
  81.     }
  84. }