monojohnny monojohnny - 9 months ago 54
Ruby Question

Script to run against stdin if no arg; otherwise input file =ARGV[0]

This works quite nicely - just wondered if there are any improvements to shorten it ?

if (ARGV[0].nil?) then

# Do something with the input here, for example:
input.each_line do |line|
puts line

Answer Source

You can eliminate the first five lines entirely.

From Pickaxe

$<: An object that provides access to the concatenation of the contents of all the files given as command-line arguments or $stdin (in the case where there are no arguments). $< supports methods similar to a File object: binmode, close, closed?, each, each_byte, each_line, eof, eof?, file, filename, fileno, getc, gets, lineno, lineno=, path, pos, pos=, read, readchar, readline, readlines, rewind, seek, skip, tell, to_a, to_i, to_io, to_s, along with the methods in Enumerable. The method file returns a File object for the file currently being read. This may change as $< reads through the files on the command line. [r/o]


print $<.read

Kernel.gets is shorthand for $<.gets, so:

while s = gets
  puts s