Sources.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\InventoryImportExport\Model\Import;

use Magento\Framework\Exception\LocalizedException;
use Magento\ImportExport\Helper\Data as DataHelper;
use Magento\ImportExport\Model\Import\Entity\AbstractEntity;
use Magento\ImportExport\Model\Import\ErrorProcessing\ProcessingErrorAggregatorInterface;
use Magento\ImportExport\Model\ResourceModel\Helper as ResourceHelper;
use Magento\ImportExport\Model\ResourceModel\Import\Data as ImportData;
use Magento\InventoryImportExport\Model\Import\Command\CommandInterface;
use Magento\InventoryImportExport\Model\Import\Serializer\Json;
use Magento\InventoryImportExport\Model\Import\Validator\ValidatorInterface;
use Magento\InventoryApi\Api\Data\SourceItemInterface;

class Sources extends AbstractEntity
{
    const COL_SKU = SourceItemInterface::SKU;
    const COL_SOURCE_CODE = SourceItemInterface::SOURCE_CODE;
    const COL_QTY = SourceItemInterface::QUANTITY;
    const COL_STATUS = SourceItemInterface::STATUS;

    protected $jsonHelper;

    private $validator;

    private $commands = [];

    public function __construct(
        Json $jsonHelper,
        ProcessingErrorAggregatorInterface $errorAggregator,
        ResourceHelper $resourceHelper,
        DataHelper $dataHelper,
        ImportData $importData,
        ValidatorInterface $validator,
        array $commands = []
    ) {
        $this->jsonHelper = $jsonHelper;
        $this->errorAggregator = $errorAggregator;
        $this->_resourceHelper = $resourceHelper;
        $this->_importExportData = $dataHelper;
        $this->_dataSourceModel = $importData;
        $this->validator = $validator;

        foreach ($commands as $command) {
            if (!$command instanceof CommandInterface) {
                throw new LocalizedException(
                    __('Source Import Commands must implement %interface.', ['interface' => CommandInterface::class])
                );
            }
        }
        $this->commands = $commands;
    }

    protected function _importData()
    {
        $command = $this->getCommandByBehavior($this->getBehavior());
        while ($bunch = $this->_dataSourceModel->getNextBunch()) {
            $command->execute($bunch);
        }

        return true;
    }

    private function getCommandByBehavior($behavior)
    {
        if (!isset($this->commands[$behavior])) {
            throw new LocalizedException(
                __('There is no command registered for behavior "%behavior".', ['behavior' => $behavior])
            );
        }

        return $this->commands[$behavior];
    }

    public function getEntityTypeCode()
    {
        return 'stock_sources';
    }

    public function validateRow(array $rowData, $rowNum)
    {
        $result = $this->validator->validate($rowData, $rowNum);
        if ($result->isValid()) {
            return true;
        }

        foreach ($result->getErrors() as $error) {
            $this->addRowError($error, $rowNum);
        }

        return false;
    }
}