Add post processing for templates

3 files changed, 64 insertions, 1 deletions

diff --git a/lib/templating/index.js b/lib/templating/index.js
index 312146a..3a8f989 100644
--- a/lib/templating/index.js
+++ b/lib/templating/index.js
@@ -1,4 +1,5 @@
 
 module.exports = {
-    render:         require('./render')
+    render:         require('./render'),
+    postRender:     require('./postRender')
 };
diff --git a/lib/templating/postRender.js b/lib/templating/postRender.js
new file mode 100644
index 0000000..6928e82
--- /dev/null
+++ b/lib/templating/postRender.js
@@ -0,0 +1,28 @@
+var Promise = require('../utils/promise');
+var replaceBlocks = require('./replaceBlocks');
+
+/**
+    Post render a template:
+        - Execute "post" for blocks
+        - Replace block content
+
+    @param {TemplateEngine} engine
+    @param {String} content
+    @return {Promise<String>}
+*/
+function postRender(engine, content) {
+    var result = replaceBlocks(content);
+
+    return Promise.forEach(result.blocks, function(blockType) {
+        var block = engine.getBlock();
+        var post = block.getPost();
+        if (!post) {
+            return;
+        }
+
+        return post();
+    })
+    .thenResolve(result.content);
+}
+
+module.exports = postRender;
diff --git a/lib/templating/replaceBlocks.js b/lib/templating/replaceBlocks.js
new file mode 100644
index 0000000..4b1c37f
--- /dev/null
+++ b/lib/templating/replaceBlocks.js
@@ -0,0 +1,34 @@
+var Immutable = require('immutable');
+var TemplateBlock = require('../models/templateBlock');
+
+/**
+    Replace position markers of blocks by body after processing
+    This is done to avoid that markdown/asciidoc processer parse the block content
+
+    @param {String} content
+    @return {Object} {blocks: Set, content: String}
+*/
+function replaceBlocks(content) {
+    var blockTypes = new Immutable.Set();
+    var newContent = content.replace(/\{\{\-\%([\s\S]+?)\%\-\}\}/g, function(match, key) {
+        var replacedWith = match;
+
+        var block = TemplateBlock.getBlockResultByKey(key);
+        if (block) {
+            var result = replaceBlocks(block.body);
+
+            blockTypes = blockTypes.add(block.name);
+            blockTypes = blockTypes.concat(result.blocks);
+            replacedWith = result.content;
+        }
+
+        return replacedWith;
+    });
+
+    return {
+        content: newContent,
+        blocks: blockTypes
+    };
+}
+
+module.exports = replaceBlocks;