Add assets inliner modifier for HTML

1 files changed, 12 insertions, 3 deletions

diff --git a/lib/output/modifiers/modifyHTML.js b/lib/output/modifiers/modifyHTML.js
index c1dad74..d46a44f 100644
--- a/lib/output/modifiers/modifyHTML.js
+++ b/lib/output/modifiers/modifyHTML.js
@@ -1,15 +1,24 @@
-
+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<Transformation>}
+    @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(resultHTML) {
+        return page.set('content', resultHTML);
+    });
 }
 
-
 module.exports = modifyHTML;