custom/plugins/ChespackTheme/src/DataResolver/CategoryListingDataResolver.php line 44

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace ChespackTheme\DataResolver;
  5. use ChespackTheme\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\ResolverContext\EntityResolverContext;
  15. use Shopware\Core\Content\Cms\DataResolver\ResolverContext\ResolverContext;
  16. use Shopware\Core\Framework\DataAbstractionLayer\Search\Criteria;
  18. class CategoryListingDataResolver extends AbstractCmsElementResolver
  19. {
  20.     private $navigationLoader;
  22.     public function __construct(NavigationLoader $navigationLoader)
  23.     {
  25.         $this->navigationLoader = $navigationLoader;
  26.     }
  28.     public function getType(): string
  29.     {
  30.         return 'category-listing';
  31.     }
  33.     public function collect(CmsSlotEntity $slot, ResolverContext $resolverContext): ?CriteriaCollection
  34.     {
  35.         $config = $slot->getFieldConfig();
  36.         $categoriesConfig = $config->get('categories');
  38.         if (!$categoriesConfig || $categoriesConfig->isStatic()) {
  39.             return null;
  40.         }
  42.         $categoryIds = $categoriesConfig->getValue();
  44.         $criteria = new Criteria($categoryIds);
  45.         $criteria->addAssociation('media');
  47.         $criteriaCollection = new CriteriaCollection();
  48.         $criteriaCollection->add('categories_' . $slot->getUniqueIdentifier(), CategoryDefinition::class, $criteria);
  50.         return $criteriaCollection;
  51.     }
  53.     public function enrich(CmsSlotEntity $slot, ResolverContext $resolverContext, ElementDataCollection $result): void
  54.     {
  55.         $categoryListing = new CategoryListingStruct();
  56.         $slot->setData($categoryListing);
  58.         $config = $slot->getFieldConfig();
  60.         $categories = $config->get('categories');
  61.         if ($categories->isMapped()) {
  62.             $categoryCollection = new CategoryCollection();
  63.             $categoryListing->setCategories($categoryCollection);;
  65.             if(empty($config->get('categories')->getValue())) {
  66.                 $this->addChildCategories($resolverContext, $categoryCollection);
  67.             } else {
  68.                 foreach ($categories->getValue() as $categoryId) {
  69.                     $this->addCategory($slot, $categoryCollection, $result, $categoryId);
  70.                 }
  71.             }
  73.         }
  75.         if($rowClass = $config->get('rowElementClassName')) {
  76.             $categoryListing->setRowElementClassName($rowClass->getValue());
  77.         }
  79.         if($colClass = $config->get('colElementClassName')) {
  80.             $categoryListing->setColElementClassName($colClass->getValue());
  81.         }
  83.         if($headingPosition = $config->get('headingPosition')) {
  84.             $categoryListing->setHeadingPosition($headingPosition->getValue());
  85.         }
  86.     }
  88.     private function addCategory(CmsSlotEntity $slot, CategoryCollection $categoryCollection, ElementDataCollection $result, string $configId): void
  89.     {
  90.         $searchResult = $result->get('categories_' . $slot->getUniqueIdentifier());
  91.         if (!$searchResult) {
  92.             return;
  93.         }
  95.         $category = $searchResult->get($configId);
  96.         if ($category instanceof CategoryEntity) {
  97.             $categoryCollection->add($category);
  98.         }
  99.     }
  101.     private function addChildCategories(ResolverContext $resolverContext, CategoryCollection $categoryCollection): void
  102.     {
  103.         if($resolverContext instanceof EntityResolverContext) {
  104.             $category = $resolverContext->getEntity();
  105.             if($category instanceof CategoryEntity) {
  106.                 $tree = $this->navigationLoader->load($category->getId(), $resolverContext->getSalesChannelContext(), $category->getId(), 1);
  107.                 foreach($tree->getTree() as $child) {
  108.                     $categoryCollection->add($child->getCategory());
  109.                 }
  110.             }
  111.         }
  112.     }
  113. }