custom/plugins/WbfkExtensions/src/CmsElementResolver/PartnerGridCmsElementResolver.php line 52

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace WbfkExtensions\CmsElementResolver;
  5. use Shopware\Core\Content\Cms\Aggregate\CmsSlot\CmsSlotEntity;
  6. use Shopware\Core\Content\Cms\DataResolver\CriteriaCollection;
  7. use Shopware\Core\Content\Cms\DataResolver\Element\AbstractCmsElementResolver;
  8. use Shopware\Core\Content\Cms\DataResolver\Element\ElementDataCollection;
  9. use Shopware\Core\Content\Cms\DataResolver\ResolverContext\ResolverContext;
  10. use Shopware\Core\Content\Media\MediaDefinition;
  11. use Shopware\Core\Framework\DataAbstractionLayer\Search\Criteria;
  13. class PartnerGridCmsElementResolver extends AbstractCmsElementResolver
  14. {
  15.     public function getType(): string
  16.     {
  17.         return 'partner-grid';
  18.     }
  20.     public function collect(CmsSlotEntity $slot, ResolverContext $resolverContext): ?CriteriaCollection
  21.     {
  23.         $partnerImageIds = [];
  24.         $partnerElements = $slot->getFieldConfig()->get('partners')?->getValue();
  26.         if (!isset($partnerElements)) {
  27.             return null;
  28.         }
  30.         foreach ($partnerElements as $element) {
  31.             if (isset($element['imageId'])) {
  32.                 if (is_array($element['imageId'])) {
  33.                     $partnerImageIds[] = $element['imageId']["id"];
  34.                 } else {
  35.                     $partnerImageIds[] = $element['imageId'];
  36.                 }
  37.             }
  38.         }
  40.         if (empty($partnerImageIds)) {
  41.             return null;
  42.         }
  44.         $criteria = new Criteria($partnerImageIds);
  46.         $criteriaCollection = new CriteriaCollection();
  47.         $criteriaCollection->add('media', MediaDefinition::class, $criteria);
  49.         return $criteriaCollection;
  50.     }
  52.     public function enrich(CmsSlotEntity $slot, ResolverContext $resolverContext, ElementDataCollection $result): void
  53.     {
  54.         $partners = [];
  55.         $partnerElements = $slot->getFieldConfig()->get('partners')?->getValue();
  57.         if (!isset($partnerElements)) {
  58.             return;
  59.         }
  61.         $partnerElementsCollection = $result->get('media')->getEntities()->getElements();
  63.         foreach ($partnerElements as $element) {
  64.             if (isset($element['imageId']) && isset($partnerElementsCollection[$element['imageId']])) {
  65.                 $partners[] = [
  66.                     'media' => $partnerElementsCollection[$element['imageId']],
  67.                     'element' => $element
  68.                 ];
  69.             }
  70.         }
  72.         $slot->assign([
  73.             'partners' => $partners
  74.         ]);
  75.     }
  76. }