lib/rouge/lexers/http.rb in rouge-0.5.3 vs lib/rouge/lexers/http.rb in rouge-0.5.4

@@ -6,10 +6,18 @@
 
       def self.methods
         @methods ||= %w(GET POST PUT DELETE HEAD OPTIONS TRACE)
       end
 
+      def content_lexer
+        return Lexers::PlainText unless @content_type
+
+        @content_lexer ||= Lexer.guess_by_mimetype(@content_type)
+      rescue Lexer::AmbiguousGuess
+        @content_lexer = Lexers::PlainText
+      end
+
       start { @content_type = 'text/plain' }
 
       state :root do
         # request
         rule %r(
@@ -57,11 +65,11 @@
 
         rule /\r?\n/, Text, :content
       end
 
       state :content do
-        rule /.+/ do |m|
-          delegate Lexer.guess_by_mimetype(@content_type)
+        rule /.+/m do |m|
+          delegate(content_lexer)
         end
       end
     end
   end
 end