1 files changed, 10 insertions, 0 deletions
diff --git a/lib/generate/page/index.js b/lib/generate/page/index.js
index f2c7b60..7f730ea 100644
--- a/lib/generate/page/index.js
+++ b/lib/generate/page/index.js
@@ -3,11 +3,21 @@ var util = require("util");
 var path = require("path");
 var Q = require("q");
 var swig = require('swig');
+var hljs = require('highlight.js');
 
 var fs = require("../fs");
 var parse = require("../../parse");
 var BaseGenerator = require("../generator");
 
+// Swig filter: highlight coloration
+swig.setFilter('code', function(code, lang) {
+    try {
+        return hljs.highlight(lang, code).value;
+    } catch(e) {
+        return hljs.highlightAuto(code).value;
+    }
+});
+
 
 /*
  *  This generator will generate a simple index.html which can be converted as a PDF