1
0
Fork 0
mirror of https://github.com/pry/pry.git synced 2022-11-09 12:35:05 -05:00
pry--pry/lib/pry/color_printer.rb

56 lines
1.4 KiB
Ruby
Raw Normal View History

# PP subclass for streaming inspect output in color.
class Pry
class ColorPrinter < ::PP
OBJ_COLOR = begin
code = CodeRay::Encoders::Terminal::TOKEN_COLORS[:keyword]
if code.start_with? "\e"
code
else
"\e[0m\e[0;#{code}m"
end
end
CodeRay::Encoders::Terminal::TOKEN_COLORS[:comment][:self] = "\e[1;34m"
def self.pp(obj, out = $>, width = 79)
q = ColorPrinter.new(out, width)
q.guard_inspect_key { q.pp obj }
q.flush
out << "\n"
end
def text(str, width = str.length)
# Don't recolorize output with color [Issue #751]
2014-05-01 04:14:33 -04:00
if str.include?("\e[")
super "#{str}\e[0m", width
elsif str.start_with?('#<') || str == '=' || str == '>'
2014-05-01 04:14:33 -04:00
super highlight_object_literal(str), width
else
2014-05-01 04:14:33 -04:00
super CodeRay.scan(str, :ruby).term, width
end
end
2013-11-02 18:09:39 -04:00
def pp(obj)
super
rescue => e
raise if e.is_a? Pry::Pager::StopPaging
2013-11-02 18:09:39 -04:00
# Read the class name off of the singleton class to provide a default
# inspect.
singleton = class << obj; self; end
ancestors = Pry::Method.safe_send(singleton, :ancestors)
klass = ancestors.reject { |k| k == singleton }.first
2013-11-02 18:09:39 -04:00
obj_id = obj.__id__.to_s(16) rescue 0
str = "#<#{klass}:0x#{obj_id}>"
2014-05-01 04:14:33 -04:00
text highlight_object_literal(str)
2013-11-02 18:09:39 -04:00
end
private
def highlight_object_literal(object_literal)
"#{OBJ_COLOR}#{object_literal}\e[0m"
end
end
end