fjord/wc.rb

61 lines
1.6 KiB
Ruby
Raw Normal View History

2024-07-02 22:44:35 +09:00
# frozen_string_literal: true
2024-07-01 09:32:50 +09:00
2024-07-02 22:44:35 +09:00
require 'optparse'
2024-07-01 09:32:50 +09:00
2024-07-25 08:33:11 +09:00
def main
2024-07-29 18:00:15 +09:00
options, filenames = parse_options
file_stats = collect_file_stats(filenames)
total_stat = calculate_total_stat(file_stats)
max_widths = calculate_max_widths(total_stat)
2024-07-30 14:30:09 +09:00
file_stats << total_stat if filenames.size > 1
2024-07-31 23:52:23 +09:00
file_stats.each { |file_stat| puts format_row(file_stat, max_widths, options) }
2024-07-25 08:33:11 +09:00
end
2024-07-12 13:18:19 +09:00
def parse_options
options = {}
OptionParser.new do |opts|
opts.on('-l') { options[:lines] = true }
opts.on('-w') { options[:words] = true }
opts.on('-c') { options[:bytes] = true }
end.parse!
options = { bytes: true, lines: true, words: true } if options.empty?
2024-07-29 18:00:15 +09:00
filenames = ARGV.empty? ? [''] : ARGV
[options, filenames]
2024-07-12 13:18:19 +09:00
end
2024-07-10 14:55:39 +09:00
2024-07-29 18:00:15 +09:00
def collect_file_stats(filenames)
filenames.map do |filename|
2024-07-30 14:30:09 +09:00
text = filename.empty? ? ARGF.read : File.read(filename)
{
filename: filename,
lines: text.lines.count,
words: text.split.size,
bytes: text.bytesize
}
2024-07-23 22:48:32 +09:00
end
end
def calculate_total_stat(file_stats)
total_stat = { filename: '合計', lines: 0, words: 0, bytes: 0 }
file_stats.each do |stats|
total_stat[:lines] += stats[:lines]
total_stat[:words] += stats[:words]
total_stat[:bytes] += stats[:bytes]
2024-07-19 14:57:01 +09:00
end
total_stat
2024-07-12 13:18:19 +09:00
end
2024-07-10 14:55:39 +09:00
def calculate_max_widths(total_stat)
%i[lines words bytes].to_h { |key| [key, total_stat[key].to_s.length] }
2024-07-12 13:18:19 +09:00
end
2024-07-10 14:55:39 +09:00
2024-07-31 23:52:23 +09:00
def format_row(stats, max_widths, options)
2024-08-01 09:47:39 +09:00
columns = %i[lines words bytes].filter_map do |key|
stats[key].to_s.rjust(max_widths[key]) if options[key]
2024-07-30 14:30:09 +09:00
end
2024-08-01 09:47:39 +09:00
[*columns, stats[:filename]].join(' ')
2024-07-10 14:55:39 +09:00
end
2024-07-20 22:45:14 +09:00
2024-07-25 08:33:11 +09:00
main