custom/plugins/HemoCustomerGroupRestriction/src/Subscriber/HttpCacheCustomerGroupSubscriber.php line 30

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace Hemo\CustomerGroupRestriction\Subscriber;
  5. use Shopware\Core\PlatformRequest;
  6. use Shopware\Core\System\SalesChannel\SalesChannelContext;
  7. use Shopware\Storefront\Framework\Cache\Event\HttpCacheGenerateKeyEvent;
  8. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  9. use Symfony\Component\HttpFoundation\Cookie;
  10. use Symfony\Component\HttpKernel\Event\ResponseEvent;
  11. use Symfony\Component\HttpKernel\KernelEvents;
  13. /**
  14.  * The storefront HTTP cache key only contains rule ids, currency, version and the logged-in
  15.  * state. Two logged-in customers from different customer groups would therefore share cached
  16.  * pages. This subscriber stores the customer group in a cookie and mixes it into the cache key.
  17.  */
  18. final class HttpCacheCustomerGroupSubscriber implements EventSubscriberInterface
  19. {
  20.     public const COOKIE_NAME = 'hemo-customer-group';
  22.     public static function getSubscribedEvents(): array
  23.     {
  24.         return [
  25.             KernelEvents::RESPONSE => 'onResponse',
  26.             HttpCacheGenerateKeyEvent::class => 'onGenerateKey',
  27.         ];
  28.     }
  30.     public function onResponse(ResponseEvent $event): void
  31.     {
  32.         if (!$event->isMainRequest()) {
  33.             return;
  34.         }
  36.         $request = $event->getRequest();
  37.         $scopes = $request->attributes->get(PlatformRequest::ATTRIBUTE_ROUTE_SCOPE, []);
  39.         if (!\is_array($scopes) || !\in_array('storefront', $scopes, true)) {
  40.             return;
  41.         }
  43.         $context = $request->attributes->get(PlatformRequest::ATTRIBUTE_SALES_CHANNEL_CONTEXT_OBJECT);
  45.         if (!$context instanceof SalesChannelContext) {
  46.             return;
  47.         }
  49.         $customerGroupId = $context->getCurrentCustomerGroup()->getId();
  51.         if ($request->cookies->get(self::COOKIE_NAME) === $customerGroupId) {
  52.             return;
  53.         }
  55.         $cookie = Cookie::create(self::COOKIE_NAME, $customerGroupId);
  56.         $cookie->setSecureDefault($request->isSecure());
  58.         $event->getResponse()->headers->setCookie($cookie);
  59.     }
  61.     public function onGenerateKey(HttpCacheGenerateKeyEvent $event): void
  62.     {
  63.         $customerGroupId = $event->getRequest()->cookies->get(self::COOKIE_NAME);
  65.         if (!\is_string($customerGroupId) || $customerGroupId === '') {
  66.             return;
  67.         }
  69.         $event->setHash(hash('sha256', $event->getHash() . '-' . $customerGroupId));
  70.     }
  71. }