class Tablr

# File lib/tablr/tablr.rb, line 4
def initialize(options = {})
  @columns            ||= TablrSpace::OrderedHash.auto
  @row_data           ||= TablrSpace::OrderedHash.new
  options[:separate]  ||= "false"
  options[:headers]   ||= "true"

  @options = options
end

# File lib/tablr/tablr.rb, line 21
def add_row(column, value)
  add_column(column) unless column_exists?(column) or !column.is_a? String

  Array(value).each do |v|
    @columns[column][:rows].push v
    adjust(column, v.length)
  end
end

# File lib/tablr/tablr.rb, line 31
def adjust(column, length)
  return nil unless column_exists?(column)
  @columns[column][:length] = length if length > @columns[column][:length]
end

# File lib/tablr/tablr.rb, line 36
  def print
    str = <<EOL
#{headers}
#{lines}
#{print_rows}#{lines}
EOL
    puts str
  end

# File lib/tablr/tablr.rb, line 49
def column_exists?(column)
  @columns.has_key? column
end

# File lib/tablr/tablr.rb, line 85
def fill_rows
  max_count = 0
  @columns.each_with_index do |v, k|
    max_count = v[1][:rows].length if v[1][:rows].length > max_count
  end
  @columns.each_with_index do |v, k|
    (0..(max_count-v[1][:rows].length)).each do |i|
      add_row v[0], ""
    end
  end
end

# File lib/tablr/tablr.rb, line 97
def get_rows
  count, array = @columns.length-1, Array.new
  fill_rows
  @columns.each_with_index do |value, key|
    value[1][:rows].each_with_index do |v, k|
      array[k] = [] if array[k].nil?
      array[k].insert -1, v
    end
  end
  array
end

# File lib/tablr/tablr.rb, line 60
def headers
  return '' if @options[:headers] == "false"
  str = lines << "\n"
  @columns.each { |k,v| str << "| #{k}" << " " * str_length(k, v[:length], -1) }
  str << "|"
end

# File lib/tablr/tablr.rb, line 53
def lines
  str = '+'
  # Get number of dashes within columns
  @columns.each { |k,v| str << '-' * v[:length] << '--+' }
  str
end

# File lib/tablr/tablr.rb, line 67
def print_rows
  str, array = '', Array.new
  # Store the max lengths of each header quickly
  @columns.each do |k,v|
    array.push v[:length]
  end

  get_rows.each_with_index do |value, i|
    str << "|"
    value.each_with_index do |v, index|
      str << " #{v}" << " " * str_length(v, array[index].to_i, 0) << " |"
    end
    str << "\n"
    str << lines if @options[:separate] == "true"
  end
  str
end

# File lib/tablr/tablr.rb, line 109
def str_length(key, length, padding=1)
  (length-(key.gsub(/\e\[(\d+)m/, '').length)-padding).abs
end