B2B-2404: Optimize Product Permissions in GraphQL resolver

- fix website visibility issue

Author: joanhe

app/code/Magento/CatalogGraphQl/Model/Resolver/Products/DataProvider/Product/CollectionProcessor/VisibilityStatusProcessor.php

@@ -35,10 +35,15 @@ public function process(
         array $attributeNames,
         ContextInterface $context = null
     ): Collection {
-        $websiteId = $context->getExtensionAttributes()->getStore()->getWebsiteId();
         $collection->joinAttribute('status', 'catalog_product/status', 'entity_id', null, 'inner');
         $collection->joinAttribute('visibility', 'catalog_product/visibility', 'entity_id', null, 'inner');
-        $collection->addWebsiteFilter([$websiteId]);
+        if ($context) {
+            $store = $context->getExtensionAttributes()->getStore();
+            if ($store) {
+                $websiteId = $store->getWebsiteId();
+                $collection->addWebsiteFilter([$websiteId]);
+            }
+        }
 
         return $collection;
     }

app/code/Magento/RelatedProductGraphQl/Model/Resolver/Batch/AbstractLikedProducts.php

@@ -80,10 +80,15 @@ abstract protected function getLinkType(): int;
      * @param \Magento\Catalog\Api\Data\ProductInterface[] $products
      * @param string[] $loadAttributes
      * @param int $linkType
+     * @param ContextInterface $context
      * @return \Magento\Catalog\Api\Data\ProductInterface[][]
      */
-    private function findRelations(array $products, array $loadAttributes, int $linkType): array
-    {
+    private function findRelations(
+        array $products,
+        array $loadAttributes,
+        int $linkType,
+        ContextInterface $context
+    ): array {
         //Loading relations
         $relations = $this->relatedProductDataProvider->getRelations($products, $linkType);
         if (!$relations) {
@@ -96,7 +101,8 @@ private function findRelations(array $products, array $loadAttributes, int $link
             $this->searchCriteriaBuilder->create(),
             $loadAttributes,
             false,
-            true
+            true,
+            $context
         );
         //Filling related products map.
         /** @var \Magento\Catalog\Api\Data\ProductInterface[] $relatedProducts */
@@ -144,7 +150,7 @@ public function resolve(ContextInterface $context, Field $field, array $requests
         $fields = array_unique(array_merge([], ...$fields));
 
         //Finding relations.
-        $related = $this->findRelations($products, $fields, $this->getLinkType());
+        $related = $this->findRelations($products, $fields, $this->getLinkType(), $context);
 
         //Matching requests with responses.
         $response = new BatchResponse();