src/Security/Voter/CompanyVoter.php line 12

Open in your IDE?
  1. <?php
  3. namespace App\Security\Voter;
  5. use App\Entity\Company;
  6. use App\Entity\User;
  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;
  12. class CompanyVoter extends Voter
  13. {
  14.     const VIEW ="ver";
  15.     const EDIT ="editar";
  16.     const DELETE = "eliminar";
  18.     private $security;
  20.     public function __construct(Security $security)
  21.     {
  22.         $this->security = $security;
  23.     }
  25.     protected function supports($attribute, $subject)
  26.     {
  27.         // if the attribute isn't one we support, return false
  28.         if (!in_array($attribute, [self::VIEW, self::EDIT])) {
  29.             return false;
  30.         }
  32.         // only vote on `Company` objects
  33.         if (!$subject instanceof Company) {
  34.             return false;
  35.         }
  36.         return true;
  37.     }
  39.     protected function voteOnAttribute($attribute, $subject, TokenInterface $token)
  40.     {
  41.         $user = $token->getUser();
  42.         // if the user is anonymous, do not grant access
  43.         if (!$user 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, $user);
  56.             case self::EDIT:
  57.                 return $this->canEdit($subject, $user);
  58.             case self::DELETE:
  59.                 return $this->canDelete($subject, $user);
  60.             break;
  61.         }
  62.         return false;        
  63.     }
  65.     private function canView(Company $company, User $user)
  66.     {        
  67.         return $user === $company->getUser();
  68.     }
  70.     private function canEdit(Company $company, User $user)
  71.     {
  72.         return $user === $company->getUser();
  73.     }
  75.     private function canDelete(Company $company, User $user)
  76.     {
  77.         return $user === $company->getUser();
  78.     }
  79. }