1 files changed, 103 insertions, 0 deletions
diff --git a/browser/components/newtab/actors/ASRouterParent.sys.mjs b/browser/components/newtab/actors/ASRouterParent.sys.mjs
new file mode 100644
index 0000000000..9c785faaa5
--- /dev/null
+++ b/browser/components/newtab/actors/ASRouterParent.sys.mjs
@@ -0,0 +1,103 @@
+/* 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/. */
+
+import { MESSAGE_TYPE_HASH } from "resource://activity-stream/common/ActorConstants.sys.mjs";
+
+import { ASRouterNewTabHook } from "resource://activity-stream/lib/ASRouterNewTabHook.sys.mjs";
+
+const { ASRouterDefaultConfig } = ChromeUtils.import(
+  "resource://activity-stream/lib/ASRouterDefaultConfig.jsm"
+);
+
+export class ASRouterTabs {
+  constructor({ asRouterNewTabHook }) {
+    this.actors = new Set();
+    this.destroy = () => {};
+    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 => {
+      if (name === MESSAGE_TYPE_HASH.BLOCK_MESSAGE_BY_ID) {
+        return Promise.all([
+          handler.handleMessage(name, data, this.getTab()),
+          // All tabs should clear messages not just preloaded, for example
+          // two different windows can display the same snippet.
+          // ASRouter blocks snippets by campaign not by id so we just tell
+          // other tabs that this specific campaign was blocked.
+          ASRouterParent.tabs.messageAll("ClearMessages", [data.campaign]),
+        ]).then(([handleMessageResult]) => handleMessageResult);
+      }
+      return handler.handleMessage(name, data, this.getTab());
+    });
+  }
+}