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

Open in your IDE?
  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\Category\Tree\TreeItem;
  11. use Shopware\Core\Content\Cms\Aggregate\CmsSlot\CmsSlotEntity;
  12. use Shopware\Core\Content\Cms\DataResolver\CriteriaCollection;
  13. use Shopware\Core\Content\Cms\DataResolver\Element\AbstractCmsElementResolver;
  14. use Shopware\Core\Content\Cms\DataResolver\Element\ElementDataCollection;
  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;
  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.         $criteria = new Criteria($categoryIds);
  46.         $criteria->addAssociation('media');
  48.         $criteriaCollection = new CriteriaCollection();
  49.         $criteriaCollection->add('categories_' . $slot->getUniqueIdentifier(), CategoryDefinition::class, $criteria);
  51.         return $criteriaCollection;
  52.     }
  54.     public function enrich(CmsSlotEntity $slot, ResolverContext $resolverContext, ElementDataCollection $result): void
  55.     {
  56.         $categoryListing = new CategoryListingStruct();
  57.         $slot->setData($categoryListing);
  59.         $config = $slot->getFieldConfig();
  61.         $categories = $config->get('categories');
  62.         if ($categories->isMapped()) {
  63.             $categoryCollection = new CategoryCollection();
  64.             $categoryListing->setCategories($categoryCollection);;
  66.             if(empty($config->get('categories')->getValue())) {
  67.                 $this->addChildCategories($resolverContext, $categoryCollection);
  68.             } else {
  69.                 foreach ($categories->getValue() as $categoryId) {
  70.                     $this->addCategory($slot, $categoryCollection, $result, $categoryId);
  71.                 }
  72.             }
  74.         }
  76.         if($rowClass = $config->get('rowElementClassName')) {
  77.             $categoryListing->setRowElementClassName($rowClass->getValue());
  78.         }
  80.         if($colClass = $config->get('colElementClassName')) {
  81.             $categoryListing->setColElementClassName($colClass->getValue());
  82.         }
  84.         if($headingPosition = $config->get('headingPosition')) {
  85.             $categoryListing->setHeadingPosition($headingPosition->getValue());
  86.         }
  87.     }
  89.     private function addCategory(CmsSlotEntity $slot, CategoryCollection $categoryCollection, ElementDataCollection $result, string $configId): void
  90.     {
  91.         $searchResult = $result->get('categories_' . $slot->getUniqueIdentifier());
  92.         if (!$searchResult) {
  93.             return;
  94.         }
  96.         $category = $searchResult->get($configId);
  97.         if ($category instanceof CategoryEntity) {
  98.             $categoryCollection->add($category);
  99.         }
  100.     }
  102.     private function addChildCategories(ResolverContext $resolverContext, CategoryCollection $categoryCollection): void
  103.     {
  104.         if($resolverContext instanceof EntityResolverContext) {
  105.             $category = $resolverContext->getEntity();
  106.             if($category instanceof CategoryEntity) {
  107.                 $tree = $this->navigationLoader->load($category->getId(), $resolverContext->getSalesChannelContext(), $category->getId(), 1);
  108.                 foreach($tree->getTree() as $child) {
  109.                     $categoryCollection->add($child->getCategory());
  110.                 }
  111.             }
  112.         }
  113.     }
  114. }