import paramiko
import sys
import base64
import time

sys.stdout.reconfigure(encoding='utf-8')

key_path = r"C:\Users\Snarf\.ssh\id_ed25519_snarfnet"
manifest_path = r"C:\Users\Snarf\OneDrive\Desktop\code\SnarfCode\homelab-gitops-repo\manifests\spaceport\dns\dns-stack.yaml"

with open(manifest_path, "rb") as f:
    b64_manifest = base64.b64encode(f.read()).decode('ascii')

c = paramiko.SSHClient()
c.set_missing_host_key_policy(paramiko.AutoAddPolicy())
c.connect("140.44.4.41", username="root", key_filename=key_path, timeout=10)

cmd = f"""
echo '{b64_manifest}' | base64 -d | k3s kubectl apply -f -
"""
stdin, stdout, stderr = c.exec_command(cmd)
print("Spaceport Apply result:")
print(stdout.read().decode('utf-8'))
err = stderr.read().decode('utf-8')
if err: print("Stderr:", err)

# Poll for pods and services
for i in range(12):
    time.sleep(5)
    stdin, stdout, stderr = c.exec_command("k3s kubectl -n dns get pvc,deploy,svc,pod -o wide")
    out = stdout.read().decode('utf-8')
    print(f"\n--- Check {i+1} ---")
    print(out)
    if "Running" in out and "140.44.4.65" in out:
        print("Spaceport DNS stack is running and VIP is allocated!")
        break

c.close()