background.js

/*
    Firefox addon "Toggle Referrer"
    Copyright (C) 2019  Manuel Reimer <manuel.reimer@gmx.de>

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

// Fired if the toolbar button is clicked.
// Toggles the referrer setting.
async function ToolbarButtonClicked() {
  const value = (await browser.privacy.websites.referrersEnabled.get({})).value;
  await browser.privacy.websites.referrersEnabled.set({value: !value});
  await UpdateBadge();
}

// Sets browserAction badge text based on referrer status.
async function UpdateBadge() {
  const value = (await browser.privacy.websites.referrersEnabled.get({})).value;
  const badgetext = value ? "!" : "";
  const title = "Referrer (" +
      browser.i18n.getMessage(value ? "title_enabled" : "title_disabled") +
      ")";

  if (browser.browserAction.setBadgeText !== undefined) // Not Android
    browser.browserAction.setBadgeText({text: badgetext});
  browser.browserAction.setTitle({title: title});

  UpdateSpoofStatus();
}

// Handles auto disable of referrer
async function DoAutoDisable() {
  // If the user enabled referrer res
  const prefs = await browser.storage.local.get();
  const autodisable = prefs.autodisable || false;
  if (autodisable) {
    await browser.privacy.websites.referrersEnabled.set({value: false});
    await UpdateBadge();
  }
}

// Get sure our badge background is red.
if (browser.browserAction.setBadgeBackgroundColor !== undefined) // Not Android
  browser.browserAction.setBadgeBackgroundColor({color: "#FF0000"});

// Register event listeners
browser.browserAction.onClicked.addListener(ToolbarButtonClicked);

// Update badge for the first time
UpdateBadge();

// Diable Referrer if the user enabled auto disabling
DoAutoDisable();

IconUpdater.Init("icons/togglereferrer.svg");