import logging

import dns.update
import dns.query
import dns.tsigkeyring
import ipaddress

config = {}

LOG = logging.getLogger(__name__)


def add_records(records):
    keyring = dns.tsigkeyring.from_text({config['TSIG_NAME']: config['TSIG_KEY']})

    for hostname, ip in records.items():
        LOG.info(f"Adding record for {hostname}({ip})")

        rrtype = "A"
        address = ipaddress.ip_address(ip)
        if isinstance(address, ipaddress.IPv4Address):
            rrtype = "A"
        if isinstance(address, ipaddress.IPv6Address):
            rrtype = "AAAA"

        update = dns.update.Update(config['DOMAIN'], keyring=keyring)
        update.add(hostname, int(config['DNS_RECORD_TTL']), rrtype, ip)
        dns.query.tcp(update, config['NAMESERVER'], timeout=2)


def delete_records(records):
    keyring = dns.tsigkeyring.from_text({config['TSIG_NAME']: config['TSIG_KEY']})

    for hostname, ip in records.items():
        LOG.info(f"Deleting record for {hostname}({ip})")

        update = dns.update.Update(config['DOMAIN'], keyring=keyring)
        update.delete(hostname)
        dns.query.tcp(update, config['NAMESERVER'], timeout=2)


def init(_config):
    global config
    config = _config