custom/plugins/AcrisFilterCS/src/Storefront/Subscriber/PropertyOptionCountSubscriber.php line 44

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace Acris\Filter\Storefront\Subscriber;
  5. use Acris\Filter\Components\FilterCount\Struct\TermCount;
  6. use Acris\Filter\Components\FilterCount\Struct\TermCountCollection;
  7. use Acris\Filter\Components\FilterCount\Struct\AggregationDocCount;
  8. use Shopware\Core\Content\Product\Events\ProductListingResultEvent;
  9. use Shopware\Core\Content\Product\Events\ProductSearchResultEvent;
  10. use Shopware\Core\Content\Property\PropertyGroupEntity;
  11. use Shopware\Core\Framework\DataAbstractionLayer\Search\AggregationResult\Bucket\TermsResult;
  12. use Shopware\Core\Framework\DataAbstractionLayer\Search\AggregationResult\Metric\EntityResult;
  13. use Shopware\Core\System\SystemConfig\SystemConfigService;
  14. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  16. class PropertyOptionCountSubscriber implements EventSubscriberInterface
  17. {
  18.     private SystemConfigService $configService;
  20.     public function __construct(SystemConfigService $configService)
  21.     {
  22.         $this->configService = $configService;
  23.     }
  25.     /**
  26.      * @var TermCountCollection
  27.      */
  28.     private TermCountCollection $termCountCollection;
  30.     public static function getSubscribedEvents()
  31.     {
  32.         return [
  33.             ProductListingResultEvent::class => [
  34.                 ['handleSearchResultBefore', 150],
  35.                 ['handleSearchResultAfter', -150]
  36.             ],
  37.             ProductSearchResultEvent::class => [
  38.                 ['handleSearchResultBefore', 150],
  39.                 ['handleSearchResultAfter', -150]
  40.             ]
  41.         ];
  42.     }
  44.     public function handleSearchResultBefore(ProductListingResultEvent $event): void
  45.     {
  46.         if(!$this->configService->get('AcrisFilterCS.config.filterCount')) {
  47.             return;
  48.         }
  50.         $aggregations = $event->getResult()->getAggregations();
  52.         $this->termCountCollection = new TermCountCollection();
  54.         /** @var TermsResult|null $properties */
  55.         $properties = $aggregations->get('properties');
  56.         $this->collectTermCount($this->termCountCollection, $properties);
  58.         /** @var TermsResult|null $options */
  59.         $options = $aggregations->get('options');
  60.         $this->collectTermCount($this->termCountCollection, $options);
  61.     }
  63.     public function handleSearchResultAfter(ProductListingResultEvent $event): void
  64.     {
  65.         if(!$this->configService->get('AcrisFilterCS.config.filterCount')) {
  66.             return;
  67.         }
  69.         $aggregations = $event->getResult()->getAggregations();
  70.         /** @var EntityResult|null $propertyGroups */
  71.         $propertyGroups = $aggregations->get('properties');
  73.         if(!$propertyGroups instanceof EntityResult) {
  74.             return;
  75.         }
  77.         /** @var PropertyGroupEntity $propertyGroup */
  78.         foreach ($propertyGroups->getEntities() as $propertyGroup) {
  79.             foreach ($propertyGroup->getOptions() as $option) {
  80.                 if($this->termCountCollection->has($option->getId()) === true) {
  81.                     $option->addExtension(AggregationDocCount::AGGREGATION_DOC_COUNT, new AggregationDocCount($this->termCountCollection->get($option->getId())->getCount()));
  82.                 }
  83.             }
  84.         }
  85.     }
  87.     private function collectTermCount(TermCountCollection $termCountCollection, ?TermsResult $termsResult): void
  88.     {
  89.         if(!$termsResult instanceof TermsResult) {
  90.             return;
  91.         }
  92.         foreach ($termsResult->getBuckets() as $bucket) {
  93.             if($termCountCollection->has($bucket->getKey()) === true) {
  94.                 $termCount = $termCountCollection->get($bucket->getKey());
  95.             } else {
  96.                 $termCount = new TermCount($bucket->getKey(), 0);
  97.                 $termCountCollection->set($bucket->getKey(), $termCount);
  98.             }
  99.             $termCount->setCount($termCount->getCount() + $bucket->getCount());
  100.         }
  101.     }
  102. }