vendor/shopware/core/Framework/Adapter/Twig/Extension/MediaExtension.php line 37

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace Shopware\Core\Framework\Adapter\Twig\Extension;
  5. use Shopware\Core\Content\Media\MediaCollection;
  6. use Shopware\Core\Framework\Context;
  7. use Shopware\Core\Framework\DataAbstractionLayer\EntityRepositoryInterface;
  8. use Shopware\Core\Framework\DataAbstractionLayer\Search\Criteria;
  9. use Twig\Extension\AbstractExtension;
  10. use Twig\TwigFunction;
  12. /**
  13.  * @package core
  14.  */
  15. class MediaExtension extends AbstractExtension
  16. {
  17.     /**
  18.      * @var EntityRepositoryInterface
  19.      */
  20.     private $mediaRepository;
  22.     /**
  23.      * @internal
  24.      */
  25.     public function __construct(EntityRepositoryInterface $mediaRepository)
  26.     {
  27.         $this->mediaRepository = $mediaRepository;
  28.     }
  30.     public function getFunctions(): array
  31.     {
  32.         return [
  33.             new TwigFunction('searchMedia', [$this, 'searchMedia']),
  34.         ];
  35.     }
  37.     public function searchMedia(array $ids, Context $context): MediaCollection
  38.     {
  39.         if (empty($ids)) {
  40.             return new MediaCollection();
  41.         }
  43.         $criteria = new Criteria($ids);
  45.         /** @var MediaCollection $media */
  46.         $media = $this->mediaRepository
  47.             ->search($criteria, $context)
  48.             ->getEntities();
  50.         return $media;
  51.     }
  52. }