#!/usr/bin/env python3
import json
import re
import urllib.request
import paramiko

PASS = "Bbt9115xty9176!"
CF = "https://api.cloudflare.com/client/v4"

c = paramiko.SSHClient()
c.set_missing_host_key_policy(paramiko.AutoAddPolicy())
c.connect("10.10.0.1", username="admin", password=PASS, timeout=15)
_, o, e = c.exec_command("cat /usr/local/etc/wan-dns-failover.env", timeout=30)
o.channel.recv_exit_status()
TOK = re.search(r"CLOUDFLARE_API_TOKEN=([^\s\"']+)", (o.read() + e.read()).decode()).group(1)
c.close()

for domain in ["justtworoommates.com", "thepinkpulse.com", "freerangefruit.com"]:
    req = urllib.request.Request(
        f"{CF}/zones?name={domain}",
        headers={"Authorization": f"Bearer {TOK}", "Content-Type": "application/json"},
    )
    with urllib.request.urlopen(req, timeout=30) as r:
        res = json.load(r)
    z = res["result"][0]
    print(f"{domain}: {z.get('name_servers')}")