Add loader for themes

3 files changed, 30 insertions, 2 deletions

diff --git a/lib/output/website/createTemplateEngine.js b/lib/output/website/createTemplateEngine.js
index 9aed746..1a70667 100644
--- a/lib/output/website/createTemplateEngine.js
+++ b/lib/output/website/createTemplateEngine.js
@@ -1,4 +1,6 @@
+var Templating = require('../../templating');
 var TemplateEngine = require('../../models/templateEngine');
+var listSearchPaths = require('./listSearchPaths');
 
 /**
     Create templating engine to render themes
@@ -7,8 +9,11 @@ var TemplateEngine = require('../../models/templateEngine');
     @return {TemplateEngine}
 */
 function createTemplateEngine(output) {
-    return new TemplateEngine({
+    var searchPaths = listSearchPaths(output);
+    var loader = new Templating.ThemesLoader(searchPaths);
 
+    return new TemplateEngine({
+        loader: loader
     });
 }
 
diff --git a/lib/output/website/listSearchPaths.js b/lib/output/website/listSearchPaths.js
new file mode 100644
index 0000000..c45f39c
--- /dev/null
+++ b/lib/output/website/listSearchPaths.js
@@ -0,0 +1,23 @@
+
+/**
+    List search paths for templates / i18n, etc
+
+    @param {Output} output
+    @return {List<String>}
+*/
+function listSearchPaths(output) {
+    var book = output.getBook();
+    var plugins = output.getPlugins();
+
+    var searchPaths = plugins
+        .valueSeq()
+        .map(function(plugin) {
+            return plugin.getPath();
+        })
+        .toList();
+
+    return searchPaths.unshift(book.getContentRoot());
+}
+
+
+module.exports = listSearchPaths;
diff --git a/lib/output/website/onAsset.js b/lib/output/website/onAsset.js
index a2342ae..8659ca6 100644
--- a/lib/output/website/onAsset.js
+++ b/lib/output/website/onAsset.js
@@ -11,7 +11,7 @@ function onAsset(output, asset) {
     var book = output.getBook();
     var options = output.getOptions();
 
-    var rootFolder = book.getRoot();
+    var rootFolder = book.getContentRoot();
     var outputFolder = options.get('root');
 
     var filePath = path.resolve(rootFolder, asset);