vendor/bitbag/elasticsearch-plugin/src/PropertyBuilder/AttributeTaxonsBuilder.php line 43

  1. <?php
  3. /*
  4.  * This file has been created by developers from BitBag.
  5.  * Feel free to contact us once you face any issues or want to start
  6.  * another great project.
  7.  * You can find more information about us on https://bitbag.io and write us
  8.  * an email on hello@bitbag.io.
  9.  */
  11. declare(strict_types=1);
  13. namespace BitBag\SyliusElasticsearchPlugin\PropertyBuilder;
  15. use BitBag\SyliusElasticsearchPlugin\Repository\TaxonRepositoryInterface;
  16. use FOS\ElasticaBundle\Event\PostTransformEvent;
  17. use Sylius\Component\Attribute\Model\AttributeInterface;
  18. use Sylius\Component\Core\Model\TaxonInterface;
  19. use Sylius\Component\Product\Model\ProductAttributeInterface;
  21. final class AttributeTaxonsBuilder extends AbstractBuilder
  22. {
  23.     protected TaxonRepositoryInterface $taxonRepository;
  25.     private string $taxonsProperty;
  27.     private array $excludedAttributes;
  29.     private bool $includeAllDescendants;
  31.     public function __construct(
  32.         TaxonRepositoryInterface $taxonRepository,
  33.         string $taxonsProperty,
  34.         bool $includeAllDescendants,
  35.         array $excludedAttributes = []
  36.     ) {
  37.         $this->taxonRepository = $taxonRepository;
  38.         $this->taxonsProperty = $taxonsProperty;
  39.         $this->includeAllDescendants = $includeAllDescendants;
  40.         $this->excludedAttributes = $excludedAttributes;
  41.     }
  43.     public function consumeEvent(PostTransformEvent $event): void
  44.     {
  45.         $documentAttribute = $event->getObject();
  47.         if (!$documentAttribute instanceof AttributeInterface
  48.             || !$documentAttribute instanceof ProductAttributeInterface
  49.             || in_array($documentAttribute->getCode(), $this->excludedAttributes)
  50.         ) {
  51.             return;
  52.         }
  54.         $taxons = $this->taxonRepository->getTaxonsByAttributeViaProduct($documentAttribute);
  55.         $taxonCodes = [];
  57.         /** @var TaxonInterface $taxon */
  58.         foreach ($taxons as $taxon) {
  59.             $taxonCodes[] = $taxon->getCode();
  61.             if (true === $this->includeAllDescendants) {
  62.                 foreach ($taxon->getAncestors() as $ancestor) {
  63.                     $taxonCodes[] = $ancestor->getCode();
  64.                 }
  65.             }
  66.         }
  68.         $document = $event->getDocument();
  70.         $document->set($this->taxonsProperty, $taxonCodes);
  71.     }
  72. }