Enable AutoUpdates from in-app with self-sign certificates on macOS (#213)

Co-authored-by: Alphrag <alphrag@pm.me>

2 files changed, 23 insertions, 7 deletions

diff --git a/src/electron/ipc-api/autoUpdate.ts b/src/electron/ipc-api/autoUpdate.ts
index d20a413b3..fb91d3515 100644
--- a/src/electron/ipc-api/autoUpdate.ts
+++ b/src/electron/ipc-api/autoUpdate.ts
@@ -1,6 +1,7 @@
-import { app, ipcMain, BrowserWindow } from 'electron';
+import { ipcMain, BrowserWindow } from 'electron';
 import { autoUpdater } from 'electron-updater';
 import { isMac, isWindows } from '../../environment';
+import { appEvents } from '../..';
 
 const debug = require('../../preload-safe-debug')('Ferdium:ipcApi:autoUpdate');
 
@@ -24,11 +25,13 @@ export default (params: { mainWindow: BrowserWindow; settings: any }) => {
             autoUpdater.checkForUpdates();
           } else if (args.action === 'install') {
             debug('installing update');
-            app.removeAllListeners('window-all-closed');
-            params.mainWindow.removeAllListeners('close');
+
+            appEvents.emit('install-update');
+
+            const openedWindows = BrowserWindow.getAllWindows();
+            for (const window of openedWindows)  window.close();
+
             autoUpdater.quitAndInstall();
-            // TODO: based on https://github.com/electron-userland/electron-builder/issues/6058#issuecomment-1130344017 (not yet tested since we don't have signed builds yet for macos)
-            app.exit();
           }
         } catch (error) {
           event.sender.send('autoUpdate', { error });
diff --git a/src/index.ts b/src/index.ts
index 59a90a3cb..fad6b9b69 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -15,6 +15,7 @@ import { join } from 'path';
 import windowStateKeeper from 'electron-window-state';
 import minimist from 'minimist';
 import ms from 'ms';
+import { EventEmitter } from 'events';
 import { enableWebContents, initializeRemote } from './electron-util';
 import { enforceMacOSAppLocation } from './enforce-macos-app-location';
 
@@ -56,6 +57,9 @@ app.userAgentFallback = userAgent();
 // be closed automatically when the JavaScript object is garbage collected.
 let mainWindow: BrowserWindow | undefined;
 let willQuitApp = false;
+let overrideAppQuitForUpdate = false;
+
+export const appEvents = new EventEmitter();
 
 // Register methods to be called once the window has been loaded.
 let onDidLoadFns: any[] | null = [];
@@ -320,7 +324,8 @@ const createWindow = () => {
         debug('Window: hide');
         mainWindow?.hide();
       }
-    } else {
+    } else if (!overrideAppQuitForUpdate) {
+      debug('Quitting the app');
       dbus.stop();
       app.quit();
     }
@@ -653,12 +658,20 @@ app.on('window-all-closed', () => {
     )
   ) {
     debug('Window: all windows closed, quit app');
-    app.quit();
+    if (!overrideAppQuitForUpdate) {
+      // TODO: based on https://github.com/electron-userland/electron-builder/issues/6058#issuecomment-1130344017 (not yet tested since we don't have signed builds yet for macos)
+      app.quit();
+    }
   } else {
     debug("Window: don't quit app");
   }
 });
 
+appEvents.on('install-update', () => {
+  willQuitApp = true;
+  overrideAppQuitForUpdate = true;
+});
+
 app.on('before-quit', event => {
   const yesButtonIndex = 0;
   let selection = yesButtonIndex;