"""Stream bounded approved AI2 helper output; sudo secret stays on stdin."""
import base64
from pathlib import Path
import shlex
import subprocess
import sys
from remote import SSH,SUDO

source=Path(sys.argv[1])
seconds=int(sys.argv[2])
assert source.parent==Path(__file__).resolve().parent and 30<=seconds<=6000
code='import base64;exec(base64.b64decode('+repr(base64.b64encode(source.read_bytes()).decode())+'))'
command='sudo -k -S -p "" -- timeout --kill-after=20s '+str(seconds)+'s python3 -B -u -c '+shlex.quote(code)
result=subprocess.run(SSH+[command],input=SUDO.read_bytes().rstrip(b'\r\n')+b'\n',timeout=seconds+40)
raise SystemExit(result.returncode)