<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
declare(strict_types=1);

namespace Magento\InventoryShippingAdminUi\Plugin\Sales\Block\Shipment;

use Magento\InventoryShippingAdminUi\Model\IsOrderSourceManageable;
use Magento\Shipping\Block\Adminhtml\Create;
use Magento\InventoryShippingAdminUi\Model\IsWebsiteInMultiSourceMode;

/**
 *  Modify back button URL on the shipment page in multi source mode
 */
class BackButtonUrlOnNewShipmentPagePlugin
{
    /**
     * @var IsWebsiteInMultiSourceMode
     */
    private $isWebsiteInMultiSourceMode;

    /**
     * @var IsOrderSourceManageable
     */
    private $isOrderSourceManageable;

    /**
     * @param IsWebsiteInMultiSourceMode $isWebsiteInMultiSourceMode
     * @param IsOrderSourceManageable $isOrderSourceManageable
     */
    public function __construct(
        IsWebsiteInMultiSourceMode $isWebsiteInMultiSourceMode,
        IsOrderSourceManageable $isOrderSourceManageable
    ) {
        $this->isWebsiteInMultiSourceMode = $isWebsiteInMultiSourceMode;
        $this->isOrderSourceManageable = $isOrderSourceManageable;
    }

    /**
     * Returns URL to Source Selection if source for order is manageable
     *
     * @param Create $subject
     * @param $result
     * @return string
     */
    public function afterGetBackUrl(Create $subject, $result)
    {
        $shipment = $subject->getShipment();
        if (empty($shipment) || !$this->isOrderSourceManageable->execute($shipment->getOrder())) {
            return $result;
        }

        $websiteId = (int)$shipment->getOrder()->getStore()->getWebsiteId();
        if ($this->isWebsiteInMultiSourceMode->execute($websiteId)) {
            return $subject->getUrl(
                'inventoryshipping/SourceSelection/index',
                [
                    'order_id' => $shipment ? $shipment->getOrderId() : null
                ]
            );
        }

        return $result;
    }
}