diff --git a/src/script.c b/src/script.c index 9cb5baa..76b5933 100644 --- a/src/script.c +++ b/src/script.c @@ -58,22 +58,15 @@ static char script_init[] = "end\n" "tio.expect = function(pattern, timeout)\n" " local str = ''\n" -" if type(pattern) ~= 'table' then\n" -" pattern = { pattern }\n" -" end\n" " while true do\n" -" local idx, pat\n" " local c = tio.read(1, timeout)\n" " if c then\n" " str = str .. c\n" -" for idx, pat in ipairs(pattern) do\n" -" local matched = { string.match(str, pat) }\n" -" if matched[1] then\n" -" return matched, str, idx\n" -" end\n" +" if string.match(str, pattern) then\n" +" return string.match(str, pattern)\n" " end\n" " else\n" -" return nil, str, 0\n" +" return nil, str\n" " end\n" " end\n" "end\n"