import os, socket, requests
from dns_patch import patch_dns
patch_dns()

hosts = {
  "Telegram": "api.telegram.org",
  "Hugging Face": "api-inference.huggingface.co"
}

token = os.getenv("HF_TOKEN")
if not token:
    raise ValueError("HF_TOKEN not set in environment!")

for name, host in hosts.items():
    ips = {ai[4][0] for ai in socket.getaddrinfo(host, None)}
    print(f"{name} DNS → {ips}")
    url = f"https://{host}/models/distilbert-base-uncased"
    headers = {"Authorization": f"Bearer {token}"} if "Hugging Face" in name else {}
    r = requests.get(url, headers=headers, timeout=5)
    print(f"{name} → {r.status_code}")