src/Security/Voter/RestaurantsVoter.php line 13

  1. <?php
  3. namespace App\Security\Voter;
  5. use App\Entity\Restaurants;
  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 RestaurantsVoter 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.     }
  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\Restaurants;
  33.     }
  35.     protected function voteOnAttribute(string $attribute, mixed $subject, TokenInterface $token): bool
  36.     // protected function voteOnAttribute($attribute, $subject, TokenInterface $token)
  37.     {
  38.         $user = $token->getUser();
  40.         /** @var Restaurants $restaurants */
  41.         $restaurants = $subject;
  43.         // if the user is anonymous, do not grant access
  44.         if (!$user instanceof UserInterface || $restaurants->getDeleted()) {
  45.             return false;
  46.         }
  48.         // ROLE_ADMIN can do anything! The power!
  49.         if ($this->security->isGranted('ROLE_ADMIN')) {
  50.             return true;
  51.         }
  53.         // ... (check conditions and return true to grant permission) ...
  54.         switch ($attribute) {
  55.             case self::EDIT:
  56.                 if ($this->security->isGranted('ROLE_CONTRACTOR')) {
  57.                     $cityIds = [];
  58.                     $city = $this->em->getRepository(City::class)->findOneBy(['contractor' => $user->getId()]);
  59.                     if ($city->getSuburb() == null) {
  60.                         $cityIdStr = $this->em->getRepository(City::class)->getSubCityIds($city->getId());
  61.                         $cityIds = [];
  62.                         if ($cityIdStr) {
  63.                             $cityIds = explode(',', $cityIdStr);
  64.                         }
  65.                     }
  66.                     array_push($cityIds, $city->getId());
  68.                     return in_array($restaurants->getCityId()->getId(), $cityIds);
  69.                 } elseif ($this->security->isGranted('ROLE_OWNER')) {
  70.                     return $user === $restaurants->getUserId();
  71.                 }
  72.                 break;
  73.             case self::VIEW:
  74.                 if ($this->security->isGranted('ROLE_CONTRACTOR')) {
  75.                     $cityIds = [];
  76.                     $city = $this->em->getRepository(City::class)->findOneBy(['contractor' => $user->getId()]);
  77.                     if ($city->getSuburb() == null) {
  78.                         $cityIdStr = $this->em->getRepository(City::class)->getSubCityIds($city->getId());
  79.                         $cityIds = [];
  80.                         if ($cityIdStr) {
  81.                             $cityIds = explode(',', $cityIdStr);
  82.                         }
  83.                     }
  84.                     array_push($cityIds, $city->getId());
  86.                     return in_array($restaurants->getCityId()->getId(), $cityIds);
  87.                 } elseif ($this->security->isGranted('ROLE_OWNER')) {
  88.                     return $user === $restaurants->getUserId();
  89.                 }
  90.                 break;
  91.         }
  93.         return false;
  94.     }
  95. }