src/Security/CompanyVoter.php line 14

  1. <?php
  2. namespace App\Security;
  4. use App\Entity\Companies;
  5. use App\Entity\Users;
  6. use App\Entity\ClientsDelegation;
  7. use App\Entity\VehicleClients;
  8. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  9. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  10. use Symfony\Component\Security\Core\Authorization\AccessDecisionManagerInterface;
  11. use Doctrine\ORM\EntityManagerInterface;
  12. use App\Service\UserHelper;
  14. class CompanyVoter extends Voter
  15. {
  16.     const ACCESS= [ 
  17.         'company_view_summary'  => 'company_view_summary',
  18.         'company_edit_info'     => 'company_edit_info',
  19.         'company_docs' => 'company_docs',
  20.         'company_bank_accounts' => 'company_bank_accounts'];
  22.     /**
  23.      * @var AccessDecisionManager|null
  24.      */
  25.     protected \Symfony\Component\Security\Core\Authorization\AccessDecisionManagerInterface $decisionManager;
  27.     /**
  28.      * @var EntityManager|null
  29.      */
  30.     protected \Doctrine\ORM\EntityManagerInterface $entityManager;
  32.     /**
  33.      * @var EntityManager|null
  34.      */
  35.     protected \App\Service\UserHelper $userHelper;
  38.     /**
  39.      * DelegateVoter constructor.
  40.      * @param AccessDecisionManager|null $decisionManager
  41.      * @param EntityManager|null $entityManager
  42.      */
  43.     public function __construct(AccessDecisionManagerInterface $decisionManager, EntityManagerInterface $entityManager, UserHelper $userHelper)
  44.     {
  45.         $this->decisionManager = $decisionManager;
  46.         $this->entityManager = $entityManager;
  47.         $this->userHelper = $userHelper;
  48.     }
  50.     /**
  51.      * determines if your voter should vote on the attribute/subject combination. If you return true,
  52.      * voteOnAttribute() will be called. Otherwise, your voter is done: some other voter should process this
  53.      */
  54.     protected function supports($attribute, $subject): bool
  55.     {
  56.         // if the attribute isn't one we support, return false
  57.         if (!in_array($attribute, self::ACCESS)) {
  58.             return false;
  59.         }
  60.         // only vote on Companies objects inside this voter
  61.         return $subject instanceof Companies;
  62.     }
  64.     /**
  65.      * If you return true from supports(), then this method is called. Your job is simple: return true to allow access
  66.      * and false to deny access
  67.      */
  68.     protected function voteOnAttribute($attribute, $subject, TokenInterface $token): bool
  69.     {
  70.         $user = $token->getUser();
  72.         if (!$user instanceof Users) {
  73.             return false; // the user must be logged in; if not, deny access
  74.         }
  75.         
  76.         // ROLE_SUPER_ADMIN can do anything! The power! Calling decide() on the AccessDecisionManager is essentially the same
  77.         // as calling isGranted() from a controller or other places (it's just a little lower-level, which is necessary for a voter).
  78.         if ($this->decisionManager->decide($token, ['ROLE_SUPER_ADMIN'])) {
  79.             return true;
  80.         }
  82.         // you know $subject is a VehicleClient object, thanks to supports
  83.         /** @var VehicleClients $vehicleClient */
  84.         $vehicleClient = $subject;
  86.         switch ($attribute) {
  87.             case self::ACCESS['company_view_summary']:
  88.                 //Admins and delegated users can view
  89.                 return $this->isDelegated($vehicleClient, $user) || $this->decisionManager->decide($token, ['ROLE_ADMIN'])|| $this->decisionManager->decide($token, ['ROLE_BENEFICIARY_REP']);
  90.             case self::ACCESS['company_edit_info']:
  91.             case self::ACCESS['company_docs']:
  92.                 return $this->decisionManager->decide($token, ['ROLE_ADMIN']); //only admins are allowed
  93.             case self::ACCESS['company_bank_accounts']:
  94.                 return $this->isDelegated($vehicleClient, $user) || $this->decisionManager->decide($token, ['ROLE_ADMIN']);
  95.         }
  97.         throw new \LogicException('This code should not be reached!');
  98.     }
  100.     /**
  101.      */
  102.     private function isDelegated(Companies $delegator, Users $user)
  103.     {
  104. //        $em = $this->entityManager;
  105. //        $delegations = $em->getRepository(ClientsDelegation::class)->findByDelegated($user->getUserId());
  106.         $delegations= $this->userHelper->getUserDelegators($user);
  107.         foreach($delegations as $delegation){
  108.             if($delegation->getDelegator() && 
  109.                     $delegation->getDelegator()->getCompany()  == $delegator){
  110.                 return true;
  111.             }
  112.         }
  113.         return false;
  114.     }
  116. }