From 9c883eebc66704684e337a7cb3796f9451089eb4 Mon Sep 17 00:00:00 2001
From: Manav Rathi <manav@mrmr.io>
Date: Thu, 1 Aug 2024 15:27:14 +0530
Subject: [PATCH] [desktop] Handle logout for utility process

---
 desktop/src/preload.ts | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/desktop/src/preload.ts b/desktop/src/preload.ts
index 3058a6376f..8472e91ff0 100644
--- a/desktop/src/preload.ts
+++ b/desktop/src/preload.ts
@@ -62,6 +62,7 @@ import { contextBridge, ipcRenderer, webUtils } from "electron/renderer";
 
 // While we can't import other code, we can import types since they're just
 // needed when compiling and will not be needed or looked around for at runtime.
+import type { IpcRendererEvent } from "electron";
 import type {
     AppUpdate,
     CollectionMapping,
@@ -199,13 +200,15 @@ const ffmpegExec = (
 // - ML
 
 const createMLWorker = () => {
-    ipcRenderer.send("createMLWorker");
-    ipcRenderer.on("createMLWorker/port", (event) => {
+    const l = (event: IpcRendererEvent) => {
         void windowLoaded.then(() => {
             // "*"" is the origin to send to.
             window.postMessage("createMLWorker/port", "*", event.ports);
+            ipcRenderer.off("createMLWorker/port", l);
         });
-    });
+    };
+    ipcRenderer.on("createMLWorker/port", l);
+    ipcRenderer.send("createMLWorker");
 };
 
 // - Watch