class Formatador

# File lib/formatador.rb, line 59
def initialize
  @indent = 1
end

# File lib/formatador.rb, line 63
def display(string = '')
  print(parse("[indent]#{string}"))
  STDOUT.flush
  nil
end

# File lib/formatador/table.rb, line 10
def display_compact_table(hashes, keys = nil, &block)
  headers = keys || []
  widths = {}
  if hashes.empty? && keys
    for key in keys
      widths[key] = key.to_s.length
    end
  else
    for hash in hashes
      next unless hash.respond_to?(:keys)

      for key in hash.keys
        unless keys
          headers << key
        end
        widths[key] = [ length(key), widths[key] || 0, hash[key] && length(hash[key]) || 0].max
      end
      headers = headers.uniq
    end
  end

  if block_given?
    headers = headers.sort(&block)
  elsif !keys
    headers = headers.sort {|x,y| x.to_s <=> y.to_s}
  end

  split = "+"
  if headers.empty?
    split << '--+'
  else
    for header in headers
      widths[header] ||= length(header)
      split << ('-' * (widths[header] + 2)) << '+'
    end
  end

  display_line(split)
  columns = []
  for header in headers
    columns << "[bold]#{header}[/]#{' ' * (widths[header] - header.to_s.length)}"
  end
  display_line("| #{columns.join(' | ')} |")
  display_line(split)

  for hash in hashes
    if hash.respond_to? :keys
      columns = []
      for header in headers
        datum = hash[header] || ''
        columns << "#{datum}#{' ' * (widths[header] - length(datum))}"
      end
      display_line("| #{columns.join(' | ')} |")
    else
      if hash == :split
        display_line(split)
      end 
    end
    nil
  end
  display_line(split)
end

# File lib/formatador.rb, line 69
def display_line(string = '')
  display(string)
  new_line
  nil
end

# File lib/formatador.rb, line 75
def display_lines(lines = [])
  for line in [*lines]
    display_line(line)
  end
  nil
end

# File lib/formatador/table.rb, line 2
def display_table(hashes, keys = nil, &block)
  new_hashes = hashes.inject([]) do |accum,item|
    accum << :split unless accum.empty?
    accum << item
  end
  display_compact_table(new_hashes, keys, &block)
end

# File lib/formatador.rb, line 90
def indent(&block)
  @indent += 1
  yield
  @indent -= 1
end

# File lib/formatador.rb, line 96
def indentation
  '  ' * @indent
end

# File lib/formatador.rb, line 112
def new_line
  print("\n")
  nil
end

# File lib/formatador.rb, line 82
def parse(string)
  if STDOUT.tty?
    string.gsub(PARSE_REGEX) { "\e[#{STYLES[$1.to_sym]}m" }.gsub(INDENT_REGEX) { indentation }
  else
    strip(string)
  end
end

# File lib/formatador.rb, line 100
def redisplay(string = '', width = 120)
  print("\r#{' ' * width}\r")
  display("#{string}")
  nil
end

# File lib/formatador.rb, line 106
def redisplay_line(string = '', width = 120)
  redisplay(string, width)
  new_line
  nil
end

# File lib/formatador/progressbar.rb, line 3
def redisplay_progressbar(current, total, options = {})
  options = { :color => 'white', :width => 50, :new_line => true }.merge!(options)
  data = progressbar(current, total, options)
  if current < total
    redisplay(data)
  else
    redisplay("#{data}")
    if options[:new_line]
      new_line
    end
    @progressbar_started_at = nil
  end
end

# File lib/formatador.rb, line 117
def strip(string)
  string.gsub(PARSE_REGEX, '').gsub(INDENT_REGEX) { indentation }
end

# File lib/formatador/table.rb, line 75
def length(value)
  value.to_s.gsub(PARSE_REGEX, '').length
end

# File lib/formatador/progressbar.rb, line 19
def progressbar(current, total, options)
  color = options[:color]
  started_at = options[:started_at]
  width = options[:width]

  output = []

  if options[:label]
    output << options[:label]
  end

  # width
  # we are going to write a string that looks like "   current/total"
  # It would be nice if it were left padded with spaces in such a way that
  # it puts the progress bar in a constant place on the page. This witdh
  # calculation allows for the "current" string to be up to two characters
  # longer than the "total" string without problems. eg- current =
  # 9.99, total = 10
  padding = total.to_s.size * 2 + 3

  output << "[#{color}]%#{padding}s[/]" % "#{current}/#{total}"

  percent = current.to_f / total.to_f
  percent = 0 if percent < 0
  percent = 1 if percent > 1

  done = '*' * (percent * width).ceil
  remaining = ' ' * (width - done.length)
  output << "[_white_]|[/][#{color}][_#{color}_]#{done}[/]#{remaining}[_white_]|[/]"

  if started_at
    elapsed = Time.now - started_at
    minutes = (elapsed / 60).truncate.to_s
    seconds = (elapsed % 60).truncate.to_s
    output << "#{minutes}:#{'0' if seconds.size < 2}#{seconds}"
  end

  output << ''
  output.join('  ')
end