Media.php

Go to the documentation of this file.

<?php
namespace Magento\Swatches\Controller\Ajax;

use Magento\Framework\App\Action\Context;
use Magento\Framework\Controller\ResultFactory;

class Media extends \Magento\Framework\App\Action\Action
{
    protected $productModelFactory;

    private $swatchHelper;

    public function __construct(
        Context $context,
        \Magento\Catalog\Model\ProductFactory $productModelFactory,
        \Magento\Swatches\Helper\Data $swatchHelper
    ) {
        $this->productModelFactory = $productModelFactory;
        $this->swatchHelper = $swatchHelper;

        parent::__construct($context);
    }

    public function execute()
    {
        $productMedia = [];
        if ($productId = (int)$this->getRequest()->getParam('product_id')) {
            $productMedia = $this->swatchHelper->getProductMediaGallery(
                $this->productModelFactory->create()->load($productId)
            );
        }

        $resultJson = $this->resultFactory->create(ResultFactory::TYPE_JSON);
        $resultJson->setData($productMedia);
        return $resultJson;
    }
}