index.js

var cheerio = require('cheerio');
var path = require('path');
var through = require('through2');
var vfs = require('vinyl-fs');
var Inky = require('./lib/inky');
var fs = require('fs');
var mkdirp = require('mkdirp');

var inky;

module.exports = function(opts, cb) {
  var stream;
  opts = opts || {};

  if (typeof inky === 'undefined') {
    inky = new Inky(opts);
  }

  // If the user passed in source files, create a stream
  if (opts.src) {
    stream = vfs
      .src(opts.src)
      .pipe(transform());

    if (opts.dest && typeof cb === 'function') {
      stream.on('finish', cb);
    }
  }
  // Otherwise, return the transform function
  else {
    return transform();
  }

  // This transform function takes in a Vinyl HTML file, converts the code from Inky to HTML, and returns the modified file.
  function transform() {
    return through.obj(function(file, enc, callback) {
      var html = cheerio.load(file.contents.toString(), opts.cheerio);
      var convertedHtml = inky.releaseTheKraken(html).html();

      file.contents = new Buffer(convertedHtml);

      // Write to disk manually if the user specified it
      if (opts.dest) {
        var outputPath = path.join(opts.dest, path.basename(file.path));
        mkdirp(opts.dest, function() {
          fs.writeFile(outputPath, convertedHtml, callback);
        });
      }
      else {
        callback(null, file);
      }
    });
  }
}

module.exports.Inky = Inky;