メソッドの再配置

This commit is contained in:
Rikuoh Tsujitani 2024-07-20 22:01:46 +09:00
parent 3e8b029d8e
commit 3df439edaf
Signed by: riq0h
GPG key ID: 010F09DEA298C717

52
wc.rb
View file

@ -2,6 +2,9 @@
require 'optparse' require 'optparse'
options, input_sources = parse_options
input_stream_sources(input_sources, options)
def parse_options def parse_options
options = {} options = {}
OptionParser.new do |opts| OptionParser.new do |opts|
@ -13,6 +16,29 @@ def parse_options
[options, ARGV.empty? ? ['-'] : ARGV] [options, ARGV.empty? ? ['-'] : ARGV]
end end
def input_stream_sources(input_sources, options)
total_stats = [0, 0, 0]
max_widths = [0, 0, 0]
input_sources.each do |source|
stats = input_stream(source)
total_stats = update_totals(total_stats, stats)
max_widths = update_max_widths(max_widths, stats)
result = format_result(stats, max_widths, options)
print_result(result, source)
end
print_total(total_stats, max_widths, options) if input_sources.size > 1
end
def input_stream(source)
if source == '-'
count_file_stats(ARGF)
else
count_file_stats(File.read(source))
end
end
def count_file_stats(input) def count_file_stats(input)
lines = words = bytes = 0 lines = words = bytes = 0
input.each_line do |line| input.each_line do |line|
@ -23,14 +49,6 @@ def count_file_stats(input)
[lines, words, bytes] [lines, words, bytes]
end end
def process_input(source)
if source == '-'
count_file_stats(ARGF)
else
count_file_stats(File.read(source))
end
end
def update_totals(totals, stats) def update_totals(totals, stats)
totals.zip(stats).map { |total, stat| total + stat } totals.zip(stats).map { |total, stat| total + stat }
end end
@ -55,21 +73,3 @@ def print_total(total_stats, max_widths, options)
total_result = format_result(total_stats, max_widths, options) total_result = format_result(total_stats, max_widths, options)
puts "#{total_result} 合計" puts "#{total_result} 合計"
end end
def process_input_sources(input_sources, options)
total_stats = [0, 0, 0]
max_widths = [0, 0, 0]
input_sources.each do |source|
stats = process_input(source)
total_stats = update_totals(total_stats, stats)
max_widths = update_max_widths(max_widths, stats)
result = format_result(stats, max_widths, options)
print_result(result, source)
end
print_total(total_stats, max_widths, options) if input_sources.size > 1
end
options, input_sources = parse_options
process_input_sources(input_sources, options)