from __future__ import annotations

from ..ddns_models import DdnsSettings, UpdateResult
from .base import DnsProviderError


def publish_rfc2136(result: UpdateResult, settings: DdnsSettings) -> None:
    if not settings.rfc2136_server:
        return

    import dns.exception
    import dns.query
    import dns.rcode
    import dns.tsigkeyring
    import dns.update

    keyring = None
    if settings.rfc2136_key_name and settings.rfc2136_key_secret:
        keyring = dns.tsigkeyring.from_text({settings.rfc2136_key_name: settings.rfc2136_key_secret})

    zone = settings.rfc2136_zone or ".".join(result.hostname.split(".")[1:])
    update = dns.update.Update(zone, keyring=keyring)
    relative = result.hostname.removesuffix("." + zone).rstrip(".") or "@"
    if result.update_ipv4 and result.ipv4:
        update.replace(relative, settings.ttl, "A", result.ipv4)
    elif result.update_ipv4:
        update.delete(relative, "A")
    if result.update_ipv6 and result.ipv6:
        update.replace(relative, settings.ttl, "AAAA", result.ipv6)
    elif result.update_ipv6:
        update.delete(relative, "AAAA")
    try:
        response = dns.query.tcp(update, settings.rfc2136_server, timeout=5)
    except dns.exception.DNSException as exc:
        raise DnsProviderError("RFC2136 DNS update failed") from exc
    rcode = response.rcode()
    if rcode != 0:
        raise DnsProviderError(f"RFC2136 update failed: {dns.rcode.to_text(rcode)}")


def delete_rfc2136_records(hostname: str, settings: DdnsSettings) -> None:
    if not settings.rfc2136_server:
        return

    import dns.exception
    import dns.query
    import dns.rcode
    import dns.tsigkeyring
    import dns.update

    keyring = None
    if settings.rfc2136_key_name and settings.rfc2136_key_secret:
        keyring = dns.tsigkeyring.from_text({settings.rfc2136_key_name: settings.rfc2136_key_secret})

    zone = settings.rfc2136_zone or ".".join(hostname.split(".")[1:])
    update = dns.update.Update(zone, keyring=keyring)
    relative = hostname.removesuffix("." + zone).rstrip(".") or "@"
    update.delete(relative, "A")
    update.delete(relative, "AAAA")
    try:
        response = dns.query.tcp(update, settings.rfc2136_server, timeout=5)
    except dns.exception.DNSException as exc:
        raise DnsProviderError("RFC2136 DNS delete failed") from exc
    rcode = response.rcode()
    if rcode != 0:
        raise DnsProviderError(f"RFC2136 delete failed: {dns.rcode.to_text(rcode)}")