vendor/bitbag/elasticsearch-plugin/src/PropertyBuilder/AttributeBuilder.php line 38

  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\Formatter\StringFormatterInterface;
  16. use BitBag\SyliusElasticsearchPlugin\PropertyNameResolver\ConcatedNameResolverInterface;
  17. use Elastica\Document;
  18. use FOS\ElasticaBundle\Event\PostTransformEvent;
  19. use function sprintf;
  20. use Sylius\Component\Attribute\Model\AttributeInterface;
  21. use Sylius\Component\Core\Model\ProductInterface;
  22. use Sylius\Component\Product\Model\ProductAttributeValue;
  24. final class AttributeBuilder extends AbstractBuilder
  25. {
  26.     private ConcatedNameResolverInterface $attributeNameResolver;
  28.     private StringFormatterInterface $stringFormatter;
  30.     public function __construct(
  31.         ConcatedNameResolverInterface $attributeNameResolver,
  32.         StringFormatterInterface $stringFormatter
  33.     ) {
  34.         $this->attributeNameResolver = $attributeNameResolver;
  35.         $this->stringFormatter = $stringFormatter;
  36.     }
  38.     public function consumeEvent(PostTransformEvent $event): void
  39.     {
  40.         $this->buildProperty(
  41.             $event,
  42.             ProductInterface::class,
  43.             function (ProductInterface $product, Document $document): void {
  44.                 $this->resolveProductAttributes($product, $document);
  45.             }
  46.         );
  47.     }
  49.     private function resolveProductAttributes(ProductInterface $product, Document $document): void
  50.     {
  51.         foreach ($product->getAttributes() as $productAttribute) {
  52.             $attribute = $productAttribute->getAttribute();
  53.             if (!$attribute) {
  54.                 continue;
  55.             }
  57.             $this->processAttribute($attribute, $productAttribute, $document);
  58.         }
  59.     }
  61.     private function resolveProductAttribute(
  62.         array $attributeConfiguration,
  63.         $attributeValue,
  64.         ProductAttributeValue $productAttribute
  65.     ): array {
  66.         if ('select' === $productAttribute->getAttribute()->getType()) {
  67.             $choices = $attributeConfiguration['choices'];
  68.             if (is_array($attributeValue)) {
  69.                 foreach ($attributeValue as $i => $item) {
  70.                     $attributeValue[$i] = $choices[$item][$productAttribute->getLocaleCode()] ?? $item;
  71.                 }
  72.             } else {
  73.                 $attributeValue = $choices[$attributeValue][$productAttribute->getLocaleCode()] ?? $attributeValue;
  74.             }
  75.         }
  77.         $attributes = [];
  78.         if (is_array($attributeValue)) {
  79.             foreach ($attributeValue as $singleElement) {
  80.                 $attributes[] = $this->stringFormatter->formatToLowercaseWithoutSpaces((string) $singleElement);
  81.             }
  82.         } else {
  83.             $attributeValue = is_string($attributeValue)
  84.                 ? $this->stringFormatter->formatToLowercaseWithoutSpaces($attributeValue) : $attributeValue;
  85.             $attributes[] = $attributeValue;
  86.         }
  88.         return $attributes;
  89.     }
  91.     private function processAttribute(
  92.         AttributeInterface $attribute,
  93.         ProductAttributeValue $productAttribute,
  94.         Document $document
  95.     ): void {
  96.         $attributeCode = $attribute->getCode();
  97.         $attributeConfiguration = $attribute->getConfiguration();
  99.         $value = $productAttribute->getValue();
  100.         $documentKey = $this->attributeNameResolver->resolvePropertyName($attributeCode);
  101.         $code = sprintf('%s_%s', $documentKey, $productAttribute->getLocaleCode());
  103.         $values = $this->resolveProductAttribute(
  104.             $attributeConfiguration,
  105.             $value,
  106.             $productAttribute
  107.         );
  109.         $document->set($code, $values);
  110.     }
  111. }