Add colorize

lib/linguist/blob_helper.rb

@@ -132,5 +132,10 @@ module Linguist
         end
       end
     end
+
+    def colorize
+      return if !text? || large?
+      lexer.colorize(data)
+    end
   end
 end

lib/linguist/language.rb

@@ -77,6 +77,10 @@ module Linguist
       !popular?
     end
 
+    def colorize(text)
+      lexer.colorize(text)
+    end
+
     def ==(other)
       eql?(other)
     end

lib/linguist/lexer.rb

@@ -1,3 +1,4 @@
+require 'albino'
 require 'yaml'
 
 module Linguist
@@ -21,6 +22,10 @@ module Linguist
       aliases.first
     end
 
+    def colorize(text)
+      Albino.colorize(text, to_s)
+    end
+
     def ==(other)
       eql?(other)
     end

test/test_blob.rb

@@ -170,4 +170,13 @@ class TestBlob < Test::Unit::TestCase
     assert_equal nil, blob("script.foo").shebang_language
     assert_equal nil, blob("foo.rb").shebang_language
   end
+
+  def test_colorize
+    assert_equal <<-HTML, blob("foo.rb").colorize
+<div class="highlight"><pre><span class="k">module</span> <span class="nn">Foo</span>
+<span class="k">end</span>
+</pre>
+</div>
+    HTML
+  end
 end

test/test_language.rb

@@ -95,4 +95,20 @@ class TestLanguage < Test::Unit::TestCase
     assert Language['Assembly'].unpopular?
     assert Language['Brainfuck'].unpopular?
   end
+
+  def test_colorize
+    assert_equal <<-HTML, Language['Text'].colorize("Hello")
+<div class="highlight"><pre>Hello
+</pre>
+</div>
+    HTML
+
+    assert_equal <<-HTML, Language['Ruby'].colorize("def foo\n  'foo'\nend\n")
+<div class="highlight"><pre><span class="k">def</span> <span class="nf">foo</span>
+  <span class="s1">&#39;foo&#39;</span>
+<span class="k">end</span>
+</pre>
+</div>
+    HTML
+  end
 end

test/test_lexer.rb

@@ -34,4 +34,20 @@ class TestLexer < Test::Unit::TestCase
     assert !Lexer['Ruby'].eql?(Lexer['Python'])
     assert !Lexer['Ruby'].eql?(Lexer.new('Ruby'))
   end
+
+  def test_colorize
+    assert_equal <<-HTML, Lexer['Text only'].colorize("Hello")
+<div class="highlight"><pre>Hello
+</pre>
+</div>
+    HTML
+
+    assert_equal <<-HTML, Lexer['Ruby'].colorize("def foo\n  'foo'\nend\n")
+<div class="highlight"><pre><span class="k">def</span> <span class="nf">foo</span>
+  <span class="s1">&#39;foo&#39;</span>
+<span class="k">end</span>
+</pre>
+</div>
+    HTML
+  end
 end