114 lines
3 KiB
Ruby
114 lines
3 KiB
Ruby
|
require "rails/code_statistics_calculator"
|
||
|
require "active_support/core_ext/enumerable"
|
||
|
|
||
|
class CodeStatistics #:nodoc:
|
||
|
TEST_TYPES = ["Controller tests",
|
||
|
"Helper tests",
|
||
|
"Model tests",
|
||
|
"Mailer tests",
|
||
|
"Job tests",
|
||
|
"Integration tests",
|
||
|
"System tests"]
|
||
|
|
||
|
HEADERS = { lines: " Lines", code_lines: " LOC", classes: "Classes", methods: "Methods" }
|
||
|
|
||
|
def initialize(*pairs)
|
||
|
@pairs = pairs
|
||
|
@statistics = calculate_statistics
|
||
|
@total = calculate_total if pairs.length > 1
|
||
|
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
|
||
|
|
||
|
private
|
||
|
def calculate_statistics
|
||
|
Hash[@pairs.map { |pair| [pair.first, calculate_directory_statistics(pair.last)] }]
|
||
|
end
|
||
|
|
||
|
def calculate_directory_statistics(directory, pattern = /^(?!\.).*?\.(rb|js|coffee|rake)$/)
|
||
|
stats = CodeStatisticsCalculator.new
|
||
|
|
||
|
Dir.foreach(directory) do |file_name|
|
||
|
path = "#{directory}/#{file_name}"
|
||
|
|
||
|
if File.directory?(path) && (/^\./ !~ file_name)
|
||
|
stats.add(calculate_directory_statistics(path, pattern))
|
||
|
elsif file_name =~ pattern
|
||
|
stats.add_by_file_path(path)
|
||
|
end
|
||
|
end
|
||
|
|
||
|
stats
|
||
|
end
|
||
|
|
||
|
def calculate_total
|
||
|
@statistics.each_with_object(CodeStatisticsCalculator.new) do |pair, total|
|
||
|
total.add(pair.last)
|
||
|
end
|
||
|
end
|
||
|
|
||
|
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_tests
|
||
|
test_loc = 0
|
||
|
@statistics.each { |k, v| test_loc += v.code_lines if TEST_TYPES.include? k }
|
||
|
test_loc
|
||
|
end
|
||
|
|
||
|
def width_for(label)
|
||
|
[@statistics.values.sum { |s| s.send(label) }.to_s.size, HEADERS[label].length].max
|
||
|
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_splitter
|
||
|
print "+----------------------"
|
||
|
HEADERS.each_key do |k|
|
||
|
print "+#{'-' * (width_for(k) + 2)}"
|
||
|
end
|
||
|
puts "+-----+-------+"
|
||
|
end
|
||
|
|
||
|
def print_line(name, statistics)
|
||
|
m_over_c = (statistics.methods / statistics.classes) rescue m_over_c = 0
|
||
|
loc_over_m = (statistics.code_lines / statistics.methods) - 2 rescue loc_over_m = 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_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
|
||
|
end
|