#!/usr/bin/env python3
from __future__ import annotations

import os
import sys

import paramiko

HOST = os.getenv("NOMAD_SSH_HOST", "82.157.112.245")
USER = os.getenv("NOMAD_SSH_USER", "ubuntu")
PASSWORD = os.getenv("NOMAD_SSH_PASSWORD", "Xiao4669805")

CMDS = [
    "echo '{pw}' | sudo -S cat /etc/docker/daemon.json 2>/dev/null || echo 'no daemon.json'",
    "echo '{pw}' | sudo -S docker info 2>/dev/null | grep -A5 'Registry Mirrors' || true",
    "echo '{pw}' | sudo -S docker images --format '{{.Repository}}:{{.Tag}}' | sort",
    "echo '{pw}' | sudo -S bash -lc 'cd /opt/nomadro-services/integrations && docker compose ps -a 2>/dev/null'",
    "echo '{pw}' | sudo -S nginx -t 2>&1",
    "curl -sI https://nomadro.cn/ | head -3",
]


def main() -> None:
    c = paramiko.SSHClient()
    c.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    c.connect(HOST, username=USER, password=PASSWORD, timeout=30)
    try:
        for cmd in CMDS:
            cmd = cmd.format(pw=PASSWORD)
            print(f"\n=== {cmd.replace(PASSWORD, '***')} ===")
            _, o, e = c.exec_command(cmd, timeout=90, get_pty=True)
            sys.stdout.buffer.write(o.read())
            sys.stdout.buffer.write(e.read())
    finally:
        c.close()


if __name__ == "__main__":
    main()