MC-42781: [Magento Cloud][Premier Support] Graphql request can add product to cart that is NOT assigned to a website of a multi site store

Author: dhorytskyi

app/code/Magento/Quote/Model/Cart/AddProductsToCart.php

@@ -7,7 +7,7 @@
 
 namespace Magento\Quote\Model\Cart;
 
-use Magento\Catalog\Api\ProductRepositoryInterface;
+use Magento\Catalog\Model\ResourceModel\Product\CollectionFactory;
 use Magento\Framework\Exception\NoSuchEntityException;
 use Magento\Quote\Api\CartRepositoryInterface;
 use Magento\Quote\Api\Data\CartInterface;
@@ -16,6 +16,7 @@
 use Magento\Quote\Model\MaskedQuoteIdToQuoteIdInterface;
 use Magento\Quote\Model\Quote;
 use Magento\Framework\Message\MessageInterface;
+use Magento\Store\Model\StoreManagerInterface;
 
 /**
  * Unified approach to add products to the Shopping Cart.
@@ -47,9 +48,14 @@ class AddProductsToCart
     ];
 
     /**
-     * @var ProductRepositoryInterface
+     * @var CollectionFactory
      */
-    private $productRepository;
+    private $productCollectionFactory;
+
+    /**
+     * @var StoreManagerInterface
+     */
+    private $storeManager;
 
     /**
      * @var array
@@ -72,18 +78,21 @@ class AddProductsToCart
     private $requestBuilder;
 
     /**
-     * @param ProductRepositoryInterface $productRepository
+     * @param CollectionFactory $productCollectionFactory
+     * @param StoreManagerInterface $storeManager
      * @param CartRepositoryInterface $cartRepository
      * @param MaskedQuoteIdToQuoteIdInterface $maskedQuoteIdToQuoteId
      * @param BuyRequestBuilder $requestBuilder
      */
     public function __construct(
-        ProductRepositoryInterface $productRepository,
+        CollectionFactory $productCollectionFactory,
+        StoreManagerInterface $storeManager,
         CartRepositoryInterface $cartRepository,
         MaskedQuoteIdToQuoteIdInterface $maskedQuoteIdToQuoteId,
         BuyRequestBuilder $requestBuilder
     ) {
-        $this->productRepository = $productRepository;
+        $this->productCollectionFactory = $productCollectionFactory;
+        $this->storeManager = $storeManager;
         $this->cartRepository = $cartRepository;
         $this->maskedQuoteIdToQuoteId = $maskedQuoteIdToQuoteId;
         $this->requestBuilder = $requestBuilder;
@@ -140,9 +149,13 @@ private function addItemToCart(CartInterface $cart, Data\CartItem $cartItem, int
             return;
         }
 
-        try {
-            $product = $this->productRepository->get($sku, false, null, true);
-        } catch (NoSuchEntityException $e) {
+        $store = $this->storeManager->getStore($cart->getStoreId());
+        $productCollection = $this->productCollectionFactory->create()
+            ->addAttributeToFilter('sku', $sku)
+            ->addWebsiteFilter([$store->getWebsiteId()])
+            ->load();
+        $product = $productCollection->getFirstItem();
+        if (!$product->getId()) {
             $this->addError(
                 __('Could not find a product with SKU "%sku"', ['sku' => $sku])->render(),
                 $cartItemPosition