SourcesSelection.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\InventoryCatalogAdminUi\ViewModel;

use Magento\Framework\Api\SearchCriteriaBuilder;
use Magento\Framework\View\Element\Block\ArgumentInterface;
use Magento\InventoryApi\Api\Data\SourceInterface;
use Magento\InventoryApi\Api\Data\SourceItemInterface;
use Magento\InventoryApi\Api\SourceRepositoryInterface;
use Magento\InventoryApi\Model\GetSourceCodesBySkusInterface;
use Magento\InventoryCatalogAdminUi\Model\BulkSessionProductsStorage;

class SourcesSelection implements ArgumentInterface
{
    private $sourceRepository;

    private $bulkSessionProductsStorage;

    private $searchCriteriaBuilder;

    private $getSourceCodesBySkus;

    public function __construct(
        SourceRepositoryInterface $sourceRepository,
        SearchCriteriaBuilder $searchCriteriaBuilder,
        GetSourceCodesBySkusInterface $getSourceCodesBySkus,
        BulkSessionProductsStorage $bulkSessionProductsStorage
    ) {
        $this->sourceRepository = $sourceRepository;
        $this->bulkSessionProductsStorage = $bulkSessionProductsStorage;
        $this->searchCriteriaBuilder = $searchCriteriaBuilder;
        $this->getSourceCodesBySkus = $getSourceCodesBySkus;
    }

    public function getSources(): array
    {
        return $this->sourceRepository->getList()->getItems();
    }

    public function getAssignedSources(): array
    {
        $skus = $this->bulkSessionProductsStorage->getProductsSkus();
        $sourceCodes = $this->getSourceCodesBySkus->execute($skus);

        $searchCriteria = $this->searchCriteriaBuilder
            ->addFilter(
                SourceItemInterface::SOURCE_CODE,
                $sourceCodes,
                'in'
            )
            ->create();

        return $this->sourceRepository->getList($searchCriteria)->getItems();
    }

    public function getProductsCount(): int
    {
        return count($this->bulkSessionProductsStorage->getProductsSkus());
    }
}