diff options
Diffstat (limited to 'trunk/ext/pty/lib/expect.rb')
-rw-r--r-- | trunk/ext/pty/lib/expect.rb | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/trunk/ext/pty/lib/expect.rb b/trunk/ext/pty/lib/expect.rb new file mode 100644 index 0000000000..08191b05b9 --- /dev/null +++ b/trunk/ext/pty/lib/expect.rb @@ -0,0 +1,36 @@ +$expect_verbose = false + +class IO + def expect(pat,timeout=9999999) + buf = '' + case pat + when String + e_pat = Regexp.new(Regexp.quote(pat)) + when Regexp + e_pat = pat + 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 + |