Merge pull request #31 from GitbookIO/feature/lang

Feature/lang

4 files changed, 38 insertions, 0 deletions

diff --git a/lib/parse/page.js b/lib/parse/page.js
index 937d84e..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) {
@@ -100,10 +103,22 @@ function parsePage(src, options) {
                 'type': 'code'
             });
 
+            // Languages in code blocks
+            var langs = _.pluck(codeNodes, 'lang').map(lnormalize);
+
+            // Check that they are all the same
+            var validLangs = _.all(_.map(langs, function(lang) {
+                return lang && lang === langs[0];
+            }));
+
+            // Main language
+            var lang = validLangs ? langs[0] : null;
+
             return {
                 id: id,
                 type: section.type,
                 content: render(nonCodeNodes),
+                lang: lang,
                 code: {
                     base: codeNodes[0].text,
                     solution: codeNodes[1].text,
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
+};
diff --git a/test/page.js b/test/page.js
index 68c0749..0f54809 100644
--- a/test/page.js
+++ b/test/page.js
@@ -45,6 +45,10 @@ describe('Page parsing', function() {
         // HRs inserted correctly
         assert.equal(HR_LEXED[0].content.match(/<hr>/g).length, 2);
     });
+
+    it('should detect an exercise\'s language', function() {
+        assert.equal(LEXED[1].lang, 'python');
+    });
 });