src/Voter/ShopVoter.php line 11

Open in your IDE?
  1. <?php
  3. namespace App\Voter;
  5. use App\Entity\Customer;
  6. use App\Model\User\User;
  7. use LogicException;
  8. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  9. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  11. class ShopVoter extends Voter
  12. {
  13.     public const SWITCH_SHOP = 'switchShop';
  15.     public const BO_VIEW = 'bo_view';
  17.     public const VIEW = 'view';
  19.     public const EDIT = 'edit';
  21.     public const REMOVE_MEMBER = 'remove_member';
  23.     public const NEW = 'new';
  25.     /** @var User */
  26.     private $user;
  28.     public function __construct(User $user)
  29.     {
  30.         $this->user = $user;
  31.     }
  33.     protected function supports($attribute, $shopEntity): bool
  34.     {
  35.         // if the attribute isn't one we support, return false
  36.         if (! in_array($attribute, [
  37.             self::SWITCH_SHOP,
  38.             self::BO_VIEW,
  39.             self::VIEW,
  40.             self::EDIT,
  41.             self::NEW,
  42.             self::REMOVE_MEMBER
  43.         ])
  44.         ) {
  45.             return false;
  46.         }
  49.         if (! $shopEntity instanceof Customer) {
  50.             return false;
  51.         }
  53.         return true;
  54.     }
  57.     protected function voteOnAttribute($attribute, $shopEntity, TokenInterface $token): bool
  58.     {
  61.         $userEntity = $token->getUser();
  62.         $this->user->setEntity($userEntity);
  64.         // Prestachef staff can do all of it
  65.         if ($this->user->isPrestachefStaff()) {
  66.             return true;
  67.         }
  69.         $shop = $this->user->getService('shopFactory')->createModel();
  70.         $shop->setEntity($shopEntity);
  73.         switch ($attribute) {
  74.             case self::VIEW:
  75.                 return $this->user->canSeeShop($shop);
  76.             case self::EDIT:
  77.                 return $this->user->canEditShop($shop);
  78.             case self::REMOVE_MEMBER:
  79.                 return $this->user->canRemoveTeamMemberFromShop($shop);
  80.             case self::NEW:
  81.                 return $this->user->canCreateShop();
  82.             case self::SWITCH_SHOP:
  83.                 return $this->user->canSwitchToShop($shop);
  84.             case self::BO_VIEW:
  85.                 // everybody see all customers
  86.                 return $this->user->isPrestachefStaff();
  87.         }
  89.         throw new LogicException('This code should not be reached!');
  90.     }
  93. }