index.js

"use strict";
const PLUGIN_NAME = "gulp-jsdom";
const through = require("through2");
const PluginError = require('plugin-error');

const jsdom = require("jsdom");
const { JSDOM } = jsdom;

module.exports = function(mutator, options, serialize) {
	options = options || {};
	serialize = serialize || true;
	
	function transform(file, encoding, callback) {
		if (file.isNull()) {
			return callback(null, file);
		}
		if (file.isStream()) {
			callback(new PluginError(PLUGIN_NAME, "Streaming not supported")); return;
		}
		
		try {
			if (file.isBuffer()) {
				const dom = new JSDOM(file.contents.toString("utf8"), options);
				
				const context = {
					file: file,
					filename: file.history[file.history.length - 1].substr(file.base.length)
				};
				const output = mutator.call(context, dom.window.document, dom.window);
				
				file.contents = Buffer.from((typeof output === "string") ? output : (serialize === true) ? dom.serialize() : dom.window.document.documentElement.outerHTML);
				this.push(file);
			}
			
		} catch (err) {
			this.emit("error", new PluginError(PLUGIN_NAME, err));
		}
		
		callback();
	}
	
	return through.obj(transform);
};