import requests
import socket

def get_dns_id(email, api_key, zone_id, record_name):
    url = f"https://api.cloudflare.com/client/v4/zones/{zone_id}/dns_records"
    headers = {
        "Content-Type": "application/json",
        "X-Auth-Email": email,
        "X-Auth-Key": api_key
    }

    try:
        response = requests.request("GET", url, headers=headers)
        data = response.json()
    except:
        print(f"Some thing went wrong while listing the DNS record: Network Error")
        exit(-1)

    if(not data["success"]):
        print(f"Some thing went wrong while listing the DNS record: {data['errors'][0]['message']}")
        exit(-1)
    else:
        for record in data["result"]:
            if(record["name"] == record_name):
                return record["id"]
        print(f"Some thing went wrong while listing the DNS record: Record doesn't exist.")
        exit(-1)

def get_dns(email, api_key, zone_id, record_id):
    url = f"https://api.cloudflare.com/client/v4/zones/{zone_id}/dns_records/{record_id}"
    headers = {
        "Content-Type": "application/json",
        "X-Auth-Email": email,
        "X-Auth-Key": api_key
    }

    try:
        response = requests.request("GET", url, headers=headers)
        data = response.json()
    except:
        print(f"Some thing went wrong while getting the DNS record: Network Error")
        exit(-1)

    if(not data["success"]):
        print(f"Some thing went wrong while getting the DNS record: {data['errors'][0]['message']}")
        exit(-1)
    else:
        return data["result"]["content"]

def update_dns(email, api_key, zone_id, record_name, record_id, record_type, value):
    url = f"https://api.cloudflare.com/client/v4/zones/{zone_id}/dns_records/{record_id}"
    payload = {
        "content": value,
        "name": record_name,
        "type": record_type
    }
    headers = {
        "Content-Type": "application/json",
        "X-Auth-Email": email,
        "X-Auth-Key": api_key
    }

    try:
        response = requests.request("PUT", url, json=payload, headers=headers)
        data = response.json()
    except:
        print(f"Some thing went wrong while updating the DNS record: Network Error")
        exit(-1)
    
    if(not data["success"]):
        print(f"Some thing went wrong while updating the DNS record: {data['errors'][0]['message']}")
        exit(-1)
    else:
        return data["result"]["content"]

def resolve_domain(domain_name):
    try:
        ip_address = socket.gethostbyname(domain_name)
        return ip_address
    except socket.gaierror as e:
        print("Something went error while resolving the domain: ", e)
        exit(-1)

email = "Your Email"
api_key = "Your API-Key (https://dash.cloudflare.com/profile/api-tokens)"
zone_id = "Domain's Zone ID"
flatten_record_name = "Record name that wanted to be flatten"
target_cname = "Target CNAME"
flatten_record_type = "A" # Fixed, do not modify

print(f"Accoount Email: {email}")
print(f"Account API-Key: {api_key}")
print(f"Zone ID: {zone_id}")

flatten_rercord_id = get_dns_id(email, api_key, zone_id, flatten_record_name)
print(f"Flatten Record Name: {flatten_record_name} (ID = {flatten_rercord_id})")

current_dns = get_dns(email, api_key, zone_id, flatten_rercord_id)
print(f"Current DNS: {current_dns}")

fastest_ip = resolve_domain(target_cname)
if(fastest_ip == current_dns):
    print(f"Fastest IP: {fastest_ip}, everything up to date")
else:
    print(f"Fastest IP: {fastest_ip}, trying to update dns...")
    print(f"Updated DNS: {update_dns(email, api_key, zone_id, flatten_record_name, flatten_rercord_id, flatten_record_type, fastest_ip)}, done.")