Server : Apache System : Linux server2.corals.io 4.18.0-348.2.1.el8_5.x86_64 #1 SMP Mon Nov 15 09:17:08 EST 2021 x86_64 User : corals ( 1002) PHP Version : 7.4.33 Disable Function : exec,passthru,shell_exec,system Directory :  /home/corals/old/vendor/amasty/feed/Model/Export/RowCustomizer/

Upload File : current_dir [ Writeable ] document_root [ Writeable ]

<?php
/**
 * @author Amasty Team
 * @copyright Copyright (c) Amasty (https://www.amasty.com)
 * @package Product Feed for Magento 2
 */

namespace Amasty\Feed\Model\Export\RowCustomizer;

use Amasty\Feed\Model\Export\Product;
use Amasty\Feed\Model\Export\Product as ExportProduct;
use Amasty\Feed\Model\Export\ProductFactory;
use Magento\Catalog\Api\Data\ProductInterface;
use Magento\CatalogImportExport\Model\Export\RowCustomizerInterface;
use Magento\Framework\EntityManager\MetadataPool;
use Magento\Store\Model\StoreManagerInterface;

class Relation implements RowCustomizerInterface
{
    /**
     * @var array
     */
    protected $parent2child;

    /**
     * @var array
     */
    protected $child2parent;

    /**
     * @var array
     */
    protected $parentData;

    /**
     * @var StoreManagerInterface
     */
    protected $storeManager;

    /**
     * @var Product
     */
    protected $export;

    /**
     * @var MetadataPool
     */
    private $metadataPool;

    /**
     * @var ProductFactory
     */
    private $productExportFactory;

    public function __construct(
        StoreManagerInterface $storeManager,
        Product $export,
        MetadataPool $metadataPool,
        ProductFactory $productExportFactory
    ) {
        $this->storeManager = $storeManager;
        $this->export = $export;
        $this->metadataPool = $metadataPool;
        $this->productExportFactory = $productExportFactory;
    }

    /**
     * @inheritdoc
     */
    public function prepareData($collection, $productIds)
    {
        $this->parentData = [];
        $parentAttributes = [
            'product' => [
                'product_id' => 'product_id'
            ],
            'url' => [
                'short' => 'short'
            ]
        ];

        if (isset($this->export->getAttributesByType(ExportProduct::PREFIX_URL_ATTRIBUTE)['configurable'])) {
            $parentAttributes['url']['configurable'] = 'configurable';
        }

        $parentAttributes = array_merge_recursive(
            $this->export->getParentAttributes(),
            $parentAttributes
        );

        if (count($parentAttributes) > 0) {
            $productMetadata = $this->metadataPool->getMetadata(ProductInterface::class);
            $linkField = $productMetadata->getLinkField();
            $identifierField = $productMetadata->getIdentifierField();
            $parent2child = $child2parent = $linkFieldMap = [];
            if ($linkField != $identifierField) {
                $selectMap = $collection->getConnection()
                    ->select()
                    ->from(
                        $productMetadata->getEntityTable(),
                        [$linkField, $identifierField]
                    )->where($identifierField . ' IN(?)', $productIds)
                    ->where($linkField . ' != ' . $identifierField);

                $linkFieldMap = $collection->getConnection()->fetchPairs($selectMap);
            }

            $select = $collection->getConnection()
                ->select()
                ->from(
                    ['r' => $collection->getTable('catalog_product_relation')],
                    ['r.parent_id', 'r.child_id']
                )->joinLeft(
                    ['pe' => $collection->getTable('catalog_product_entity')],
                    'pe.' . $linkField . ' = r.parent_id',
                    ['pe.type_id']
                )->where('r.child_id IN(?)', $productIds);

            foreach ($collection->getConnection()->fetchAll($select) as $row) {
                $childId = $row['child_id'] ?? null;
                $parentId = $row['parent_id'] ?? null;
                if (isset($parentId) && !isset($parent2child[$parentId])) {
                    $parent2child[$parentId] = [];

                    if (isset($linkFieldMap[$parentId])) {
                        $parentId = $linkFieldMap[$parentId];
                    }
                }

                if (isset($childId) && !isset($child2parent[$childId])) {
                    $child2parent[$childId] = [];
                }

                $parent2child[$parentId][$childId] = $childId;
                if ($row['type_id'] === $this->export->getParentPriority()) {
                    // Add priority element to top of an array.
                    array_unshift($child2parent[$childId], $parentId);
                } else {
                    $child2parent[$childId][$parentId] = $parentId;
                }
            }

            $this->parent2child = $parent2child;
            $this->child2parent = $child2parent;

            $parentsExport = $this->productExportFactory->create(['storeId' => $collection->getStoreId()]);

            $exportData = $parentsExport
                ->setAttributes($parentAttributes)
                ->setStoreId($collection->getStoreId())
                ->setUtmParams($this->export->getUtmParams())
                ->setExcludeDisabledParents($this->export->getExcludeDisabledParents())
                ->exportParents(array_keys($this->parent2child));

            foreach ($exportData as $item) {
                if (array_key_exists('product_link_id', $item)) {
                    $this->parentData[(int)$item['product_link_id']] = $item;
                }
            }
        }
    }

    /**
     * @inheritdoc
     */
    public function addHeaderColumns($columns)
    {
        return $columns;
    }

    /**
     * @inheritdoc
     */
    public function addData($dataRow, $productId)
    {
        $customData = &$dataRow['amasty_custom_data'];

        if (isset($this->child2parent[$productId])) {
            foreach ($this->child2parent[$productId] as $parentId) {
                if (isset($this->parentData[$parentId])) {
                    $customData['parent_data'] = $this->parentData[$parentId];
                    break;
                }
            }
        }

        return $dataRow;
    }

    /**
     * @inheritdoc
     */
    public function getAdditionalRowsCount($additionalRowsCount, $productId)
    {
        return $additionalRowsCount;
    }

    /**
     * @param array $dataRow
     * @param array $parentRow
     */
    protected function _fillParentData(&$dataRow, $parentRow)
    {
        foreach ($parentRow as $key => $value) {
            if (isset($dataRow[$key])) {
                if (is_array($value)) {
                    $this->_fillParentData($dataRow[$key], $parentRow[$key]);
                } else {
                    if ($dataRow[$key] == "" && !empty($value)) {
                        $dataRow[$key] = $value;
                    }
                }
            } else {
                $dataRow[$key] = $value;
            }
        }
    }
}