diff options
Diffstat (limited to 'lib/output/modifiers/modifyHTML.js')
-rw-r--r-- | lib/output/modifiers/modifyHTML.js | 15 |
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; |