InventoryRequest.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\InventorySourceSelection\Model\Request;

use Magento\InventorySourceSelectionApi\Api\Data\InventoryRequestExtensionInterface;
use Magento\InventorySourceSelectionApi\Api\Data\InventoryRequestInterface;
use Magento\InventorySourceSelectionApi\Api\Data\ItemRequestInterface;
use Magento\InventorySourceSelectionApi\Api\Data\ItemRequestInterfaceFactory;
use Magento\Framework\Model\AbstractExtensibleModel;

class InventoryRequest extends AbstractExtensibleModel implements InventoryRequestInterface
{
    private $stockId;

    private $items;

    private $itemRequestFactory;

    public function __construct(
        \Magento\Framework\Model\Context $context,
        \Magento\Framework\Registry $registry,
        \Magento\Framework\Api\ExtensionAttributesFactory $extensionFactory,
        \Magento\Framework\Api\AttributeValueFactory $customAttributeFactory,
        ItemRequestInterfaceFactory $itemRequestFactory,
        int $stockId,
        array $items,
        \Magento\Framework\Model\ResourceModel\AbstractResource $resource = null,
        \Magento\Framework\Data\Collection\AbstractDb $resourceCollection = null,
        array $data = []
    ) {
        parent::__construct(
            $context,
            $registry,
            $extensionFactory,
            $customAttributeFactory,
            $resource,
            $resourceCollection,
            $data
        );

        $this->itemRequestFactory = $itemRequestFactory;
        $this->stockId = $stockId;

        //TODO: Temporary fix for resolving issue with webApi (https://github.com/magento-engcom/msi/issues/1524)
        foreach ($items as $item) {
            if (false === $item instanceof ItemRequestInterface) {
                $this->items[] = $this->itemRequestFactory->create([
                    'sku' => $item['sku'],
                    'qty' => $item['qty']
                ]);
            } else {
                $this->items[] = $item;
            }
        }
    }

    public function getStockId(): int
    {
        return $this->stockId;
    }

    public function getItems(): array
    {
        return $this->items;
    }

    public function setStockId(int $stockId): void
    {
        $this->stockId = $stockId;
    }

    public function setItems(array $items): void
    {
        $this->items = $items;
    }

    public function getExtensionAttributes(): ?InventoryRequestExtensionInterface
    {
        $extensionAttributes = $this->_getExtensionAttributes();
        if (null === $extensionAttributes) {
            $extensionAttributes = $this->extensionAttributesFactory->create(InventoryRequestInterface::class);
            $this->setExtensionAttributes($extensionAttributes);
        }
        return $extensionAttributes;
    }

    public function setExtensionAttributes(InventoryRequestExtensionInterface $extensionAttributes): void
    {
        $this->_setExtensionAttributes($extensionAttributes);
    }
}