added support for Rouge syntax highlighting

.gitignore

@@ -1,5 +1,6 @@
 *.gem
 *.rbc
+*.swp
 .bundle
 .config
 .yardoc

lib/peek/rblineprof/controller_helpers.rb

@@ -1,8 +1,4 @@
-begin
-  require 'pygments.rb'
-rescue LoadError
-  # Doesn't have pygments.rb installed
-end
+require 'peek/rblineprof/syntax_highlighter'
 
 require 'rack/utils'
 
@@ -17,13 +13,9 @@ module ControllerHelpers
 
       protected
 
-      def pygmentize?
-        defined?(Pygments)
-      end
-
-      def pygmentize(file_name, code, lexer = nil)
-        if pygmentize? && lexer.present?
-          Pygments.highlight(code, :lexer => lexer_for_filename(file_name))
+      def highlight(file_name, code, lexer = false)
+        if lexer
+          SyntaxHighlighter.highlight(code, lexer_for_filename(file_name))
         else
           "<pre>#{Rack::Utils.escape_html(code)}</pre>"
         end
@@ -35,8 +27,8 @@ def rblineprof_enabled?
 
       def lexer_for_filename(file_name)
         case file_name
-        when /\.rb$/ then 'ruby'
-        when /\.erb$/ then 'erb'
+        when /\.rb$/  then :ruby
+        when /\.erb$/ then :erb
         end
       end
 
@@ -122,7 +114,7 @@ def inject_rblineprof
               end
             end
             output << "<pre class='duration'>#{times.join("\n")}</pre>"
-            output << "<pre class='code highlight'>#{pygmentize(file_name, code.join, 'ruby')}</pre>"
+            output << "<pre class='code highlight'>#{highlight(file_name, code.join, true)}</pre>"
             output << "</div></div>" # .data then .peek-rblineprof-file
           end
 

lib/peek/rblineprof/highlighters/pygments_highlighter.rb

@@ -0,0 +1,9 @@
+module Peek
+  module Rblineprof
+    class PygmentsHighlighter
+      def self.process(code, lexer)
+        Pygments.highlight(code, :lexer => lexer)
+      end
+    end
+  end
+end

lib/peek/rblineprof/highlighters/rouge_highlighter.rb

@@ -0,0 +1,23 @@
+module Peek
+  module Rblineprof
+    class RougeHighlighter
+      def self.process(code, lexer)
+        lexer = find_lexer(lexer)
+
+        return code if lexer == :no_lexer
+
+        formatter = Rouge::Formatters::HTML.new(:line_numbers => true)
+        formatter.format(lexer.new.lex(code))
+      end
+
+      def self.find_lexer(lexer)
+        lexers = {
+          ruby: Rouge::Lexers::Ruby,
+          erb:  Rouge::Lexers::ERB
+        }
+
+        lexers.fetch(lexer.to_sym, :no_lexer)
+      end
+    end
+  end
+end

lib/peek/rblineprof/syntax_highlighter.rb

@@ -0,0 +1,38 @@
+require 'peek/rblineprof/highlighters/pygments_highlighter'
+require 'peek/rblineprof/highlighters/rouge_highlighter'
+
+module Peek
+  module Rblineprof
+    class SyntaxHighlighter
+      class_attribute :highlighter
+
+      def self.highlight(code, lexer)
+        return code unless highlighter
+
+        highlighter.process(code, lexer)
+      end
+    end
+  end
+end
+
+module Peek
+  module Rblineprof
+    begin
+      require 'pygments.rb'
+
+      SyntaxHighlighter.highlighter = PygmentsHighlighter
+    rescue LoadError
+      # Doesn't have pygments.rb installed
+    end
+
+    if SyntaxHighlighter.highlighter.nil?
+      begin
+        require 'rouge'
+
+        SyntaxHighlighter.highlighter = RougeHighlighter
+      rescue LoadError
+        # Doesn't have rouge installed
+      end
+    end
+  end
+end