37 lines
629 B
Ruby
37 lines
629 B
Ruby
$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 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
|
|
|