bitwarden-estensione-browser/src/notification/bar.js

require('./bar.scss');

document.addEventListener('DOMContentLoaded', () => {
    var i18n = {};
    var lang = window.navigator.language;

    i18n.appName = chrome.i18n.getMessage('appName');
    i18n.close = chrome.i18n.getMessage('close');
    i18n.yes = chrome.i18n.getMessage('yes');
    i18n.never = chrome.i18n.getMessage('never');
    i18n.folder = chrome.i18n.getMessage('folder');
    i18n.notificationAddSave = chrome.i18n.getMessage('notificationAddSave');
    i18n.notificationNeverSave = chrome.i18n.getMessage('notificationNeverSave');
    i18n.notificationAddDesc = chrome.i18n.getMessage('notificationAddDesc');
    i18n.notificationChangeSave = chrome.i18n.getMessage('notificationChangeSave');
    i18n.notificationChangeDesc = chrome.i18n.getMessage('notificationChangeDesc');
    lang = chrome.i18n.getUILanguage();

    // delay 50ms so that we get proper body dimensions
    setTimeout(load, 50);

    function load() {
        const isVaultLocked = getQueryVariable('isVaultLocked') == 'true';
        document.getElementById('logo').src = isVaultLocked
            ? chrome.runtime.getURL('images/icon38_locked.png')
            : chrome.runtime.getURL('images/icon38.png');

        document.getElementById('logo-link').title = i18n.appName;

        var neverButton = document.querySelector('#template-add .never-save');
        neverButton.setAttribute('short-text', i18n.never);
        neverButton.setAttribute('full-text', i18n.notificationNeverSave);

        var selectFolder = document.querySelector('#template-add .select-folder');
        selectFolder.setAttribute('aria-label', i18n.folder);

        var addButton = document.querySelector('#template-add .add-save');
        addButton.setAttribute('short-text', i18n.yes);
        addButton.setAttribute('full-text', i18n.notificationAddSave);

        var changeButton = document.querySelector('#template-change .change-save');
        changeButton.setAttribute('short-text', i18n.yes);
        changeButton.setAttribute('full-text', i18n.notificationChangeSave);

        var closeIcon = document.getElementById('close');
        closeIcon.src = chrome.runtime.getURL('images/close.png');
        closeIcon.alt = i18n.close;

        var closeButton = document.getElementById('close-button')
        closeButton.title = i18n.close;
        closeButton.setAttribute('aria-label', i18n.close);

        document.querySelector('#template-add .add-text').textContent = i18n.notificationAddDesc;
        document.querySelector('#template-change .change-text').textContent = i18n.notificationChangeDesc;

        if (getQueryVariable('add')) {
            setContent(document.getElementById('template-add'));

            var addButton = document.querySelector('#template-add-clone .add-save'),
                neverButton = document.querySelector('#template-add-clone .never-save');

            addButton.addEventListener('click', (e) => {
                e.preventDefault();

                const folderId = document.querySelector('#template-add-clone .select-folder').value;

                const bgAddSaveMessage = {
                    command: 'bgAddSave',
                    folder: folderId,
                };
                sendPlatformMessage(bgAddSaveMessage);
            });

            neverButton.addEventListener('click', (e) => {
                e.preventDefault();
                sendPlatformMessage({
                    command: 'bgNeverSave'
                });
            });

            if (!isVaultLocked) {
                const responseFoldersCommand = 'notificationBarGetFoldersList';
                chrome.runtime.onMessage.addListener((msg) => {
                    if (msg.command === responseFoldersCommand && msg.data) {
                        fillSelectorWithFolders(msg.data.folders);
                    }
                });
                sendPlatformMessage({
                    command: 'bgGetDataForTab',
                    responseCommand: responseFoldersCommand
                });
            }
        } else if (getQueryVariable('change')) {
            setContent(document.getElementById('template-change'));
            var changeButton = document.querySelector('#template-change-clone .change-save');
            changeButton.addEventListener('click', (e) => {
                e.preventDefault();

                const bgChangeSaveMessage = {
                    command: 'bgChangeSave'
                };
                sendPlatformMessage(bgChangeSaveMessage);
            });
        }

        closeButton.addEventListener('click', (e) => {
            e.preventDefault();
            sendPlatformMessage({
                command: 'bgCloseNotificationBar'
            });
        });

        window.addEventListener("resize", adjustHeight);
        adjustHeight();
    }

    function getQueryVariable(variable) {
        var query = window.location.search.substring(1);
        var vars = query.split('&');

        for (var i = 0; i < vars.length; i++) {
            var pair = vars[i].split('=');
            if (pair[0] === variable) {
                return pair[1];
            }
        }

        return null;
    }

    function setContent(element) {
        const content = document.getElementById('content');
        while (content.firstChild) {
            content.removeChild(content.firstChild);
        }

        var newElement = element.cloneNode(true);
        newElement.id = newElement.id + '-clone';
        content.appendChild(newElement);
    }

    function sendPlatformMessage(msg) {
        chrome.runtime.sendMessage(msg);
    }

    function fillSelectorWithFolders(folders) {
        const select = document.querySelector('#template-add-clone .select-folder');
        select.appendChild(new Option(chrome.i18n.getMessage('selectFolder'), null, true));
        folders.forEach((folder) => {
            //Select "No Folder" (id=null) folder by default
            select.appendChild(new Option(folder.name, folder.id || '', false));
        });
    }

    function adjustHeight() {
        sendPlatformMessage({
            command: 'bgAdjustNotificationBar',
            data: {
                height: document.querySelector('body').scrollHeight
            }
        });
    }
});