Revert "Get rid of Code#highlighted"

This reverts commit 5599b0a976.

And fixes jruby-head.

https://travis-ci.org/pry/pry/jobs/30420112

lib/pry/code.rb

@@ -253,7 +253,12 @@ class Pry
     # @return [String] a formatted representation (based on the configuration of
     #   the object).
     def to_s
-      print_to_output("")
+      print_to_output("", false)
+    end
+
+    # @return [String] a (possibly highlighted) copy of the source code.
+    def highlighted
+      print_to_output("", true)
     end
 
     # Writes a formatted representation (based on the configuration of the
@@ -261,7 +266,7 @@ class Pry
     def print_to_output(output, color=false)
       @lines.each do |loc|
         loc = loc.dup
-        loc.colorize(@code_type)
+        loc.colorize(@code_type)              if color
         loc.add_line_number(max_lineno_width, color) if @with_line_numbers
         loc.add_marker(@marker_lineno)        if @with_marker
         loc.indent(@indentation_num)          if @with_indentation

lib/pry/commands/cat/abstract_formatter.rb

@@ -8,7 +8,7 @@ class Pry
       def decorate(content)
         content.code_type = code_type
         content.between(*between_lines).
-          with_line_numbers(use_line_numbers?)
+          with_line_numbers(use_line_numbers?).highlighted
       end
 
       def code_type

lib/pry/commands/show_source.rb

@@ -40,7 +40,7 @@ class Pry
     # The source for code_object prepared for display.
     def content_for(code_object)
       Code.new(code_object.source, start_line_for(code_object)).
-        with_line_numbers(use_line_numbers?)
+        with_line_numbers(use_line_numbers?).highlighted
     end
   end
 

lib/pry/commands/watch_expression.rb

@@ -88,7 +88,7 @@ class Pry
 
     def add_expression(arguments)
       expressions << Expression.new(_pry_, target, arg_string)
-      output.puts "Watching #{Code.new(arg_string)}"
+      output.puts "Watching #{Code.new(arg_string).highlighted}"
     end
 
     def add_hook

lib/pry/commands/watch_expression/expression.rb

@@ -6,7 +6,7 @@ class Pry
       def initialize(_pry_, target, source)
         @_pry_ = _pry_
         @target = target
-        @source = source.strip
+        @source = Code.new(source).strip
       end
 
       def eval!
@@ -15,7 +15,7 @@ class Pry
       end
 
       def to_s
-        "#{Code.new(source).strip} => #{value}"
+        "#{Code.new(source).highlighted.strip} => #{value}"
       end
 
       # Has the value of the expression changed?

lib/pry/commands/whereami.rb

@@ -89,7 +89,7 @@ class Pry
       set_file_and_dir_locals(@file)
 
       out = "\n#{text.bold('From:')} #{location}:\n\n" <<
-        code.with_line_numbers(use_line_numbers?).with_marker(marker) << "\n"
+        code.with_line_numbers(use_line_numbers?).with_marker(marker).highlighted << "\n"
 
       _pry_.pager.page out
     end