src/Security/Voter/UserVoter.php line 11

Open in your IDE?
  1. <?php
  3. namespace App\Security\Voter;
  5. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  6. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  7. use Symfony\Component\Security\Core\User\UserInterface;
  8. use App\Entity\User;
  9. use Symfony\Component\Security\Core\Security;
  11. class UserVoter extends Voter
  12. {
  13.     const VIEW ="ver";
  14.     const EDIT ="editar";
  15.     const DELETE = "eliminar";
  17.     private $security;
  19.     public function __construct(Security $security)
  20.     {
  21.         $this->security = $security;
  22.     }
  24.     protected function supports($attribute, $subject)
  25.     {
  26.         // if the attribute isn't one we support, return false
  27.         if (!in_array($attribute, [self::VIEW, self::EDIT, self::DELETE])) {
  28.             return false;
  29.         }
  31.         // only vote on `User` objects
  32.         if (!$subject instanceof User) {
  33.             return false;
  34.         }
  35.         return true;
  36.     }
  38.     protected function voteOnAttribute($attribute, $subject, TokenInterface $token)
  39.     {
  40.         $admin = $token->getUser();
  42.         // if the user is anonymous, do not grant access
  43.         if (!$admin 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::VIEW:
  55.                 return $this->canView($subject, $admin);
  56.             case self::EDIT:
  57.                 return $this->canEdit($subject, $admin);
  58.             case self::DELETE:
  59.                 return $this->canDelete($subject, $admin);
  60.         }
  61.         return false; 
  62.     }
  63.     
  64.     private function canView(User $user, User $admin)
  65.     {        
  66.         return $admin->getCompany() === $user->getCompany();
  67.     }
  69.     private function canEdit(User $user, User $admin)
  70.     {
  71.         
  72.         // ROLE_ADMIN can do anything! The power!
  73.         if ( $admin->getRole() == 'ROLE_CHIEF' || $admin->getRole() == 'ROLE_SUPPLIER_CHIEF') {
  74.             
  75.             return $admin->getCompany() === $user->getCompany();
  76.         }else{
  77.             
  78.             return $admin->getId() === $user->getId();        
  79.         }
  80.         
  81.     }
  83.     private function canDelete(User $user, User $admin)
  84.     {
  85.         if ($admin->getId() !== $user->getId())
  86.         {
  87.             return $admin->getCompany() === $user->getCompany();
  88.         }
  90.         return false;
  91.         
  92.     }
  93. }