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-25 20:20:47 +09:00
|
|
|
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)
|
2024-07-28 20:53:15 +09:00
|
|
|
print_file_stats([{ filename: '合計', **total_stats }], max_widths, options) if sources.size > 1
|
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-25 08:33:11 +09:00
|
|
|
sources = ARGV.empty? ? [''] : ARGV
|
2024-07-21 09:50:07 +09:00
|
|
|
[options, sources]
|
2024-07-12 13:18:19 +09:00
|
|
|
end
|
2024-07-10 14:55:39 +09:00
|
|
|
|
2024-07-25 20:20:47 +09:00
|
|
|
def collect_file_stats(sources)
|
|
|
|
sources.map do |source|
|
2024-07-25 08:33:11 +09:00
|
|
|
input = source.empty? ? ARGF.read : File.read(source)
|
2024-07-28 20:53:15 +09:00
|
|
|
{ filename: source, lines: input.lines.count, words: input.split.size, bytes: input.bytesize }
|
2024-07-23 22:48:32 +09:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2024-07-25 20:20:47 +09:00
|
|
|
def calculate_total_stats(file_stats)
|
2024-07-28 20:53:15 +09:00
|
|
|
total_stats = { lines: 0, words: 0, bytes: 0 }
|
|
|
|
file_stats.each do |stats|
|
|
|
|
total_stats[:lines] += stats[:lines]
|
|
|
|
total_stats[:words] += stats[:words]
|
|
|
|
total_stats[:bytes] += stats[:bytes]
|
2024-07-19 14:57:01 +09:00
|
|
|
end
|
2024-07-28 20:53:15 +09:00
|
|
|
total_stats
|
2024-07-12 13:18:19 +09:00
|
|
|
end
|
2024-07-10 14:55:39 +09:00
|
|
|
|
2024-07-25 20:20:47 +09:00
|
|
|
def calculate_max_widths(file_stats, total_stats)
|
2024-07-28 20:53:15 +09:00
|
|
|
all_stats = file_stats + [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
|
2024-07-20 22:01:46 +09:00
|
|
|
end
|
2024-07-12 13:18:19 +09:00
|
|
|
end
|
2024-07-10 14:55:39 +09:00
|
|
|
|
2024-07-12 13:18:19 +09:00
|
|
|
def format_result(stats, max_widths, options)
|
2024-07-28 20:53:15 +09:00
|
|
|
%i[lines words bytes].filter_map do |key|
|
|
|
|
stats[key].to_s.rjust(max_widths[key]) if options[key]
|
|
|
|
end.join(' ')
|
2024-07-12 13:18:19 +09:00
|
|
|
end
|
2024-07-02 22:44:35 +09:00
|
|
|
|
2024-07-25 20:20:47 +09:00
|
|
|
def print_file_stats(file_stats, max_widths, options)
|
2024-07-28 20:53:15 +09:00
|
|
|
file_stats.each do |stats|
|
2024-07-25 08:33:11 +09:00
|
|
|
result = format_result(stats, max_widths, options)
|
2024-07-28 20:53:15 +09:00
|
|
|
puts "#{result} #{stats[:filename]}"
|
2024-07-25 08:33:11 +09:00
|
|
|
end
|
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
|