background.js

"use strict";

// Open welcome screen when installing addon.
browser.runtime.onInstalled.addListener(function (details) {
    if (details.reason == "install" || details.reason == "update") {
        browser.storage.sync.get("username").then(function (storage) {
            if (!("username" in storage) || !storage["username"]) {
                var options = {"url": "options/welcome.html",
                               "width": 750, "height": 360, "type": "popup"};
                browser.windows.create(options).then(function (welcomeWindow) {
                    browser.windows.onRemoved.addListener(function handler(id) {
                        if (id == welcomeWindow.id) {
                            browser.windows.onRemoved.removeListener(handler);
                            browser.runtime.openOptionsPage();
                        }
                    });
                });
            }
        });
    }
});

browser.menus.create({
    id: "paste-email",
    contexts: ["editable"],
    title: browser.i18n.getMessage("menuPasteAddress")
});

function openCreateAddress(parent_tab, frameId) {
    var options = {"url": "../create-address/create-address.html",
                   "type": "popup", "width": 750, "height": 490};
    browser.windows.create(options).then(function (window) {
        // (FF 56) Security policy blocks running code until tab has completed loading.
        browser.tabs.onUpdated.addListener(function handler(tabId, changeInfo, tab) {
            if (tabId == window.tabs[0].id && changeInfo.status == "complete") {
                browser.tabs.onUpdated.removeListener(handler);
                // Send the parent url and window ID through to the new window.
                browser.tabs.sendMessage(
                    tab.id, [parent_tab.url, parent_tab.windowId, parent_tab.id, frameId]);
            }
        });
    });
}

browser.menus.onClicked.addListener(function(info, parent_tab) {
    if (info.menuItemId == "paste-email") {
        openCreateAddress(parent_tab, info.frameId);
    } else {
        // Paste previous email.
        browser.tabs.sendMessage(parent_tab.id, info.menuItemId,
                                 {"frameId": info.frameId});
    }
});

// Open create address window on keyboard shortcut.
browser.commands.onCommand.addListener(function (command) {
    browser.tabs.query({
        currentWindow: true,
        active: true
    }).then(function (tabs) {
        // This won't work if user is inside an iframe (as we don't have the frame ID).
        browser.tabs.sendMessage(tabs[0].id, "check_editable").then(function (is_editable) {
            if (is_editable)
                openCreateAddress(tabs[0], 0);
        });
    });
});



/**
 * Paste previous address context menus.
 */

// Track current status of previous address menus.
var current_domain = "";
var previous_address_menus = [];
var previous_addresses = {};
browser.storage.local.get("previous_addresses").then(function (addresses) {
    previous_addresses = addresses;
});
browser.storage.onChanged.addListener(function (changes, area) {
    if ("previous_addresses" in changes) {
        previous_addresses = changes["previous_addresses"].newValue;
        current_domain = "*invalid*";  // Force context menu to reload.
    }
});

// Update the currently displayed previous addresses context menu items.
browser.menus.onShown.addListener(function (info, tab) {
    if (!info.editable)
        return;

    let domain = (new URL(tab.url)).hostname;

    if (domain == current_domain)
        return;

    current_domain = domain;

    // Remove previous menu items.
    for (const id of previous_address_menus)
        browser.menus.remove(id);
    previous_address_menus = [];

    if (!current_domain) {
        // Refresh in case any menu items have been removed.
        browser.menus.refresh();
        return;
    }

    // Add any new ones for this domain.
    let addresses = [];
    let p = current_domain.length;
    while (p >= 0) {
        p = current_domain.lastIndexOf(".", p - 1);
        let domain = current_domain.slice(p + 1);
        if (domain in previous_addresses) {
            addresses = previous_addresses[domain];
            break;
        }
    }

    for (let [email, url] of addresses) {
        url = new URL(url);
        let url_detail = current_domain == url.hostname ? url.pathname : url.hostname;
        let id = browser.menus.create({
            id: email,
            contexts: ["editable"],
            title: browser.i18n.getMessage("menuPastePrevious", email) + " (" + url_detail + ")"
        });
        previous_address_menus.push(id);
    }

    browser.menus.refresh();
});


// Update some settings each time the addon is loaded.
browser.storage.sync.get(["username", "password"]).then(function (storage) {
    let data = {
        "cmd": "login",
        "fe-login-user": storage["username"],
        "fe-login-pass": storage["password"]
    };

    return callAPI(data);
}).then(function (login) {
    // Update list of domains and real emails.
    browser.storage.local.set({
        "domains": login["domain_name_list"],
        "real_emails": Object.keys(login["real_email_list"])
    });

    let data = {
        "cmd": "read_dea",
        "session_id": login["session_id"]
    };

    // Load public suffix data from file as needed.
    let suffixes = fetch(browser.runtime.getURL("public_suffix.json")).then(function (response) {
        if (response.ok)
            return response.json();
    });

    return Promise.all([callAPI(data), suffixes]);
}).then(function (values) {
    // Update local storage of existing disposable addresses.
    let current_prev_addresses = {};
    let [addresses, [rules, exceptions]] = values;
    for (const address of addresses) {
        if (address["website"]) {
            try {
                var domain = new URL(address["website"]);
            } catch (e) {
                if (e instanceof TypeError)
                    continue;  // Not a valid URL.
                throw e;
            }
            domain = org_domain(domain, rules, exceptions);
            let email = [address["disposable_name"] + "@" + address["disposable_domain"],
                         address["website"]];

            if (domain in current_prev_addresses)
                current_prev_addresses[domain].push(email);
            else
                current_prev_addresses[domain] = [email];
        }
    }

    previous_addresses = current_prev_addresses;
    browser.storage.local.set({"previous_addresses": current_prev_addresses});
});