Move buildMenu to appWindow (#277)

src/main-process/appWindow.ts

@@ -1,4 +1,4 @@
-import { BrowserWindow, screen, app, shell, ipcMain, Tray, Menu } from 'electron';
+import { BrowserWindow, screen, app, shell, ipcMain, Tray, Menu, dialog, MenuItem } from 'electron';
 import path from 'node:path';
 import Store from 'electron-store';
 import { StoreType } from '../store';
@@ -46,6 +46,7 @@ export class AppWindow {
     this.setupWindowEvents();
     this.sendQueuedEventsOnReady();
     this.setupTray();
+    this.buildMenu();
   }
 
   public isReady(): boolean {
@@ -206,4 +207,35 @@ export class AppWindow {
     // If we want to make it more dynamic return tray so we can access it later
     return tray;
   }
+
+  buildMenu() {
+    const menu = Menu.getApplicationMenu();
+    if (menu) {
+      const aboutMenuItem = {
+        label: 'About ComfyUI',
+        click: () => {
+          dialog.showMessageBox({
+            title: 'About',
+            message: `ComfyUI v${app.getVersion()}`,
+            detail: 'Created by Comfy Org\nCopyright © 2024',
+            buttons: ['OK'],
+          });
+        },
+      };
+      const helpMenuItem = menu.items.find((item) => item.role === 'help');
+      if (helpMenuItem && helpMenuItem.submenu) {
+        helpMenuItem.submenu.append(new MenuItem(aboutMenuItem));
+        Menu.setApplicationMenu(menu);
+      } else {
+        // If there's no Help menu, add one
+        menu.append(
+          new MenuItem({
+            label: 'Help',
+            submenu: [aboutMenuItem],
+          })
+        );
+        Menu.setApplicationMenu(menu);
+      }
+    }
+  }
 }

src/main.ts

@@ -13,7 +13,6 @@ import { DownloadManager } from './models/DownloadManager';
 import { findAvailablePort, getModelsDirectory, rotateLogFiles } from './utils';
 import { ComfySettings } from './config/comfySettings';
 import dotenv from 'dotenv';
-import { buildMenu } from './menu/menu';
 import { ComfyConfigManager } from './config/comfyConfigManager';
 import { AppWindow } from './main-process/appWindow';
 import { getAppResourcesPath, getBasePath, getPythonInstallPath } from './install/resourcePaths';
@@ -147,7 +146,7 @@ if (!gotTheLock) {
     log.info('App ready');
 
     try {
-      createWindow();
+      appWindow = new AppWindow();
       new PathHandlers().registerHandlers();
       new AppInfoHandlers().registerHandlers();
 
@@ -259,14 +258,6 @@ function restartApp({ customMessage, delay }: { customMessage?: string; delay?:
     });
 }
 
-/**
- * Creates the main application window.
- */
-export const createWindow = (): void => {
-  appWindow = new AppWindow();
-  buildMenu();
-};
-
 const isComfyServerReady = async (host: string, port: number): Promise<boolean> => {
   const url = `http://${host}:${port}/queue`;