PWA-806: Localize emails sent through GraphQL application

Author: davemacaulay

app/code/Magento/GraphQl/Plugin/StoreResolver.php

@@ -4,12 +4,10 @@
 
 use Magento\Framework\App\AreaInterface;
 use Magento\Framework\App\AreaList;
-use Magento\Framework\App\ObjectManager;
 use Magento\Framework\App\State;
-use Magento\Store\Model\StoreManagerInterface;
 
 /**
- * Load translations for GraphQL requests
+ * Load translations on the first instance of a translated string
  */
 class TranslationLoader
 {
@@ -23,6 +21,11 @@ class TranslationLoader
      */
     private $appState;
 
+    /**
+     * @var bool
+     */
+    private $translationsLoaded = false;
+
     /**
      * @param AreaList $areaList
      * @param State $appState
@@ -36,11 +39,16 @@ public function __construct(
     }
 
     /**
-     * Before rendering any string ensure the translation aspect of area is loaded
+     * Before render of any localized string ensure the translation data is loaded
+     *
+     * @throws \Magento\Framework\Exception\LocalizedException
      */
-    public function beforeRender(\Magento\Framework\Phrase $subject)
+    public function beforeRender()
     {
-        $area = $this->areaList->getArea($this->appState->getAreaCode());
-        $area->load(AreaInterface::PART_TRANSLATE);
+        if ($this->translationsLoaded === false) {
+            $area = $this->areaList->getArea($this->appState->getAreaCode());
+            $area->load(AreaInterface::PART_TRANSLATE);
+            $this->translationsLoaded = true;
+        }
     }
 }

app/code/Magento/GraphQl/Plugin/TranslationLoader.php

@@ -44,7 +44,4 @@
     <type name="Magento\Framework\Phrase">
         <plugin name="graphQlLocalizationPhrasePlugin" type="Magento\GraphQl\Plugin\TranslationLoader" />
     </type>
-    <type name="Magento\Store\Model\StoreResolver">
-        <plugin name="graphQlLocalizationStoreResolver" type="Magento\GraphQl\Plugin\StoreResolver" />
-    </type>
 </config>

app/code/Magento/GraphQl/etc/graphql/di.xml

@@ -0,0 +1,44 @@
+<?php
+
+namespace Magento\StoreGraphQl\Plugin;
+
+use Magento\Framework\App\Request\Http as HttpRequest;
+use Magento\Framework\App\RequestInterface;
+use Magento\Store\Model\Resolver\Store;
+
+/**
+ * Ensure the store resolver gets the correct scope based on the GraphQl header
+ */
+class StoreResolver
+{
+    /**
+     * @var RequestInterface|HttpRequest
+     */
+    private $request;
+
+    /**
+     * @param RequestInterface $request
+     */
+    public function __construct(
+        RequestInterface $request
+    ) {
+        $this->request = $request;
+    }
+
+    /**
+     * If no scope is provided and there is a Store header, ensure the correct store code is used
+     *
+     * @param Store $subject
+     * @param null $scopeId
+     * @return array
+     * @SuppressWarnings(PHPMD.UnusedFormalParameter)
+     */
+    public function beforeGetScope(Store $subject, $scopeId = null)
+    {
+        $storeCode = $this->request->getHeader('Store');
+
+        if ($scopeId === null && $storeCode) {
+            return [$storeCode];
+        }
+    }
+}

app/code/Magento/StoreGraphQl/Plugin/StoreResolver.php

@@ -23,4 +23,7 @@
             </argument>
         </arguments>
     </type>
+    <type name="Magento\Store\Model\Resolver\Store">
+        <plugin name="graphQlLocalizationStoreResolver" type="Magento\StoreGraphQl\Plugin\StoreResolver" />
+    </type>
 </config>