custom/plugins/JkwebShopwareCategoryListingPlugin/src/DataResolver/CategoryListingDataResolver.php line 74

  1. <?php declare(strict_types=1);
  3. namespace Jkweb\Shopware\Plugin\CategoryListing\DataResolver;
  5. use Jkweb\Shopware\Plugin\CategoryListing\Struct\CategoryListingStruct;
  6. use Shopware\Core\Content\Category\CategoryCollection;
  7. use Shopware\Core\Content\Category\CategoryDefinition;
  8. use Shopware\Core\Content\Category\CategoryEntity;
  9. use Shopware\Core\Content\Category\Service\NavigationLoader;
  10. use Shopware\Core\Content\Cms\Aggregate\CmsSlot\CmsSlotEntity;
  11. use Shopware\Core\Content\Cms\DataResolver\CriteriaCollection;
  12. use Shopware\Core\Content\Cms\DataResolver\Element\AbstractCmsElementResolver;
  13. use Shopware\Core\Content\Cms\DataResolver\Element\ElementDataCollection;
  14. use Shopware\Core\Content\Cms\DataResolver\FieldConfigCollection;
  15. use Shopware\Core\Content\Cms\DataResolver\ResolverContext\EntityResolverContext;
  16. use Shopware\Core\Content\Cms\DataResolver\ResolverContext\ResolverContext;
  17. use Shopware\Core\Framework\DataAbstractionLayer\Search\Criteria;
  19. class CategoryListingDataResolver extends AbstractCmsElementResolver
  20. {
  21.     private NavigationLoader $navigationLoader;
  23.     public function __construct(NavigationLoader $navigationLoader)
  24.     {
  26.         $this->navigationLoader = $navigationLoader;
  27.     }
  29.     public function getType(): string
  30.     {
  31.         return 'category-listing';
  32.     }
  34.     public function collect(CmsSlotEntity $slot, ResolverContext $resolverContext): ?CriteriaCollection
  35.     {
  36.         $config = $slot->getFieldConfig();
  37.         $categoriesConfig = $config->get('categories');
  39.         if (!$categoriesConfig || $categoriesConfig->isStatic()) {
  40.             return null;
  41.         }
  43.         $categoryIds = $categoriesConfig->getValue();
  45.         if (!$categoryIds) {
  46.             return null;
  47.         }
  49.         $criteria = new Criteria($categoryIds);
  50.         $criteria->addAssociation('media');
  52.         $criteriaCollection = new CriteriaCollection();
  53.         $criteriaCollection->add('categories_' . $slot->getUniqueIdentifier(), CategoryDefinition::class, $criteria);
  55.         return $criteriaCollection;
  56.     }
  58.     public function enrich(CmsSlotEntity $slot, ResolverContext $resolverContext, ElementDataCollection $result): void
  59.     {
  60.         $categoryListing = new CategoryListingStruct();
  61.         $slot->setData($categoryListing);
  63.         /** @var FieldConfigCollection $config */
  64.         $config = $slot->getFieldConfig();
  66.         $categories = $config->get('categories');
  67.         if ($categories->isMapped()) {
  68.             $categoryCollection = new CategoryCollection();
  69.             $categoryListing->setCategories($categoryCollection);;
  72.             $categoryIds = $categories->getValue();
  73.             if (!$categoryIds) {
  74.                 $this->addChildCategories($resolverContext, $categoryCollection);
  75.             } else {
  76.                 foreach ($categoryIds as $categoryId) {
  77.                     $this->addCategory($slot, $categoryCollection, $result, $categoryId);
  78.                 }
  79.             }
  81.         }
  83.         if($rowClass = $config->get('rowElementClassName')) {
  84.             $categoryListing->setRowElementClassName($rowClass->getValue());
  85.         }
  87.         if($colClass = $config->get('colElementClassName')) {
  88.             $categoryListing->setColElementClassName($colClass->getValue());
  89.         }
  91.         if($headingPosition = $config->get('headingPosition')) {
  92.             $categoryListing->setHeadingPosition($headingPosition->getValue());
  93.         }
  94.     }
  96.     private function addCategory(CmsSlotEntity $slot, CategoryCollection $categoryCollection, ElementDataCollection $result, string $configId): void
  97.     {
  98.         $searchResult = $result->get('categories_' . $slot->getUniqueIdentifier());
  99.         if (!$searchResult) {
  100.             return;
  101.         }
  103.         $category = $searchResult->get($configId);
  104.         if ($category instanceof CategoryEntity) {
  105.             $categoryCollection->add($category);
  106.         }
  107.     }
  109.     private function addChildCategories(ResolverContext $resolverContext, CategoryCollection $categoryCollection): void
  110.     {
  111.         if($resolverContext instanceof EntityResolverContext) {
  112.             $category = $resolverContext->getEntity();
  113.             if($category instanceof CategoryEntity) {
  114.                 $tree = $this->navigationLoader->load($category->getId(), $resolverContext->getSalesChannelContext(), $category->getId(), 1);
  115.                 foreach($tree->getTree() as $child) {
  116.                     $categoryCollection->add($child->getCategory());
  117.                 }
  118.             }
  119.         }
  120.     }
  121. }