1 files changed, 51 insertions, 20 deletions

diff --git a/packages/main/src/stores/MainStore.ts b/packages/main/src/stores/MainStore.ts
index 93cabb9..d717bed 100644
--- a/packages/main/src/stores/MainStore.ts
+++ b/packages/main/src/stores/MainStore.ts
@@ -19,7 +19,8 @@
  */
 
 import type { Action, BrowserViewBounds } from '@sophie/shared';
-import { applySnapshot, Instance, types } from 'mobx-state-tree';
+import type { ResourceKey } from 'i18next';
+import { applySnapshot, flow, Instance, types } from 'mobx-state-tree';
 
 import type I18nStore from '../i18n/I18nStore';
 import type { UseTranslationResult } from '../i18n/I18nStore';
@@ -67,6 +68,12 @@ const MainStore = types
     useTranslation(ns?: string): UseTranslationResult {
       return self.i18n?.useTranslation(ns) ?? { ready: false };
     },
+    getTranslation(language: string, namespace: string): Promise<ResourceKey> {
+      if (self.i18n === undefined) {
+        throw new Error('i18next has not been set');
+      }
+      return self.i18n.getTranslation(language, namespace);
+    },
   }))
   .volatile(
     (): {
@@ -79,10 +86,43 @@ const MainStore = types
     setBrowserViewBounds(bounds: BrowserViewBounds): void {
       applySnapshot(self.browserViewBounds, bounds);
     },
+    setMainWindow(mainWindow: MainWindow | undefined): void {
+      self.mainWindow = mainWindow;
+    },
+    openWebpageInBrowser() {
+      getEnv(self).openURLInExternalBrowser(
+        'https://gitlab.com/say-hi-to-sophie/shophie',
+      );
+    },
+    openAboutDialog() {
+      getEnv(self).openAboutDialog();
+    },
+    beforeDestroy(): void {
+      self.mainWindow?.dispose();
+    },
+    setI18n(i18n: I18nStore): void {
+      self.i18n = i18n;
+    },
+    addMissingTranslation(
+      languages: string[],
+      namespace: string,
+      key: string,
+      value: string,
+    ): void {
+      self.i18n?.addMissingTranslation(languages, namespace, key, value);
+    },
+    reloadTranslations: flow(function* reloadTranslations() {
+      if (self.i18n !== undefined) {
+        yield self.i18n.reloadTranslations();
+        self.mainWindow?.reloadTranslations();
+      }
+    }),
+  }))
+  .actions((self) => ({
     dispatch(action: Action): void {
       switch (action.action) {
         case 'set-browser-view-bounds':
-          this.setBrowserViewBounds(action.browserViewBounds);
+          self.setBrowserViewBounds(action.browserViewBounds);
           break;
         case 'set-selected-service-id':
           self.settings.setSelectedServiceId(action.serviceId);
@@ -98,11 +138,19 @@ const MainStore = types
           break;
         case 'reload-all-translations':
           if (self.i18n !== undefined) {
-            self.i18n.reloadTranslations().catch((error) => {
+            self.reloadTranslations().catch((error) => {
               log.error('Failed to reload translations', error);
             });
           }
           break;
+        case 'add-missing-translation':
+          self.addMissingTranslation(
+            action.languages,
+            action.namespace,
+            action.key,
+            action.value,
+          );
+          break;
         case 'dispatch-service-action': {
           const { serviceId, serviceAction } = action;
           const service = self.shared.servicesById.get(serviceId);
@@ -123,23 +171,6 @@ const MainStore = types
           break;
       }
     },
-    setMainWindow(mainWindow: MainWindow | undefined): void {
-      self.mainWindow = mainWindow;
-    },
-    openWebpageInBrowser() {
-      getEnv(self).openURLInExternalBrowser(
-        'https://gitlab.com/say-hi-to-sophie/shophie',
-      );
-    },
-    openAboutDialog() {
-      getEnv(self).openAboutDialog();
-    },
-    beforeDestroy(): void {
-      self.mainWindow?.dispose();
-    },
-    setI18n(i18n: I18nStore): void {
-      self.i18n = i18n;
-    },
   }));
 
 /*