2 files changed, 221 insertions, 0 deletions
diff --git a/browser/components/asrouter/actors/ASRouterChild.sys.mjs b/browser/components/asrouter/actors/ASRouterChild.sys.mjs
new file mode 100644
index 0000000000..8e5fd5ccf5
--- /dev/null
+++ b/browser/components/asrouter/actors/ASRouterChild.sys.mjs
@@ -0,0 +1,123 @@
+/* vim: set ts=2 sw=2 sts=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/. */
+
+// We use importESModule here instead of static import so that
+// the Karma test environment won't choke on this module. This
+// is because the Karma test environment already stubs out
+// MESSAGE_TYPE_LIST and MESSAGE_TYPE_HASH, and overrides importESModule
+// to be a no-op (which can't be done for a static import statement).
+
+// eslint-disable-next-line mozilla/use-static-import
+const { MESSAGE_TYPE_LIST, MESSAGE_TYPE_HASH: msg } =
+  ChromeUtils.importESModule(
+    "resource:///modules/asrouter/ActorConstants.sys.mjs"
+  );
+
+const VALID_TYPES = new Set(MESSAGE_TYPE_LIST);
+
+export class ASRouterChild extends JSWindowActorChild {
+  constructor() {
+    super();
+    this.observers = new Set();
+  }
+
+  didDestroy() {
+    this.observers.clear();
+  }
+
+  actorCreated() {
+    // NOTE: DOMDocElementInserted may be called multiple times per
+    // PWindowGlobal due to the initial about:blank document's window global
+    // being re-used.
+    const window = this.contentWindow;
+    Cu.exportFunction(this.asRouterMessage.bind(this), window, {
+      defineAs: "ASRouterMessage",
+    });
+    Cu.exportFunction(this.addParentListener.bind(this), window, {
+      defineAs: "ASRouterAddParentListener",
+    });
+    Cu.exportFunction(this.removeParentListener.bind(this), window, {
+      defineAs: "ASRouterRemoveParentListener",
+    });
+  }
+
+  handleEvent(event) {
+    // DOMDocElementCreated is only used to create the actor.
+  }
+
+  addParentListener(listener) {
+    this.observers.add(listener);
+  }
+
+  removeParentListener(listener) {
+    this.observers.delete(listener);
+  }
+
+  receiveMessage({ name, data }) {
+    switch (name) {
+      case "UpdateAdminState":
+      case "ClearProviders": {
+        this.observers.forEach(listener => {
+          let result = Cu.cloneInto(
+            {
+              type: name,
+              data,
+            },
+            this.contentWindow
+          );
+          listener(result);
+        });
+        break;
+      }
+    }
+  }
+
+  wrapPromise(promise) {
+    return new this.contentWindow.Promise((resolve, reject) =>
+      promise.then(resolve, reject)
+    );
+  }
+
+  sendQuery(aName, aData = null) {
+    return this.wrapPromise(
+      new Promise(resolve => {
+        super.sendQuery(aName, aData).then(result => {
+          resolve(Cu.cloneInto(result, this.contentWindow));
+        });
+      })
+    );
+  }
+
+  asRouterMessage({ type, data }) {
+    // Some legacy privileged addons send this message, but it got removed from
+    // VALID_TYPES in bug 1715158. Thankfully, these addons don't appear to
+    // require any actions from this message - just a Promise that resolves.
+    if (type === "NEWTAB_MESSAGE_REQUEST") {
+      return this.wrapPromise(Promise.resolve());
+    }
+
+    if (VALID_TYPES.has(type)) {
+      switch (type) {
+        case msg.DISABLE_PROVIDER:
+        case msg.ENABLE_PROVIDER:
+        case msg.EXPIRE_QUERY_CACHE:
+        case msg.FORCE_WHATSNEW_PANEL:
+        case msg.CLOSE_WHATSNEW_PANEL:
+        case msg.FORCE_PRIVATE_BROWSING_WINDOW:
+        case msg.IMPRESSION:
+        case msg.RESET_PROVIDER_PREF:
+        case msg.SET_PROVIDER_USER_PREF:
+        case msg.USER_ACTION: {
+          return this.sendAsyncMessage(type, data);
+        }
+        default: {
+          // these messages need a response
+          return this.sendQuery(type, data);
+        }
+      }
+    }
+    throw new Error(`Unexpected type "${type}"`);
+  }
+}
diff --git a/browser/components/asrouter/actors/ASRouterParent.sys.mjs b/browser/components/asrouter/actors/ASRouterParent.sys.mjs
new file mode 100644
index 0000000000..aab909df05
--- /dev/null
+++ b/browser/components/asrouter/actors/ASRouterParent.sys.mjs
@@ -0,0 +1,98 @@
+/* vim: set ts=2 sw=2 sts=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/. */
+
+// We use importESModule here instead of static import so that
+// the Karma test environment won't choke on this module, since
+// it doesn't seem to understand using static import for sys.mjs
+// files.
+// eslint-disable-next-line mozilla/use-static-import
+const { ASRouterNewTabHook } = ChromeUtils.importESModule(
+  "resource:///modules/asrouter/ASRouterNewTabHook.sys.mjs"
+);
+
+import { ASRouterDefaultConfig } from "resource:///modules/asrouter/ASRouterDefaultConfig.sys.mjs";
+
+export class ASRouterTabs {
+  constructor({ asRouterNewTabHook }) {
+    this.actors = new Set();
+    this.destroy = () => {};
+    // This is one of several entrypoints to ASRouter Initialization. There is
+    // another one in BrowserGlue, and another in BackgroundTaskUtils.
+    asRouterNewTabHook.createInstance(ASRouterDefaultConfig());
+    this.loadingMessageHandler = asRouterNewTabHook
+      .getInstance()
+      .then(initializer => {
+        const parentProcessMessageHandler = initializer.connect({
+          clearChildMessages: ids => this.messageAll("ClearMessages", ids),
+          clearChildProviders: ids => this.messageAll("ClearProviders", ids),
+          updateAdminState: state => this.messageAll("UpdateAdminState", state),
+        });
+        this.destroy = () => {
+          initializer.disconnect();
+        };
+        return parentProcessMessageHandler;
+      });
+  }
+
+  get size() {
+    return this.actors.size;
+  }
+
+  messageAll(message, data) {
+    return Promise.all(
+      [...this.actors].map(a => a.sendAsyncMessage(message, data))
+    );
+  }
+
+  registerActor(actor) {
+    this.actors.add(actor);
+  }
+
+  unregisterActor(actor) {
+    this.actors.delete(actor);
+  }
+}
+
+const defaultTabsFactory = () =>
+  new ASRouterTabs({ asRouterNewTabHook: ASRouterNewTabHook });
+
+export class ASRouterParent extends JSWindowActorParent {
+  static tabs = null;
+
+  static nextTabId = 0;
+
+  constructor({ tabsFactory } = { tabsFactory: defaultTabsFactory }) {
+    super();
+    this.tabsFactory = tabsFactory;
+  }
+
+  actorCreated() {
+    ASRouterParent.tabs = ASRouterParent.tabs || this.tabsFactory();
+    this.tabsFactory = null;
+    this.tabId = ++ASRouterParent.nextTabId;
+    ASRouterParent.tabs.registerActor(this);
+  }
+
+  didDestroy() {
+    ASRouterParent.tabs.unregisterActor(this);
+    if (ASRouterParent.tabs.size < 1) {
+      ASRouterParent.tabs.destroy();
+      ASRouterParent.tabs = null;
+    }
+  }
+
+  getTab() {
+    return {
+      id: this.tabId,
+      browser: this.browsingContext.embedderElement,
+    };
+  }
+
+  receiveMessage({ name, data }) {
+    return ASRouterParent.tabs.loadingMessageHandler.then(handler => {
+      return handler.handleMessage(name, data, this.getTab());
+    });
+  }
+}