Permalink
Browse files

Added expect.rb from 1.9.3p0.

  • Loading branch information...
1 parent 7f701a7 commit 981a2efde02cb71903fbf437f11036783b015d57 @brixen brixen committed Nov 22, 2011
Showing with 51 additions and 0 deletions.
  1. 0 lib/{ → 18}/expect.rb
  2. +51 −0 lib/19/expect.rb
File renamed without changes.
View
@@ -0,0 +1,51 @@
+$expect_verbose = false
+
+class IO
+ # Reads from the IO until pattern +pat+ 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 the pattern. If the timeout expires or eof is found, nil is returned
+ # or yielded. 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
+ @unusedBuf ||= ''
+ while true
+ if not @unusedBuf.empty?
+ c = @unusedBuf.slice!(0).chr
+ elsif !IO.select([self],nil,nil,timeout) or eof? then
+ result = nil
+ @unusedBuf = buf
+ break
+ else
+ c = getc.chr
+ end
+ 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
+

0 comments on commit 981a2ef

Please sign in to comment.