tabs/src/Application.ts

import {app, dialog, Menu, shell, Tray} from "electron";
import Meta from "./Meta";
import Config from "./Config";
import Updater from "./Updater";
import MainWindow from "./windows/MainWindow";
import * as os from "os";

export default class Application {
    private readonly devMode: boolean;
    private readonly config: Config;
    private readonly updater: Updater;
    private readonly mainWindow: MainWindow;
    private tray?: Tray;

    public constructor(devMode: boolean) {
        this.devMode = devMode;
        this.config = new Config();
        this.updater = new Updater(this.config, this);
        this.mainWindow = new MainWindow(this);
    }

    public async start(): Promise<void> {
        console.log('Starting app');

        this.setupSystemTray();
        this.mainWindow.setup();
        this.setupElectronTweaks();

        // Check for updates
        if (os.platform() === 'win32') {
            this.updater.checkAndPromptForUpdates(this.mainWindow.getWindow()).then(() => {
                console.log('Update check successful.');
            }).catch(console.error);
        }

        console.log('App started');
    }

    public async stop(): Promise<void> {
        this.mainWindow.teardown();
    }

    public getConfig(): Config {
        return this.config;
    }

    public getUpdater(): Updater {
        return this.updater;
    }

    public isDevMode(): boolean {
        return this.devMode;
    }

    public async openExternalLink(url: string): Promise<void> {
        if (url.startsWith('https://')) {
            console.log('Opening link', url);
            await shell.openExternal(url);
        } else {
            const {response} = await dialog.showMessageBox({
                message: 'Are you sure you want to open this link?\n' + url,
                type: 'question',
                buttons: ['Cancel', 'Open link'],
            });

            if (response === 1) {
                console.log('Opening link', url);
                await shell.openExternal(url);
            }
        }
    }

    private setupElectronTweaks() {
        // Open external links in default OS browser
        app.on('web-contents-created', (e, contents) => {
            if (contents.getType() === 'webview') {
                console.log('Setting external links to open in default OS browser');
                contents.setWindowOpenHandler(details => {
                    if (details.url.startsWith(details.referrer.url)) return {action: 'allow'};

                    const url = details.url;
                    this.openExternalLink(url)
                        .catch(console.error);
                    return {action: 'deny'};
                });
            }
        });

        // Disable unused features
        app.on('web-contents-created', (e, contents) => {
            contents.on('will-attach-webview', (e, webPreferences) => {
                delete webPreferences.preload;
                webPreferences.nodeIntegration = false;

                // TODO: Here would be a good place to filter accessed urls (params.src).
                //  Also consider 'will-navigate' event on contents.
            });
        });
    }

    private setupSystemTray() {
        console.log('Loading system Tray');
        this.tray = new Tray(Meta.ICON_PATH);
        this.tray.setToolTip('Tabs');
        this.tray.setContextMenu(Menu.buildFromTemplate([
            {label: 'Tabs', enabled: false},
            {label: 'Open Tabs', click: () => this.mainWindow.getWindow().show()},
            {type: 'separator'},
            {label: 'Quit', role: 'quit'},
        ]));
        this.tray.on('click', () => this.mainWindow.toggle());
    }
}