src/Security/Voter.php line 13

Open in your IDE?
  1. <?php
  2. namespace App\Security;
  4. use App\Entity\Structure;
  5. use App\Entity\User;
  6. use App\Entity\Chat;
  7. use App\Entity\HousingUnit;
  8. use App\Entity\Attachment;
  9. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  10. use Symfony\Component\Security\Core\Authorization\Voter\Voter as BaseVoter;
  11. use Symfony\Component\Security\Core\Security;
  13. class Voter extends BaseVoter
  14. {
  15.     // attributes for administrators
  16.     const VIEW = 'view';
  17.     const EDIT = 'edit';
  18.     const ATTACHMENT = 'attachment';
  20.     private $security;
  22.     public function __construct(Security $security)
  23.     {
  24.         $this->security = $security;
  25.     }
  26.     
  27.     protected function supports(string $attribute, $subject): bool
  28.     {
  29.         // if the attribute isn't one we support, return false
  30.         if ($attribute === self::ATTACHMENT && $subject instanceof Attachment) return true;
  31.         if (!in_array($attribute, [self::VIEW, self::EDIT])) {
  32.             return false;
  33.         }
  35.         if ($subject instanceof Structure) {return true;}
  36.         if ($subject instanceof Chat) {return true;}
  37.         if ($subject instanceof HousingUnit) {return true;}
  38.         if ($subject instanceof Attachment) {return true;}
  40.         return false;
  41.     }
  43.     protected function voteOnAttribute(string $attribute, $subject, TokenInterface $token): bool
  44.     { 
  45.         $user = $token->getUser();
  47.         if (!$user instanceof User) {
  48.             // the user must be logged in; if not, deny access
  49.             return false;
  50.         }
  52.         if ($this->security->isGranted('ROLE_SUPER_ADMIN')) {
  53.             return true;
  54.         }
  56.         if ($subject instanceof Structure) {return $this->voteOnStructure($attribute, $subject, $user);}
  57.         if ($subject instanceof Chat) {return $this->voteOnChat($attribute, $subject, $user);}
  58.         if ($subject instanceof HousingUnit) {return $this->voteOnHousingUnit($attribute, $subject, $user);}
  59.         if ($subject instanceof Attachment) {return $this->voteOnAttachment($attribute, $subject, $user);}
  61.     }
  62.     
  63.     private function voteOnStructure($attribute, Structure $structure, User $user) : bool
  64.     {
  65.         switch ($attribute) {
  66.             case self::VIEW:
  67.                 return $this->canViewStructure($structure, $user);
  68.             case self::EDIT:
  69.                 return $this->canEditStructure($structure, $user);
  70.         }
  72.         throw new \LogicException('This code should not be reached!');
  73.     }
  75.     private function canViewStructure(Structure $structure, User $user): bool
  76.     {
  77.         // if they can edit, they can view
  78.         if ($this->canEditStructure($structure, $user)) {
  79.             return true;
  80.         }
  82.         // the object could have, for example, a method `isPrivate()`
  83.         //return !$post->isPrivate();
  84.         return false;
  85.     }
  87.     private function canEditStructure(Structure $structure, User $user): bool
  88.     {
  89.         $userFirm = $user->getFirm();
  90.         $structureFirm = $structure->getFirm();
  91.         return $userFirm === $structureFirm;
  92.     }
  93.     
  94.     private function voteOnChat($attribute, Chat $chat, User $user) : bool
  95.     {
  96.         $userFirm = $user->getFirm();
  97.         $chatFirm = $chat->getHousingUnit()->getStructure()->getFirm();
  98.         return $userFirm === $chatFirm;
  99.     }
  101.     private function voteOnHousingUnit($attribute, HousingUnit $hu, User $user) : bool
  102.     {
  103.         $userFirm = $user->getFirm();
  104.         $huFirm = $hu->getStructure()->getFirm();
  105.         return $userFirm === $huFirm;
  106.     }
  108.     private function voteOnAttachment($attribute, Attachment $attach, User $user) : bool
  109.     {
  110.         if ($attach->getBoardMessage() !== null) return $this->canViewStructure($attach->getBoardMessage()->getBoard()->getStructure(), $user);
  111.         if ($attach->getChatMessage() !== null) return $this->voteOnChat($attribute, $attach->getChatMessage()->getChat(), $user);
  112.         return false;
  113.     }
  114.     
  115. }