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'
def main
options, input_sources = parse_options
all_stats = collect_all_stats(input_sources)
total_stats = calculate_total_stats(all_stats)
max_widths = calculate_max_widths(all_stats, total_stats)
print_all_stats(all_stats, max_widths, options)
print_all_stats([['合計', total_stats]], max_widths, options) if input_sources.size > 1
options, sources = parse_options
file_stats = collect_file_stats(sources)
total_stats = calculate_total_stats(file_stats)
max_widths = calculate_max_widths(file_stats, total_stats)
print_file_stats(file_stats, max_widths, options)
print_file_stats([['合計', total_stats]], max_widths, options) if sources.size > 1
end
def parse_options
@ -23,22 +23,22 @@ def parse_options
[options, sources]
end
def collect_all_stats(input_sources)
input_sources.map do |source|
def collect_file_stats(sources)
sources.map do |source|
input = source.empty? ? ARGF.read : File.read(source)
stats = { lines: input.lines.count, words: input.split.size, bytes: input.bytesize }
[source, stats]
end
end
def calculate_total_stats(all_stats)
all_stats.reduce({ lines: 0, words: 0, bytes: 0 }) do |total, (_, stats)|
def calculate_total_stats(file_stats)
file_stats.reduce({ lines: 0, words: 0, bytes: 0 }) do |total, (_, stats)|
total.merge(stats) { |_, a, b| a + b }
end
end
def calculate_max_widths(all_stats, total_stats)
(all_stats.map(&:last) + [total_stats]).each_with_object({ lines: 0, words: 0, bytes: 0 }) do |stats, max_widths|
def calculate_max_widths(file_stats, total_stats)
(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 }
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(' ')
end
def print_all_stats(all_stats, max_widths, options)
all_stats.each do |source, stats|
def print_file_stats(file_stats, max_widths, options)
file_stats.each do |source, stats|
result = format_result(stats, max_widths, options)
puts "#{result} #{source}"
end