3 files changed, 171 insertions, 76 deletions

diff --git a/packages/preload/src/contextBridge/SharedStoreConnector.ts b/packages/preload/src/contextBridge/SharedStoreConnector.ts
new file mode 100644
index 0000000..13f9ac1
--- /dev/null
+++ b/packages/preload/src/contextBridge/SharedStoreConnector.ts
@@ -0,0 +1,88 @@
+/*
+ * Copyright (C)  2021-2022 Kristóf Marussy <kristof@marussy.com>
+ *
+ * This file is part of Sophie.
+ *
+ * Sophie is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as
+ * published by the Free Software Foundation, version 3.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ *
+ * SPDX-License-Identifier: AGPL-3.0-only
+ */
+
+import {
+  Action,
+  MainToRendererIpcMessage,
+  RendererToMainIpcMessage,
+  SharedStoreListener,
+  SharedStoreSnapshotIn,
+} from '@sophie/shared';
+import { ipcRenderer } from 'electron';
+import log from 'loglevel';
+import type { IJsonPatch } from 'mobx-state-tree';
+
+export default class SharedStoreConnector {
+  private onSharedStoreChangeCalled = false;
+
+  private listener: SharedStoreListener | undefined;
+
+  constructor(private readonly allowReplaceListener: boolean) {
+    ipcRenderer.on(
+      MainToRendererIpcMessage.SharedStorePatch,
+      (_event, patch) => {
+        try {
+          // `mobx-state-tree` will validate the patch, so we can safely cast here.
+          this.listener?.onPatch(patch as IJsonPatch[]);
+        } catch (error) {
+          log.error('Shared store listener onPatch failed', error);
+          this.listener = undefined;
+        }
+      },
+    );
+  }
+
+  async onSharedStoreChange(listener: SharedStoreListener): Promise<void> {
+    if (this.onSharedStoreChangeCalled && !this.allowReplaceListener) {
+      throw new Error('Shared store change listener was already set');
+    }
+    this.onSharedStoreChangeCalled = true;
+    let success = false;
+    let snapshot: unknown;
+    try {
+      snapshot = await ipcRenderer.invoke(
+        RendererToMainIpcMessage.GetSharedStoreSnapshot,
+      );
+      success = true;
+    } catch (error) {
+      log.error('Failed to get initial shared store snapshot', error);
+    }
+    if (!success) {
+      throw new Error('Failed to connect to shared store');
+    }
+    // `mobx-state-tree` will validate the snapshot, so we can safely cast here.
+    listener.onSnapshot(snapshot as SharedStoreSnapshotIn);
+    this.listener = listener;
+  }
+}
+
+export function dispatchAction(actionToDispatch: Action): void {
+  // Let the full zod parse error bubble up to the main world,
+  // since all data it may contain was provided from the main world.
+  const parsedAction = Action.parse(actionToDispatch);
+  try {
+    ipcRenderer.send(RendererToMainIpcMessage.DispatchAction, parsedAction);
+  } catch (error) {
+    // Do not leak IPC failure details into the main world.
+    const message = 'Failed to dispatch action';
+    log.error(message, actionToDispatch, error);
+    throw new Error(message);
+  }
+}
diff --git a/packages/preload/src/contextBridge/TranslationsConnector.ts b/packages/preload/src/contextBridge/TranslationsConnector.ts
new file mode 100644
index 0000000..284b793
--- /dev/null
+++ b/packages/preload/src/contextBridge/TranslationsConnector.ts
@@ -0,0 +1,71 @@
+/*
+ * Copyright (C)  2022 Kristóf Marussy <kristof@marussy.com>
+ *
+ * This file is part of Sophie.
+ *
+ * Sophie is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as
+ * published by the Free Software Foundation, version 3.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ *
+ * SPDX-License-Identifier: AGPL-3.0-only
+ */
+
+import {
+  MainToRendererIpcMessage,
+  RendererToMainIpcMessage,
+  Translation,
+} from '@sophie/shared';
+import { ipcRenderer } from 'electron';
+import type { ResourceKey } from 'i18next';
+import log from 'loglevel';
+
+export default class TranslationsConnector {
+  private listener: (() => void) | undefined;
+
+  constructor(private readonly devMode: boolean) {
+    ipcRenderer.on(MainToRendererIpcMessage.ReloadTranslations, () => {
+      try {
+        this.listener?.();
+      } catch (error) {
+        log.error('Translations listener onReloadTranslations failed', error);
+        this.listener = undefined;
+      }
+    });
+  }
+
+  onReloadTranslations(listener: () => void): void {
+    if (!this.devMode) {
+      throw new Error(
+        'Translation reloading is only supported in development mode',
+      );
+    }
+    this.listener = listener;
+  }
+}
+
+export async function getTranslation(
+  translation: Translation,
+): Promise<ResourceKey> {
+  const parsedTranslation = Translation.parse(translation);
+  try {
+    // We don't have any way to validate translations,
+    // but they are coming from a trusted source and will be validated
+    // in the renderer anyways, so we should be fine.
+    return (await ipcRenderer.invoke(
+      RendererToMainIpcMessage.GetTranslation,
+      parsedTranslation,
+    )) as ResourceKey;
+  } catch (error) {
+    const message = 'Failed to get translation';
+    log.error(message, translation, error);
+    throw new Error(message);
+  }
+}
diff --git a/packages/preload/src/contextBridge/createSophieRenderer.ts b/packages/preload/src/contextBridge/createSophieRenderer.ts
index 8bdf07e..3de9d9e 100644
--- a/packages/preload/src/contextBridge/createSophieRenderer.ts
+++ b/packages/preload/src/contextBridge/createSophieRenderer.ts
@@ -18,85 +18,21 @@
  * SPDX-License-Identifier: AGPL-3.0-only
  */
 
-import {
-  Action,
-  MainToRendererIpcMessage,
-  RendererToMainIpcMessage,
-  SharedStoreListener,
-  SharedStoreSnapshotIn,
-  SophieRenderer,
-} from '@sophie/shared';
-import { ipcRenderer } from 'electron';
-import log from 'loglevel';
-import type { IJsonPatch } from 'mobx-state-tree';
+import { SophieRenderer } from '@sophie/shared';
 
-class SharedStoreConnector {
-  readonly #allowReplaceListener: boolean;
+import SharedStoreConnector, { dispatchAction } from './SharedStoreConnector';
+import TranslationsConnector, { getTranslation } from './TranslationsConnector';
 
-  #onSharedStoreChangeCalled = false;
-
-  #listener: SharedStoreListener | undefined;
-
-  constructor(allowReplaceListener: boolean) {
-    this.#allowReplaceListener = allowReplaceListener;
-    ipcRenderer.on(
-      MainToRendererIpcMessage.SharedStorePatch,
-      (_event, patch) => {
-        try {
-          // `mobx-state-tree` will validate the patch, so we can safely cast here.
-          this.#listener?.onPatch(patch as IJsonPatch[]);
-        } catch (error) {
-          log.error('Shared store listener onPatch failed', error);
-          this.#listener = undefined;
-        }
-      },
-    );
-  }
-
-  async onSharedStoreChange(listener: SharedStoreListener): Promise<void> {
-    if (this.#onSharedStoreChangeCalled && !this.#allowReplaceListener) {
-      throw new Error('Shared store change listener was already set');
-    }
-    this.#onSharedStoreChangeCalled = true;
-    let success = false;
-    let snapshot: unknown;
-    try {
-      snapshot = await ipcRenderer.invoke(
-        RendererToMainIpcMessage.GetSharedStoreSnapshot,
-      );
-      success = true;
-    } catch (error) {
-      log.error('Failed to get initial shared store snapshot', error);
-    }
-    if (!success) {
-      throw new Error('Failed to connect to shared store');
-    }
-    // `mobx-state-tree` will validate the snapshot, so we can safely cast here.
-    listener.onSnapshot(snapshot as SharedStoreSnapshotIn);
-    this.#listener = listener;
-  }
-}
-
-function dispatchAction(actionToDispatch: Action): void {
-  // Let the full zod parse error bubble up to the main world,
-  // since all data it may contain was provided from the main world.
-  const parsedAction = Action.parse(actionToDispatch);
-  try {
-    ipcRenderer.send(RendererToMainIpcMessage.DispatchAction, parsedAction);
-  } catch (error) {
-    // Do not leak IPC failure details into the main world.
-    const message = 'Failed to dispatch action';
-    log.error(message, actionToDispatch, error);
-    throw new Error(message);
-  }
-}
-
-export default function createSophieRenderer(
-  allowReplaceListener: boolean,
-): SophieRenderer {
-  const connector = new SharedStoreConnector(allowReplaceListener);
+export default function createSophieRenderer(devMode: boolean): SophieRenderer {
+  const sharedStoreConnector = new SharedStoreConnector(devMode);
+  const translationsConnector = new TranslationsConnector(devMode);
   return {
-    onSharedStoreChange: connector.onSharedStoreChange.bind(connector),
+    onSharedStoreChange:
+      sharedStoreConnector.onSharedStoreChange.bind(sharedStoreConnector),
     dispatchAction,
+    getTranslation,
+    onReloadTranslations: translationsConnector.onReloadTranslations.bind(
+      translationsConnector,
+    ),
   };
 }