1 files changed, 57 insertions, 0 deletions
diff --git a/browser/components/pagedata/PageDataParent.sys.mjs b/browser/components/pagedata/PageDataParent.sys.mjs
new file mode 100644
index 0000000000..c3e7743b99
--- /dev/null
+++ b/browser/components/pagedata/PageDataParent.sys.mjs
@@ -0,0 +1,57 @@
+/* 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/. */
+
+const lazy = {};
+
+ChromeUtils.defineESModuleGetters(lazy, {
+  PageDataService: "resource:///modules/pagedata/PageDataService.sys.mjs",
+  PromiseUtils: "resource://gre/modules/PromiseUtils.sys.mjs",
+});
+
+/**
+ * Receives messages from PageDataChild and passes them to the PageData service.
+ */
+export class PageDataParent extends JSWindowActorParent {
+  #deferredCollection = null;
+
+  /**
+   * Starts data collection in the child process. Returns a promise that
+   * resolves to the page data or null if the page is closed before data
+   * collection completes.
+   *
+   * @returns {Promise<PageData|null>}
+   */
+  collectPageData() {
+    if (!this.#deferredCollection) {
+      this.#deferredCollection = lazy.PromiseUtils.defer();
+      this.sendQuery("PageData:Collect").then(
+        this.#deferredCollection.resolve,
+        this.#deferredCollection.reject
+      );
+    }
+
+    return this.#deferredCollection.promise;
+  }
+
+  /**
+   * Called when the page is destroyed.
+   */
+  didDestroy() {
+    this.#deferredCollection?.resolve(null);
+  }
+
+  /**
+   * Called when a message is received from the content process.
+   *
+   * @param {ReceiveMessageArgument} msg
+   *   The received message.
+   */
+  receiveMessage(msg) {
+    switch (msg.name) {
+      case "PageData:DocumentReady":
+        lazy.PageDataService.pageLoaded(this, msg.data.url);
+        break;
+    }
+  }
+}