class Bundler::UI::Shell

def add_color(string, *color)
  @shell.set_color(string, *color)
end

def ask(msg)
  @shell.ask(msg)
end

def confirm(msg = nil, newline = nil)
  return unless confirm?
  tell_me(msg || yield, :green, newline)
end

def confirm?
  level("confirm")
end

def debug(msg = nil, newline = nil)
  return unless debug?
  tell_me(msg || yield, nil, newline)
end

def debug?
  level("debug")
end

def error(msg = nil, newline = nil, color = :red)
  return unless error?
  tell_err(msg || yield, color, newline)
end

def error?
  level("error")
end

def info(msg = nil, newline = nil)
  return unless info?
  tell_me(msg || yield, nil, newline)
end

def info?
  level("info")
end

def initialize(options = {})
  Thor::Base.shell = options["no-color"] ? Thor::Shell::Basic : nil
  @shell = Thor::Base.shell.new
  @level = ENV["DEBUG"] ? "debug" : "info"
  @warning_history = []
end

def level(name = nil)
  return @level unless name
  unless index = LEVELS.index(name)
    raise "#{name.inspect} is not a valid level"
  end
  index <= LEVELS.index(@level)
end

def level=(level)
  raise ArgumentError unless LEVELS.include?(level.to_s)
  @level = level.to_s
end

def no?
  @shell.no?(msg)
end

def quiet?
  level("quiet")
end

def silence(&blk)
  with_level("silent", &blk)
end

def strip_leading_spaces(text)
  spaces = text[/\A\s+/, 0]
  spaces ? text.gsub(/#{spaces}/, "") : text
end

def tell_err(message, color = nil, newline = nil)
  return if @shell.send(:stderr).closed?
  newline ||= message.to_s !~ /( |\t)\Z/
  message = word_wrap(message) if newline.is_a?(Hash) && newline[:wrap]
  color = nil if color && !$stderr.tty?
  buffer = @shell.send(:prepare_message, message, *color)
  buffer << "\n" if newline && !message.to_s.end_with?("\n")
  @shell.send(:stderr).print(buffer)
  @shell.send(:stderr).flush
end

def tell_me(msg, color = nil, newline = nil)
  msg = word_wrap(msg) if newline.is_a?(Hash) && newline[:wrap]
  if newline.nil?
    @shell.say(msg, color)
  else
    @shell.say(msg, color, newline)
  end
end

def trace(e, newline = nil, force = false)
  return unless debug? || force
  msg = "#{e.class}: #{e.message}\n#{e.backtrace.join("\n  ")}"
  tell_err(msg, nil, newline)
end

def unprinted_warnings
  []
end

def warn(msg = nil, newline = nil, color = :yellow)
  return unless warn?
  return if @warning_history.include? msg
  @warning_history << msg
  tell_err(msg || yield, color, newline)
end

def warn?
  level("warn")
end

def with_level(level)
  original = @level
  @level = level
  yield
ensure
  @level = original
end

def word_wrap(text, line_width = @shell.terminal_width)
  strip_leading_spaces(text).split("\n").collect do |line|
    line.length > line_width ? line.gsub(/(.{1,#{line_width}})(\s+|$)/, "\\1\n").strip : line
  end * "\n"
end

def yes?(msg)
  @shell.yes?(msg)
end