class Pry::ColorPrinter

def self.default(_output, value, pry_instance)
  pry_instance.pager.open do |pager|
    pager.print pry_instance.config.output_prefix
    pp(value, pager, pry_instance.output.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 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

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