#!/usr/bin/python

from dnsimple import DNSimple
from urllib import urlopen
from sys import stderr, exit
from apikey import API_EMAIL, API_KEY

domains = [{"domain": 'jshaver.net', "names": ["", "bc"]},
            {"domain": 'theamazingshavers.com', "names": [""]}]

ttl = 600

dns = DNSimple(email=API_EMAIL, api_token=API_KEY)

response = urlopen("http://icanhazip.com/")

if response.getcode() != 200:
    stderr.write("Non 200 status when retrieving ip! Status: {}".format(response.getcode()))
    exit(-1)

ip = response.readline().rstrip()

def get_a_record(domain, name):
    records = dns.records(domain)
    for entry in records:
        record = entry["record"]
        if record["name"] is name and ["record_type"] is 'A':
            return record
    return None

def update_record(domain, name, record, ip):
    new = {"name": name, "content": ip}
    response = dns.update_record(domain, record["id"], new)
    return response["record"]

def create_record(domain, name, ip):
    new = {
            "name": name,
            "content": ip,
            "record_type": "A",
            "ttl": ttl}
    print new
    print domain
    response = dns.add_record(domain, new)
    return response["record"]

def update_domain_ip(domain, name, ip):
    record = get_a_record(domain, name)

    if record is None:
        return create_record(domain, name, ip)

    if record["content"] is ip:
        return record

    return update_record(record, name, ip)

def main(domainList, ip):
    for domain in domainList:
        for name in domain["names"]:
            update_domain_ip(domain["domain"], name,  ip)

main(domains, ip)