Adapt themes website templates

1 files changed, 9 insertions, 0 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();
 };