class CodeStatistics

def calculate_code
  code_loc = 0
  @statistics.each { |k, v| code_loc += v.code_lines unless TEST_TYPES.include? k }
  code_loc
end

def calculate_directory_statistics(directory, pattern = /^(?!\.).*?\.(rb|js|ts|css|scss|coffee|rake|erb)$/)
  stats = CodeStatisticsCalculator.new
  Dir.foreach(directory) do |file_name|
    path = "#{directory}/#{file_name}"
    if File.directory?(path) && !file_name.start_with?(".")
      stats.add(calculate_directory_statistics(path, pattern))
    elsif file_name&.match?(pattern)
      stats.add_by_file_path(path)
    end
  end
  stats
end

def calculate_statistics
  Hash[@pairs.map { |pair| [pair.first, calculate_directory_statistics(pair.last)] }]
end

def calculate_tests
  test_loc = 0
  @statistics.each { |k, v| test_loc += v.code_lines if TEST_TYPES.include? k }
  test_loc
end

def calculate_total
  @statistics.each_with_object(CodeStatisticsCalculator.new) do |pair, total|
    total.add(pair.last)
  end
end

def initialize(*pairs)
  @pairs      = pairs
  @statistics = calculate_statistics
  @total      = calculate_total if pairs.length > 1
end

def print_code_test_stats
  code  = calculate_code
  tests = calculate_tests
  puts "  Code LOC: #{code}     Test LOC: #{tests}     Code to Test Ratio: 1:#{sprintf("%.1f", tests.to_f / code)}"
  puts ""
end

def print_header
  print_splitter
  print "| Name                "
  HEADERS.each do |k, v|
    print " | #{v.rjust(width_for(k))}"
  end
  puts " | M/C | LOC/M |"
  print_splitter
end

def print_line(name, statistics)
  m_over_c   = (statistics.methods / statistics.classes) rescue 0
  loc_over_m = (statistics.code_lines / statistics.methods) - 2 rescue 0
  print "| #{name.ljust(20)} "
  HEADERS.each_key do |k|
    print "| #{statistics.send(k).to_s.rjust(width_for(k))} "
  end
  puts "| #{m_over_c.to_s.rjust(3)} | #{loc_over_m.to_s.rjust(5)} |"
end

def print_splitter
  print "+----------------------"
  HEADERS.each_key do |k|
    print "+#{'-' * (width_for(k) + 2)}"
  end
  puts "+-----+-------+"
end

def to_s
  print_header
  @pairs.each { |pair| print_line(pair.first, @statistics[pair.first]) }
  print_splitter
  if @total
    print_line("Total", @total)
    print_splitter
  end
  print_code_test_stats
end

def width_for(label)
  [@statistics.values.sum { |s| s.public_send(label) }.to_s.size, HEADERS[label].length].max
end