DisallowCreateShipmentPlugin.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\InventoryShippingAdminUi\Plugin\Sales\Block\Order\Invoice\Create;

use Magento\InventoryApi\Api\GetSourcesAssignedToStockOrderedByPriorityInterface;
use Magento\Sales\Block\Adminhtml\Order\Invoice\Create\Form;
use Magento\Framework\Exception\LocalizedException;
use Magento\InventorySalesApi\Model\StockByWebsiteIdResolverInterface;

class DisallowCreateShipmentPlugin
{
    private $getSourcesAssignedToStockOrderedByPriority;

    private $stockByWebsiteIdResolver;

    public function __construct(
        GetSourcesAssignedToStockOrderedByPriorityInterface $getSourcesAssignedToStockOrderedByPriority,
        StockByWebsiteIdResolverInterface $stockByWebsiteIdResolver
    ) {
        $this->getSourcesAssignedToStockOrderedByPriority = $getSourcesAssignedToStockOrderedByPriority;
        $this->stockByWebsiteIdResolver = $stockByWebsiteIdResolver;
    }

    public function afterCanCreateShipment(Form $subject, bool $result)
    {
        try {
            $websiteId = $subject->getOrder()->getStore()->getWebsiteId();
            $stockId = $this->stockByWebsiteIdResolver->execute((int)$websiteId)->getStockId();
            $sources = $this->getSourcesAssignedToStockOrderedByPriority->execute((int)$stockId);
            if (count($sources) > 1) {
                return false;
            }
        } catch (LocalizedException $e) {
            return false;
        }

        return $result;
    }
}