import sys

from gpu_rent.config import load_config
from gpu_rent.state import load_state
from gpu_rent.ssh_ops import run_ssh

cfg = load_config(require_auth=True)
state = load_state()
cmd = sys.argv[1] if len(sys.argv) > 1 else "systemctl is-active swarmui"
out = run_ssh(cfg, state.floating_ip, cmd, check=False, timeout=120)
sys.stdout.write(out)