Your IP : 216.73.216.43

Current Path : /home/rtorresani/www/vendor/magento/module-adobe-stock-asset/Model/
Upload File :
Current File : //home/rtorresani/www/vendor/magento/module-adobe-stock-asset/Model/AppendAttributes.php

<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
declare(strict_types=1);

namespace Magento\AdobeStockAsset\Model;

use Magento\AdobeStockAssetApi\Model\Asset\Command\LoadByIdsInterface;
use Magento\Framework\Api\AttributeValueFactory;
use Magento\Framework\Api\Search\Document;
use Magento\Framework\Api\Search\SearchResultInterface;
use Magento\MediaGalleryApi\Api\GetAssetsByIdsInterface;

/**
 * Class is used for adding an additional assets attributes such as is_downloaded or path to the search results
 */
class AppendAttributes
{
    private const ATTRIBUTE_CODE_IS_DOWNLOADED = 'is_downloaded';
    private const ATTRIBUTE_CODE_IS_LICENSED_LOCALLY = 'is_licensed_locally';
    private const ATTRIBUTE_CODE_PATH = 'path';

    /**
     * @var AttributeValueFactory
     */
    private $attributeValueFactory;

    /**
     * @var LoadByIdsInterface
     */
    private $loadByIds;

    /**
     * @var GetAssetsByIdsInterface
     */
    private $getMediaGalleryAssetsByIds;

    /**
     * @param AttributeValueFactory $attributeValueFactory
     * @param LoadByIdsInterface $loadByIds
     * @param GetAssetsByIdsInterface $getMediaGalleryAssetById
     */
    public function __construct(
        AttributeValueFactory $attributeValueFactory,
        LoadByIdsInterface $loadByIds,
        GetAssetsByIdsInterface $getMediaGalleryAssetById
    ) {
        $this->attributeValueFactory = $attributeValueFactory;
        $this->loadByIds = $loadByIds;
        $this->getMediaGalleryAssetsByIds = $getMediaGalleryAssetById;
    }

    /**
     * Add additional asset attributes
     *
     * @param SearchResultInterface $searchResult
     *
     * @return SearchResultInterface
     */
    public function execute(SearchResultInterface $searchResult): SearchResultInterface
    {
        $items = $searchResult->getItems();

        if (empty($items)) {
            return $searchResult;
        }

        $ids = array_map(
            static function ($item) {
                return $item->getId();
            },
            $items
        );

        $assets = $this->loadByIds->execute($ids);

        foreach ($items as $item) {
            if (!isset($assets[$item->getId()])) {
                $this->addAttributes(
                    $item,
                    [
                        self::ATTRIBUTE_CODE_IS_DOWNLOADED => 0,
                        self::ATTRIBUTE_CODE_PATH => '',
                        self::ATTRIBUTE_CODE_IS_LICENSED_LOCALLY => 0
                    ]
                );
                continue;
            }

            $path = $this->getMediaGalleryAssetsByIds->execute(
                [$assets[$item->getId()]->getMediaGalleryId()]
            )[0]->getPath();

            $this->addAttributes(
                $item,
                [
                    self::ATTRIBUTE_CODE_IS_DOWNLOADED => 1,
                    self::ATTRIBUTE_CODE_PATH => $path,
                    self::ATTRIBUTE_CODE_IS_LICENSED_LOCALLY => $assets[$item->getId()]->getIsLicensed() ?? 0
                ]
            );
        }

        return $searchResult;
    }

    /**
     * Add attributes to document
     *
     * @param Document $document
     * @param array $attributes [code => value]
     * @return Document
     */
    private function addAttributes(Document $document, array $attributes): Document
    {
        $customAttributes = $document->getCustomAttributes();

        foreach ($attributes as $code => $value) {
            $attribute = $this->attributeValueFactory->create();
            $attribute->setAttributeCode($code);
            $attribute->setValue($value);
            $customAttributes[$code] = $attribute;
        }

        $document->setCustomAttributes($customAttributes);

        return $document;
    }
}