summaryrefslogtreecommitdiff
path: root/ext/pty/lib/expect.rb
blob: 52396d90bdd2d379fe7789bb4fd4ae8706c20c6a (plain)
1
2
3
4
5
6
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
40
41
42
43
44
$expect_verbose = false

class IO
  # Reads ios until pattern matches or the timeout is over. It returns
  # an array with the read buffer, followed by the matches. If a block is given,
  # the result is yielded to the block and returns nil. The optional timeout parameter defines,
  # in seconds, the total time to wait for pattern. If it is over of eof is found, it 
  # returns/yields nil. However, the buffer in a timeout session is kept for the next expect call.
  # The default timeout is 9999999 seconds.
  def expect(pat,timeout=9999999)
    buf = ''
    case pat
    when String
      e_pat = Regexp.new(Regexp.quote(pat))
    when Regexp
      e_pat = pat
    else
      raise TypeError, "unsupported pattern class: #{pattern.class}"
    end
    while true
      if !IO.select([self],nil,nil,timeout) or eof? then
        result = nil
        break
      end
      c = getc.chr
      buf << c
      if $expect_verbose
        STDOUT.print c
        STDOUT.flush
      end
      if mat=e_pat.match(buf) then
        result = [buf,*mat.to_a[1..-1]]
        break
      end
    end
    if block_given? then
      yield result
    else
      return result
    end
    nil
  end
end