src/Twig/Extension/CategoryExtension.php line 55

Open in your IDE?
  1. <?php
  3. /**
  4.  * Pimcore
  5.  *
  6.  * This source file is available under two different licenses:
  7.  * - GNU General Public License version 3 (GPLv3)
  8.  * - Pimcore Enterprise License (PEL)
  9.  * Full copyright and license information is available in
  10.  * LICENSE.md which is distributed with this source code.
  11.  *
  12.  *  @copyright  Copyright (c) Pimcore GmbH (http://www.pimcore.org)
  13.  *  @license    http://www.pimcore.org/license     GPLv3 and PEL
  14.  */
  16. namespace App\Twig\Extension;
  18. use App\Model\Product\Category;
  19. use Pimcore\Model\DataObject\Folder;
  20. use App\Website\LinkGenerator\CategoryLinkGenerator;
  21. use Twig\Extension\AbstractExtension;
  22. use Twig\TwigFunction;
  24. class CategoryExtension extends AbstractExtension
  25. {
  26.     /**
  27.      * @var CategoryLinkGenerator
  28.      */
  29.     protected $categoryLinkGenerator;
  31.     /**
  32.      * CategoryFilterExtension constructor.
  33.      *
  34.      * @param CategoryLinkGenerator $categoryLinkGenerator
  35.      */
  36.     public function __construct(CategoryLinkGenerator $categoryLinkGenerator)
  37.     {
  38.         $this->categoryLinkGenerator = $categoryLinkGenerator;
  39.     }
  41.     /**
  42.      * @return TwigFunction[]
  43.      */
  44.     public function getFunctions()
  45.     {
  46.         return [
  47.             new TwigFunction('app_get_main_categories', [$this, 'prepareData']),
  48.         ];
  49.     }
  51.     /**
  52.      * @param Document $document
  53.      * @return \stdClass
  54.      */
  55.     public function prepareData($document = null)
  56.     {
  57.         $root = $document->getProperty('products_root_category');
  59.         $categories = [];
  61.         $categories = array_filter($root->getChildren(), function ($item) {
  62.             return $item instanceof Category && $item->isPublished();
  63.         });
  65.         usort($categories, function($a, $b) {return $a->getId() >= $b->getId();});
  67.         return $categories;
  68.     }
  69. }