src/Security/Voter/BusinessVoter.php line 13

  1. <?php
  3. namespace App\Security\Voter;
  5. use App\Entity\Business;
  6. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  7. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  8. use Symfony\Component\Security\Core\User\UserInterface;
  9. use Symfony\Component\Security\Core\Security;
  10. use App\Entity\City;
  11. use Doctrine\Persistence\ManagerRegistry;
  13. class BusinessVoter extends Voter
  14. {
  15.     private $em;
  16.     // these strings are just invented: you can use anything
  17.     const VIEW = 'view';
  18.     const EDIT = 'edit';
  20.     public function __construct(protected Security $security, protected ManagerRegistry $doctrine)
  21.     {
  22.         $this->em = $this->doctrine->getManager();
  23.     }
  25.     #https://symfony.com/doc/6.2/security/voters.html
  26.     // protected function supports($attribute, $subject)
  27.     protected function supports(string $attribute, mixed $subject): bool
  28.     {
  29.         // replace with your own logic
  30.         // https://symfony.com/doc/current/security/voters.html
  31.         return in_array($attribute, [self::EDIT, self::VIEW])
  32.             && $subject instanceof \App\Entity\Business;
  33.     }
  35.     // protected function voteOnAttribute($attribute, $subject, TokenInterface $token)
  36.     protected function voteOnAttribute(string $attribute, mixed $subject, TokenInterface $token): bool
  37.     {
  38.         $user = $token->getUser();
  39.         /** @var Business $business */
  40.         $business = $subject;
  41.         // if the user is anonymous, do not grant access
  42.         if (!$user instanceof UserInterface || $business->getDeleted()) {
  43.             return false;
  44.         }
  46.         // ROLE_ADMIN can do anything! The power!
  47.         if ($this->security->isGranted('ROLE_ADMIN')) {
  48.             return true;
  49.         }
  51.         // ... (check conditions and return true to grant permission) ...
  52.         switch ($attribute) {
  53.             case self::EDIT:
  54.                 if ($this->security->isGranted('ROLE_CONTRACTOR')) {
  55.                     $cityIds = [];
  56.                     $city = $this->em->getRepository(City::class)->findOneBy(['contractor' => $user->getId()]);
  57.                     if ($city->getSuburb() == null) {
  58.                         $cityIdStr = $this->em->getRepository(City::class)->getSubCityIds($city->getId());
  59.                         $cityIds = [];
  60.                         if ($cityIdStr) {
  61.                             $cityIds = explode(',', $cityIdStr);
  62.                         }
  63.                     }
  64.                     array_push($cityIds, $city->getId());
  66.                     return in_array($business->getCityId()->getId(), $cityIds);
  67.                 }
  68.                 break;
  69.             case self::VIEW:
  70.                 if ($this->security->isGranted('ROLE_CONTRACTOR')) {
  71.                     $cityIds = [];
  72.                     $city = $this->em->getRepository(City::class)->findOneBy(['contractor' => $user->getId()]);
  73.                     if ($city->getSuburb() == null) {
  74.                         $cityIdStr = $this->em->getRepository(City::class)->getSubCityIds($city->getId());
  75.                         $cityIds = [];
  76.                         if ($cityIdStr) {
  77.                             $cityIds = explode(',', $cityIdStr);
  78.                         }
  79.                     }
  80.                     array_push($cityIds, $city->getId());
  82.                     return in_array($business->getCityId()->getId(), $cityIds);
  83.                 }
  84.                 break;
  85.         }
  87.         return false;
  88.     }
  89. }