たぶん完成

wc.rb

@@ -8,29 +8,24 @@ bytes = 0
 
 options = {}
 OptionParser.new do |opts|
-  opts.on('-l') do
+  opts.on('-l') { options[:lines] = true }
-    options[:lines] = true
+  opts.on('-w') { options[:words] = true }
-  end
+  opts.on('-c') { options[:bytes] = true }
-  opts.on('-w') do
-    options[:words] = true
-  end
-  opts.on('-c') do
-    options[:bytes] = true
-  end
 end.parse!
 
-file_path = ARGV[0]
+input = if !ARGV.empty?
-if file_path.nil?
+          File.read(ARGV[0])
-  puts 'ファイルパスがありません。'
+        elsif !$stdin.tty?
-  exit
+          $stdin.read
-end
+        else
+          puts 'ファイルパスまたはパイプラインからの入力がありません。'
+          exit
+        end
 
-File.open(file_path) do |file|
+input.each_line do |line|
-  file.each_line do |line|
+  lines += 1
-    lines += 1
+  words += line.split.size
-    words += line.split.size
+  bytes += line.bytesize
-    bytes += line.bytesize
-  end
 end
 
 options = { bytes: true, lines: true, words: true } if options.empty?
@@ -39,4 +34,5 @@ result << lines if options[:lines]
 result << words if options[:words]
 result << bytes if options[:bytes]
 
-puts "#{result.join(' ')} #{file_path}"
+file_name = ARGV[0]
+puts "#{result.join(' ')} #{file_name}"