From eb09f6102946a16251b44bc37a3d803d4e8cf168 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Samy=20Pess=C3=A9?= <samypesse@gmail.com>
Date: Tue, 1 Apr 2014 23:00:22 -0700
Subject: [PATCH] Fix usage of highlight.js

---
 lib/parse/page.js | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/lib/parse/page.js b/lib/parse/page.js
index b28135c67..09670b809 100644
--- a/lib/parse/page.js
+++ b/lib/parse/page.js
@@ -1,12 +1,17 @@
 var _ = require('lodash');
 var marked = require('marked');
+var hljs = require('highlight.js');
 
 var renderer = require('./renderer');
 
 // Synchronous highlighting with highlight.js
 marked.setOptions({
-    highlight: function (code) {
-        return require('highlight.js').highlightAuto(code).value;
+    highlight: function (code, lang) {
+        try {
+            return hljs.highlight(lang, code).value;
+        } catch(e) {
+            return hljs.highlightAuto(code).value;
+        }
     }
 });