Refactor auto updating

Author: Mikunj

.editorconfig

@@ -5,3 +5,7 @@ charset = utf-8
 end_of_line = lf
 insert_final_newline = true
 trim_trailing_whitespace = true
+
+[*.js]
+indent_size = 2
+indent_style = space

package-lock.json

@@ -24,7 +24,6 @@
   "dependencies": {
     "axios": "^0.18.1",
     "electron-is-dev": "^1.0.1",
-    "electron-progressbar": "^1.2.0",
     "electron-updater": "^4.2.0",
     "electron-window-state": "^5.0.3",
     "flag-icon-css": "^3.3.0",

package.json

@@ -1,89 +1,91 @@
 import { dialog } from "electron";
 import isDev from "electron-is-dev";
 import { autoUpdater } from "electron-updater";
-import ProgressBar from "electron-progressbar";
 
-let progressBar = null;
 let isUpdating = false;
-let downloadAndInstall = false;
 
-function checkForUpdate(onQuitAndInstall) {
+async function checkForUpdate(getMainWindow, onQuitAndInstall) {
   // Disable for development
   if (isDev) {
     return;
   }
 
-  autoUpdater.logger = console;
-  autoUpdater.autoDownload = false;
-
-  autoUpdater.on("error", err => {
-    if (isUpdating) {
-      dialog.showErrorBox("Update Error: ", err == null ? "unknown" : err.message);
-      isUpdating = false;
-      console.error("Error in auto-updater.", err.message);
-    }
-  });
-
-  autoUpdater.on("update-available", info => {
-    console.log(`Update available: ${info.version}`);
+  if (isUpdating) {
+    return;
+  }
 
-    const message = `Update ${info.version} found. Do you want to download the update?`;
-    const detail = `View the release notes at: https://github.com/loki-project/loki-electron-gui-wallet/releases/tag/v${info.version}`;
+  autoUpdater.logger = console;
 
-    dialog.showMessageBox(
-      {
-        type: "info",
-        title: "Update available",
-        message,
-        detail,
-        buttons: ["Download and Install", "Download and Install Later", "No"],
-        defaultId: 0
-      },
-      buttonIndex => {
-        // Download and install
-        if (buttonIndex === 0) {
-          downloadAndInstall = true;
-          if (!progressBar) {
-            progressBar = new ProgressBar({
-              indeterminate: false,
-              title: "Downloading...",
-              text: `Downloading wallet v${info.version}`
-            });
-          }
-        }
+  try {
+    // Get the update using electron-updater
+    try {
+      const info = await autoUpdater.checkForUpdates();
+      if (!info.downloadPromise) {
+        console.info("auto-update: no update to download");
 
-        // Download
-        if (buttonIndex !== 2) {
-          isUpdating = true;
-          autoUpdater.downloadUpdate();
-        }
+        return;
       }
-    );
-  });
-
-  autoUpdater.on("download-progress", progress => {
-    if (progressBar) {
-      progressBar.value = progress.percent;
+      await info.downloadPromise;
+    } catch (error) {
+      await showCannotUpdateDialog(getMainWindow());
+      throw error;
     }
-  });
 
-  autoUpdater.on("update-downloaded", () => {
-    console.log("Update downloaded");
-    isUpdating = false;
-
-    if (progressBar) {
-      progressBar.setCompleted();
-      progressBar = null;
+    // Update downloaded successfully, we should ask the user to update
+    console.info("auto-update: showing update dialog...");
+    const shouldUpdate = await showUpdateDialog(getMainWindow());
+    if (!shouldUpdate) {
+      return;
     }
 
-    // If download and install was selected then quit and install
-    if (downloadAndInstall && onQuitAndInstall) {
+    console.info("auto-update: calling quitAndInstall...");
+    if (onQuitAndInstall) {
       onQuitAndInstall(autoUpdater);
-      downloadAndInstall = false;
     }
+  } catch (error) {
+    console.error("auto-update error:", getPrintableError(error));
+  } finally {
+    isUpdating = false;
+  }
+}
+
+function getPrintableError(error) {
+  return error && error.stack ? error.stack : error;
+}
+
+async function showUpdateDialog(mainWindow) {
+  const RESTART_BUTTON = 0;
+  const LATER_BUTTON = 1;
+  const options = {
+    type: "info",
+    buttons: ["Restart Wallet", "Later"],
+    title: "Loki Electron Wallet update available",
+    message: "There is a new version of Loki Electron Wallet available.",
+    detail: "Press Restart Wallet to apply the update",
+    defaultId: LATER_BUTTON,
+    cancelId: RESTART_BUTTON
+  };
+  return new Promise(resolve => {
+    dialog.showMessageBox(mainWindow, options, response => {
+      resolve(response === RESTART_BUTTON);
+    });
   });
+}
 
-  autoUpdater.checkForUpdates();
+async function showCannotUpdateDialog(mainWindow) {
+  const options = {
+    type: "error",
+    buttons: ["Ok"],
+    title: "Cannot update",
+    message:
+      "Loki Electron Wallet failed to update but there is a new version available. Please go to https://loki.network/ and install the new version manually."
+  };
+
+  return new Promise(resolve => {
+    dialog.showMessageBox(mainWindow, options, () => {
+      resolve();
+    });
+  });
 }
 
 export { checkForUpdate };

src-electron/main-process/auto-updater.js

@@ -155,18 +155,21 @@ function createWindow() {
 }
 
 app.on("ready", () => {
-  checkForUpdate(autoUpdater => {
-    if (mainWindow) {
-      mainWindow.webContents.send("showQuitScreen");
-    }
+  checkForUpdate(
+    () => mainWindow,
+    autoUpdater => {
+      if (mainWindow) {
+        mainWindow.webContents.send("showQuitScreen");
+      }
 
-    const promise = backend ? backend.quit() : Promise.resolve();
-    promise.then(() => {
-      installUpdate = true;
-      backend = null;
-      autoUpdater.quitAndInstall();
-    });
-  });
+      const promise = backend ? backend.quit() : Promise.resolve();
+      promise.then(() => {
+        installUpdate = true;
+        backend = null;
+        autoUpdater.quitAndInstall();
+      });
+    }
+  );
   if (process.platform === "darwin") {
     const menu = Menu.buildFromTemplate(menuTemplate);
     Menu.setApplicationMenu(menu);