StockSourceLinkProcessor.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\InventoryAdminUi\Model\Stock;

use Magento\Framework\Api\DataObjectHelper;
use Magento\Framework\Exception\InputException;
use Magento\Framework\Api\SearchCriteriaBuilder;
use Magento\InventoryApi\Api\Data\StockSourceLinkInterfaceFactory;
use Magento\InventoryApi\Api\Data\StockSourceLinkInterface;
use Magento\InventoryApi\Api\GetStockSourceLinksInterface;
use Magento\InventoryApi\Api\StockSourceLinksDeleteInterface;
use Magento\InventoryApi\Api\StockSourceLinksSaveInterface;

class StockSourceLinkProcessor
{
    private $searchCriteriaBuilder;

    private $stockSourceLinkFactory;

    private $stockSourceLinksSave;

    private $stockSourceLinksDelete;

    private $getStockSourceLinks;

    private $dataObjectHelper;

    public function __construct(
        SearchCriteriaBuilder $searchCriteriaBuilder,
        StockSourceLinkInterfaceFactory $stockSourceLinkFactory,
        StockSourceLinksSaveInterface $stockSourceLinksSave,
        StockSourceLinksDeleteInterface $stockSourceLinksDelete,
        GetStockSourceLinksInterface $getStockSourceLinks,
        DataObjectHelper $dataObjectHelper
    ) {
        $this->searchCriteriaBuilder = $searchCriteriaBuilder;
        $this->stockSourceLinkFactory = $stockSourceLinkFactory;
        $this->stockSourceLinksSave = $stockSourceLinksSave;
        $this->stockSourceLinksDelete = $stockSourceLinksDelete;
        $this->getStockSourceLinks = $getStockSourceLinks;
        $this->dataObjectHelper = $dataObjectHelper;
    }

    public function process(int $stockId, array $linksData)
    {
        $linksForDelete = $this->getAssignedLinks($stockId);
        $linksForSave = [];

        foreach ($linksData as $linkData) {
            $sourceCode = $linkData[StockSourceLinkInterface::SOURCE_CODE];

            if (isset($linksForDelete[$sourceCode])) {
                $link = $linksForDelete[$sourceCode];
            } else {
                $link = $this->stockSourceLinkFactory->create();
            }

            $linkData[StockSourceLinkInterface::STOCK_ID] = $stockId;
            $this->dataObjectHelper->populateWithArray($link, $linkData, StockSourceLinkInterface::class);

            $linksForSave[] = $link;
            unset($linksForDelete[$sourceCode]);
        }

        if (count($linksForSave) > 0) {
            $this->stockSourceLinksSave->execute($linksForSave);
        }
        if (count($linksForDelete) > 0) {
            $this->stockSourceLinksDelete->execute($linksForDelete);
        }
    }

    private function getAssignedLinks(int $stockId): array
    {
        $searchCriteria = $this->searchCriteriaBuilder
            ->addFilter(StockSourceLinkInterface::STOCK_ID, $stockId)
            ->create();

        $result = [];
        foreach ($this->getStockSourceLinks->execute($searchCriteria)->getItems() as $link) {
            $result[$link->getSourceCode()] = $link;
        }
        return $result;
    }
}