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

namespace Magento\QuoteGraphQl\Model\Cart;

use Magento\Framework\Exception\InputException;
use Magento\Framework\Exception\NoSuchEntityException;
use Magento\Framework\GraphQl\Exception\GraphQlInputException;
use Magento\Framework\GraphQl\Exception\GraphQlNoSuchEntityException;
use Magento\Quote\Api\Data\AddressInterface;
use Magento\Quote\Api\Data\CartInterface;
use Magento\Quote\Model\ShippingAddressManagementInterface;

/**
 * Assigning shipping address to cart
 */
class AssignShippingAddressToCart
{
    /**
     * @var ShippingAddressManagementInterface
     */
    private $shippingAddressManagement;

    /**
     * @param ShippingAddressManagementInterface $shippingAddressManagement
     */
    public function __construct(
        ShippingAddressManagementInterface $shippingAddressManagement
    ) {
        $this->shippingAddressManagement = $shippingAddressManagement;
    }

    /**
     * Assign shipping address to cart
     *
     * @param CartInterface $cart
     * @param AddressInterface $shippingAddress
     * @throws GraphQlInputException
     * @throws GraphQlNoSuchEntityException
     */
    public function execute(
        CartInterface $cart,
        AddressInterface $shippingAddress
    ): void {
        try {
            $this->shippingAddressManagement->assign($cart->getId(), $shippingAddress);
        } catch (NoSuchEntityException $e) {
            if ($cart->getIsVirtual()) {
                throw new GraphQlNoSuchEntityException(
                    __('Shipping address is not allowed on cart: cart contains no items for shipment.'),
                    $e
                );
            } else {
                throw new GraphQlNoSuchEntityException(__($e->getMessage()), $e);
            }
        } catch (InputException $e) {
            throw new GraphQlInputException(__($e->getMessage()), $e);
        }
    }
}