MC-42970: [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/QuoteGraphQl/Model/Cart/AddSimpleProductToCart.php

@@ -8,7 +8,8 @@
 namespace Magento\QuoteGraphQl\Model\Cart;
 
 use Exception;
-use Magento\Catalog\Api\ProductRepositoryInterface;
+use Magento\Catalog\Api\Data\ProductInterface;
+use Magento\Catalog\Model\ResourceModel\Product\CollectionFactory as ProductCollectionFactory;
 use Magento\Framework\Exception\NoSuchEntityException;
 use Magento\Framework\GraphQl\Exception\GraphQlInputException;
 use Magento\Framework\GraphQl\Exception\GraphQlNoSuchEntityException;
@@ -21,24 +22,24 @@
 class AddSimpleProductToCart
 {
     /**
-     * @var ProductRepositoryInterface
+     * @var ProductCollectionFactory
      */
-    private $productRepository;
+    private $productCollectionFactory;
 
     /**
      * @var BuyRequestBuilder
      */
     private $buyRequestBuilder;
 
     /**
-     * @param ProductRepositoryInterface $productRepository
+     * @param ProductCollectionFactory $productCollectionFactory
      * @param BuyRequestBuilder $buyRequestBuilder
      */
     public function __construct(
-        ProductRepositoryInterface $productRepository,
-        BuyRequestBuilder $buyRequestBuilder
+        ProductCollectionFactory $productCollectionFactory,
+        BuyRequestBuilder        $buyRequestBuilder
     ) {
-        $this->productRepository = $productRepository;
+        $this->productCollectionFactory = $productCollectionFactory;
         $this->buyRequestBuilder = $buyRequestBuilder;
     }
 
@@ -55,12 +56,13 @@ public function execute(Quote $cart, array $cartItemData): void
         $cartItemData['model'] = $cart;
         $sku = $this->extractSku($cartItemData);
 
-        try {
-            $product = $this->productRepository->get($sku, false, null, true);
-        } catch (NoSuchEntityException $e) {
-            throw new GraphQlNoSuchEntityException(__('Could not find a product with SKU "%sku"', ['sku' => $sku]));
-        }
-        if (!in_array($cart->getStoreId(), $product->getStoreIds())) {
+        $productCollection = $this->productCollectionFactory->create()
+            ->addAttributeToFilter(ProductInterface::SKU, $sku)
+            ->addWebsiteFilter([$cart->getStore()->getWebsiteId()])
+            ->load();
+        /** @var ProductInterface $product */
+        $product = $productCollection->getFirstItem();
+        if (!$product->getId()) {
             throw new GraphQlNoSuchEntityException(__('Could not find a product with SKU "%sku"', ['sku' => $sku]));
         }