Merge pull request #153 from pchaigno/linguist-lexers

Use Linguist lexers
gjtorikian · Oct 5, 2014 · 68b876f · 68b876f
2 parents aeadfbd + 956be5b
commit 68b876f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/html/pipeline/syntax_highlight_filter.rb b/lib/html/pipeline/syntax_highlight_filter.rb
@@ -13,7 +13,7 @@ def call
         doc.search('pre').each do |node|
           default = context[:highlight] && context[:highlight].to_s
           next unless lang = node['lang'] || default
-          next unless lexer = Pygments::Lexer[lang]
+          next unless lexer = lexer_for(lang)
           text = node.inner_text
 
           html = highlight_with_timeout_handling(lexer, text)
@@ -34,6 +34,10 @@ def highlight_with_timeout_handling(lexer, text)
       rescue Timeout::Error => boom
         nil
       end
+
+      def lexer_for(lang)
+        (Linguist::Language[lang] && Linguist::Language[lang].lexer) || Pygments::Lexer[lang]
+      end
     end
   end
 end