1 files changed, 83 insertions, 0 deletions
diff --git a/comm/mail/components/extensions/child/ext-extensionScripts.js b/comm/mail/components/extensions/child/ext-extensionScripts.js
new file mode 100644
index 0000000000..5d5f364c3e
--- /dev/null
+++ b/comm/mail/components/extensions/child/ext-extensionScripts.js
@@ -0,0 +1,83 @@
+/* -*- 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";
+
+var { ExtensionError } = ExtensionUtils;
+
+/**
+ * Represents (in the child extension process) a script registered
+ * programmatically (instead of being included in the addon manifest).
+ *
+ * @param {ExtensionPageContextChild} context
+ *        The extension context which has registered the script.
+ * @param {string} scriptId
+ *        An unique id that represents the registered script
+ *        (generated and used internally to identify it across the different processes).
+ */
+class ExtensionScriptChild {
+  constructor(type, context, scriptId) {
+    this.type = type;
+    this.context = context;
+    this.scriptId = scriptId;
+    this.unregistered = false;
+  }
+
+  async unregister() {
+    if (this.unregistered) {
+      throw new ExtensionError("script already unregistered");
+    }
+
+    this.unregistered = true;
+
+    await this.context.childManager.callParentAsyncFunction(
+      "extensionScripts.unregister",
+      [this.scriptId]
+    );
+
+    this.context = null;
+  }
+
+  api() {
+    const { context } = this;
+
+    return {
+      unregister: () => {
+        return context.wrapPromise(this.unregister());
+      },
+    };
+  }
+}
+
+this.extensionScripts = class extends ExtensionAPI {
+  getAPI(context) {
+    let api = {
+      register(options) {
+        return context.cloneScope.Promise.resolve().then(async () => {
+          const scriptId = await context.childManager.callParentAsyncFunction(
+            "extensionScripts.register",
+            [this.type, options]
+          );
+
+          const registeredScript = new ExtensionScriptChild(
+            this.type,
+            context,
+            scriptId
+          );
+
+          return Cu.cloneInto(registeredScript.api(), context.cloneScope, {
+            cloneFunctions: true,
+          });
+        });
+      },
+    };
+
+    return {
+      composeScripts: { type: "compose", ...api },
+      messageDisplayScripts: { type: "messageDisplay", ...api },
+    };
+  }
+};