#!/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 ls /www/server/panel/vhost/nginx/ | grep nomadro",
    "echo '{pw}' | sudo -S head -40 /www/server/panel/vhost/nginx/lounge.nomadro.cn.conf 2>/dev/null || echo '{pw}' | sudo -S head -40 /www/server/panel/vhost/nginx/mirotalk.nomadro.cn.conf",
    "echo '{pw}' | sudo -S ls /etc/letsencrypt/live/ 2>/dev/null | head -20",
    "echo '{pw}' | sudo -S docker images | head -20",
    "echo '{pw}' | sudo -S cat /opt/nomadro-services/integrations/docker-compose.yml | head -30",
]


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=60, get_pty=True)
            out = o.read().decode("utf-8", errors="replace")
            err = e.read().decode("utf-8", errors="replace")
            sys.stdout.buffer.write(out.encode("utf-8", errors="replace"))
            sys.stdout.buffer.write(err.encode("utf-8", errors="replace"))
    finally:
        c.close()


if __name__ == "__main__":
    main()