src/Security/Voter/ContractVoter.php line 12

Open in your IDE?
  1. <?php
  3. namespace App\Security\Voter;
  5. use App\Entity\Authorization;
  6. use App\Entity\Contract;
  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\Security;
  12. class ContractVoter extends Voter
  13. {
  14.     public const CREATE = 'CAN_CREATE';
  15.     public const READ = 'CAN_READ';
  16.     public const EDIT = 'CAN_EDIT';
  17.     public const DELETE = 'CAN_DELETE';
  19.     private Security $security;
  21.     public function __construct(Security $security)
  22.     {
  23.         $this->security = $security;
  24.     }
  26.     protected function supports($attribute, $subject): bool
  27.     {
  28.         $supportsAttribute = in_array($attribute, [self::CREATE, self::DELETE, self::EDIT, self::READ]);
  29.         $supportsSubject = $subject instanceof Contract;
  31.         return $supportsAttribute && $supportsSubject;
  32.     }
  34.     /**
  35.      * @param Contract $subject
  36.      */
  37.     protected function voteOnAttribute(string $attribute, $subject, TokenInterface $token): bool
  38.     {
  39.         $user = $this->security->getUser();
  40.         if (!$user) {
  41.             return false;
  42.         }
  44.         switch ($attribute) {
  45.             case self::CREATE:
  46.                 return $this->canCreate($subject, $user);
  47.             case self::READ:
  48.                 return $this->canRead($subject, $user);
  49.             case self::EDIT:
  50.                 return $this->canEdit($subject, $user);
  51.             case self::DELETE:
  52.                 return $this->canDelete($subject, $user);
  53.         }
  55.         return false;
  56.     }
  58.     private function canCreate(Contract $contract, User $user): bool
  59.     {
  60.         if ($this->security->isGranted(Authorization::ROLE_ADMIN)
  61.             || $this->security->isGranted(Authorization::ROLE_OWNER_ADMIN)
  62.         ) {
  63.             return true;
  64.         }
  66.         return false;
  67.     }
  69.     private function canRead(Contract $contract, User $user): bool
  70.     {
  71.         if ($this->security->isGranted(Authorization::ROLE_ADMIN)) {
  72.             return true;
  73.         }
  75.         if (
  76.             $this->security->isGranted(Authorization::ROLE_OWNER_REQUESTER)
  77.             && $user->getCompany()->getId() === $contract->getOwnerCompany()->getId()
  78.         ) {
  79.             return true;
  80.         }
  82.         if (
  83.             $this->security->isGranted(Authorization::ROLE_SERVICE_PROVIDER)
  84.             && $user->getCompany()->getId() === $contract->getServiceProviderCompany()->getId()
  85.         ) {
  86.             return true;
  87.         }
  89.         return false;
  90.     }
  92.     private function canEdit(Contract $contract, User $user): bool
  93.     {
  94.         if (!$this->canRead($contract, $user)) {
  95.             return false;
  96.         }
  98.         if (!$this->security->isGranted(Authorization::ROLE_OWNER_ADMIN)) {
  99.             return false;
  100.         }
  102.         return true;
  103.     }
  105.     private function canDelete(Contract $contract, User $user): bool
  106.     {
  107.         return $this->canEdit($contract, $user);
  108.     }
  109. }