src/Voter/OrderVoter.php line 12

Open in your IDE?
  1. <?php
  3. namespace App\Voter;
  5. use App\Entity\Order;
  6. use App\Model\OrderFactory;
  7. use App\Model\User\User;
  8. use LogicException;
  9. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  10. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  12. class OrderVoter extends Voter
  13. {
  14.     public const VIEW = 'view';
  16.     public const BO_VIEW = 'bo_view';
  18.     public const EDIT = 'edit';
  20.     public const CANCEL = 'cancel';
  22.     public const BUY = 'buy';
  24.     private $orderFactory;
  26.     private $user;
  29.     public function __construct(OrderFactory $orderFactory, User $user)
  30.     {
  31.         $this->orderFactory = $orderFactory;
  32.         $this->user         = $user;
  34.     }
  36.     protected function supports($attribute, $orderEntity): bool
  37.     {
  40.         // if the attribute isn't one we support, return false
  41.         if (! in_array($attribute, [
  42.             self::VIEW,
  43.             self::BO_VIEW,
  44.             self::EDIT,
  45.             self::CANCEL,
  46.             self::BUY,
  47.         ])
  48.         ) {
  49.             return false;
  50.         }
  53.         if (! $orderEntity instanceof Order) {
  54.             return false;
  55.         }
  57.         return true;
  58.     }
  61.     protected function voteOnAttribute($attribute, $orderEntity, TokenInterface $token): bool
  62.     {
  64.         $roles      = [];
  65.         $userEntity = $token->getUser();
  66.         if ($userEntity != 'anon.') {
  67.             $this->user->setEntity($userEntity);
  68.             $roles = $userEntity->getRoles();
  69.         }
  72.         $order = $this->orderFactory->createModel();
  73.         $order->setEntity($orderEntity);
  76.         if (in_array('ROLE_ADMIN', $roles)) {
  77.             return true;
  78.         }
  82.         switch ($attribute) {
  83.             case self::VIEW:
  84.                 return $order->canBeViewedBy($this->user);
  85.             case self::EDIT:
  86.                 return $order->canBeEditedBy($this->user);
  87.             case self::BO_VIEW:
  88.                 return $order->canBeBoViewedBy($this->user);
  89.             case self::CANCEL:
  90.                 return $order->canBeCancelledBy($this->user);
  91.         }
  93.         throw new LogicException('This code should not be reached!');
  94.     }
  97. }