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

Open in your IDE?
  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\AttributeType\DateAttributeType;
  21. use Sylius\Component\Attribute\AttributeType\DatetimeAttributeType;
  22. use Sylius\Component\Attribute\AttributeType\SelectAttributeType;
  23. use Sylius\Component\Attribute\Model\AttributeInterface;
  24. use Sylius\Component\Core\Model\ProductInterface;
  25. use Sylius\Component\Product\Model\ProductAttributeValue;
  27. final class AttributeBuilder extends AbstractBuilder
  28. {
  29.     public const DEFAULT_DATE_TIME_FORMAT = 'Y-m-d H:i:s';
  31.     public const DEFAULT_DATE_FORMAT = 'Y-m-d';
  33.     private ConcatedNameResolverInterface $attributeNameResolver;
  35.     private StringFormatterInterface $stringFormatter;
  37.     public function __construct(
  38.         ConcatedNameResolverInterface $attributeNameResolver,
  39.         StringFormatterInterface $stringFormatter
  40.     ) {
  41.         $this->attributeNameResolver = $attributeNameResolver;
  42.         $this->stringFormatter = $stringFormatter;
  43.     }
  45.     public function consumeEvent(PostTransformEvent $event): void
  46.     {
  47.         $this->buildProperty(
  48.             $event,
  49.             ProductInterface::class,
  50.             function (ProductInterface $product, Document $document): void {
  51.                 $this->resolveProductAttributes($product, $document);
  52.             }
  53.         );
  54.     }
  56.     private function resolveProductAttributes(ProductInterface $product, Document $document): void
  57.     {
  58.         foreach ($product->getAttributes() as $productAttribute) {
  59.             $attribute = $productAttribute->getAttribute();
  60.             if (!$attribute) {
  61.                 continue;
  62.             }
  64.             $this->processAttribute($attribute, $productAttribute, $document);
  65.         }
  66.     }
  68.     private function resolveProductAttribute(
  69.         array $attributeConfiguration,
  70.         $attributeValue,
  71.         ProductAttributeValue $productAttribute
  72.     ): array {
  73.         if (SelectAttributeType::TYPE === $productAttribute->getAttribute()->getType()) {
  74.             $choices = $attributeConfiguration['choices'];
  75.             if (is_array($attributeValue)) {
  76.                 foreach ($attributeValue as $i => $item) {
  77.                     $attributeValue[$i] = $choices[$item][$productAttribute->getLocaleCode()] ?? $item;
  78.                 }
  79.             } else {
  80.                 $attributeValue = $choices[$attributeValue][$productAttribute->getLocaleCode()] ?? $attributeValue;
  81.             }
  82.         }
  84.         $attributes = [];
  85.         if (is_array($attributeValue)) {
  86.             foreach ($attributeValue as $singleElement) {
  87.                 $attributes[] = $this->stringFormatter->formatToLowercaseWithoutSpaces((string) $singleElement);
  88.             }
  89.         } else {
  90.             switch (true) {
  91.                 case is_string($attributeValue):
  92.                     $attributes[] = $this->stringFormatter->formatToLowercaseWithoutSpaces($attributeValue);
  94.                     break;
  95.                 case $attributeValue instanceof \DateTime:
  96.                     $attributeFormat = $productAttribute->getAttribute()->getConfiguration()['format'] ?? null;
  97.                     $defaultFormat = DateAttributeType::TYPE === $productAttribute->getAttribute()->getStorageType() ?
  98.                         self::DEFAULT_DATE_FORMAT :
  99.                         self::DEFAULT_DATE_TIME_FORMAT
  100.                     ;
  102.                     $attributes[] = $attributeValue->format($attributeFormat ?? $defaultFormat);
  104.                     break;
  105.                 default:
  106.                     $attributes[] = $attributeValue;
  107.             }
  108.         }
  110.         return $attributes;
  111.     }
  113.     private function processAttribute(
  114.         AttributeInterface $attribute,
  115.         ProductAttributeValue $productAttribute,
  116.         Document $document
  117.     ): void {
  118.         $attributeCode = $attribute->getCode();
  119.         $attributeConfiguration = $attribute->getConfiguration();
  121.         $value = $productAttribute->getValue();
  122.         $documentKey = $this->attributeNameResolver->resolvePropertyName($attributeCode);
  123.         $code = sprintf('%s_%s', $documentKey, $productAttribute->getLocaleCode());
  125.         $values = $this->resolveProductAttribute(
  126.             $attributeConfiguration,
  127.             $value,
  128.             $productAttribute
  129.         );
  131.         $values = in_array($attribute->getStorageType(), [DateAttributeType::TYPE, DatetimeAttributeType::TYPE]) ?
  132.             ($values[0] ?? $values) :
  133.             $values
  134.         ;
  136.         $document->set($code, $values);
  137.     }
  138. }