bitwarden-browser/src/background/commands.background.ts

import { BrowserApi } from '../browser/browserApi';

import MainBackground from './main.background';

import { PasswordGenerationService } from 'jslib-common/abstractions/passwordGeneration.service';
import { PlatformUtilsService } from 'jslib-common/abstractions/platformUtils.service';
import { VaultTimeoutService } from 'jslib-common/abstractions/vaultTimeout.service';
import LockedVaultPendingNotificationsItem from './models/lockedVaultPendingNotificationsItem';

export default class CommandsBackground {
    private isSafari: boolean;

    constructor(private main: MainBackground, private passwordGenerationService: PasswordGenerationService,
        private platformUtilsService: PlatformUtilsService, private vaultTimeoutService: VaultTimeoutService) {
        this.isSafari = this.platformUtilsService.isSafari();
    }

    async init() {
        BrowserApi.messageListener('commands.background', async (msg: any, sender: chrome.runtime.MessageSender, sendResponse: any) => {
            if (msg.command === 'unlockCompleted' && msg.data.target === 'commands.background') {
                await this.processCommand(msg.data.commandToRetry.msg.command, msg.data.commandToRetry.sender);
            }

            if (msg.command === 'keyboardShortcutTriggered' && msg.shortcut) {
                await this.processCommand(msg.shortcut, sender);
            }
        });

        if (chrome && chrome.commands) {
            chrome.commands.onCommand.addListener(async (command: string) => {
                await this.processCommand(command);
            });
        }
    }

    private async processCommand(command: string, sender?: chrome.runtime.MessageSender) {
        switch (command) {
            case 'generate_password':
                await this.generatePasswordToClipboard();
                break;
            case 'autofill_login':
                await this.autoFillLogin(sender ? sender.tab : null);
                break;
            case 'open_popup':
                await this.openPopup();
                break;
            case 'lock_vault':
                await this.vaultTimeoutService.lock(true);
                break;
            default:
                break;
        }
    }

    private async generatePasswordToClipboard() {
        const options = (await this.passwordGenerationService.getOptions())[0];
        const password = await this.passwordGenerationService.generatePassword(options);
        this.platformUtilsService.copyToClipboard(password, { window: window });
        this.passwordGenerationService.addHistory(password);
    }

    private async autoFillLogin(tab?: chrome.tabs.Tab) {
        if (!tab) {
            tab = await BrowserApi.getTabFromCurrentWindowId();
        }

        if (tab == null) {
            return;
        }

        if (await this.vaultTimeoutService.isLocked()) {
            const retryMessage: LockedVaultPendingNotificationsItem = {
                commandToRetry: {
                    msg: { command: 'autofill_login' },
                    sender: { tab: tab },
                },
                target: 'commands.background',
            };
            await BrowserApi.tabSendMessageData(tab, 'addToLockedVaultPendingNotifications', retryMessage);

            BrowserApi.tabSendMessageData(tab, 'promptForLogin');
            return;
        }

        await this.main.collectPageDetailsForContentScript(tab, 'autofill_cmd');
    }

    private async openPopup() {
        // Chrome APIs cannot open popup
        if (!this.isSafari) {
            return;
        }

        this.main.openPopup();
    }
}