class Thor::Shell::Basic

def ask(statement, color=nil)
  say("#{statement} ", color)
  $stdin.gets.strip
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)
  return true if @always_force
  options = block_given? ? "[Ynaqdh]" : "[Ynaqh]"
  while true
    answer = ask %[Overwrite #{destination}? (enter "h" for help) #{options}]
    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...'
        raise 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:
 overwrite
do not overwrite
 overwrite this and all others
, abort
, show the differences between the old and the new
, show this help
end

def initialize #:nodoc:
  @base, @padding = nil, 0
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 mute
  @mute = true
  yield
ensure
  @mute = false
end

def mute?
  @mute
end

def no?(statement, color=nil)
  !yes?(statement, color)
end

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

def print_table(table, options={})
  return if table.empty?
  formats, ident, colwidth = [], options[:ident].to_i, options[:colwidth]
  options[:truncate] = terminal_width if options[:truncate] == true
  formats << "%-#{colwidth + 2}s" if colwidth
  start = colwidth ? 1 : 0
  start.upto(table.first.length - 2) do |i|
    maxima ||= table.max{|a,b| a[i].size <=> b[i].size }[i].size
    formats << "%-#{maxima + 2}s"
  end
  formats[0] = formats[0].insert(0, " " * ident)
  formats << "%s"
  table.each do |row|
    sentence = ""
    row.each_with_index do |column, i|
      sentence << formats[i] % column.to_s
    end
    sentence = truncate(sentence, options[:truncate]) if options[:truncate]
    $stdout.puts sentence
  end
end

def print_wrapped(message, options={})
  ident = options[:ident] || 0
  width = terminal_width - ident
  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, " " * ident)
    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)$/))
  message = message.to_s
  message = set_color(message, color) if color
  spaces = "  " * padding
  if force_new_line
    $stdout.puts(spaces + message)
  else
    $stdout.print(spaces + message)
  end
  $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
  $stdout.puts "#{status}#{spaces}#{message}"
  $stdout.flush
end

def set_color(string, color, bold=false) #: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 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)
  if string.length <= width
    string
  else
    ( string[0, width-3] || "" ) + "..."
  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) =~ is?(:yes))
end