atom/packages/about/lib/update-manager.js

const { Emitter, CompositeDisposable } = require('atom');

const Unsupported = 'unsupported';
const Idle = 'idle';
const CheckingForUpdate = 'checking';
const DownloadingUpdate = 'downloading';
const UpdateAvailableToInstall = 'update-available';
const UpToDate = 'no-update-available';
const ErrorState = 'error';

let UpdateManager = class UpdateManager {
  constructor() {
    this.emitter = new Emitter();
    this.currentVersion = atom.getVersion();
    this.availableVersion = atom.getVersion();
    this.resetState();
    this.listenForAtomEvents();
  }

  listenForAtomEvents() {
    this.subscriptions = new CompositeDisposable();

    this.subscriptions.add(
      atom.autoUpdater.onDidBeginCheckingForUpdate(() => {
        this.setState(CheckingForUpdate);
      }),
      atom.autoUpdater.onDidBeginDownloadingUpdate(() => {
        this.setState(DownloadingUpdate);
      }),
      atom.autoUpdater.onDidCompleteDownloadingUpdate(({ releaseVersion }) => {
        this.setAvailableVersion(releaseVersion);
      }),
      atom.autoUpdater.onUpdateNotAvailable(() => {
        this.setState(UpToDate);
      }),
      atom.autoUpdater.onUpdateError(() => {
        this.setState(ErrorState);
      }),
      atom.config.observe('core.automaticallyUpdate', value => {
        this.autoUpdatesEnabled = value;
        this.emitDidChange();
      })
    );

    // TODO: When https://github.com/atom/electron/issues/4587 is closed we can add this support.
    // atom.autoUpdater.onUpdateAvailable =>
    //   @find('.about-updates-item').removeClass('is-shown')
    //   @updateAvailable.addClass('is-shown')
  }

  dispose() {
    this.subscriptions.dispose();
  }

  onDidChange(callback) {
    return this.emitter.on('did-change', callback);
  }

  emitDidChange() {
    this.emitter.emit('did-change');
  }

  getAutoUpdatesEnabled() {
    return (
      this.autoUpdatesEnabled && this.state !== UpdateManager.State.Unsupported
    );
  }

  setAutoUpdatesEnabled(enabled) {
    return atom.config.set('core.automaticallyUpdate', enabled);
  }

  getErrorMessage() {
    return atom.autoUpdater.getErrorMessage();
  }

  getState() {
    return this.state;
  }

  setState(state) {
    this.state = state;
    this.emitDidChange();
  }

  resetState() {
    this.state = atom.autoUpdater.platformSupportsUpdates()
      ? atom.autoUpdater.getState()
      : Unsupported;
    this.emitDidChange();
  }

  getAvailableVersion() {
    return this.availableVersion;
  }

  setAvailableVersion(version) {
    this.availableVersion = version;

    if (this.availableVersion !== this.currentVersion) {
      this.state = UpdateAvailableToInstall;
    } else {
      this.state = UpToDate;
    }

    this.emitDidChange();
  }

  checkForUpdate() {
    atom.autoUpdater.checkForUpdate();
  }

  restartAndInstallUpdate() {
    atom.autoUpdater.restartAndInstallUpdate();
  }

  getReleaseNotesURLForCurrentVersion() {
    return this.getReleaseNotesURLForVersion(this.currentVersion);
  }

  getReleaseNotesURLForAvailableVersion() {
    return this.getReleaseNotesURLForVersion(this.availableVersion);
  }

  getReleaseNotesURLForVersion(appVersion) {
    // Dev versions will not have a releases page
    if (appVersion.indexOf('dev') > -1) {
      return 'https://atom.io/releases';
    }

    if (!appVersion.startsWith('v')) {
      appVersion = `v${appVersion}`;
    }

    const releaseRepo =
      appVersion.indexOf('nightly') > -1 ? 'atom-nightly-releases' : 'atom';
    return `https://github.com/atom/${releaseRepo}/releases/tag/${appVersion}`;
  }
};

UpdateManager.State = {
  Unsupported: Unsupported,
  Idle: Idle,
  CheckingForUpdate: CheckingForUpdate,
  DownloadingUpdate: DownloadingUpdate,
  UpdateAvailableToInstall: UpdateAvailableToInstall,
  UpToDate: UpToDate,
  Error: ErrorState
};

module.exports = UpdateManager;