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-25 20:20:47 +09:00
parent 056d23d6b2
commit bff969520f
Signed by: riq0h
GPG key ID: 010F09DEA298C717
1 changed files with 14 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
wc.rb
View file

@ -3,12 +3,12 @@
require 'optparse' require 'optparse'
def main def main
options, input_sources = parse_options options, sources = parse_options
all_stats = collect_all_stats(input_sources) file_stats = collect_file_stats(sources)
total_stats = calculate_total_stats(all_stats) total_stats = calculate_total_stats(file_stats)
max_widths = calculate_max_widths(all_stats, total_stats) max_widths = calculate_max_widths(file_stats, total_stats)
print_all_stats(all_stats, max_widths, options) print_file_stats(file_stats, max_widths, options)
print_all_stats([['合計', total_stats]], max_widths, options) if input_sources.size > 1 print_file_stats([['合計', total_stats]], max_widths, options) if sources.size > 1
end end
def parse_options def parse_options
@ -23,22 +23,22 @@ def parse_options
[options, sources] [options, sources]
end end
def collect_all_stats(input_sources) def collect_file_stats(sources)
input_sources.map do |source| sources.map do |source|
input = source.empty? ? ARGF.read : File.read(source) input = source.empty? ? ARGF.read : File.read(source)
stats = { lines: input.lines.count, words: input.split.size, bytes: input.bytesize } stats = { lines: input.lines.count, words: input.split.size, bytes: input.bytesize }
[source, stats] [source, stats]
end end
end end
def calculate_total_stats(all_stats) def calculate_total_stats(file_stats)
all_stats.reduce({ lines: 0, words: 0, bytes: 0 }) do |total, (_, stats)| file_stats.reduce({ lines: 0, words: 0, bytes: 0 }) do |total, (_, stats)|
total.merge(stats) { |_, a, b| a + b } total.merge(stats) { |_, a, b| a + b }
end end
end end
def calculate_max_widths(all_stats, total_stats) def calculate_max_widths(file_stats, total_stats)
(all_stats.map(&:last) + [total_stats]).each_with_object({ lines: 0, words: 0, bytes: 0 }) do |stats, max_widths| (file_stats.map(&:last) + [total_stats]).each_with_object({ lines: 0, words: 0, bytes: 0 }) do |stats, max_widths|
max_widths.merge!(stats) { |_, max, stat| [max, stat.to_s.length].max } max_widths.merge!(stats) { |_, max, stat| [max, stat.to_s.length].max }
end end
end end
@ -47,8 +47,8 @@ def format_result(stats, max_widths, options)
%i[lines words bytes].map { |key| stats[key].to_s.rjust(max_widths[key]) if options[key] }.compact.join(' ') %i[lines words bytes].map { |key| stats[key].to_s.rjust(max_widths[key]) if options[key] }.compact.join(' ')
end end
def print_all_stats(all_stats, max_widths, options) def print_file_stats(file_stats, max_widths, options)
all_stats.each do |source, stats| file_stats.each do |source, stats|
result = format_result(stats, max_widths, options) result = format_result(stats, max_widths, options)
puts "#{result} #{source}" puts "#{result} #{source}"
end end