1 files changed, 54 insertions, 0 deletions
diff --git a/devtools/shared/fronts/device.js b/devtools/shared/fronts/device.js
new file mode 100644
index 0000000000..28f7a096a6
--- /dev/null
+++ b/devtools/shared/fronts/device.js
@@ -0,0 +1,54 @@
+/* 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 {Cc, Ci, Cu} = require("chrome");
+const {deviceSpec} = require("devtools/shared/specs/device");
+const protocol = require("devtools/shared/protocol");
+const defer = require("devtools/shared/defer");
+
+const DeviceFront = protocol.FrontClassWithSpec(deviceSpec, {
+  initialize: function (client, form) {
+    protocol.Front.prototype.initialize.call(this, client);
+    this.actorID = form.deviceActor;
+    this.manage(this);
+  },
+
+  screenshotToBlob: function () {
+    return this.screenshotToDataURL().then(longstr => {
+      return longstr.string().then(dataURL => {
+        let deferred = defer();
+        longstr.release().then(null, Cu.reportError);
+        let req = Cc["@mozilla.org/xmlextras/xmlhttprequest;1"]
+            .createInstance(Ci.nsIXMLHttpRequest);
+        req.open("GET", dataURL, true);
+        req.responseType = "blob";
+        req.onload = () => {
+          deferred.resolve(req.response);
+        };
+        req.onerror = () => {
+          deferred.reject(req.status);
+        };
+        req.send();
+        return deferred.promise;
+      });
+    });
+  },
+});
+
+const _knownDeviceFronts = new WeakMap();
+
+exports.getDeviceFront = function (client, form) {
+  if (!form.deviceActor) {
+    return null;
+  }
+
+  if (_knownDeviceFronts.has(client)) {
+    return _knownDeviceFronts.get(client);
+  }
+
+  let front = new DeviceFront(client, form);
+  _knownDeviceFronts.set(client, front);
+  return front;
+};