ShippingAddressAssignment.php

Go to the documentation of this file.

<?php
namespace Magento\Quote\Model;

class ShippingAddressAssignment
{
    private $shippingAssignmentProcessor;

    private $cartExtensionFactory;

    public function __construct(
        \Magento\Quote\Api\Data\CartExtensionFactory $cartExtensionFactory,
        \Magento\Quote\Model\Quote\ShippingAssignment\ShippingAssignmentProcessor $shippingAssignmentProcessor
    ) {
        $this->shippingAssignmentProcessor = $shippingAssignmentProcessor;
        $this->cartExtensionFactory = $cartExtensionFactory;
    }

    public function setAddress(
        \Magento\Quote\Api\Data\CartInterface $quote,
        \Magento\Quote\Api\Data\AddressInterface $address,
        $useForShipping = false
    ) {
        if ($useForShipping) {
            $quote->removeAddress($quote->getShippingAddress()->getId());
            $address->setSameAsBilling(1);
            $address->setCollectShippingRates(true);
        } else {
            $address = $quote->getShippingAddress()->setSameAsBilling(0);
        }

        $quote->setShippingAddress($address);
        $cartExtension = $quote->getExtensionAttributes();
        if ($cartExtension === null) {
            $cartExtension = $this->cartExtensionFactory->create();
        }
        $shippingAssignment = $this->shippingAssignmentProcessor->create($quote);
        $cartExtension->setShippingAssignments([$shippingAssignment]);
        $quote->setExtensionAttributes($cartExtension);
    }
}