Add basic code language normalization

3 files changed, 23 insertions, 1 deletions

diff --git a/lib/parse/page.js b/lib/parse/page.js
index cfe1da8..75eca9d 100644
--- a/lib/parse/page.js
+++ b/lib/parse/page.js
@@ -4,6 +4,9 @@ var hljs = require('highlight.js');
 
 var renderer = require('./renderer');
 
+var lnormalize = require('../utils/lang').normalize;
+
+
 // Synchronous highlighting with highlight.js
 marked.setOptions({
     highlight: function (code, lang) {
@@ -101,7 +104,7 @@ function parsePage(src, options) {
             });
 
             // Languages in code blocks
-            var langs = _.pluck(codeNodes, 'lang');
+            var langs = _.pluck(codeNodes, 'lang').map(lnormalize);
 
             // Check that they are all the same
             var validLangs = _.all(_.map(langs, function(lang) {
diff --git a/lib/utils/index.js b/lib/utils/index.js
new file mode 100644
index 0000000..155e723
--- /dev/null
+++ b/lib/utils/index.js
@@ -0,0 +1,3 @@
+module.exports = {
+    lang: require('./lang'),
+};
diff --git a/lib/utils/lang.js b/lib/utils/lang.js
new file mode 100644
index 0000000..7fd71e1
--- /dev/null
+++ b/lib/utils/lang.js
@@ -0,0 +1,16 @@
+var MAP = {
+    'py': 'python',
+    'js': 'javascript',
+    'rb': 'ruby',
+};
+
+function normalize(lang) {
+    var lower = lang.toLowerCase();
+    return MAP[lower] || lower;
+}
+
+// Exports
+module.exports = {
+    normalize: normalize,
+    MAP: MAP
+};