1 files changed, 199 insertions, 0 deletions
diff --git a/remote/domains/parent/Target.jsm b/remote/domains/parent/Target.jsm
new file mode 100644
index 0000000000..0d572cd517
--- /dev/null
+++ b/remote/domains/parent/Target.jsm
@@ -0,0 +1,199 @@
+/* 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";
+
+var EXPORTED_SYMBOLS = ["Target"];
+
+const { XPCOMUtils } = ChromeUtils.import(
+  "resource://gre/modules/XPCOMUtils.jsm"
+);
+XPCOMUtils.defineLazyServiceGetter(
+  this,
+  "UUIDGen",
+  "@mozilla.org/uuid-generator;1",
+  "nsIUUIDGenerator"
+);
+
+const { ContextualIdentityService } = ChromeUtils.import(
+  "resource://gre/modules/ContextualIdentityService.jsm"
+);
+const { Domain } = ChromeUtils.import(
+  "chrome://remote/content/domains/Domain.jsm"
+);
+const { MainProcessTarget } = ChromeUtils.import(
+  "chrome://remote/content/targets/MainProcessTarget.jsm"
+);
+const { TabManager } = ChromeUtils.import(
+  "chrome://remote/content/TabManager.jsm"
+);
+const { TabSession } = ChromeUtils.import(
+  "chrome://remote/content/sessions/TabSession.jsm"
+);
+const { WindowManager } = ChromeUtils.import(
+  "chrome://remote/content/WindowManager.jsm"
+);
+
+let browserContextIds = 1;
+
+class Target extends Domain {
+  constructor(session) {
+    super(session);
+
+    this._onTargetCreated = this._onTargetCreated.bind(this);
+    this._onTargetDestroyed = this._onTargetDestroyed.bind(this);
+  }
+
+  getBrowserContexts() {
+    return {
+      browserContextIds: [],
+    };
+  }
+
+  createBrowserContext() {
+    const identity = ContextualIdentityService.create(
+      "remote-agent-" + browserContextIds++
+    );
+    return { browserContextId: identity.userContextId };
+  }
+
+  disposeBrowserContext({ browserContextId }) {
+    ContextualIdentityService.remove(browserContextId);
+    ContextualIdentityService.closeContainerTabs(browserContextId);
+  }
+
+  getTargets() {
+    const { targets } = this.session.target;
+
+    const targetInfos = [];
+    for (const target of targets) {
+      if (target instanceof MainProcessTarget) {
+        continue;
+      }
+
+      targetInfos.push(this._getTargetInfo(target));
+    }
+
+    return { targetInfos };
+  }
+
+  setDiscoverTargets({ discover }) {
+    const { targets } = this.session.target;
+    if (discover) {
+      targets.on("target-created", this._onTargetCreated);
+      targets.on("target-destroyed", this._onTargetDestroyed);
+    } else {
+      targets.off("target-created", this._onTargetCreated);
+      targets.off("target-destroyed", this._onTargetDestroyed);
+    }
+    for (const target of targets) {
+      this._onTargetCreated("target-created", target);
+    }
+  }
+
+  async createTarget({ browserContextId }) {
+    const { targets } = this.session.target;
+    const onTarget = targets.once("target-created");
+    const tab = TabManager.addTab({ userContextId: browserContextId });
+    const target = await onTarget;
+    if (tab.linkedBrowser != target.browser) {
+      throw new Error(
+        "Unexpected tab opened: " + tab.linkedBrowser.currentURI.spec
+      );
+    }
+    return { targetId: target.id };
+  }
+
+  closeTarget({ targetId }) {
+    const { targets } = this.session.target;
+    const target = targets.getById(targetId);
+
+    if (!target) {
+      throw new Error(`Unable to find target with id '${targetId}'`);
+    }
+
+    TabManager.removeTab(target.tab);
+  }
+
+  async activateTarget({ targetId }) {
+    const { targets, window } = this.session.target;
+    const target = targets.getById(targetId);
+
+    if (!target) {
+      throw new Error(`Unable to find target with id '${targetId}'`);
+    }
+
+    // Focus the window, and select the corresponding tab
+    await WindowManager.focus(window);
+    TabManager.selectTab(target.tab);
+  }
+
+  attachToTarget({ targetId }) {
+    const { targets } = this.session.target;
+    const target = targets.getById(targetId);
+
+    if (!target) {
+      throw new Error(`Unable to find target with id '${targetId}'`);
+    }
+
+    const tabSession = new TabSession(
+      this.session.connection,
+      target,
+      UUIDGen.generateUUID()
+        .toString()
+        .slice(1, -1)
+    );
+    this.session.connection.registerSession(tabSession);
+
+    this._emitAttachedToTarget(target, tabSession);
+
+    return {
+      sessionId: tabSession.id,
+    };
+  }
+
+  setAutoAttach() {}
+
+  sendMessageToTarget({ sessionId, message }) {
+    const { connection } = this.session;
+    connection.sendMessageToTarget(sessionId, message);
+  }
+
+  /**
+   * Internal methods: the following methods are not part of CDP;
+   * note the _ prefix.
+   */
+
+  _emitAttachedToTarget(target, tabSession) {
+    const targetInfo = this._getTargetInfo(target);
+    this.emit("Target.attachedToTarget", {
+      targetInfo,
+      sessionId: tabSession.id,
+      waitingForDebugger: false,
+    });
+  }
+
+  _getTargetInfo(target) {
+    return {
+      targetId: target.id,
+      type: target.type,
+      title: target.title,
+      url: target.url,
+      // TODO: Correctly determine if target is attached (bug 1680780)
+      attached: target.id == this.session.target.id,
+      browserContextId: target.browserContextId,
+    };
+  }
+
+  _onTargetCreated(eventName, target) {
+    const targetInfo = this._getTargetInfo(target);
+    this.emit("Target.targetCreated", { targetInfo });
+  }
+
+  _onTargetDestroyed(eventName, target) {
+    this.emit("Target.targetDestroyed", {
+      targetId: target.id,
+    });
+  }
+}