class Thor::Shell::Basic

def as_unicode
  yield
end

def as_unicode
  old, $KCODE = $KCODE, "U"
  yield
ensure
  $KCODE = old
end

def ask(statement, *args)
  options = args.last.is_a?(Hash) ? args.pop : {}
  color = args.first
  if options[:limited_to]
    ask_filtered(statement, color, options)
  else
    ask_simply(statement, color, options)
  end
end

def ask_filtered(statement, color, options)
  answer_set = options[:limited_to]
  correct_answer = nil
  until correct_answer
    answers = answer_set.join(", ")
    answer = ask_simply("#{statement} [#{answers}]", color, options)
    correct_answer = answer_set.include?(answer) ? answer : nil
    say("Your response must be one of: [#{answers}]. Please try again.") unless correct_answer
  end
  correct_answer
end

def ask_simply(statement, color, options)
  default = options[:default]
  message = [statement, ("(#{default})" if default), nil].uniq.join(" ")
  message = prepare_message(message, color)
  result = Thor::LineEditor.readline(message, options)
  return unless result
  result.strip!
  if default && result == ""
    default
  else
    result
  end
end

def can_display_colors?
  false
end

def dynamic_width
  @dynamic_width ||= (dynamic_width_stty.nonzero? || dynamic_width_tput)
end

def dynamic_width_stty
  %x(stty size 2>/dev/null).split[1].to_i
end

def dynamic_width_tput
  %x(tput cols 2>/dev/null).to_i
end

def error(statement)
  stderr.puts statement
end

def file_collision(destination) # rubocop:disable MethodLength
  return true if @always_force
  options = block_given? ? "[Ynaqdh]" : "[Ynaqh]"
  loop do
    answer = ask(
      %[Overwrite #{destination}? (enter "h" for help) #{options}],
      :add_to_history => false
    )
    case answer
    when is?(:yes), is?(:force), ""
      return true
    when is?(:no), is?(:skip)
      return false
    when is?(:always)
      return @always_force = true
    when is?(:quit)
      say "Aborting..."
      fail SystemExit
    when is?(:diff)
      show_diff(destination, yield) if block_given?
      say "Retrying..."
    else
      say file_collision_help
    end
  end
end

def file_collision_help #:nodoc:
  <<-HELP
  Y - yes, overwrite
  n - no, do not overwrite
  a - all, overwrite this and all others
  q - quit, abort
  d - diff, show the differences between the old and the new
  h - help, show this help
  HELP
end

def initialize #:nodoc:
  @base, @mute, @padding, @always_force = nil, false, 0, false
end

def is?(value) #:nodoc:
  value = value.to_s
  if value.size == 1
    /\A#{value}\z/i
  else
    /\A(#{value}|#{value[0, 1]})\z/i
  end
end

def lookup_color(color)
  return color unless color.is_a?(Symbol)
  self.class.const_get(color.to_s.upcase)
end

def mute
  @mute = true
  yield
ensure
  @mute = false
end

def mute? # rubocop:disable TrivialAccessors
  @mute
end

def no?(statement, color = nil)
  !!(ask(statement, color, :add_to_history => false) =~ is?(:no))
end

def padding=(value)
  @padding = [0, value].max
end

def prepare_message(message, *color)
  spaces = "  " * padding
  spaces + set_color(message.to_s, *color)
end

def print_in_columns(array)
  return if array.empty?
  colwidth = (array.map { |el| el.to_s.size }.max || 0) + 2
  array.each_with_index do |value, index|
    # Don't output trailing spaces when printing the last column
    if ((((index + 1) % (terminal_width / colwidth))).zero? && !index.zero?) || index + 1 == array.length
      stdout.puts value
    else
      stdout.printf("%-#{colwidth}s", value)
    end
  end
end

def print_table(array, options = {}) # rubocop:disable MethodLength
  return if array.empty?
  formats, indent, colwidth = [], options[:indent].to_i, options[:colwidth]
  options[:truncate] = terminal_width if options[:truncate] == true
  formats << "%-#{colwidth + 2}s" if colwidth
  start = colwidth ? 1 : 0
  colcount = array.max { |a, b| a.size <=> b.size }.size
  maximas = []
  start.upto(colcount - 1) do |index|
    maxima = array.map { |row| row[index] ? row[index].to_s.size : 0 }.max
    maximas << maxima
    if index == colcount - 1
      # Don't output 2 trailing spaces when printing the last column
      formats << "%-s"
    else
      formats << "%-#{maxima + 2}s"
    end
  end
  formats[0] = formats[0].insert(0, " " * indent)
  formats << "%s"
  array.each do |row|
    sentence = ""
    row.each_with_index do |column, index|
      maxima = maximas[index]
      if column.is_a?(Numeric)
        if index == row.size - 1
          # Don't output 2 trailing spaces when printing the last column
          f = "%#{maxima}s"
        else
          f = "%#{maxima}s  "
        end
      else
        f = formats[index]
      end
      sentence << f % column.to_s
    end
    sentence = truncate(sentence, options[:truncate]) if options[:truncate]
    stdout.puts sentence
  end
end

def print_wrapped(message, options = {})
  indent = options[:indent] || 0
  width = terminal_width - indent
  paras = message.split("\n\n")
  paras.map! do |unwrapped|
    unwrapped.strip.gsub(/\n/, " ").squeeze(" ").gsub(/.{1,#{width}}(?:\s|\Z)/) { ($& + 5.chr).gsub(/\n\005/, "\n").gsub(/\005/, "\n") }
  end
  paras.each do |para|
    para.split("\n").each do |line|
      stdout.puts line.insert(0, " " * indent)
    end
    stdout.puts unless para == paras.last
  end
end

def quiet? #:nodoc:
  mute? || (base && base.options[:quiet])
end

def say(message = "", color = nil, force_new_line = (message.to_s !~ /( |\t)\Z/))
  buffer = prepare_message(message, *color)
  buffer << "\n" if force_new_line && !message.to_s.end_with?("\n")
  stdout.print(buffer)
  stdout.flush
end

def say_status(status, message, log_status = true)
  return if quiet? || log_status == false
  spaces = "  " * (padding + 1)
  color  = log_status.is_a?(Symbol) ? log_status : :green
  status = status.to_s.rjust(12)
  status = set_color status, color, true if color
  buffer = "#{status}#{spaces}#{message}"
  buffer << "\n" unless buffer.end_with?("\n")
  stdout.print(buffer)
  stdout.flush
end

def set_color(string, *args) #:nodoc:
  string
end

def show_diff(destination, content) #:nodoc:
  diff_cmd = ENV["THOR_DIFF"] || ENV["RAILS_DIFF"] || "diff -u"
  Tempfile.open(File.basename(destination), File.dirname(destination)) do |temp|
    temp.write content
    temp.rewind
    system %(#{diff_cmd} "#{destination}" "#{temp.path}")
  end
end

def stderr
  $stderr
end

def stdout
  $stdout
end

def terminal_width
  if ENV["THOR_COLUMNS"]
    result = ENV["THOR_COLUMNS"].to_i
  else
    result = unix? ? dynamic_width : 80
  end
  result < 10 ? 80 : result
rescue
  80
end

def truncate(string, width)
  as_unicode do
    chars = string.chars.to_a
    if chars.length <= width
      chars.join
    else
      ( chars[0, width - 3].join) + "..."
    end
  end
end

def unix?
  RUBY_PLATFORM =~ /(aix|darwin|linux|(net|free|open)bsd|cygwin|solaris|irix|hpux)/i
end

def yes?(statement, color = nil)
  !!(ask(statement, color, :add_to_history => false) =~ is?(:yes))
end