diff options
Diffstat (limited to 'ext/pty/lib/expect.rb')
-rw-r--r-- | ext/pty/lib/expect.rb | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/ext/pty/lib/expect.rb b/ext/pty/lib/expect.rb new file mode 100644 index 0000000000..54c69edadb --- /dev/null +++ b/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).nil? then + result = nil + break + end + c = getc.chr + buf << c + if $expect_verbose + STDOUT.print c + STDOUT.flush + end + if buf =~ e_pat then + result = [buf,$1,$2,$3,$4,$5,$6,$7,$8,$9] + break + end + end + if iterator? then + yield result + else + return result + end + nil + end +end + |