Adapt themes website templates

2 files changed, 12 insertions, 1 deletions

diff --git a/lib/generators/site.js b/lib/generators/site.js
index c8b9e91..a19da25 100644
--- a/lib/generators/site.js
+++ b/lib/generators/site.js
@@ -2,7 +2,11 @@ var util = require("util");
 var path = require("path");
 var Q = require("q");
 var _ = require("lodash");
+
 var nunjucks = require("nunjucks");
+var ParentExtension = require("nunjucks-parent");
+var AutoEscapeExtension = require("nunjucks-autoescape");
+var FilterExtension = require("nunjucks-filter");
 
 var fs = require("../utils/fs");
 var BaseGenerator = require("../generator");
@@ -74,6 +78,11 @@ Generator.prototype.prepareTemplates = function() {
 		}
 	);
 
+    // Add extension
+    this.env.addExtension('ParentExtension', new ParentExtension());
+    this.env.addExtension('AutoEscapeExtension', new AutoEscapeExtension(this.env));
+    this.env.addExtension('FilterExtension', new FilterExtension(this.env));
+
 	return Q();
 };
 
diff --git a/lib/template.js b/lib/template.js
index e4df84c..27234f2 100644
--- a/lib/template.js
+++ b/lib/template.js
@@ -1,7 +1,9 @@
 var _ = require("lodash");
 var Q = require("q");
+
 var nunjucks = require("nunjucks");
 
+
 var TemplateEngine = function(book) {
 	this.book = book;
 
@@ -22,7 +24,7 @@ TemplateEngine.prototype.renderFile = function(filename) {
 	return that.book.statFile(filename)
 	.then(function(stat) {
 		var context = {
-			// Variabels from book.json
+			// Variables from book.json
 			book: that.book.options.variables,
 
 			// infos about the file