src/Security/Voter/InvoiceVoter.php line 12

Open in your IDE?
  1. <?php
  3. namespace App\Security\Voter;
  5. use App\Entity\Invoice;
  6. use App\Entity\InvoiceStatus;
  7. use App\Entity\User;
  8. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  9. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  10. use Symfony\Component\Security\Core\User\UserInterface;
  12. class InvoiceVoter extends Voter
  13. {
  14.     public const EDIT = 'EDIT';
  15.     public const VIEW = 'VIEW';
  16.     public const DELETE = 'DELETE';
  17.     public const GENERATE_INVOICE = 'GENERATE_INVOICE';
  19.     protected function supports(string $attribute, $subject): bool
  20.     {
  21.         return in_array($attribute, [self::EDIT, self::VIEW, self::GENERATE_INVOICE, self::DELETE])
  22.             && $subject instanceof Invoice;
  23.     }
  25.     protected function voteOnAttribute(string $attribute, $subject, TokenInterface $token): bool
  26.     {
  27.         /** @var User $user */
  28.         $user = $token->getUser();
  30.         if (!$user instanceof UserInterface) {
  31.             return false;
  32.         }
  34.         /** @var Invoice $subject */
  36.         if ($user->getId() !== $subject->getAuthor()->getId()) {
  37.             return false;
  38.         }
  40.         if ($subject->getStatus() === InvoiceStatus::Draft) {
  41.             return true;
  42.         }
  44.         return $attribute === self::VIEW;
  45.     }
  46. }