Add callback registration for frontend

Comfy-Org · Dec 1, 2024 · a03569a · a03569a
1 parent d90615e
commit a03569a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/preload.ts b/src/preload.ts
@@ -215,6 +215,12 @@ const electronAPI = {
   restoreCustomNodes: () => {
     return ipcRenderer.invoke(IPC_CHANNELS.RESTORE_CUSTOM_NODES);
   },
+  onRestoreCustomNodes: (callback: (message: { total: number, index?: number, exitCode?: number }) => void) => {
+    ipcRenderer.on(IPC_CHANNELS.RESTORE_CUSTOM_NODES_PROGRESS, (_event, value) => {
+      console.info(`Received ${IPC_CHANNELS.RESTORE_CUSTOM_NODES_PROGRESS} event`, value);
+      callback(value);
+    });
+  },
 } as const;
 
 export type ElectronAPI = typeof electronAPI;