From c19296efb753167619a66532ecb6481339415b9d Mon Sep 17 00:00:00 2001
From: Ben Ogle <ogle.ben@gmail.com>
Date: Thu, 25 Feb 2016 16:01:23 -0800
Subject: [PATCH] Add isEnabled function to AutoUpdateManager

---
 spec/auto-update-manager-spec.js | 32 ++++++++++++++++++++++++++++++++
 src/auto-update-manager.js       | 27 +++++++++++++++++++++++++--
 2 files changed, 57 insertions(+), 2 deletions(-)

diff --git a/spec/auto-update-manager-spec.js b/spec/auto-update-manager-spec.js
index 499e2cb3b..6e3c037ef 100644
--- a/spec/auto-update-manager-spec.js
+++ b/spec/auto-update-manager-spec.js
@@ -63,6 +63,38 @@ describe('AutoUpdateManager (renderer)', () => {
     })
   })
 
+  describe('::isEnabled', () => {
+    let platform, releaseChannel
+    it('returns true on OS X and Windows, when in stable', () => {
+      spyOn(autoUpdateManager, 'getPlatform').andCallFake(() =>  platform)
+      spyOn(autoUpdateManager, 'getReleaseChannel').andCallFake(() => releaseChannel)
+
+      platform = 'win32'
+      releaseChannel = 'stable'
+      expect(autoUpdateManager.isEnabled()).toBe(true)
+
+      platform = 'win32'
+      releaseChannel = 'dev'
+      expect(autoUpdateManager.isEnabled()).toBe(false)
+
+      platform = 'darwin'
+      releaseChannel = 'stable'
+      expect(autoUpdateManager.isEnabled()).toBe(true)
+
+      platform = 'darwin'
+      releaseChannel = 'dev'
+      expect(autoUpdateManager.isEnabled()).toBe(false)
+
+      platform = 'linux'
+      releaseChannel = 'stable'
+      expect(autoUpdateManager.isEnabled()).toBe(false)
+
+      platform = 'linux'
+      releaseChannel = 'dev'
+      expect(autoUpdateManager.isEnabled()).toBe(false)
+    })
+  })
+
   describe('::dispose', () => {
     it('subscribes to "update-not-available" event', () => {
       const spy = jasmine.createSpy('spy')
diff --git a/src/auto-update-manager.js b/src/auto-update-manager.js
index e7852f2d9..e99ce83d9 100644
--- a/src/auto-update-manager.js
+++ b/src/auto-update-manager.js
@@ -31,6 +31,18 @@ export default class AutoUpdateManager {
     this.emitter.dispose()
   }
 
+  checkForUpdate () {
+    ipcRenderer.send('check-for-update')
+  }
+
+  quitAndInstallUpdate () {
+    ipcRenderer.send('install-update')
+  }
+
+  isEnabled () {
+    return this.getReleaseChannel() == 'stable' && (this.getPlatform() === 'darwin' || this.getPlatform() === 'win32')
+  }
+
   onDidBeginCheckingForUpdate (callback) {
     return this.emitter.on('did-begin-checking-for-update', callback)
   }
@@ -47,7 +59,18 @@ export default class AutoUpdateManager {
     return this.emitter.on('update-not-available', callback)
   }
 
-  checkForUpdate () {
-    ipcRenderer.send('check-for-update')
+  getPlatform () {
+    return process.platform
+  }
+
+  // TODO: We should move this into atom env or something.
+  getReleaseChannel () {
+    let version = atom.getVersion()
+    if (version.indexOf('beta') > -1) {
+      return 'beta'
+    } else if (version.indexOf('dev') > -1) {
+      return 'dev'
+    }
+    return 'stable'
   }
 }