Module: Polyfill::V2_4::StringIO::ClassMethods

Defined in:
lib/polyfill/v2_4/string_io.rb

Instance Method Summary collapse

Instance Method Details

#foreach(name, *args) ⇒ Object



7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
# File 'lib/polyfill/v2_4/string_io.rb', line 7

def foreach(name, *args)
  hash, others = args.partition { |arg| arg.is_a?(::Hash) }
  chomps = hash[0] && hash[0][:chomp]

  unless block_given?
    return super(name, *others) unless chomps

    separator = others.find do |other|
      other.respond_to?(:to_str)
    end || $INPUT_RECORD_SEPARATOR

    return ::Enumerator.new do |yielder|
      super(name, *others) do |line|
        yielder.yield(line.chomp(separator))
      end
    end
  end

  block =
    if chomps
      separator = others.find do |other|
        other.respond_to?(:to_str)
      end || $INPUT_RECORD_SEPARATOR

      proc do |line|
        yield(line.chomp(separator))
      end
    else
      ::Proc.new
    end

  super(name, *others, &block)
end

#readlines(file_name, *args) ⇒ Object



41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
# File 'lib/polyfill/v2_4/string_io.rb', line 41

def readlines(file_name, *args)
  hash, others = args.partition { |arg| arg.is_a?(::Hash) }

  inputs = super(file_name, *others)

  if hash[0] && hash[0][:chomp]
    separator = others.find do |other|
      other.respond_to?(:to_str)
    end || $INPUT_RECORD_SEPARATOR

    inputs.each { |input| input.chomp!(separator) }
  end

  inputs
end