src/Security/Voter/UserVoter.php line 15

  1. <?php
  3. namespace App\Security\Voter;
  5. use App\Entity\User;
  6. use phpDocumentor\Reflection\DocBlock\Tags\Uses;
  7. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  8. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  9. use Symfony\Component\Security\Core\User\UserInterface;
  10. use Symfony\Component\Security\Core\Security;
  11. use App\Entity\City;
  12. use Doctrine\Persistence\ManagerRegistry;
  13. use App\Entity\Orders;
  15. class UserVoter extends Voter
  16. {
  17.     private $em;
  18.     // these strings are just invented: you can use anything
  19.     const VIEW = 'view';
  20.     const EDIT = 'edit';
  22.     public function __construct(protected Security $security, protected ManagerRegistry $doctrine)
  23.     {
  24.         $this->em = $this->doctrine->getManager();
  25.     }
  27.     // protected function supports($attribute, $subject)
  28.     protected function supports(string $attribute, mixed $subject): bool
  29.     {
  30.         // replace with your own logic
  31.         // https://symfony.com/doc/current/security/voters.html
  32.         return in_array($attribute, [self::EDIT, self::VIEW])
  33.             && $subject instanceof \App\Entity\User;
  34.     }
  36.     // protected function voteOnAttribute($attribute, $subject, TokenInterface $token)
  37.     protected function voteOnAttribute(string $attribute, mixed $subject, TokenInterface $token): bool
  38.     {
  39.         $loginUser = $token->getUser();
  40.         /** @var Uses $user */
  41.         $user = $subject;
  42.         // if the user is anonymous, do not grant access
  43.         if (!$loginUser instanceof UserInterface) {
  44.             return false;
  45.         }
  47.         // ROLE_ADMIN can do anything! The power!
  48.         if ($this->security->isGranted('ROLE_ADMIN')) {
  49.             return true;
  50.         }
  52.         // ... (check conditions and return true to grant permission) ...
  53.         switch ($attribute) {
  54.             case self::EDIT:
  55.                 if ($this->security->isGranted('ROLE_CONTRACTOR')) {
  56.                     $cityIds = [];
  57.                     $city = $this->em->getRepository(City::class)->findOneBy(['contractor' => $loginUser->getId()]);
  58.                     if ($city->getSuburb() == null) {
  59.                         $cityIdStr = $this->em->getRepository(City::class)->getSubCityIds($city->getId());
  60.                         $cityIds = [];
  61.                         if ($cityIdStr) {
  62.                             $cityIds = explode(',', $cityIdStr);
  63.                         }
  64.                     }
  65.                     array_push($cityIds, $city->getId());
  67.                     $orderUserId = [];
  68.                     $orderUserStr = $this->em->getRepository(Orders::class)->getOrderUserByCity($cityIds);
  70.                     if ($orderUserStr) {
  71.                         $orderUserId = explode(',', $orderUserStr);
  72.                     }
  74.                     return in_array($user->getId(), $orderUserId);
  75.                 }
  76.                 break;
  77.             case self::VIEW:
  78.                 if ($this->security->isGranted('ROLE_CONTRACTOR')) {
  79.                     $cityIds = [];
  80.                     $city = $this->em->getRepository(City::class)->findOneBy(['contractor' => $loginUser->getId()]);
  81.                     if ($city->getSuburb() == null) {
  82.                         $cityIdStr = $this->em->getRepository(City::class)->getSubCityIds($city->getId());
  83.                         $cityIds = [];
  84.                         if ($cityIdStr) {
  85.                             $cityIds = explode(',', $cityIdStr);
  86.                         }
  87.                     }
  88.                     array_push($cityIds, $city->getId());
  90.                     $orderUserId = [];
  91.                     $orderUserStr = $this->em->getRepository(Orders::class)->getOrderUserByCity($cityIds);
  93.                     if ($orderUserStr) {
  94.                         $orderUserId = explode(',', $orderUserStr);
  95.                     }
  97.                     return in_array($user->getId(), $orderUserId);
  99.                 }
  100.                 break;
  101.         }
  103.         return false;
  104.     }
  105. }