ShippingConfiguration.php

Go to the documentation of this file.

<?php
namespace Magento\InstantPurchase\Model\QuoteManagement;

use Magento\Framework\Exception\LocalizedException;
use Magento\InstantPurchase\Model\ShippingMethodChoose\DeferredShippingMethodChooserInterface;
use Magento\InstantPurchase\Model\ShippingMethodChoose\DeferredShippingMethodChooserPool;
use Magento\Quote\Api\Data\ShippingMethodInterface;
use Magento\Quote\Model\Quote;
use Magento\Quote\Model\Quote\Address;

class ShippingConfiguration
{
    private $deferredShippingMethodChooserPool;

    public function __construct(
        DeferredShippingMethodChooserPool $deferredShippingMethodChooserPool
    ) {
        $this->deferredShippingMethodChooserPool = $deferredShippingMethodChooserPool;
    }

    public function configureShippingMethod(
        Quote $quote,
        ShippingMethodInterface $shippingMethod
    ): Quote {
        if ($quote->isVirtual()) {
            return $quote;
        }

        $shippingAddress = $quote->getShippingAddress();
        $shippingMethodCode = $this->getShippingMethodCodeToUse($shippingAddress, $shippingMethod);
        $shippingAddress->setShippingMethod($shippingMethodCode);

        return $quote;
    }

    private function getShippingMethodCodeToUse(
        Address $address,
        ShippingMethodInterface $shippingMethod
    ): string {
        if ($shippingMethod->getCarrierCode() === DeferredShippingMethodChooserInterface::CARRIER) {
            return $this->resolveDeferredShippingMethodChoose($address, $shippingMethod);
        } else {
            return $this->getCorrespondingShippingRateCode($address, $shippingMethod);
        }
    }

    private function getCorrespondingShippingRateCode(
        Address $address,
        ShippingMethodInterface $shippingMethod
    ): string {
        $address->setCollectShippingRates(true);
        $address->collectShippingRates();
        $shippingRates = $address->getAllShippingRates();
        foreach ($shippingRates as $shippingRate) {
            if ($shippingRate->getCarrier() === $shippingMethod->getCarrierCode()
                &&
                $shippingRate->getMethod() === $shippingMethod->getMethodCode()
            ) {
                return $shippingRate->getCode();
            }
        }
        throw new LocalizedException(__('Specified shipping method is not available.'));
    }

    private function resolveDeferredShippingMethodChoose(
        Address $address,
        ShippingMethodInterface $shippingMethod
    ): string {
        $deferredShippingMethodChooser = $this->deferredShippingMethodChooserPool->get(
            $shippingMethod->getMethodCode()
        );

        $shippingMethodCode = $deferredShippingMethodChooser->choose($address);
        if (empty($shippingMethodCode)) {
            throw new LocalizedException(__('Appropriate shipping method is not available.'));
        }

        return $shippingMethodCode;
    }
}