riq0h/fjord
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

さらに修正

Browse source
This commit is contained in:
Rikuoh Tsujitani 2024-07-29 22:05:22 +09:00
parent 9ca583442a
commit 16e5867588
Signed by: riq0h
GPG key ID: 010F09DEA298C717
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
wc.rb
View file

@ -6,7 +6,7 @@ def main
options, filenames = parse_options
file_stats = collect_file_stats(filenames)
total_stats = calculate_total_stats(file_stats)
max_widths = calculate_max_widths(file_stats + [total_stats])
max_widths = calculate_max_widths(total_stats)
print_file_stats(file_stats, max_widths, options)
print_file_stats([total_stats], max_widths, options) if filenames.size > 1
end
@ -26,12 +26,12 @@ end
def collect_file_stats(filenames)
filenames.map do |filename|
input = filename.empty? ? ARGF.read : File.read(filename)
{ filename: filename, lines: input.lines.count, words: input.split.size, bytes: input.bytesize }
{ filenames: filename, lines: input.lines.count, words: input.split.size, bytes: input.bytesize }
end
end
def calculate_total_stats(file_stats)
total_stats = { filename: '合計', lines: 0, words: 0, bytes: 0 }
total_stats = { filenames: '合計', lines: 0, words: 0, bytes: 0 }
file_stats.each do |stats|
total_stats[:lines] += stats[:lines]
total_stats[:words] += stats[:words]
@ -40,9 +40,9 @@ def calculate_total_stats(file_stats)
total_stats
end
def calculate_max_widths(all_stats)
def calculate_max_widths(total_stats)
%i[lines words bytes].each_with_object({}) do |key, max_widths|
max_widths[key] = all_stats.map { |stats| stats[key].to_s.length }.max
max_widths[key] = total_stats[key].to_s.length
end
end
@ -55,7 +55,7 @@ end
def print_file_stats(file_stats, max_widths, options)
file_stats.each do |stats|
result = format_result(stats, max_widths, options)
puts "#{result} #{stats[:filename]}"
puts [result, stats[:filenames]].join(' ')
end
end