Merge branch 'fixes'

1 files changed, 25 insertions, 0 deletions

diff --git a/lib/output/modifiers/modifyHTML.js b/lib/output/modifiers/modifyHTML.js
new file mode 100644
index 0000000..0fcf994
--- /dev/null
+++ b/lib/output/modifiers/modifyHTML.js
@@ -0,0 +1,25 @@
+var cheerio = require('cheerio');
+var Promise = require('../../utils/promise');
+
+/**
+    Apply a list of operations to a page and
+    output the new page.
+
+    @param {Page}
+    @param {List|Array<Transformation>}
+    @return {Promise<Page>}
+*/
+function modifyHTML(page, operations) {
+    var html = page.getContent();
+    var $ = cheerio.load(html);
+
+    return Promise.forEach(operations, function(op) {
+        op($);
+    })
+    .then(function() {
+        var resultHTML = $.html();
+        return page.set('content', resultHTML);
+    });
+}
+
+module.exports = modifyHTML;