pry--pry/lib/pry/output.rb

class Pry::Output
  attr_reader :_pry_

  def initialize(_pry_)
    @_pry_ = _pry_
    @boxed_io = _pry_.config.output
  end

  def puts(*objs)
    return print "\n" if objs.empty?

    objs.each do |obj|
      if (ary = Array.try_convert(obj))
        puts(*ary)
      else
        print "#{obj.to_s.chomp}\n"
      end
    end
    nil
  end

  def print(*objs)
    objs.each do |obj|
      @boxed_io.print decolorize_maybe(obj.to_s)
    end
    nil
  end
  alias << print
  alias write print

  def tty?
    @boxed_io.respond_to?(:tty?) and @boxed_io.tty?
  end

  def method_missing(name, *args, &block)
    @boxed_io.__send__(name, *args, &block)
  end

  def respond_to_missing?(m, include_all=false)
    @boxed_io.respond_to?(m, include_all)
  end

  def decolorize_maybe(str)
    if _pry_.config.color
      str
    else
      Pry::Helpers::Text.strip_color str
    end
  end
end