Add virtual products to cart

Author: rogyar

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

@@ -7,19 +7,36 @@
 
 namespace Magento\QuoteGraphQl\Model\Cart;
 
+use Magento\Framework\Exception\NoSuchEntityException;
 use Magento\Quote\Model\Quote;
 use Magento\Quote\Model\Quote\Item as QuoteItem;
+use Magento\Quote\Model\QuoteIdToMaskedQuoteIdInterface;
 
 /**
  * Extract data from cart
  */
 class ExtractDataFromCart
 {
+    /**
+     * @var QuoteIdToMaskedQuoteIdInterface
+     */
+    private $quoteIdToMaskedQuoteId;
+
+    /**
+     * @param QuoteIdToMaskedQuoteIdInterface $quoteIdToMaskedQuoteId
+     */
+    public function __construct(
+        QuoteIdToMaskedQuoteIdInterface $quoteIdToMaskedQuoteId
+    ) {
+        $this->quoteIdToMaskedQuoteId = $quoteIdToMaskedQuoteId;
+    }
+
     /**
      * Extract data from cart
      *
      * @param Quote $cart
      * @return array
+     * @throws NoSuchEntityException
      */
     public function execute(Quote $cart): array
     {
@@ -41,7 +58,8 @@ public function execute(Quote $cart): array
         }
 
         return [
-            'items' => $items,
+            'cart_id' => $this->quoteIdToMaskedQuoteId->execute((int)$cart->getId()),
+            'items' => $items
         ];
     }
 }

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

@@ -0,0 +1,90 @@
+<?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\QuoteGraphQl\Model\Cart\AddProductsToCart;
+use Magento\QuoteGraphQl\Model\Cart\ExtractDataFromCart;
+use Magento\QuoteGraphQl\Model\Cart\GetCartForUser;
+
+/**
+ * Add virtual products to cart GraphQl resolver
+ *
+ * {@inheritdoc}
+ */
+class AddVirtualProductsToCart implements ResolverInterface
+{
+    /**
+     * @var ArrayManager
+     */
+    private $arrayManager;
+
+    /**
+     * @var GetCartForUser
+     */
+    private $getCartForUser;
+
+    /**
+     * @var AddProductsToCart
+     */
+    private $addProductsToCart;
+
+    /**
+     * @var ExtractDataFromCart
+     */
+    private $extractDataFromCart;
+
+    /**
+     * @param ArrayManager $arrayManager
+     * @param GetCartForUser $getCartForUser
+     * @param AddProductsToCart $addProductsToCart
+     * @param ExtractDataFromCart $extractDataFromCart
+     */
+    public function __construct(
+        ArrayManager $arrayManager,
+        GetCartForUser $getCartForUser,
+        AddProductsToCart $addProductsToCart,
+        ExtractDataFromCart $extractDataFromCart
+    ) {
+        $this->arrayManager = $arrayManager;
+        $this->getCartForUser = $getCartForUser;
+        $this->addProductsToCart = $addProductsToCart;
+        $this->extractDataFromCart = $extractDataFromCart;
+    }
+
+    /**
+     * @inheritdoc
+     */
+    public function resolve(Field $field, $context, ResolveInfo $info, array $value = null, array $args = null)
+    {
+        $cartHash = $this->arrayManager->get('input/cart_id', $args);
+        $cartItems = $this->arrayManager->get('input/cartItems', $args);
+
+        if (!isset($cartHash)) {
+            throw new GraphQlInputException(__('Missing key "cart_id" in cart data'));
+        }
+
+        if (!isset($cartItems) || !is_array($cartItems) || empty($cartItems)) {
+            throw new GraphQlInputException(__('Missing key "cartItems" in cart data'));
+        }
+
+        $currentUserId = $context->getUserId();
+        $cart = $this->getCartForUser->execute((string)$cartHash, $currentUserId);
+
+        $this->addProductsToCart->execute($cart, $cartItems);
+        $cartData = $this->extractDataFromCart->execute($cart);
+
+        return [
+            'cart' => $cartData,
+        ];
+    }
+}

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

@@ -15,6 +15,7 @@ type Mutation {
     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")
+    addVirtualProductsToCart(input: AddVirtualProductsToCartInput): AddVirtualProductsToCartOutput @resolver(class: "Magento\\QuoteGraphQl\\Model\\Resolver\\AddVirtualProductsToCart")
 }
 
 input SetShippingAddressesOnCartInput {
@@ -167,11 +168,21 @@ input AddSimpleProductsToCartInput {
     cartItems: [SimpleProductCartItemInput!]!
 }
 
+input AddVirtualProductsToCartInput {
+    cart_id: String!
+    cartItems: [VirtualProductCartItemInput!]!
+}
+
 input SimpleProductCartItemInput {
     data: CartItemInput!
     customizable_options:[CustomizableOptionInput!]
 }
 
+input VirtualProductCartItemInput {
+    data: CartItemInput!
+    customizable_options:[CustomizableOptionInput!]
+}
+
 input CustomizableOptionInput {
     id: Int!
     value: String!
@@ -181,6 +192,10 @@ type AddSimpleProductsToCartOutput {
     cart: Cart!
 }
 
+type AddVirtualProductsToCartOutput {
+    cart: Cart!
+}
+
 type SimpleCartItem implements CartItemInterface @doc(description: "Simple Cart Item") {
     customizable_options: [SelectedCustomizableOption] @resolver(class: "Magento\\QuoteGraphQl\\Model\\Resolver\\CustomizableOptions")
 }