1 files changed, 53 insertions, 0 deletions
diff --git a/packages/gitbook/lib/templating/postRender.js b/packages/gitbook/lib/templating/postRender.js
new file mode 100644
index 0000000..f464f86
--- /dev/null
+++ b/packages/gitbook/lib/templating/postRender.js
@@ -0,0 +1,53 @@
+var Promise = require('../utils/promise');
+
+
+/**
+ * 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, blocks) {
+    var newContent = content.replace(/\{\{\-\%([\s\S]+?)\%\-\}\}/g, function(match, key) {
+        var replacedWith = match;
+
+        var block = blocks.get(key);
+        if (block) {
+            replacedWith = replaceBlocks(block.get('body'), blocks);
+        }
+
+        return replacedWith;
+    });
+
+    return newContent;
+}
+
+/**
+ * Post render a template:
+ *     - Execute "post" for blocks
+ *     - Replace block content
+ *
+ * @param {TemplateEngine} engine
+ * @param {TemplateOutput} content
+ * @return {Promise<String>}
+ */
+function postRender(engine, output) {
+    var content = output.getContent();
+    var blocks = output.getBlocks();
+
+    var result = replaceBlocks(content, blocks);
+
+    return Promise.forEach(blocks, function(block) {
+        var post = block.get('post');
+
+        if (!post) {
+            return;
+        }
+
+        return post();
+    })
+    .thenResolve(result);
+}
+
+module.exports = postRender;