GetAssignedStockIdsBySku.php 1.61 KB
Newer Older
Ketan's avatar
Ketan committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58
<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
declare(strict_types=1);

namespace Magento\InventorySalesAdminUi\Model\ResourceModel;

use Magento\InventoryApi\Api\GetSourceItemsBySkuInterface;
use Magento\InventorySalesAdminUi\Model\GetStockSourceLinksBySourceCode;

/**
 * Get all stocks Ids by sku
 */
class GetAssignedStockIdsBySku
{
    /**
     * @var GetSourceItemsBySkuInterface
     */
    private $getSourceItemsBySku;

    /**
     * @var GetStockSourceLinksBySourceCode
     */
    private $getStockSourceLinksBySourceCode;

    /**
     * @param GetSourceItemsBySkuInterface $getSourceItemsBySku
     * @param GetStockSourceLinksBySourceCode $getStockSourceLinksBySourceCode
     */
    public function __construct(
        GetSourceItemsBySkuInterface $getSourceItemsBySku,
        GetStockSourceLinksBySourceCode $getStockSourceLinksBySourceCode
    ) {
        $this->getSourceItemsBySku = $getSourceItemsBySku;
        $this->getStockSourceLinksBySourceCode = $getStockSourceLinksBySourceCode;
    }

    /**
     * @param string $sku
     * @return array
     */
    public function execute(string $sku): array
    {
        $sourceItems = $this->getSourceItemsBySku->execute($sku);

        $stocksIds = [];
        foreach ($sourceItems as $sourceItem) {
            $stockSourceLinks = $this->getStockSourceLinksBySourceCode->execute($sourceItem->getSourceCode());
            foreach ($stockSourceLinks as $stockSourceLink) {
                $stocksIds[] = (int)$stockSourceLink->getStockId();
            }
        }

        return array_unique($stocksIds);
    }
}