GraphQl shipping address coverage

app/code/Magento/QuoteGraphQl/Model/Cart/SetShippingAddressOnCart.php

@@ -0,0 +1,96 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\QuoteGraphQl\Model\Cart;
+
+use Magento\Authorization\Model\UserContextInterface;
+use Magento\Customer\Api\Data\AddressInterface;
+use Magento\Framework\GraphQl\Exception\GraphQlAuthorizationException;
+use Magento\Framework\GraphQl\Exception\GraphQlInputException;
+use Magento\Framework\GraphQl\Query\Resolver\ContextInterface;
+use Magento\Quote\Api\Data\CartInterface;
+use Magento\Quote\Model\Quote\Address;
+use Magento\Quote\Model\ShippingAddressManagementInterface;
+use Magento\Customer\Api\AddressRepositoryInterface;
+
+/**
+ * Set single shipping address for a specified shopping cart
+ */
+class SetShippingAddressOnCart implements SetShippingAddressesOnCartInterface
+{
+    /**
+     * @var ShippingAddressManagementInterface
+     */
+    private $shippingAddressManagement;
+
+    /**
+     * @var AddressRepositoryInterface
+     */
+    private $addressRepository;
+
+    /**
+     * @var Address
+     */
+    private $addressModel;
+
+    /**
+     * @param ShippingAddressManagementInterface $shippingAddressManagement
+     * @param AddressRepositoryInterface $addressRepository
+     * @param Address $addressModel
+     */
+    public function __construct(
+        ShippingAddressManagementInterface $shippingAddressManagement,
+        AddressRepositoryInterface $addressRepository,
+        Address $addressModel
+    ) {
+        $this->shippingAddressManagement = $shippingAddressManagement;
+        $this->addressRepository = $addressRepository;
+        $this->addressModel = $addressModel;
+    }
+
+    /**
+     * @inheritdoc
+     */
+    public function execute(ContextInterface $context, CartInterface $cart, array $shippingAddresses): void
+    {
+        if (count($shippingAddresses) > 1) {
+            throw new GraphQlInputException(
+                __('Multiple addresses do not allowed here!')
+            );
+        }
+        $shippingAddress = current($shippingAddresses);
+        $customerAddressId = $shippingAddress['customer_address_id'] ?? null;
+        $addressInput = $shippingAddress['address'] ?? null;
+
+        if (!$customerAddressId && !$addressInput) {
+            throw new GraphQlInputException(
+                __('Shipping address should contain either "customer_address_id" or "address" input.')
+            );
+        }
+        if ($customerAddressId && $addressInput) {
+            throw new GraphQlInputException(
+                __('Shipping address can\'t contain "customer_address_id" and "address" input at the same time.')
+            );
+        }
+        if ($customerAddressId) {
+            if ((!$context->getUserId()) || $context->getUserType() == UserContextInterface::USER_TYPE_GUEST) {
+                throw new GraphQlAuthorizationException(
+                    __(
+                        'Address management allowed only for authorized customers.'
+                    )
+                );
+            }
+            /** @var AddressInterface $customerAddress */
+            $customerAddress = $this->addressRepository->getById($customerAddressId);
+            $shippingAddress = $this->addressModel->importCustomerAddressData($customerAddress);
+        } else {
+            $shippingAddress = $this->addressModel->addData($addressInput);
+        }
+
+        $this->shippingAddressManagement->assign($cart->getId(), $shippingAddress);
+    }
+}

app/code/Magento/QuoteGraphQl/Model/Cart/SetShippingAddressesOnCartInterface.php

@@ -0,0 +1,32 @@
+<?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\GraphQl\Exception\GraphQlInputException;
+use Magento\Framework\GraphQl\Query\Resolver\ContextInterface;
+use Magento\Quote\Api\Data\CartInterface;
+
+/**
+ * Extension point for setting shipping addresses for a specified shopping cart
+ *
+ * All objects that are responsible for set shipping addresses on cart via GraphQl should implement this interface.
+ */
+interface SetShippingAddressesOnCartInterface
+{
+
+    /**
+     * Set shipping addresses for a specified shopping cart
+     *
+     * @param ContextInterface $context
+     * @param CartInterface $cart
+     * @param array $shippingAddresses
+     * @return void
+     * @throws GraphQlInputException
+     */
+    public function execute(ContextInterface $context, CartInterface $cart, array $shippingAddresses): void;
+}

app/code/Magento/QuoteGraphQl/Model/Resolver/SetShippingAddressesOnCart.php

@@ -0,0 +1,100 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\QuoteGraphQl\Model\Resolver;
+
+use Magento\Framework\GraphQl\Config\Element\Field;
+use Magento\Framework\GraphQl\Exception\GraphQlInputException;
+use Magento\Framework\GraphQl\Query\ResolverInterface;
+use Magento\Framework\GraphQl\Schema\Type\ResolveInfo;
+use Magento\Framework\Stdlib\ArrayManager;
+use Magento\Quote\Model\MaskedQuoteIdToQuoteIdInterface;
+use Magento\Quote\Model\ShippingAddressManagementInterface;
+use Magento\QuoteGraphQl\Model\Cart\GetCartForUser;
+use Magento\QuoteGraphQl\Model\Cart\SetShippingAddressesOnCartInterface;
+
+/**
+ * Class SetShippingAddressesOnCart
+ *
+ * Mutation resolver for setting shipping addresses for shopping cart
+ */
+class SetShippingAddressesOnCart implements ResolverInterface
+{
+    /**
+     * @var MaskedQuoteIdToQuoteIdInterface
+     */
+    private $maskedQuoteIdToQuoteId;
+
+    /**
+     * @var ShippingAddressManagementInterface
+     */
+    private $shippingAddressManagement;
+
+    /**
+     * @var GetCartForUser
+     */
+    private $getCartForUser;
+
+    /**
+     * @var ArrayManager
+     */
+    private $arrayManager;
+
+    /**
+     * @var SetShippingAddressesOnCartInterface
+     */
+    private $setShippingAddressesOnCart;
+
+    /**
+     * @param MaskedQuoteIdToQuoteIdInterface $maskedQuoteIdToQuoteId
+     * @param ShippingAddressManagementInterface $shippingAddressManagement
+     * @param GetCartForUser $getCartForUser
+     * @param ArrayManager $arrayManager
+     * @param SetShippingAddressesOnCartInterface $setShippingAddressesOnCart
+     */
+    public function __construct(
+        MaskedQuoteIdToQuoteIdInterface $maskedQuoteIdToQuoteId,
+        ShippingAddressManagementInterface $shippingAddressManagement,
+        GetCartForUser $getCartForUser,
+        ArrayManager $arrayManager,
+        SetShippingAddressesOnCartInterface $setShippingAddressesOnCart
+    ) {
+        $this->maskedQuoteIdToQuoteId = $maskedQuoteIdToQuoteId;
+        $this->shippingAddressManagement = $shippingAddressManagement;
+        $this->getCartForUser = $getCartForUser;
+        $this->arrayManager = $arrayManager;
+        $this->setShippingAddressesOnCart = $setShippingAddressesOnCart;
+    }
+
+    /**
+     * @inheritdoc
+     */
+    public function resolve(Field $field, $context, ResolveInfo $info, array $value = null, array $args = null)
+    {
+        $shippingAddresses = $this->arrayManager->get('input/shipping_addresses', $args);
+        $maskedCartId = $this->arrayManager->get('input/cart_id', $args);
+
+        if (!$maskedCartId) {
+            throw new GraphQlInputException(__('Required parameter "cart_id" is missing'));
+        }
+        if (!$shippingAddresses) {
+            throw new GraphQlInputException(__('Required parameter "shipping_addresses" is missing'));
+        }
+
+        $maskedCartId = $args['input']['cart_id'];
+        $cart = $this->getCartForUser->execute($maskedCartId, $context->getUserId());
+
+        $this->setShippingAddressesOnCart->execute($context, $cart, $shippingAddresses);
+
+        return [
+            'cart' => [
+                'cart_id' => $maskedCartId,
+                'model' => $cart
+            ]
+        ];
+    }
+}

app/code/Magento/QuoteGraphQl/composer.json

@@ -8,7 +8,9 @@
         "magento/module-quote": "*",
         "magento/module-checkout": "*",
         "magento/module-catalog": "*",
-        "magento/module-store": "*"
+        "magento/module-store": "*",
+        "magento/module-customer": "*",
+        "magento/module-authorization": "*"
     },
     "suggest": {
         "magento/module-graph-ql": "*"

app/code/Magento/QuoteGraphQl/etc/graphql/di.xml

@@ -0,0 +1,28 @@
+<?xml version="1.0"?>
+<!--
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+-->
+<config xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:noNamespaceSchemaLocation="urn:magento:framework:ObjectManager/etc/config.xsd">
+    <preference for="Magento\QuoteGraphQl\Model\Cart\SetShippingAddressesOnCartInterface"
+                type="Magento\QuoteGraphQl\Model\Cart\SetShippingAddressOnCart" />
+    <virtualType name="multishippingPaymentSpecification" type="Magento\Payment\Model\Method\Specification\Composite">
+        <arguments>
+            <argument name="specifications" xsi:type="array">
+                <item name="enabled" xsi:type="string">Magento\Multishipping\Model\Payment\Method\Specification\Enabled</item>
+            </argument>
+        </arguments>
+    </virtualType>
+    <type name="Magento\Multishipping\Block\Checkout\Billing">
+        <arguments>
+            <argument name="paymentSpecification" xsi:type="object">multishippingPaymentSpecification</argument>
+        </arguments>
+    </type>
+    <type name="Magento\Multishipping\Model\Checkout\Type\Multishipping">
+        <arguments>
+            <argument name="paymentSpecification" xsi:type="object">multishippingPaymentSpecification</argument>
+        </arguments>
+    </type>
+</config>

app/code/Magento/QuoteGraphQl/etc/schema.graphqls

@@ -6,17 +6,24 @@ type Query {
 }
 
 type Mutation {
+    createEmptyCart: String @resolver(class: "\\Magento\\QuoteGraphQl\\Model\\Resolver\\Cart\\CreateEmptyCart") @doc(description:"Creates empty shopping cart for guest or logged in user")
+    applyCouponToCart(input: ApplyCouponToCartInput): ApplyCouponToCartOutput @resolver(class: "\\Magento\\QuoteGraphQl\\Model\\Resolver\\Coupon\\ApplyCouponToCart")
+    removeCouponFromCart(input: RemoveCouponFromCartInput): RemoveCouponFromCartOutput @resolver(class: "\\Magento\\QuoteGraphQl\\Model\\Resolver\\Coupon\\RemoveCouponFromCart")
+    setShippingAddressesOnCart(input: SetShippingAddressesOnCartInput): SetShippingAddressesOnCartOutput @resolver(class: "\\Magento\\QuoteGraphQl\\Model\\Resolver\\SetShippingAddressesOnCart")
     createEmptyCart: String @resolver(class: "\\Magento\\QuoteGraphQl\\Model\\Resolver\\CreateEmptyCart") @doc(description:"Creates empty shopping cart for guest or logged in user")
     applyCouponToCart(input: ApplyCouponToCartInput): ApplyCouponToCartOutput @resolver(class: "\\Magento\\QuoteGraphQl\\Model\\Resolver\\ApplyCouponToCart")
     removeCouponFromCart(input: RemoveCouponFromCartInput): RemoveCouponFromCartOutput @resolver(class: "\\Magento\\QuoteGraphQl\\Model\\Resolver\\RemoveCouponFromCart")
-    setShippingAddressesOnCart(input: SetShippingAddressesOnCartInput): SetShippingAddressesOnCartOutput
     setBillingAddressOnCart(input: SetBillingAddressOnCartInput): SetBillingAddressOnCartOutput
     setShippingMethodsOnCart(input: SetShippingMethodsOnCartInput): SetShippingMethodsOnCartOutput @resolver(class: "\\Magento\\QuoteGraphQl\\Model\\Resolver\\SetShippingMethodsOnCart")
     addSimpleProductsToCart(input: AddSimpleProductsToCartInput): AddSimpleProductsToCartOutput @resolver(class: "Magento\\QuoteGraphQl\\Model\\Resolver\\AddSimpleProductsToCart")
 }
 
 input SetShippingAddressesOnCartInput {
     cart_id: String!
+    shipping_addresses: [ShippingAddressInput!]!
+}
+
+input ShippingAddressInput {
     customer_address_id: Int # Can be provided in one-page checkout and is required for multi-shipping checkout
     address: CartAddressInput
     cart_items: [CartItemQuantityInput!]