lib/rouge/formatters/terminal256.rb in rouge-0.2.1 vs lib/rouge/formatters/terminal256.rb in rouge-0.2.2

@@ -144,15 +144,24 @@
       end
 
     # private
       def escape_sequence(token)
         @escape_sequences ||= {}
-        @escape_sequences[token.name] ||= begin
-          esc = EscapeSequence.new(theme.get_style(token))
-          # don't highlight text backgrounds
-          esc.style.delete(:bg) if token.name == 'Text'
-          esc
-        end
+        @escape_sequences[token.name] ||=
+          EscapeSequence.new(get_style(token))
+      end
+
+      def get_style(token)
+        return text_style if token.name == 'Text'
+
+        theme.get_own_style(token) || text_style
+      end
+
+      def text_style
+        style = theme.get_style(Token['Text'])
+        # don't highlight text backgrounds
+        style.delete :bg
+        style
       end
     end
   end
 end