From 63f7775804dd24ce45530e509c73b1974f72250b Mon Sep 17 00:00:00 2001
From: Martin Fournier <martin@mfournier.com>
Date: Thu, 16 Dec 2021 20:21:25 -0500
Subject: [PATCH] Remove process handler when doing a hard reload

---
 electron/main.js | 52 ++++++++++++++++++++++++++++--------------------
 1 file changed, 30 insertions(+), 22 deletions(-)

diff --git a/electron/main.js b/electron/main.js
index f155f1d54..7cd6c619f 100644
--- a/electron/main.js
+++ b/electron/main.js
@@ -9,22 +9,12 @@ if (greenworks.init()) {
 
 const debug = false;
 
-let win;
-let intervalID;
-
 function createWindow(killall) {
-  win = new BrowserWindow({
+  const win = new BrowserWindow({
     show: false,
     backgroundThrottling: false,
   });
 
-  win.on('closed', function() {
-    clearInterval(intervalID);
-    win = null;
-    app.quit();
-    process.exit(0);
-  });
-
   win.removeMenu();
   win.maximize();
   noScripts = killall ? { query: { noScripts: killall } } : {};
@@ -47,7 +37,7 @@ function createWindow(killall) {
   // This is backward but the game fills in an array called `document.achievements` and we retrieve it from
   // here. Hey if it works it works.
   const achievements = greenworks.getAchievementNames();
-  intervalID = setInterval(async () => {
+  const intervalID = setInterval(async () => {
     const achs = await win.webContents.executeJavaScript("document.achievements");
     console.log(achs);
     for (const ach of achs) {
@@ -55,6 +45,7 @@ function createWindow(killall) {
       greenworks.activateAchievement(ach, () => undefined);
     }
   }, 1000);
+  win.achievementsIntervalID = intervalID;
 
   // Create the Application's main menu
   Menu.setApplicationMenu(
@@ -84,6 +75,7 @@ function createWindow(killall) {
           {
             label: "reload & kill all scripts",
             click: () => {
+              setStopProcessHandler(app, win, false);
               if (intervalID) clearInterval(intervalID);
               win.webContents.forcefullyCrashRenderer();
               win.close();
@@ -119,19 +111,35 @@ function createWindow(killall) {
       },
     ]),
   );
+
+  return win;
 }
 
-app.on('window-all-closed', function () {
-  // On OS X it is common for applications and their menu bar
-  // to stay active until the user quits explicitly with Cmd + Q
-  if (process.platform !== 'darwin') {
-    clearInterval(intervalID);
-    win = null;
-    app.quit()
-    process.exit(0);
+function setStopProcessHandler(app, window, enabled) {
+  const clearWindowHandler = () => {
+    if (window.achievementsIntervalID) {
+      clearInterval(window.achievementsIntervalID);
+    }
+    window = null;
   }
-});
+
+  const stopProcessHandler = () => {
+    if (process.platform !== 'darwin') {
+      app.quit()
+      process.exit(0);
+    }
+  }
+
+  if (enabled) {
+    window.on('closed', clearWindowHandler);
+    app.on('window-all-closed', stopProcessHandler);
+  } else {
+    window.removeListener('closed', clearWindowHandler);
+    app.removeListener('window-all-closed', stopProcessHandler);
+  }
+}
 
 app.whenReady().then(() => {
-  createWindow(false);
+  const win = createWindow(false);
+  setStopProcessHandler(app, win, true);
 });