fune/browser/base/content/browser-siteProtections.js

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

/* eslint-env mozilla/browser-window */

/**
 * Utility object to handle manipulations of the protections indicators in the UI
 */
var gProtectionsHandler = {
  // smart getters
  get _protectionsPopup() {
    delete this._protectionsPopup;
    return this._protectionsPopup = document.getElementById("protections-popup");
  },
  get _protectionsIconBox() {
    delete this._protectionsIconBox;
    return this._protectionsIconBox = document.getElementById("tracking-protection-icon-animatable-box");
  },
  get _protectionsPopupMainViewHeaderLabel() {
    delete this._protectionsPopupMainViewHeaderLabel;
    return this._protectionsPopupMainViewHeaderLabel =
      document.getElementById("protections-popup-mainView-panel-header-span");
  },
  get _protectionsPopupTPSwitch() {
    delete this._protectionsPopupTPSwitch;
    return this._protectionsPopupTPSwitch =
      document.getElementById("protections-popup-tp-switch");
  },
  get _protectionPopupSettingsButton() {
    delete this._protectionPopupSettingsButton;
    return this._protectionPopupSettingsButton =
      document.getElementById("protections-popup-settings-button");
  },
  get _protectionPopupFooter() {
    delete this._protectionPopupFooter;
    return this._protectionPopupFooter =
      document.getElementById("protections-popup-footer");
  },
  get _protectionPopupTrackersCounterDescription() {
    delete this._protectionPopupTrackersCounterDescription;
    return this._protectionPopupTrackersCounterDescription =
      document.getElementById("protections-popup-trackers-blocked-counter-description");
  },

  handleProtectionsButtonEvent(event) {
    event.stopPropagation();
    if ((event.type == "click" && event.button != 0) ||
        (event.type == "keypress" && event.charCode != KeyEvent.DOM_VK_SPACE &&
         event.keyCode != KeyEvent.DOM_VK_RETURN)) {
      return; // Left click, space or enter only
    }

    // Make sure that the display:none style we set in xul is removed now that
    // the popup is actually needed
    this._protectionsPopup.hidden = false;

    // Refresh strings.
    this.refreshProtectionsPopup();

    // Now open the popup, anchored off the primary chrome element
    PanelMultiView.openPopup(this._protectionsPopup, gIdentityHandler._identityIcon, {
      position: "bottomcenter topleft",
      triggerEvent: event,
    }).catch(Cu.reportError);
  },

  onPopupShown(event) {
    if (event.target == this._protectionsPopup) {
      window.addEventListener("focus", this, true);
    }
  },

  onPopupHidden(event) {
    if (event.target == this._protectionsPopup) {
      window.removeEventListener("focus", this, true);
      this._protectionsPopup.removeAttribute("open");
    }
  },

  handleEvent(event) {
    let elem = document.activeElement;
    let position = elem.compareDocumentPosition(this._protectionsPopup);

    if (!(position & (Node.DOCUMENT_POSITION_CONTAINS |
                      Node.DOCUMENT_POSITION_CONTAINED_BY)) &&
        !this._protectionsPopup.hasAttribute("noautohide")) {
      // Hide the panel when focusing an element that is
      // neither an ancestor nor descendant unless the panel has
      // @noautohide (e.g. for a tour).
      PanelMultiView.hidePopup(this._protectionsPopup);
    }
  },

  refreshProtectionsPopup() {
    // Refresh the state of the TP toggle switch.
    this._protectionsPopupTPSwitch.toggleAttribute("enabled",
      !this._protectionsPopup.hasAttribute("hasException"));

    let host = gIdentityHandler.getHostForDisplay();

    // Push the appropriate strings out to the UI.
    this._protectionsPopupMainViewHeaderLabel.textContent =
      // gNavigatorBundle.getFormattedString("protections.header", [host]);
      `Tracking Protections for ${host}`;

    let currentlyEnabled =
      !this._protectionsPopup.hasAttribute("hasException");

    this._protectionsPopupTPSwitch.toggleAttribute("enabled", currentlyEnabled);

    // Set the counter of the 'Trackers blocked This Week'.
    // We need to get the statistics of trackers. So far, we haven't implemented
    // this yet. So we use a fake number here. Should be resolved in
    // Bug 1555231.
    this.setTrackersBlockedCounter(244051);
  },

  async onTPSwitchCommand(event) {
    // When the switch is clicked, we wait 500ms and then disable/enable
    // protections, causing the page to refresh, and close the popup.
    // We need to ensure we don't handle more clicks during the 500ms delay,
    // so we keep track of state and return early if needed.
    if (this._TPSwitchCommanding) {
      return;
    }

    this._TPSwitchCommanding = true;

    let currentlyEnabled =
      !this._protectionsPopup.hasAttribute("hasException");

    this._protectionsPopupTPSwitch.toggleAttribute("enabled", !currentlyEnabled);
    await new Promise((resolve) => setTimeout(resolve, 500));

    if (currentlyEnabled) {
      ContentBlocking.disableForCurrentPage();
      gIdentityHandler.recordClick("unblock");
    } else {
      ContentBlocking.enableForCurrentPage();
      gIdentityHandler.recordClick("block");
    }

    PanelMultiView.hidePopup(this._protectionsPopup);
    delete this._TPSwitchCommanding;
  },

  setTrackersBlockedCounter(trackerCount) {
    this._protectionPopupTrackersCounterDescription.textContent =
      // gNavigatorBundle.getFormattedString(
      //   "protections.trackers_counter", [cnt]);
      `Trackers blocked this week: ${trackerCount.toLocaleString()}`;
  },
};