Move auto-updater-win to ES6

This commit is contained in:
Samuel Attard 2016-07-26 11:40:55 +10:00
parent c730800f0c
commit 20e26a9639

View file

@ -1,17 +1,11 @@
'use strict'
const app = require('electron').app
const EventEmitter = require('events').EventEmitter
const {app} = require('electron')
const {EventEmitter} = require('events')
const squirrelUpdate = require('./squirrel-update-win')
const util = require('util')
function AutoUpdater () {
EventEmitter.call(this)
}
util.inherits(AutoUpdater, EventEmitter)
AutoUpdater.prototype.quitAndInstall = function () {
class AutoUpdater extends EventEmitter {
quitAndInstall () {
if (!this.updateAvailable) {
return this.emitError('No update available, can\'t quit and install')
}
@ -19,15 +13,15 @@ AutoUpdater.prototype.quitAndInstall = function () {
return app.quit()
}
AutoUpdater.prototype.getFeedURL = function () {
getFeedURL () {
return this.updateURL
}
AutoUpdater.prototype.setFeedURL = function (updateURL, headers) {
setFeedURL (updateURL, headers) {
this.updateURL = updateURL
}
AutoUpdater.prototype.checkForUpdates = function () {
checkForUpdates () {
if (!this.updateURL) {
return this.emitError('Update URL is not set')
}
@ -64,8 +58,9 @@ AutoUpdater.prototype.checkForUpdates = function () {
// Private: Emit both error object and message, this is to keep compatibility
// with Old APIs.
AutoUpdater.prototype.emitError = function (message) {
emitError (message) {
return this.emit('error', new Error(message), message)
}
}
module.exports = new AutoUpdater()