import os
import sys
import paramiko

sys.stdout.reconfigure(encoding="utf-8")
pkey = paramiko.Ed25519Key.from_private_key_file(os.path.expanduser('~/.ssh/id_ed25519_snarfnet'))

for user in ["root", "dietpi", "snarf"]:
    for auth_type in ["key", "pass"]:
        client = paramiko.SSHClient()
        client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        try:
            if auth_type == "key":
                client.connect("140.44.4.21", username=user, pkey=pkey, timeout=5)
            else:
                pw = "dietpi" if user == "dietpi" else "Katzir476!"
                client.connect("140.44.4.21", username=user, password=pw, timeout=5)
            print(f"SUCCESS: user={user}, auth={auth_type}")
            stdin, stdout, stderr = client.exec_command("whoami; hostname")
            print("  Output:", stdout.read().decode().strip())
            client.close()
            break
        except Exception as e:
            # print(f"Fail {user} {auth_type}: {e}")
            pass