zulip-desktop/app/main/autoupdater.ts

import {app, dialog, session} from 'electron';
import util from 'util';

import isDev from 'electron-is-dev';
import log from 'electron-log';
import {UpdateDownloadedEvent, UpdateInfo, autoUpdater} from 'electron-updater';

import * as ConfigUtil from '../renderer/js/utils/config-util';
import * as LinkUtil from '../renderer/js/utils/link-util';

import {linuxUpdateNotification} from './linuxupdater';	// Required only in case of linux

const sleep = util.promisify(setTimeout);

export async function appUpdater(updateFromMenu = false): Promise<void> {
	// Don't initiate auto-updates in development
	if (isDev) {
		return;
	}

	if (process.platform === 'linux' && !process.env.APPIMAGE) {
		const ses = session.fromPartition('persist:webviewsession');
		await linuxUpdateNotification(ses);
		return;
	}

	let updateAvailable = false;

	// Create Logs directory
	const LogsDir = `${app.getPath('userData')}/Logs`;

	// Log whats happening
	log.transports.file.file = `${LogsDir}/updates.log`;
	log.transports.file.level = 'info';
	autoUpdater.logger = log;

	// Handle auto updates for beta/pre releases
	const isBetaUpdate = ConfigUtil.getConfigItem('betaUpdate');

	autoUpdater.allowPrerelease = isBetaUpdate || false;

	const eventsListenerRemove = ['update-available', 'update-not-available'];
	autoUpdater.on('update-available', async (info: UpdateInfo) => {
		if (updateFromMenu) {
			updateAvailable = true;

			// This is to prevent removal of 'update-downloaded' and 'error' event listener.
			eventsListenerRemove.forEach(event => {
				autoUpdater.removeAllListeners(event);
			});

			await dialog.showMessageBox({
				message: `A new version ${info.version}, of Zulip Desktop is available`,
				detail: 'The update will be downloaded in the background. You will be notified when it is ready to be installed.'
			});
		}
	});

	autoUpdater.on('update-not-available', async () => {
		if (updateFromMenu) {
			// Remove all autoUpdator listeners so that next time autoUpdator is manually called these
			// listeners don't trigger multiple times.
			autoUpdater.removeAllListeners();

			await dialog.showMessageBox({
				message: 'No updates available',
				detail: `You are running the latest version of Zulip Desktop.\nVersion: ${app.getVersion()}`
			});
		}
	});

	autoUpdater.on('error', async (error: Error) => {
		if (updateFromMenu) {
			// Remove all autoUpdator listeners so that next time autoUpdator is manually called these
			// listeners don't trigger multiple times.
			autoUpdater.removeAllListeners();

			const messageText = (updateAvailable) ? ('Unable to download the updates') : ('Unable to check for updates');
			const {response} = await dialog.showMessageBox({
				type: 'error',
				buttons: ['Manual Download', 'Cancel'],
				message: messageText,
				detail: `Error: ${error.message}\n\nThe latest version of Zulip Desktop is available at -\nhttps://zulip.com/apps/.\n
				Current Version: ${app.getVersion()}`
			});
			if (response === 0) {
				await LinkUtil.openBrowser(new URL('https://zulip.com/apps/'));
			}
		}
	});

	// Ask the user if update is available
	autoUpdater.on('update-downloaded', async (event: UpdateDownloadedEvent) => {
		// Ask user to update the app
		const {response} = await dialog.showMessageBox({
			type: 'question',
			buttons: ['Install and Relaunch', 'Install Later'],
			defaultId: 0,
			message: `A new update ${event.version} has been downloaded`,
			detail: 'It will be installed the next time you restart the application'
		});
		if (response === 0) {
			await sleep(1000);
			autoUpdater.quitAndInstall();
			// Force app to quit. This is just a workaround, ideally autoUpdater.quitAndInstall() should relaunch the app.
			app.quit();
		}
	});
	// Init for updates
	await autoUpdater.checkForUpdates();
}