forked from mirrors/gecko-dev
		
	
		
			
				
	
	
		
			383 lines
		
	
	
	
		
			12 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			383 lines
		
	
	
	
		
			12 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
/* -*- Mode: indent-tabs-mode: nil; js-indent-level: 2 -*- */
 | 
						|
/* vim: set sts=2 sw=2 et tw=80: */
 | 
						|
/* 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/. */
 | 
						|
 | 
						|
"use strict";
 | 
						|
 | 
						|
ChromeUtils.defineESModuleGetters(this, {
 | 
						|
  BrowserUsageTelemetry: "resource:///modules/BrowserUsageTelemetry.sys.mjs",
 | 
						|
  ExtensionTelemetry: "resource://gre/modules/ExtensionTelemetry.sys.mjs",
 | 
						|
  PageActions: "resource:///modules/PageActions.sys.mjs",
 | 
						|
  PanelPopup: "resource:///modules/ExtensionPopups.sys.mjs",
 | 
						|
});
 | 
						|
 | 
						|
var { DefaultWeakMap } = ExtensionUtils;
 | 
						|
 | 
						|
var { ExtensionParent } = ChromeUtils.importESModule(
 | 
						|
  "resource://gre/modules/ExtensionParent.sys.mjs"
 | 
						|
);
 | 
						|
var { PageActionBase } = ChromeUtils.importESModule(
 | 
						|
  "resource://gre/modules/ExtensionActions.sys.mjs"
 | 
						|
);
 | 
						|
 | 
						|
// WeakMap[Extension -> PageAction]
 | 
						|
let pageActionMap = new WeakMap();
 | 
						|
 | 
						|
class PageAction extends PageActionBase {
 | 
						|
  constructor(extension, buttonDelegate) {
 | 
						|
    let tabContext = new TabContext(tab => this.getContextData(null));
 | 
						|
    super(tabContext, extension);
 | 
						|
    this.buttonDelegate = buttonDelegate;
 | 
						|
  }
 | 
						|
 | 
						|
  updateOnChange(target) {
 | 
						|
    this.buttonDelegate.updateButton(target.ownerGlobal);
 | 
						|
  }
 | 
						|
 | 
						|
  dispatchClick(tab, clickInfo) {
 | 
						|
    this.buttonDelegate.emit("click", tab, clickInfo);
 | 
						|
  }
 | 
						|
 | 
						|
  getTab(tabId) {
 | 
						|
    if (tabId !== null) {
 | 
						|
      return tabTracker.getTab(tabId);
 | 
						|
    }
 | 
						|
    return null;
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
this.pageAction = class extends ExtensionAPIPersistent {
 | 
						|
  static for(extension) {
 | 
						|
    return pageActionMap.get(extension);
 | 
						|
  }
 | 
						|
 | 
						|
  static onUpdate(id, manifest) {
 | 
						|
    if (!("page_action" in manifest)) {
 | 
						|
      // If the new version has no page action then mark this widget as hidden
 | 
						|
      // in the telemetry. If it is already marked hidden then this will do
 | 
						|
      // nothing.
 | 
						|
      BrowserUsageTelemetry.recordWidgetChange(makeWidgetId(id), null, "addon");
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  static onDisable(id) {
 | 
						|
    BrowserUsageTelemetry.recordWidgetChange(makeWidgetId(id), null, "addon");
 | 
						|
  }
 | 
						|
 | 
						|
  static onUninstall(id) {
 | 
						|
    // If the telemetry already has this widget as hidden then this will not
 | 
						|
    // record anything.
 | 
						|
    BrowserUsageTelemetry.recordWidgetChange(makeWidgetId(id), null, "addon");
 | 
						|
  }
 | 
						|
 | 
						|
  async onManifestEntry(entryName) {
 | 
						|
    let { extension } = this;
 | 
						|
    let options = extension.manifest.page_action;
 | 
						|
 | 
						|
    this.action = new PageAction(extension, this);
 | 
						|
    await this.action.loadIconData();
 | 
						|
 | 
						|
    let widgetId = makeWidgetId(extension.id);
 | 
						|
    this.id = widgetId + "-page-action";
 | 
						|
 | 
						|
    this.tabManager = extension.tabManager;
 | 
						|
 | 
						|
    this.browserStyle = options.browser_style;
 | 
						|
 | 
						|
    pageActionMap.set(extension, this);
 | 
						|
 | 
						|
    this.lastValues = new DefaultWeakMap(() => ({}));
 | 
						|
 | 
						|
    if (!this.browserPageAction) {
 | 
						|
      let onPlacedHandler = (buttonNode, isPanel) => {
 | 
						|
        // eslint-disable-next-line mozilla/balanced-listeners
 | 
						|
        buttonNode.addEventListener("auxclick", event => {
 | 
						|
          if (event.button !== 1 || event.target.disabled) {
 | 
						|
            return;
 | 
						|
          }
 | 
						|
 | 
						|
          // The panel is not automatically closed when middle-clicked.
 | 
						|
          if (isPanel) {
 | 
						|
            buttonNode.closest("#pageActionPanel").hidePopup();
 | 
						|
          }
 | 
						|
          let window = event.target.ownerGlobal;
 | 
						|
          let tab = window.gBrowser.selectedTab;
 | 
						|
          this.tabManager.addActiveTabPermission(tab);
 | 
						|
          this.action.dispatchClick(tab, {
 | 
						|
            button: event.button,
 | 
						|
            modifiers: clickModifiersFromEvent(event),
 | 
						|
          });
 | 
						|
        });
 | 
						|
      };
 | 
						|
 | 
						|
      this.browserPageAction = PageActions.addAction(
 | 
						|
        new PageActions.Action({
 | 
						|
          id: widgetId,
 | 
						|
          extensionID: extension.id,
 | 
						|
          title: this.action.getProperty(null, "title"),
 | 
						|
          iconURL: this.action.getProperty(null, "icon"),
 | 
						|
          pinnedToUrlbar: this.action.getPinned(),
 | 
						|
          disabled: !this.action.getProperty(null, "enabled"),
 | 
						|
          onCommand: (event, buttonNode) => {
 | 
						|
            this.handleClick(event.target.ownerGlobal, {
 | 
						|
              button: event.button || 0,
 | 
						|
              modifiers: clickModifiersFromEvent(event),
 | 
						|
            });
 | 
						|
          },
 | 
						|
          onBeforePlacedInWindow: browserWindow => {
 | 
						|
            if (
 | 
						|
              this.extension.hasPermission("menus") ||
 | 
						|
              this.extension.hasPermission("contextMenus")
 | 
						|
            ) {
 | 
						|
              browserWindow.document.addEventListener("popupshowing", this);
 | 
						|
            }
 | 
						|
          },
 | 
						|
          onPlacedInPanel: buttonNode => onPlacedHandler(buttonNode, true),
 | 
						|
          onPlacedInUrlbar: buttonNode => onPlacedHandler(buttonNode, false),
 | 
						|
          onRemovedFromWindow: browserWindow => {
 | 
						|
            browserWindow.document.removeEventListener("popupshowing", this);
 | 
						|
          },
 | 
						|
        })
 | 
						|
      );
 | 
						|
 | 
						|
      if (this.extension.startupReason != "APP_STARTUP") {
 | 
						|
        // Make sure the browser telemetry has the correct state for this widget.
 | 
						|
        // Defer loading BrowserUsageTelemetry until after startup is complete.
 | 
						|
        ExtensionParent.browserStartupPromise.then(() => {
 | 
						|
          BrowserUsageTelemetry.recordWidgetChange(
 | 
						|
            widgetId,
 | 
						|
            this.browserPageAction.pinnedToUrlbar
 | 
						|
              ? "page-action-buttons"
 | 
						|
              : null,
 | 
						|
            "addon"
 | 
						|
          );
 | 
						|
        });
 | 
						|
      }
 | 
						|
 | 
						|
      // If the page action is only enabled in some URLs, do pattern matching in
 | 
						|
      // the active tabs and update the button if necessary.
 | 
						|
      if (this.action.getProperty(null, "enabled") === undefined) {
 | 
						|
        for (let window of windowTracker.browserWindows()) {
 | 
						|
          let tab = window.gBrowser.selectedTab;
 | 
						|
          if (this.action.isShownForTab(tab)) {
 | 
						|
            this.updateButton(window);
 | 
						|
          }
 | 
						|
        }
 | 
						|
      }
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  onShutdown(isAppShutdown) {
 | 
						|
    pageActionMap.delete(this.extension);
 | 
						|
    this.action.onShutdown();
 | 
						|
 | 
						|
    // Removing the browser page action causes PageActions to forget about it
 | 
						|
    // across app restarts, so don't remove it on app shutdown, but do remove
 | 
						|
    // it on all other shutdowns since there's no guarantee the action will be
 | 
						|
    // coming back.
 | 
						|
    if (!isAppShutdown && this.browserPageAction) {
 | 
						|
      this.browserPageAction.remove();
 | 
						|
      this.browserPageAction = null;
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  // Updates the page action button in the given window to reflect the
 | 
						|
  // properties of the currently selected tab:
 | 
						|
  //
 | 
						|
  // Updates "tooltiptext" and "aria-label" to match "title" property.
 | 
						|
  // Updates "image" to match the "icon" property.
 | 
						|
  // Enables or disables the icon, based on the "enabled" and "patternMatching" properties.
 | 
						|
  updateButton(window) {
 | 
						|
    let tab = window.gBrowser.selectedTab;
 | 
						|
    let tabData = this.action.getContextData(tab);
 | 
						|
    let last = this.lastValues.get(window);
 | 
						|
 | 
						|
    window.requestAnimationFrame(() => {
 | 
						|
      // If we get called just before shutdown, we might have been destroyed by
 | 
						|
      // this point.
 | 
						|
      if (!this.browserPageAction) {
 | 
						|
        return;
 | 
						|
      }
 | 
						|
 | 
						|
      let title = tabData.title || this.extension.name;
 | 
						|
      if (last.title !== title) {
 | 
						|
        this.browserPageAction.setTitle(title, window);
 | 
						|
        last.title = title;
 | 
						|
      }
 | 
						|
 | 
						|
      let enabled =
 | 
						|
        tabData.enabled != null ? tabData.enabled : tabData.patternMatching;
 | 
						|
      if (last.enabled !== enabled) {
 | 
						|
        this.browserPageAction.setDisabled(!enabled, window);
 | 
						|
        last.enabled = enabled;
 | 
						|
      }
 | 
						|
 | 
						|
      let icon = tabData.icon;
 | 
						|
      if (last.icon !== icon) {
 | 
						|
        this.browserPageAction.setIconURL(icon, window);
 | 
						|
        last.icon = icon;
 | 
						|
      }
 | 
						|
    });
 | 
						|
  }
 | 
						|
 | 
						|
  /**
 | 
						|
   * Triggers this page action for the given window, with the same effects as
 | 
						|
   * if it were clicked by a user.
 | 
						|
   *
 | 
						|
   * This has no effect if the page action is hidden for the selected tab.
 | 
						|
   *
 | 
						|
   * @param {Window} window
 | 
						|
   */
 | 
						|
  triggerAction(window) {
 | 
						|
    this.handleClick(window, { button: 0, modifiers: [] });
 | 
						|
  }
 | 
						|
 | 
						|
  handleEvent(event) {
 | 
						|
    switch (event.type) {
 | 
						|
      case "popupshowing":
 | 
						|
        const menu = event.target;
 | 
						|
        const trigger = menu.triggerNode;
 | 
						|
        const getActionId = () => {
 | 
						|
          let actionId = trigger.getAttribute("actionid");
 | 
						|
          if (actionId) {
 | 
						|
            return actionId;
 | 
						|
          }
 | 
						|
          // When a page action is clicked, triggerNode will be an ancestor of
 | 
						|
          // a node corresponding to an action. triggerNode will be the page
 | 
						|
          // action node itself when a page action is selected with the
 | 
						|
          // keyboard. That's because the semantic meaning of page action is on
 | 
						|
          // an hbox that contains an <image>.
 | 
						|
          for (let n = trigger; n && !actionId; n = n.parentElement) {
 | 
						|
            if (n.id == "page-action-buttons" || n.localName == "panelview") {
 | 
						|
              // We reached the page-action-buttons or panelview container.
 | 
						|
              // Stop looking; no action was found.
 | 
						|
              break;
 | 
						|
            }
 | 
						|
            actionId = n.getAttribute("actionid");
 | 
						|
          }
 | 
						|
          return actionId;
 | 
						|
        };
 | 
						|
        if (
 | 
						|
          menu.id === "pageActionContextMenu" &&
 | 
						|
          trigger &&
 | 
						|
          getActionId() === this.browserPageAction.id &&
 | 
						|
          !this.browserPageAction.getDisabled(trigger.ownerGlobal)
 | 
						|
        ) {
 | 
						|
          global.actionContextMenu({
 | 
						|
            extension: this.extension,
 | 
						|
            onPageAction: true,
 | 
						|
            menu: menu,
 | 
						|
          });
 | 
						|
        }
 | 
						|
        break;
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  // Handles a click event on the page action button for the given
 | 
						|
  // window.
 | 
						|
  // If the page action has a |popup| property, a panel is opened to
 | 
						|
  // that URL. Otherwise, a "click" event is emitted, and dispatched to
 | 
						|
  // the any click listeners in the add-on.
 | 
						|
  async handleClick(window, clickInfo) {
 | 
						|
    const { extension } = this;
 | 
						|
 | 
						|
    ExtensionTelemetry.pageActionPopupOpen.stopwatchStart(extension, this);
 | 
						|
    let tab = window.gBrowser.selectedTab;
 | 
						|
    let popupURL = this.action.triggerClickOrPopup(tab, clickInfo);
 | 
						|
 | 
						|
    // If the widget has a popup URL defined, we open a popup, but do not
 | 
						|
    // dispatch a click event to the extension.
 | 
						|
    // If it has no popup URL defined, we dispatch a click event, but do not
 | 
						|
    // open a popup.
 | 
						|
    if (popupURL) {
 | 
						|
      if (this.popupNode && this.popupNode.panel.state !== "closed") {
 | 
						|
        // The panel is being toggled closed.
 | 
						|
        ExtensionTelemetry.pageActionPopupOpen.stopwatchCancel(extension, this);
 | 
						|
        window.BrowserPageActions.togglePanelForAction(
 | 
						|
          this.browserPageAction,
 | 
						|
          this.popupNode.panel
 | 
						|
        );
 | 
						|
        return;
 | 
						|
      }
 | 
						|
 | 
						|
      this.popupNode = new PanelPopup(
 | 
						|
        extension,
 | 
						|
        window.document,
 | 
						|
        popupURL,
 | 
						|
        this.browserStyle
 | 
						|
      );
 | 
						|
      // Remove popupNode when it is closed.
 | 
						|
      this.popupNode.panel.addEventListener(
 | 
						|
        "popuphiding",
 | 
						|
        () => {
 | 
						|
          this.popupNode = undefined;
 | 
						|
        },
 | 
						|
        { once: true }
 | 
						|
      );
 | 
						|
      await this.popupNode.contentReady;
 | 
						|
      window.BrowserPageActions.togglePanelForAction(
 | 
						|
        this.browserPageAction,
 | 
						|
        this.popupNode.panel
 | 
						|
      );
 | 
						|
      ExtensionTelemetry.pageActionPopupOpen.stopwatchFinish(extension, this);
 | 
						|
    } else {
 | 
						|
      ExtensionTelemetry.pageActionPopupOpen.stopwatchCancel(extension, this);
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  PERSISTENT_EVENTS = {
 | 
						|
    onClicked({ context, fire }) {
 | 
						|
      const { extension } = this;
 | 
						|
      const { tabManager } = extension;
 | 
						|
 | 
						|
      let listener = async (_event, tab, clickInfo) => {
 | 
						|
        if (fire.wakeup) {
 | 
						|
          await fire.wakeup();
 | 
						|
        }
 | 
						|
        // TODO: we should double-check if the tab is already being closed by the time
 | 
						|
        // the background script got started and we converted the primed listener.
 | 
						|
        context?.withPendingBrowser(tab.linkedBrowser, () =>
 | 
						|
          fire.sync(tabManager.convert(tab), clickInfo)
 | 
						|
        );
 | 
						|
      };
 | 
						|
 | 
						|
      this.on("click", listener);
 | 
						|
      return {
 | 
						|
        unregister: () => {
 | 
						|
          this.off("click", listener);
 | 
						|
        },
 | 
						|
        convert(newFire, extContext) {
 | 
						|
          fire = newFire;
 | 
						|
          context = extContext;
 | 
						|
        },
 | 
						|
      };
 | 
						|
    },
 | 
						|
  };
 | 
						|
 | 
						|
  getAPI(context) {
 | 
						|
    const { action } = this;
 | 
						|
 | 
						|
    return {
 | 
						|
      pageAction: {
 | 
						|
        ...action.api(context),
 | 
						|
 | 
						|
        onClicked: new EventManager({
 | 
						|
          context,
 | 
						|
          module: "pageAction",
 | 
						|
          event: "onClicked",
 | 
						|
          inputHandling: true,
 | 
						|
          extensionApi: this,
 | 
						|
        }).api(),
 | 
						|
 | 
						|
        openPopup: () => {
 | 
						|
          let window = windowTracker.topWindow;
 | 
						|
          this.triggerAction(window);
 | 
						|
        },
 | 
						|
      },
 | 
						|
    };
 | 
						|
  }
 | 
						|
};
 | 
						|
 | 
						|
global.pageActionFor = this.pageAction.for;
 |