1 files changed, 102 insertions, 0 deletions
diff --git a/devtools/client/fronts/source.js b/devtools/client/fronts/source.js
new file mode 100644
index 0000000000..d8a195bbb2
--- /dev/null
+++ b/devtools/client/fronts/source.js
@@ -0,0 +1,102 @@
+/* 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";
+
+const { sourceSpec } = require("resource://devtools/shared/specs/source.js");
+const {
+  FrontClassWithSpec,
+  registerFront,
+} = require("resource://devtools/shared/protocol.js");
+const {
+  ArrayBufferFront,
+} = require("resource://devtools/client/fronts/array-buffer.js");
+
+/**
+ * A SourceFront provides a way to access the source text of a script.
+ *
+ * @param client DevToolsClient
+ *        The DevTools Client instance.
+ * @param form Object
+ *        The form sent across the remote debugging protocol.
+ */
+class SourceFront extends FrontClassWithSpec(sourceSpec) {
+  constructor(client, form) {
+    super(client);
+    if (form) {
+      this._url = form.url;
+      // this is here for the time being, until the source front is managed
+      // via protocol.js marshalling
+      this.actorID = form.actor;
+    }
+  }
+
+  form(json) {
+    this._url = json.url;
+  }
+
+  get actor() {
+    return this.actorID;
+  }
+
+  get url() {
+    return this._url;
+  }
+
+  // Alias for source.blackbox to avoid changing protocol.js packets
+  blackBox(range) {
+    return this.blackbox(range);
+  }
+
+  // Alias for source.unblackbox to avoid changing protocol.js packets
+  unblackBox() {
+    return this.unblackbox();
+  }
+
+  /**
+   * Get a Front for either an ArrayBuffer or LongString
+   * for this SourceFront's source.
+   */
+  async source() {
+    const response = await super.source();
+    return this._onSourceResponse(response);
+  }
+
+  _onSourceResponse(response) {
+    const { contentType, source } = response;
+    if (source instanceof ArrayBufferFront) {
+      return source.slice(0, source.length).then(function (resp) {
+        if (resp.error) {
+          return resp;
+        }
+        // Keeping str as a string, ArrayBuffer/Uint8Array will not survive
+        // setIn/mergeIn operations.
+        const str = atob(resp.encoded);
+        const newResponse = {
+          source: {
+            binary: str,
+            toString: () => "[wasm]",
+          },
+          contentType,
+        };
+        return newResponse;
+      });
+    }
+
+    return source.substring(0, source.length).then(function (resp) {
+      if (resp.error) {
+        return resp;
+      }
+
+      const newResponse = {
+        source: resp,
+        contentType,
+      };
+      return newResponse;
+    });
+  }
+}
+
+exports.SourceFront = SourceFront;
+registerFront(SourceFront);