src/Security/Voter/SocietyV2Voter.php line 11

Open in your IDE?
  1. <?php
  3. namespace App\Security\Voter;
  5. use App\Entity\User;
  6. use App\Entity\V2\SocietyV2;
  7. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  8. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  9. use Symfony\Component\Security\Core\Security;
  11. class SocietyV2Voter extends Voter
  12. {
  13.     const VIEW = 'view';
  14.     const EDIT = 'edit';
  15.     const DELETE = 'delete';
  17.     private Security $security;
  19.     public function __construct(Security $security)
  20.     {
  21.         $this->security = $security;
  22.     }
  24.     protected function supports(string $attribute, $subject): bool
  25.     {
  26.         return in_array($attribute, [self::VIEW, self::EDIT, self::DELETE])
  27.             && $subject instanceof SocietyV2;
  28.     }
  30.     protected function voteOnAttribute(string $attribute, $subject, TokenInterface $token): bool
  31.     {
  32.         $user = $token->getUser();
  34.         if (!$user instanceof User) {
  35.             return false;
  36.         }
  38.         /** @var SocietyV2 $society */
  39.         $society = $subject;
  41.         // Admin can do everything
  42.         if ($this->security->isGranted('ROLE_ADMIN')) {
  43.             return true;
  44.         }
  46.         switch ($attribute) {
  47.             case self::VIEW:
  48.                 return $this->canView($society, $user);
  49.             case self::EDIT:
  50.                 return $this->canEdit($society, $user);
  51.             case self::DELETE:
  52.                 return $this->canDelete($society, $user);
  53.         }
  55.         return false;
  56.     }
  58.     private function canView(SocietyV2 $society, User $user): bool
  59.     {
  60.         // Owner can view
  61.         if ($society->getUser() && $society->getUser()->getId() === $user->getId()) {
  62.             return true;
  63.         }
  65.         // Users from the same client can view
  66.         if ($society->getClient() && $user->getClient()) {
  67.             if ($society->getClient()->getId() === $user->getClient()->getId()) {
  68.                 return true;
  69.             }
  70.         }
  72.         return false;
  73.     }
  75.     private function canEdit(SocietyV2 $society, User $user): bool
  76.     {
  77.         // Same as view for now - owner or same client
  78.         return $this->canView($society, $user);
  79.     }
  81.     private function canDelete(SocietyV2 $society, User $user): bool
  82.     {
  83.         // Only owner can delete
  84.         if ($society->getUser() && $society->getUser()->getId() === $user->getId()) {
  85.             return true;
  86.         }
  88.         return false;
  89.     }
  90. }