[ProblemSet] Add more langs.

refs #55

Signed-off-by: Eric Wang <skygragon@gmail.com>

Author: skygragon

lib/helper.js

@@ -78,14 +78,18 @@ h.statusToName = function(sc) {
 
 h.langToExt = function(lang) {
   switch (lang) {
+    case 'bash':       return '.sh';
     case 'c':          return '.c';
     case 'cpp':        return '.cpp';
     case 'csharp':     return '.cs';
     case 'golang':     return '.go';
     case 'java':       return '.java';
     case 'javascript': return '.js';
+    case 'mysql':      return '.sql';
     case 'python':     return '.py';
+    case 'python3':    return '.py3';
     case 'ruby':       return '.rb';
+    case 'scala':      return '.scala';
     case 'swift':      return '.swift';
     default:           return '.raw';
   }
@@ -101,7 +105,11 @@ h.extToLang = function(fullpath) {
     case '.java':  return 'java';
     case '.js':    return 'javascript';
     case '.py':    return 'python';
-    case '.rb'   : return 'ruby';
+    case '.py3':   return 'python3';
+    case '.rb':    return 'ruby';
+    case '.scala': return 'scala';
+    case '.sh':    return 'bash';
+    case '.sql':   return 'mysql';
     case '.swift': return 'swift';
     default:       return 'unknown';
   }
@@ -115,14 +123,18 @@ h.langToCommentStyle = function(lang) {
     case 'golang':
     case 'java':
     case 'javascript':
+    case 'scala':
     case 'swift':
     default:
       return {
         commentHeader: '/*',
         commentLine:   ' *',
         commentFooter: ' */'
       };
+    case 'bash':
+    case 'mysql':
     case 'python':
+    case 'python3':
     case 'ruby':
       return {
         commentHeader: '#',

test/test_helper.js

@@ -96,30 +96,38 @@ describe('helper', function() {
 
   describe('#langToExt', function() {
     it('should ok', function() {
+      assert.equal(h.langToExt('bash'), '.sh');
       assert.equal(h.langToExt('c'), '.c');
       assert.equal(h.langToExt('cpp'), '.cpp');
       assert.equal(h.langToExt('csharp'), '.cs');
       assert.equal(h.langToExt('golang'), '.go');
       assert.equal(h.langToExt('java'), '.java');
       assert.equal(h.langToExt('javascript'), '.js');
+      assert.equal(h.langToExt('mysql'), '.sql');
       assert.equal(h.langToExt('python'), '.py');
+      assert.equal(h.langToExt('python3'), '.py3');
       assert.equal(h.langToExt('ruby'), '.rb');
+      assert.equal(h.langToExt('scala'), '.scala');
       assert.equal(h.langToExt('swift'), '.swift');
       assert.equal(h.langToExt('rust'), '.raw');
     });
   }); // #langToExt
 
   describe('#extToLang', function() {
     it('should ok', function() {
+      assert.equal(h.extToLang('/usr/bin/file.sh'), 'bash');
       assert.equal(h.extToLang('/home/skygragon/file.c'), 'c');
       assert.equal(h.extToLang('/var/log/file.cpp'), 'cpp');
       assert.equal(h.extToLang('./file.cs'), 'csharp');
       assert.equal(h.extToLang('../file.go'), 'golang');
       assert.equal(h.extToLang('file.java'), 'java');
       assert.equal(h.extToLang('c:/file.js'), 'javascript');
       assert.equal(h.extToLang('c:/Users/skygragon/file.py'), 'python');
+      assert.equal(h.extToLang('c:/Users/skygragon/file.py3'), 'python3');
       assert.equal(h.extToLang('~/file.rb'), 'ruby');
+      assert.equal(h.extToLang('/tmp/file.scala'), 'scala');
       assert.equal(h.extToLang('~/leetcode/file.swift'), 'swift');
+      assert.equal(h.extToLang('~/leetcode/../file.sql'), 'mysql');
       assert.equal(h.extToLang('/home/skygragon/file.dat'), 'unknown');
     });
   }); // #extToLang
@@ -137,14 +145,18 @@ describe('helper', function() {
         commentFooter: '#'
       };
 
+      assert.deepEqual(h.langToCommentStyle('bash'), RUBY_STYLE);
       assert.deepEqual(h.langToCommentStyle('c'), C_STYLE);
       assert.deepEqual(h.langToCommentStyle('cpp'), C_STYLE);
       assert.deepEqual(h.langToCommentStyle('csharp'), C_STYLE);
       assert.deepEqual(h.langToCommentStyle('golang'), C_STYLE);
       assert.deepEqual(h.langToCommentStyle('java'), C_STYLE);
       assert.deepEqual(h.langToCommentStyle('javascript'), C_STYLE);
+      assert.deepEqual(h.langToCommentStyle('mysql'), RUBY_STYLE);
       assert.deepEqual(h.langToCommentStyle('python'), RUBY_STYLE);
+      assert.deepEqual(h.langToCommentStyle('python3'), RUBY_STYLE);
       assert.deepEqual(h.langToCommentStyle('ruby'), RUBY_STYLE);
+      assert.deepEqual(h.langToCommentStyle('scala'), C_STYLE);
       assert.deepEqual(h.langToCommentStyle('swift'), C_STYLE);
     });
   }); // #langToCommentStyle