src/LawFirm/Security/ContactQualityVoter.php line 14

Open in your IDE?
  1. <?php
  3. namespace App\LawFirm\Security;
  5. use AppBundle\Entity\LawFirm;
  6. use AppBundle\Entity\User;
  7. use CRMBundle\Entity\ContactQuality;
  8. use CRMBundle\Enum\ContactQualityEnum;
  9. use CRMBundle\Repository\ContactRepository;
  10. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  11. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  12. use InvalidArgumentException;
  14. class ContactQualityVoter extends Voter
  15. {
  16.     public const EDIT = 'edit';
  17.     public const REMOVE = 'remove';
  18.     
  19.     private ContactRepository $contactRepository;
  21.     public function __construct(ContactRepository $contactRepository)
  22.     {
  23.         $this->contactRepository = $contactRepository;
  24.     }
  26.     /**
  27.      * @inheritDoc
  28.      */
  29.     protected function supports($attribute, $subject)
  30.     {
  31.         if (!in_array(
  32.             $attribute,
  33.             [
  34.                 self::EDIT,
  35.                 self::REMOVE,
  36.             ]
  37.         )) {
  38.             return false;
  39.         }
  41.         if (!$subject instanceof ContactQuality) {
  42.             return false;
  43.         }
  45.         return true;
  46.     }
  48.     /**
  49.      * {@inheritdoc}
  50.      * @param ContactQuality $subject
  51.      */
  52.     protected function voteOnAttribute($attribute, $subject, TokenInterface $token)
  53.     {
  54.         $user = $token->getUser();
  56.         if (!$user instanceof User) {
  57.             return false;
  58.         }
  60.         /** @var LawFirm $lawFirm */
  61.         $lawFirm = $user->getLawFirm();
  63.         $isLockedQuality = in_array($subject->getQualityRelated(), [
  64.             ContactQualityEnum::NONE,
  65.             ContactQualityEnum::JURISDICTION,
  66.             ContactQualityEnum::CUSTOMER,
  67.         ]);
  69.         switch ($attribute) {
  70.             case self::EDIT:
  71.                 return !$isLockedQuality;
  72.             case self::REMOVE:
  73.                 return !$isLockedQuality && !$this->usedByContact($subject, $lawFirm);
  74.             default:
  75.                 throw new InvalidArgumentException('Invalid attribute');
  76.         }
  77.     }
  79.     private function usedByContact(ContactQuality $contactQuality, LawFirm $lawFirm): bool
  80.     {
  81.         return $this->contactRepository->count([
  82.             'lawFirm' => $lawFirm,
  83.             'qualityNew' => $contactQuality,
  84.         ]) > 0;
  85.     }
  86. }