import sys
from pathlib import Path

from gpu_rent.config import load_config
from gpu_rent.state import load_state
from gpu_rent.ssh_ops import _connect, put_file_on, run_ssh_on

cfg = load_config(require_auth=True)
state = load_state()
client = _connect(cfg, state.floating_ip)
try:
    put_file_on(client, Path("tmp_smoke.sh"), "/tmp/tmp_smoke.sh")
    out = run_ssh_on(
        client, "sed -i 's/\\r$//' /tmp/tmp_smoke.sh && bash /tmp/tmp_smoke.sh", check=False, timeout=120
    )
    sys.stdout.write(out)
finally:
    client.close()