/* -*- 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 },
    };
  }
};