bitwarden-desktop/src/main/menu.main.ts

import {
    app,
    BrowserWindow,
    clipboard,
    dialog,
    ipcMain,
    Menu,
    MenuItem,
    MenuItemConstructorOptions,
    shell,
} from 'electron';

import { Main } from '../main';

import { isMacAppStore, isSnapStore, isWindowsStore } from 'jslib/electron/utils';

import { ConstantsService } from 'jslib/services/constants.service';

export class MenuMain {
    menu: Menu;
    updateMenuItem: MenuItem;
    addNewLogin: MenuItem;
    addNewItem: MenuItem;
    addNewFolder: MenuItem;
    syncVault: MenuItem;
    settings: MenuItem;
    lockNow: MenuItem;
    logOut: MenuItem;
    twoStepLogin: MenuItem;
    changeEmail: MenuItem;
    changeMasterPass: MenuItem;
    premiumMembership: MenuItem;
    passwordGenerator: MenuItem;
    passwordHistory: MenuItem;
    searchVault: MenuItem;
    unlockedRequiredMenuItems: MenuItem[] = [];

    constructor(private main: Main) { }

    init() {
        this.initContextMenu();
        this.initApplicationMenu();

        this.updateMenuItem = this.menu.getMenuItemById('checkForUpdates');
        this.addNewLogin = this.menu.getMenuItemById('addNewLogin');
        this.addNewItem = this.menu.getMenuItemById('addNewItem');
        this.addNewFolder = this.menu.getMenuItemById('addNewFolder');
        this.syncVault = this.menu.getMenuItemById('syncVault');
        this.settings = this.menu.getMenuItemById('settings');
        this.lockNow = this.menu.getMenuItemById('lockNow');
        this.logOut = this.menu.getMenuItemById('logOut');
        this.twoStepLogin = this.menu.getMenuItemById('twoStepLogin');
        this.changeEmail = this.menu.getMenuItemById('changeEmail');
        this.changeMasterPass = this.menu.getMenuItemById('changeMasterPass');
        this.premiumMembership = this.menu.getMenuItemById('premiumMembership');
        this.passwordGenerator = this.menu.getMenuItemById('passwordGenerator');
        this.passwordHistory = this.menu.getMenuItemById('passwordHistory');
        this.searchVault = this.menu.getMenuItemById('searchVault');

        this.unlockedRequiredMenuItems = [
            this.addNewLogin, this.addNewItem, this.addNewFolder,
            this.syncVault, this.settings, this.lockNow, this.twoStepLogin, this.changeEmail,
            this.changeMasterPass, this.premiumMembership, this.passwordGenerator, this.passwordHistory,
            this.searchVault];
        this.updateApplicationMenuState(false, true);
    }

    updateApplicationMenuState(isAuthenticated: boolean, isLocked: boolean) {
        this.unlockedRequiredMenuItems.forEach((mi: MenuItem) => {
            mi.enabled = isAuthenticated && !isLocked;
        });

        this.logOut.enabled = isAuthenticated;
    }

    private initContextMenu() {
        if (this.main.windowMain.win == null) {
            return;
        }

        const selectionMenu = Menu.buildFromTemplate([
            {
                label: this.main.i18nService.t('copy'),
                role: 'copy',
            },
            { type: 'separator' },
            {
                label: this.main.i18nService.t('selectAll'),
                role: 'selectall',
            },
        ]);

        const inputMenu = Menu.buildFromTemplate([
            {
                label: this.main.i18nService.t('undo'),
                role: 'undo',
            },
            {
                label: this.main.i18nService.t('redo'),
                role: 'redo',
            },
            { type: 'separator' },
            {
                label: this.main.i18nService.t('cut'),
                role: 'cut',
                enabled: false,
            },
            {
                label: this.main.i18nService.t('copy'),
                role: 'copy',
                enabled: false,
            },
            {
                label: this.main.i18nService.t('paste'),
                role: 'paste',
            },
            { type: 'separator' },
            {
                label: this.main.i18nService.t('selectAll'),
                role: 'selectall',
            },
        ]);

        const inputSelectionMenu = Menu.buildFromTemplate([
            {
                label: this.main.i18nService.t('cut'),
                role: 'cut',
            },
            {
                label: this.main.i18nService.t('copy'),
                role: 'copy',
            },
            {
                label: this.main.i18nService.t('paste'),
                role: 'paste',
            },
            { type: 'separator' },
            {
                label: this.main.i18nService.t('selectAll'),
                role: 'selectall',
            },
        ]);

        this.main.windowMain.win.webContents.on('context-menu', (e, props) => {
            const selected = props.selectionText && props.selectionText.trim() !== '';
            if (props.isEditable && selected) {
                inputSelectionMenu.popup(this.main.windowMain.win);
            } else if (props.isEditable) {
                inputMenu.popup(this.main.windowMain.win);
            } else if (selected) {
                selectionMenu.popup(this.main.windowMain.win);
            }
        });
    }

    private initApplicationMenu() {
        const accountSubmenu: MenuItemConstructorOptions[] = [
            {
                label: this.main.i18nService.t('changeMasterPass'),
                id: 'changeMasterPass',
                click: async () => {
                    const result = dialog.showMessageBox(this.main.windowMain.win, {
                        title: this.main.i18nService.t('changeMasterPass'),
                        message: this.main.i18nService.t('changeMasterPass'),
                        detail: this.main.i18nService.t('changeMasterPasswordConfirmation'),
                        buttons: [this.main.i18nService.t('yes'), this.main.i18nService.t('no')],
                        cancelId: 1,
                        defaultId: 0,
                        noLink: true,
                    });
                    if (result === 0) {
                        await this.openWebVault();
                    }
                },
            },
            {
                label: this.main.i18nService.t('changeEmail'),
                id: 'changeEmail',
                click: async () => {
                    const result = dialog.showMessageBox(this.main.windowMain.win, {
                        title: this.main.i18nService.t('changeEmail'),
                        message: this.main.i18nService.t('changeEmail'),
                        detail: this.main.i18nService.t('changeEmailConfirmation'),
                        buttons: [this.main.i18nService.t('yes'), this.main.i18nService.t('no')],
                        cancelId: 1,
                        defaultId: 0,
                        noLink: true,
                    });
                    if (result === 0) {
                        await this.openWebVault();
                    }
                },
            },
            {
                label: this.main.i18nService.t('twoStepLogin'),
                id: 'twoStepLogin',
                click: async () => {
                    const result = dialog.showMessageBox(this.main.windowMain.win, {
                        title: this.main.i18nService.t('twoStepLogin'),
                        message: this.main.i18nService.t('twoStepLogin'),
                        detail: this.main.i18nService.t('twoStepLoginConfirmation'),
                        buttons: [this.main.i18nService.t('yes'), this.main.i18nService.t('no')],
                        cancelId: 1,
                        defaultId: 0,
                        noLink: true,
                    });
                    if (result === 0) {
                        await this.openWebVault();
                    }
                },
            },
            { type: 'separator' },
            {
                label: this.main.i18nService.t('logOut'),
                id: 'logOut',
                click: () => {
                    const result = dialog.showMessageBox(this.main.windowMain.win, {
                        title: this.main.i18nService.t('logOut'),
                        message: this.main.i18nService.t('logOut'),
                        detail: this.main.i18nService.t('logOutConfirmation'),
                        buttons: [this.main.i18nService.t('logOut'), this.main.i18nService.t('cancel')],
                        cancelId: 1,
                        defaultId: 0,
                        noLink: true,
                    });
                    if (result === 0) {
                        this.main.messagingService.send('logout');
                    }
                },
            },
        ];

        if (!isMacAppStore() && !isWindowsStore()) {
            accountSubmenu.unshift({
                label: this.main.i18nService.t('premiumMembership'),
                click: () => this.main.messagingService.send('openPremium'),
                id: 'premiumMembership',
            });
        }

        const template: MenuItemConstructorOptions[] = [
            {
                label: this.main.i18nService.t('file'),
                submenu: [
                    {
                        label: this.main.i18nService.t('addNewLogin'),
                        click: () => this.main.messagingService.send('newLogin'),
                        accelerator: 'CmdOrCtrl+N',
                        id: 'addNewLogin',
                    },
                    {
                        label: this.main.i18nService.t('addNewItem'),
                        id: 'addNewItem',
                        submenu: [
                            {
                                label: this.main.i18nService.t('typeLogin'),
                                click: () => this.main.messagingService.send('newLogin'),
                                accelerator: 'CmdOrCtrl+Shift+L',
                            },
                            {
                                label: this.main.i18nService.t('typeCard'),
                                click: () => this.main.messagingService.send('newCard'),
                                accelerator: 'CmdOrCtrl+Shift+C',
                            },
                            {
                                label: this.main.i18nService.t('typeIdentity'),
                                click: () => this.main.messagingService.send('newIdentity'),
                                accelerator: 'CmdOrCtrl+Shift+I',
                            },
                            {
                                label: this.main.i18nService.t('typeSecureNote'),
                                click: () => this.main.messagingService.send('newSecureNote'),
                                accelerator: 'CmdOrCtrl+Shift+S',
                            },
                        ],
                    },
                    {
                        label: this.main.i18nService.t('addNewFolder'),
                        id: 'addNewFolder',
                        click: () => this.main.messagingService.send('newFolder'),
                    },
                    { type: 'separator' },
                    {
                        label: this.main.i18nService.t('syncVault'),
                        id: 'syncVault',
                        click: () => this.main.messagingService.send('syncVault'),
                    },
                ],
            },
            {
                label: this.main.i18nService.t('edit'),
                submenu: [
                    {
                        label: this.main.i18nService.t('undo'),
                        role: 'undo',
                    },
                    {
                        label: this.main.i18nService.t('redo'),
                        role: 'redo',
                    },
                    { type: 'separator' },
                    {
                        label: this.main.i18nService.t('cut'),
                        role: 'cut',
                    },
                    {
                        label: this.main.i18nService.t('copy'),
                        role: 'copy',
                    },
                    {
                        label: this.main.i18nService.t('paste'),
                        role: 'paste',
                    },
                    { type: 'separator' },
                    {
                        label: this.main.i18nService.t('selectAll'),
                        role: 'selectall',
                    },
                ],
            },
            {
                label: this.main.i18nService.t('view'),
                submenu: [
                    {
                        label: this.main.i18nService.t('searchVault'),
                        id: 'searchVault',
                        click: () => this.main.messagingService.send('focusSearch'),
                        accelerator: 'CmdOrCtrl+F',
                    },
                    { type: 'separator' },
                    {
                        label: this.main.i18nService.t('passwordGenerator'),
                        id: 'passwordGenerator',
                        click: () => this.main.messagingService.send('openPasswordGenerator'),
                        accelerator: 'CmdOrCtrl+G',
                    },
                    {
                        label: this.main.i18nService.t('passwordHistory'),
                        id: 'passwordHistory',
                        click: () => this.main.messagingService.send('openPasswordHistory'),
                    },
                    { type: 'separator' },
                    {
                        label: this.main.i18nService.t('zoomIn'),
                        role: 'zoomin', accelerator: 'CmdOrCtrl+=',
                    },
                    {
                        label: this.main.i18nService.t('zoomOut'),
                        role: 'zoomout', accelerator: 'CmdOrCtrl+-',
                    },
                    {
                        label: this.main.i18nService.t('resetZoom'),
                        role: 'resetzoom', accelerator: 'CmdOrCtrl+0',
                    },
                    { type: 'separator' },
                    {
                        label: this.main.i18nService.t('toggleFullScreen'),
                        role: 'togglefullscreen',
                    },
                    { type: 'separator' },
                    {
                        label: this.main.i18nService.t('reload'),
                        role: 'forcereload',
                    },
                    {
                        label: this.main.i18nService.t('toggleDevTools'),
                        role: 'toggledevtools',
                        accelerator: 'F12',
                    },
                ],
            },
            {
                label: this.main.i18nService.t('account'),
                submenu: accountSubmenu,
            },
            {
                label: this.main.i18nService.t('window'),
                role: 'window',
                submenu: [
                    {
                        label: this.main.i18nService.t('minimize'),
                        role: 'minimize',
                    },
                    {
                        label: this.main.i18nService.t('close'),
                        role: 'close',
                    },
                ],
            },
            {
                label: this.main.i18nService.t('help'),
                role: 'help',
                submenu: [
                    {
                        label: this.main.i18nService.t('emailUs'),
                        click: () => shell.openExternal('mailTo:hello@bitwarden.com'),
                    },
                    {
                        label: this.main.i18nService.t('visitOurWebsite'),
                        click: () => shell.openExternal('https://bitwarden.com/contact'),
                    },
                    {
                        label: this.main.i18nService.t('fileBugReport'),
                        click: () => shell.openExternal('https://github.com/bitwarden/desktop'),
                    },
                    { type: 'separator' },
                    {
                        label: this.main.i18nService.t('followUs'),
                        submenu: [
                            {
                                label: this.main.i18nService.t('blog'),
                                click: () => shell.openExternal('https://blog.bitwarden.com'),
                            },
                            {
                                label: 'Twitter',
                                click: () => shell.openExternal('https://twitter.com/bitwarden_app'),
                            },
                            {
                                label: 'Facebook',
                                click: () => shell.openExternal('https://www.facebook.com/bitwarden/'),
                            },
                            {
                                label: 'Google+',
                                click: () => shell.openExternal('https://plus.google.com/114869903467947368993'),
                            },
                            {
                                label: 'GitHub',
                                click: () => shell.openExternal('https://github.com/bitwarden'),
                            },
                        ],
                    },
                    { type: 'separator' },
                    {
                        label: this.main.i18nService.t('goToWebVault'),
                        click: async () => await this.openWebVault(),
                    },
                    {
                        label: this.main.i18nService.t('getMobileApp'),
                        submenu: [
                            {
                                label: 'iOS',
                                click: () => {
                                    shell.openExternal('https://itunes.apple.com/app/' +
                                        'bitwarden-free-password-manager/id1137397744?mt=8');
                                },
                            },
                            {
                                label: 'Android',
                                click: () => {
                                    shell.openExternal('https://play.google.com/store/apps/' +
                                        'details?id=com.x8bit.bitwarden');
                                },
                            },
                        ],
                    },
                    {
                        label: this.main.i18nService.t('getBrowserExtension'),
                        submenu: [
                            {
                                label: 'Chrome',
                                click: () => {
                                    shell.openExternal('https://chrome.google.com/webstore/detail/' +
                                        'bitwarden-free-password-m/nngceckbapebfimnlniiiahkandclblb');
                                },
                            },
                            {
                                label: 'Firefox',
                                click: () => {
                                    shell.openExternal('https://addons.mozilla.org/firefox/addon/' +
                                        'bitwarden-password-manager/');
                                },
                            },
                            {
                                label: 'Opera',
                                click: () => {
                                    shell.openExternal('https://addons.opera.com/extensions/details/' +
                                        'bitwarden-free-password-manager/');
                                },
                            },
                            {
                                label: 'Edge',
                                click: () => {
                                    shell.openExternal('https://www.microsoft.com/store/p/' +
                                        'bitwarden-free-password-manager/9p6kxl0svnnl');
                                },
                            },
                            {
                                label: 'Safari',
                                click: () => {
                                    shell.openExternal('https://safari-extensions.apple.com/details/' +
                                        '?id=com.bitwarden.safari-LTZ2PFU5D6');
                                },
                            },
                        ],
                    },
                ],
            },
        ];

        const firstMenuOptions: MenuItemConstructorOptions[] = [
            { type: 'separator' },
            {
                label: this.main.i18nService.t('settings'),
                id: 'settings',
                click: () => this.main.messagingService.send('openSettings'),
            },
            {
                label: this.main.i18nService.t('lockNow'),
                id: 'lockNow',
                click: () => this.main.messagingService.send('lockVault'),
                accelerator: 'CmdOrCtrl+L',
            },
        ];

        const updateMenuItem = {
            label: this.main.i18nService.t('checkForUpdates'),
            click: () => this.main.updaterMain.checkForUpdate(true),
            id: 'checkForUpdates',
        };

        if (process.platform === 'darwin') {
            const firstMenuPart: MenuItemConstructorOptions[] = [
                {
                    label: this.main.i18nService.t('aboutBitwarden'),
                    role: 'about',
                },
            ];

            if (!isMacAppStore()) {
                firstMenuPart.push(updateMenuItem);
            }

            template.unshift({
                label: 'Bitwarden',
                submenu: firstMenuPart.concat(firstMenuOptions, [
                    { type: 'separator' },
                    {
                        label: this.main.i18nService.t('services'),
                        role: 'services', submenu: [],
                    },
                    { type: 'separator' },
                    {
                        label: this.main.i18nService.t('hideBitwarden'),
                        role: 'hide',
                    },
                    {
                        label: this.main.i18nService.t('hideOthers'),
                        role: 'hideothers',
                    },
                    {
                        label: this.main.i18nService.t('showAll'),
                        role: 'unhide',
                    },
                    { type: 'separator' },
                    {
                        label: this.main.i18nService.t('quitBitwarden'),
                        role: 'quit',
                    },
                ]),
            });

            // Window menu
            template[template.length - 2].submenu = [
                {
                    label: this.main.i18nService.t('close'),
                    role: isMacAppStore() ? 'quit' : 'close',
                },
                {
                    label: this.main.i18nService.t('minimize'),
                    role: 'minimize',
                },
                {
                    label: this.main.i18nService.t('zoom'),
                    role: 'zoom',
                },
                { type: 'separator' },
                {
                    label: this.main.i18nService.t('bringAllToFront'),
                    role: 'front',
                },
            ];
        } else {
            // File menu
            template[0].submenu = (template[0].submenu as MenuItemConstructorOptions[]).concat(
                firstMenuOptions);

            // About menu
            const aboutMenuAdditions: MenuItemConstructorOptions[] = [
                { type: 'separator' },
            ];

            if (!isWindowsStore() && !isSnapStore()) {
                aboutMenuAdditions.push(updateMenuItem);
            }

            aboutMenuAdditions.push({
                label: this.main.i18nService.t('aboutBitwarden'),
                click: () => {
                    const aboutInformation = this.main.i18nService.t('version', app.getVersion()) +
                        '\nShell ' + process.versions.electron +
                        '\nRenderer ' + process.versions.chrome +
                        '\nNode ' + process.versions.node +
                        '\nArchitecture ' + process.arch;
                    const result = dialog.showMessageBox(this.main.windowMain.win, {
                        title: 'Bitwarden',
                        message: 'Bitwarden',
                        detail: aboutInformation,
                        type: 'info',
                        noLink: true,
                        buttons: [this.main.i18nService.t('ok'), this.main.i18nService.t('copy')],
                    });
                    if (result === 1) {
                        clipboard.writeText(aboutInformation);
                    }
                },
            });

            template[template.length - 1].submenu =
                (template[template.length - 1].submenu as MenuItemConstructorOptions[]).concat(aboutMenuAdditions);
        }

        this.menu = Menu.buildFromTemplate(template);
        Menu.setApplicationMenu(this.menu);
    }

    private async openWebVault() {
        let webUrl = 'https://vault.bitwarden.com';
        const urlsObj: any = await this.main.storageService.get(ConstantsService.environmentUrlsKey);
        if (urlsObj != null) {
            if (urlsObj.base != null) {
                webUrl = urlsObj.base;
            } else if (urlsObj.webVault != null) {
                webUrl = urlsObj.webVault;
            }
        }
        shell.openExternal(webUrl);
    }
}