app/Plugin/CustomerGroup/Security/EventSubscriber/LoginSubscriber.php line 59

Open in your IDE?
  1. <?php
  2. /**
  3.  * This file is part of CustomerGroup
  4.  *
  5.  * Copyright(c) Akira Kurozumi <info@a-zumi.net>
  6.  *
  7.  * https://a-zumi.net
  8.  *
  9.  * For the full copyright and license information, please view the LICENSE
  10.  * file that was distributed with this source code.
  11.  */
  13. namespace Plugin\CustomerGroup\Security\EventSubscriber;
  16. use Doctrine\ORM\EntityManagerInterface;
  17. use Eccube\Entity\Category;
  18. use Eccube\Entity\Customer;
  19. use Eccube\Entity\Product;
  20. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  21. use Symfony\Component\Routing\Generator\UrlGeneratorInterface;
  22. use Symfony\Component\Security\Http\Event\InteractiveLoginEvent;
  23. use Symfony\Component\Security\Http\SecurityEvents;
  25. /**
  26.  * Class LoginSubscriber
  27.  * @package Plugin\CustomerGroup\Security\EventSubscriber
  28.  *
  29.  *
  30.  */
  31. class LoginSubscriber implements EventSubscriberInterface
  32. {
  33.     /**
  34.      * @var UrlGeneratorInterface
  35.      */
  36.     private $urlGenerator;
  38.     /**
  39.      * @var EntityManagerInterface
  40.      */
  41.     private $entityManager;
  43.     public function __construct(
  44.         UrlGeneratorInterface $urlGenerator,
  45.         EntityManagerInterface $entityManager
  46.     )
  47.     {
  48.         $this->urlGenerator = $urlGenerator;
  49.         $this->entityManager = $entityManager;
  50.     }
  52.     public static function getSubscribedEvents()
  53.     {
  54.         return [
  55.             SecurityEvents::INTERACTIVE_LOGIN => 'onInteractiveLogin'
  56.         ];
  57.     }
  59.     public function onInteractiveLogin(InteractiveLoginEvent $event)
  60.     {
  61.         $user = $event->getAuthenticationToken()->getUser();
  63.         if (!$user instanceof Customer) {
  64.             return;
  65.         }
  67.         $event->getAuthenticationToken()->setAttribute('canViewProducts', array_map(function (Product $product) {
  68.             return $product->getId();
  69.         }, $user->getGroupProducts()->toArray()));
  71.         $event->getAuthenticationToken()->setAttribute('canViewCategories', array_map(function (Category $category) {
  72.             return $category->getId();
  73.         }, $user->getGroupCategories()->toArray()));
  74.     }
  75. }