Curry IpcMainEventChannel functions with WebContents where applicable

1 files changed, 45 insertions, 3 deletions

diff --git a/gui/src/main/ipc-event-channel.ts b/gui/src/main/ipc-event-channel.ts
index d883f52152..d189634950 100644
--- a/gui/src/main/ipc-event-channel.ts
+++ b/gui/src/main/ipc-event-channel.ts
@@ -1,6 +1,48 @@
-import { ipcMain } from 'electron';
+import { ipcMain, WebContents } from 'electron';
 
-import { createIpcMain } from '../shared/ipc-helpers';
+import { createIpcMain, IpcMain, Schema } from '../shared/ipc-helpers';
 import { ipcSchema } from '../shared/ipc-schema';
 
-export const IpcMainEventChannel = createIpcMain(ipcSchema, ipcMain);
+// Type where the notify functions have been replaced with either `undefined` if no `WebContents` is
+// available, or with the function curried with the `WebContents`.
+type IpcMainBootstrappedWithWebContents<S extends Schema> = {
+  [GK in keyof IpcMain<S>]: {
+    [CK in keyof IpcMain<S>[GK]]: CK extends `notify${string}`
+      ? undefined | ((arg: Parameters<IpcMain<S>[GK][CK]>[1]) => ReturnType<IpcMain<S>[GK][CK]>)
+      : IpcMain<S>[GK][CK];
+  };
+};
+
+const ipcMainFromSchema = createIpcMain(ipcSchema, ipcMain);
+// eslint-disable-next-line @typescript-eslint/naming-convention
+export let IpcMainEventChannel = bootstrapIpcMainWithWebContents();
+
+// Curries all notify functions with `WebContents` if it's not `undefined`. If it is `undefined`
+// then the whole function will be replaced with `undefined`.
+function bootstrapIpcMainWithWebContents(
+  webContents?: WebContents,
+): IpcMainBootstrappedWithWebContents<typeof ipcSchema> {
+  return Object.fromEntries(
+    Object.entries(ipcMainFromSchema).map(([groupKey, group]) => {
+      const newGroup = Object.fromEntries(
+        Object.entries(group).map(([callKey, call]) => {
+          if (callKey.startsWith('notify')) {
+            const newCall = webContents
+              ? (arg: Parameters<typeof call>[1]) => call(webContents, arg)
+              : undefined;
+            return [callKey, newCall];
+          } else {
+            return [callKey, call];
+          }
+        }),
+      );
+
+      return [groupKey, newGroup];
+    }),
+  ) as IpcMainBootstrappedWithWebContents<typeof ipcSchema>;
+}
+
+// Change the `IpcMainEventChannel` for a new one with a new `WebContents`.
+export function changeIpcWebContents(webContents?: WebContents) {
+  IpcMainEventChannel = bootstrapIpcMainWithWebContents(webContents);
+}