[Refactor] Move registerIpcHandlers call out of getInstance (#177)

Comfy-Org · Nov 5, 2024 · 4d06e9b · 4d06e9b
1 parent 4da5a5d
commit 4d06e9b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/main.ts b/src/main.ts
@@ -181,6 +181,8 @@ if (!gotTheLock) {
         return;
       }
       downloadManager = DownloadManager.getInstance(mainWindow!, getModelsDirectory(basePath));
+      downloadManager.registerIpcHandlers();
+
       port = await findAvailablePort(8000, 9999).catch((err) => {
         log.error(`ERROR: Failed to find available port: ${err}`);
         throw err;

diff --git a/src/models/DownloadManager.ts b/src/models/DownloadManager.ts
@@ -258,12 +258,11 @@ export class DownloadManager {
   public static getInstance(mainWindow: BrowserWindow, modelsDirectory: string): DownloadManager {
     if (!DownloadManager.instance) {
       DownloadManager.instance = new DownloadManager(mainWindow, modelsDirectory);
-      DownloadManager.instance.registerIpcHandlers();
     }
     return DownloadManager.instance;
   }
 
-  private registerIpcHandlers() {
+  public registerIpcHandlers() {
     ipcMain.handle(IPC_CHANNELS.START_DOWNLOAD, (event, { url, path, filename }) =>
       this.startDownload(url, path, filename)
     );