Replace shortcuts in templating

4 files changed, 102 insertions, 0 deletions

diff --git a/lib/templating/index.js b/lib/templating/index.js
new file mode 100644
index 0000000..312146a
--- /dev/null
+++ b/lib/templating/index.js
@@ -0,0 +1,4 @@
+
+module.exports = {
+    render:         require('./render')
+};
diff --git a/lib/templating/listShortcuts.js b/lib/templating/listShortcuts.js
new file mode 100644
index 0000000..8f2388b
--- /dev/null
+++ b/lib/templating/listShortcuts.js
@@ -0,0 +1,31 @@
+var Immutable = require('immutable');
+var parsers = require('../parsers');
+
+/**
+    Return a list of all shortcuts that can apply
+    to a file for a TemplatEngine
+
+    @param {TemplateEngine} engine
+    @param {String} filePath
+    @return {List<Shortcut>}
+*/
+function listShortcuts(engine, filePath) {
+    var blocks = engine.getBlocks();
+    var parser = parsers.getForFile(filePath);
+    if (!parser) {
+        return Immutable.List();
+    }
+
+    return blocks
+        .map(function(block) {
+            var shortcuts = block.getShortcuts();
+
+            return shortcuts.filter(function(shortcut) {
+                var parsers = shortcut.get('parsers');
+                return parsers.includes(parser.name);
+            });
+        })
+        .flatten(1);
+}
+
+module.exports = listShortcuts;
diff --git a/lib/templating/render.js b/lib/templating/render.js
new file mode 100644
index 0000000..87d5096
--- /dev/null
+++ b/lib/templating/render.js
@@ -0,0 +1,30 @@
+var Promise = require('../utils/promise');
+
+var replaceShortcuts = require('./replaceShortcuts');
+
+/**
+    Render a template
+
+    @param {TemplateEngine} engine
+    @param {String} filePath
+    @param {String} content
+    @param {Object} ctx
+    @return {Promise<String>}
+*/
+function renderTemplate(engine, filePath, content, context) {
+    context = context || {};
+    var env = engine.toNunjucks();
+
+    content = replaceShortcuts(engine, filePath, content);
+
+    return Promise.nfcall(
+        env.renderString.bind(env),
+        content,
+        context,
+        {
+            path: filePath
+        }
+    );
+}
+
+module.exports = renderTemplate;
diff --git a/lib/templating/replaceShortcuts.js b/lib/templating/replaceShortcuts.js
new file mode 100644
index 0000000..f6a51cb
--- /dev/null
+++ b/lib/templating/replaceShortcuts.js
@@ -0,0 +1,37 @@
+var escapeStringRegexp = require('escape-string-regexp');
+var listShortcuts = require('./listShortcuts');
+
+/*
+    Apply a shortcut of block to a template
+    @param {String} content
+    @param {Shortcut} shortcut
+    @return {String}
+*/
+function applyShortcut(content, shortcut) {
+    var tags = shortcut.get('tag');
+    var start = shortcut.get('start');
+    var end = shortcut.get('end');
+
+    var regex = new RegExp(
+        escapeStringRegexp(start) + '([\\s\\S]*?[^\\$])' + escapeStringRegexp(end),
+       'g'
+    );
+    return content.replace(regex, function(all, match) {
+        return '{% ' + tags.start + ' %}' + match + '{% ' + tags.end + ' %}';
+    });
+}
+
+/**
+    Replace shortcuts from blocks in a string
+
+    @param {TemplateEngine} engine
+    @param {String} filePath
+    @param {String} content
+    @return {String}
+*/
+function replaceShortcuts(engine, filePath, content) {
+    var shortcuts = listShortcuts(engine, filePath);
+    return shortcuts.reduce(applyShortcut, content);
+}
+
+module.exports = replaceShortcuts;