Dialog before update

Author: Emmanouil Konstantinidis

main.js

@@ -7,36 +7,13 @@ require('crash-reporter').start();
 
 var Menu = require('menu');
 var Tray = require('tray');
-var BrowserWindow = require('browser-window');
 var AutoLaunch = require('auto-launch');
+var BrowserWindow = require('browser-window');
+var dialog = require('dialog');
 
 var iconIdle = path.join(__dirname, 'images', 'tray-idleTemplate.png');
 var iconActive = path.join(__dirname, 'images', 'tray-active.png');
 
-autoUpdater
-  .on('error', function(event, message) {
-    console.log('ERRORED.');
-    console.log('Event: ' + JSON.stringify(event) + '. MESSAGE: ' + message);
-  })
-  .on('checking-for-update', function () {
-    console.log('Checking for update');
-  })
-  .on('update-available', function () {
-    console.log('Update available');
-  })
-  .on('update-not-available', function () {
-    console.log('Update not available');
-  })
-  .on('update-downloaded', function (event, releaseNotes, releaseName,
-    releaseDate, updateUrl, quitAndUpdate) {
-    console.log('Update downloaded');
-    quitAndUpdate();
-  });
-
-autoUpdater.setFeedUrl('https://raw.githubusercontent.com/' +
-  'ekonstantinidis/gitify/master/auto-update.json');
-autoUpdater.checkForUpdates();
-
 var autoStart = new AutoLaunch({
   name: 'Gitify',
   path: process.execPath.match(/.*?\.app/)[0]
@@ -70,6 +47,7 @@ app.on('ready', function(){
     appIcon.window.setVisibleOnAllWorkspaces(true);
 
     initMenu();
+    checkAutoUpdate();
   }
 
   function showWindow (bounds) {
@@ -114,6 +92,51 @@ app.on('ready', function(){
     appIcon.window.hide();
   }
 
+  function checkAutoUpdate() {
+
+    autoUpdater
+      .on('error', function(event, message) {
+        console.log('ERRORED.');
+        console.log('Event: ' + JSON.stringify(event) + '. MESSAGE: ' + message);
+      })
+      .on('checking-for-update', function () {
+        console.log('Checking for update');
+      })
+      .on('update-available', function () {
+        console.log('Update available');
+      })
+      .on('update-not-available', function () {
+        console.log('Update not available');
+        app.dock.hide();
+      })
+      .on('update-downloaded', function (event, releaseNotes, releaseName,
+        releaseDate, updateUrl, quitAndUpdate) {
+        console.log('Update downloaded');
+        confirmAutoUpdate(quitAndUpdate);
+      });
+
+    autoUpdater.setFeedUrl('https://raw.githubusercontent.com/' +
+      'ekonstantinidis/gitify/master/auto-update.json');
+    autoUpdater.checkForUpdates();
+  }
+
+  function confirmAutoUpdate(quitAndUpdate) {
+    dialog.showMessageBox({
+      type: 'question',
+      buttons: ['Update & Restart', 'Cancel'],
+      title: 'Update Available',
+      cancelId: 99,
+      message: 'There is an update available. Would you like to update the app now?'
+    }, function (response) {
+        console.log('Exit: ' + response);
+        app.dock.hide();
+        if (response === 0) {
+          quitAndUpdate();
+        }
+      }
+    );
+  }
+
   ipc.on('reopen-window', function() {
     appIcon.window.show();
   });
@@ -138,6 +161,5 @@ app.on('ready', function(){
     app.quit();
   });
 
-  app.dock.hide();
   appIcon.setToolTip('GitHub Notifications on your menu bar.');
 });