B2B-2221: Magento\CatalogGraphQl\Plugin\DesignLoader plugin loads too much data

- Changes on plugin are added

Author: avattam06

app/code/Magento/CatalogGraphQl/Plugin/DesignLoader.php

@@ -0,0 +1,92 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\CatalogGraphQl\Plugin;
+
+use Magento\Catalog\Model\Product;
+use Magento\Framework\Message\MessageInterface;
+use Magento\Framework\View\DesignLoader as ViewDesignLoader;
+use Magento\Framework\Message\ManagerInterface;
+use Magento\Catalog\Block\Product\ImageFactory;
+use Magento\Framework\App\AreaList;
+use Magento\Framework\App\State;
+
+/**
+ * Load necessary design files for GraphQL
+ */
+class DesignLoader
+{
+    /**
+     * @var DesignLoader
+     */
+    private $designLoader;
+
+    /**
+     * @var ManagerInterface
+     */
+    private ManagerInterface $messageManager;
+
+    /**
+     * Application
+     *
+     * @var AreaList
+     */
+    private AreaList $areaList;
+
+    /**
+     * Layout
+     *
+     * @var State
+     */
+    private State $appState;
+
+    /**
+     * @param ViewDesignLoader $designLoader
+     * @param ManagerInterface $messageManager
+     */
+    public function __construct(
+        ViewDesignLoader $designLoader,
+        ManagerInterface $messageManager,
+        AreaList $areaList,
+        State $appState
+    ) {
+        $this->designLoader = $designLoader;
+        $this->messageManager = $messageManager;
+        $this->areaList = $areaList;
+        $this->appState = $appState;
+    }
+
+    /**
+     * Before create load the design files
+     *
+     * @param ImageFactory $subject
+     * @param Product $product
+     * @param string $imageId
+     * @param array|null $attributes
+     *
+     * @SuppressWarnings(PHPMD.UnusedFormalParameter)
+     */
+    public function beforeCreate(
+        ImageFactory $subject,
+        Product $product,
+        string $imageId,
+        array $attributes = null
+    ) {
+        try {
+            $area = $this->areaList->getArea($this->appState->getAreaCode());
+            $area->load(\Magento\Framework\App\Area::PART_DESIGN);
+        } catch (\Magento\Framework\Exception\LocalizedException $e) {
+            if ($e->getPrevious() instanceof \Magento\Framework\Config\Dom\ValidationException) {
+                /** @var MessageInterface $message */
+                $message = $this->messageManager
+                    ->createMessage(MessageInterface::TYPE_ERROR)
+                    ->setText($e->getMessage());
+                $this->messageManager->addUniqueMessages([$message]);
+            }
+        }
+    }
+}