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 || {};
  opts.cheerio = Inky.mergeCheerioOpts(opts.cheerio);

  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 convertedHtml = inky.releaseTheKraken(
        file.contents.toString(),
        opts.cheerio
      );

      file.contents = new Buffer.from(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;