67 lines
1.9 KiB
Ruby
67 lines
1.9 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'pp'
|
|
require 'English'
|
|
|
|
class Pry
|
|
# PP subclass for streaming inspect output in color.
|
|
class ColorPrinter < ::PP
|
|
Pry::SyntaxHighlighter.overwrite_coderay_comment_token!
|
|
|
|
def self.default(_output, value, pry_instance)
|
|
pry_instance.pager.open do |pager|
|
|
pager.print pry_instance.config.output_prefix
|
|
pp(value, pager, Pry::Terminal.width! - 1)
|
|
end
|
|
end
|
|
|
|
def self.pp(obj, output = $DEFAULT_OUTPUT, max_width = 79)
|
|
queue = ColorPrinter.new(output, max_width, "\n")
|
|
queue.guard_inspect_key { queue.pp(obj) }
|
|
queue.flush
|
|
output << "\n"
|
|
end
|
|
|
|
def pp(object)
|
|
return super unless object.is_a?(String)
|
|
|
|
# Avoid calling Ruby 2.4+ String#pretty_print that prints multiline
|
|
# Strings prettier
|
|
text(object.inspect)
|
|
rescue StandardError => exception
|
|
raise if exception.is_a?(Pry::Pager::StopPaging)
|
|
|
|
text(highlight_object_literal(inspect_object(object)))
|
|
end
|
|
|
|
def text(str, max_width = str.length)
|
|
if str.include?("\e[")
|
|
super("#{str}\e[0m", max_width)
|
|
elsif str.start_with?('#<') || %w[= >].include?(str)
|
|
super(highlight_object_literal(str), max_width)
|
|
else
|
|
super(SyntaxHighlighter.highlight(str), max_width)
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def highlight_object_literal(object_literal)
|
|
code = Pry::SyntaxHighlighter.keyword_token_color
|
|
obj_color = code.start_with?("\e") ? code : "\e[0m\e[0;#{code}m"
|
|
"#{obj_color}#{object_literal}\e[0m"
|
|
end
|
|
|
|
def inspect_object(object)
|
|
object.inspect
|
|
rescue StandardError
|
|
# Read the class name off of the singleton class to provide a default
|
|
# inspect.
|
|
singleton = class << object; self; end
|
|
ancestors = Pry::Method.safe_send(singleton, :ancestors)
|
|
klass = ancestors.find { |k| k != singleton }
|
|
"#<#{klass}:0x#{object.__id__.to_s(16)}>"
|
|
end
|
|
end
|
|
end
|