fune/browser/components/ssb/ImageTools.jsm

/* 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/. */

var EXPORTED_SYMBOLS = ["ImageTools"];

const { XPCOMUtils } = ChromeUtils.import(
  "resource://gre/modules/XPCOMUtils.jsm"
);

XPCOMUtils.defineLazyModuleGetters(this, {
  FileUtils: "resource://gre/modules/FileUtils.jsm",
  NetUtil: "resource://gre/modules/NetUtil.jsm",
  Services: "resource://gre/modules/Services.jsm",
});

XPCOMUtils.defineLazyServiceGetter(
  this,
  "ImgTools",
  "@mozilla.org/image/tools;1",
  Ci.imgITools
);

XPCOMUtils.defineLazyGlobalGetters(this, ["Blob"]);

const ImageTools = {
  /**
   * Given a data URI decodes the data into an object with "type" which is the
   * found mimetype and "container" which is an imgIContainer.
   *
   * @param {nsIURI} dataURI the URI to load.
   * @return {Promise<object>} the image info.
   */
  loadImage(dataURI) {
    return new Promise((resolve, reject) => {
      if (!dataURI.schemeIs("data")) {
        reject(new Error("Should only be loading data URIs."));
        return;
      }

      let channel = NetUtil.newChannel({
        uri: dataURI,
        loadUsingSystemPrincipal: true,
      });

      ImgTools.decodeImageFromChannelAsync(
        dataURI,
        channel,
        (container, status) => {
          if (Components.isSuccessCode(status)) {
            resolve({
              type: channel.contentType,
              container,
            });
          } else {
            reject(Components.Exception("Failed to load image.", status));
          }
        },
        null
      );
    });
  },

  scaleImage(container, width, height) {
    return new Promise((resolve, reject) => {
      let stream = ImgTools.encodeScaledImage(
        container,
        "image/png",
        width,
        height,
        ""
      );

      try {
        stream.QueryInterface(Ci.nsIAsyncInputStream);
      } catch (e) {
        reject(
          Components.Exception(
            "imgIEncoder must implement nsIAsyncInputStream",
            e
          )
        );
      }

      let binaryStream = Cc["@mozilla.org/binaryinputstream;1"].createInstance(
        Ci.nsIBinaryInputStream
      );
      binaryStream.setInputStream(stream);

      let buffers = [];
      let callback = () => {
        try {
          let available = binaryStream.available();
          if (available) {
            let buffer = new ArrayBuffer(available);
            binaryStream.readArrayBuffer(available, buffer);
            buffers.push(buffer);

            stream.asyncWait(callback, 0, 0, Services.tm.mainThread);
            return;
          }

          // No data available, assume the encoding is done.
          resolve(new Blob(buffers));
        } catch (e) {
          reject(e);
        }
      };

      try {
        stream.asyncWait(callback, 0, 0, Services.tm.mainThread);
      } catch (e) {
        reject(e);
      }
    });
  },

  saveIcon(container, width, height, target) {
    return new Promise((resolve, reject) => {
      let output = FileUtils.openFileOutputStream(target);
      let stream = ImgTools.encodeScaledImage(
        container,
        "image/vnd.microsoft.icon",
        width,
        height,
        ""
      );
      NetUtil.asyncCopy(stream, output, status => {
        if (Components.isSuccessCode(status)) {
          resolve();
        } else {
          reject(Components.Exception("Failed to save icon.", status));
        }
      });
    });
  },
};