custom/plugins/CioSalesChannelSwitcher/src/Subscriber/HeaderSalesChannelSelectorSubscriber.php line 83

Open in your IDE?
  1. <?php
  3. namespace CioSalesChannelSwitcher\Subscriber;
  5. use CioCustomerPermissionGroups\Service\CustomerPermissionService;
  6. use CioSalesChannelSwitcher\Service\SwitchSalesChannelService;
  7. use Shopware\Core\Checkout\Customer\CustomerEntity;
  8. use Shopware\Core\Checkout\Customer\Event\CustomerLoginEvent;
  9. use Shopware\Core\Checkout\Customer\Event\CustomerLogoutEvent;
  10. use Shopware\Core\Framework\Context;
  11. use Shopware\Core\Framework\DataAbstractionLayer\EntityRepositoryInterface;
  12. use Shopware\Core\Framework\DataAbstractionLayer\Search\Criteria;
  13. use Shopware\Core\Framework\DataAbstractionLayer\Search\Filter\EqualsFilter;
  14. use Shopware\Core\Framework\DataAbstractionLayer\Search\Filter\NotFilter;
  15. use Shopware\Core\System\SalesChannel\SalesChannelEntity;
  16. use Shopware\Storefront\Event\StorefrontRenderEvent;
  17. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  18. use Symfony\Component\HttpFoundation\RedirectResponse;
  19. use Symfony\Component\HttpFoundation\Session\Session;
  21. class HeaderSalesChannelSelectorSubscriber implements EventSubscriberInterface
  22. {
  23.     private EntityRepositoryInterface $salesChannelRepository;
  24.     private CustomerPermissionService $customerPermissionService;
  25.     private SwitchSalesChannelService $switchSalesChannelService;
  26.     private Session $session;
  28.     public function __construct(EntityRepositoryInterface $salesChannelRepository, CustomerPermissionService $customerPermissionService, SwitchSalesChannelService $switchSalesChannelService, Session $session)
  29.     {
  30.         $this->salesChannelRepository = $salesChannelRepository;
  31.         $this->customerPermissionService = $customerPermissionService;
  32.         $this->switchSalesChannelService = $switchSalesChannelService;
  33.         $this->session = $session;
  34.     }
  36.     public static function getSubscribedEvents(): array
  37.     {
  38.         return [
  39.             StorefrontRenderEvent::class => 'onStorefrontRender',
  40.             CustomerLoginEvent::class => 'onCustomerLoginEvent',
  41.             CustomerLogoutEvent::class => 'onCustomerLogoutEvent'
  42.         ];
  43.     }
  45.     public function onStorefrontRender(StorefrontRenderEvent $event)
  46.     {
  47.         $customer = $event->getSalesChannelContext()->getCustomer();
  49.         if ($customer) {
  50.             $salesChannels = $this->getAllowedSalesChannels($customer, $event->getSalesChannelContext()->getContext());
  52.             $event->setParameter('salesChannels', $salesChannels);
  53.             $event->setParameter('currentSalesChannelId', $event->getSalesChannelContext()->getSalesChannel()->getId());
  54.         }
  55.     }
  57.     public function onCustomerLoginEvent(CustomerLoginEvent $event)
  58.     {
  59.         /** @var string[] $allowedSalesChannelIds */
  60.         $allowedSalesChannelIds = array_map(function (SalesChannelEntity $salesChannelEntity) {
  61.             return $salesChannelEntity->getId();
  62.         }, $this->getAllowedSalesChannels($event->getCustomer(), $event->getSalesChannelContext()->getContext()));
  64.         if ($this->session->has('cio_master_login_current') && $this->session->has('cio_master_login_sales_channel_id')) {
  65.             $this->session->remove('cio_master_login_current');
  66.             $this->session->save();
  67.             (new RedirectResponse($this->switchSalesChannelService->getSalesChannelDomainEntityAndLogIn($this->session->get('cio_master_login_sales_channel_id'), $event->getSalesChannelId(), $event->getSalesChannelContext())))->send();
  68.         }
  70.         // user can't access the current sales channel
  71.         if (!in_array($event->getSalesChannelId(), $allowedSalesChannelIds)) {
  73.             // if user has access to other sales channels, then login to the first allowed
  74.             if (count($allowedSalesChannelIds) > 0) {
  75.                 if (!$this->session->get('ignore_sales_channel_switch_during_login')) {
  76.                     (new RedirectResponse($this->switchSalesChannelService->getSalesChannelDomainEntityAndLogIn($allowedSalesChannelIds[array_key_first($allowedSalesChannelIds)], $event->getSalesChannelId(), $event->getSalesChannelContext())))->send();
  77.                 }
  78.             }
  80.         }
  81.     }
  83.     public function onCustomerLogoutEvent(CustomerLogoutEvent $event)
  84.     {
  85.         $this->session->set('ignore_sales_channel_switch_during_login', false);
  86.     }
  88.     protected function getAllowedSalesChannels(CustomerEntity $customer, Context $context): array
  89.     {
  90.         $salesChannels = [];
  91.         $salesChannelEntities = $this->salesChannelRepository
  92.             ->search((new Criteria())->addFilter(new NotFilter(NotFilter::CONNECTION_AND, [new EqualsFilter('name', 'Headless')])), $context)
  93.             ->getEntities();
  95.         foreach ($salesChannelEntities as $salesChannelEntity) {
  96.             /** @var $salesChannelEntity SalesChannelEntity */
  98.             foreach ($this->customerPermissionService->getCustomerPermissionGroups($customer, $context) as $customerPermissionGroup) {
  99.                 if ($customerPermissionGroup->getName() == $salesChannelEntity->getName()) {
  100.                     $salesChannels[] = $salesChannelEntity;
  101.                 }
  102.             }
  103.         }
  105.         return $salesChannels;
  106.     }
  107. }