#! /usr/bin/env python3
# vim:set noet sw=8 ts=8 sts=8:

import sys, socket
import dns.query
import dns.zone
import dns.rdatatype

def resolve(hostname):
	try:
		addrs = socket.getaddrinfo(
				hostname, 53, proto=socket.IPPROTO_UDP)
		return list(dict.fromkeys([addr[4][0] for addr in addrs]))
	except Exception as e:
		raise (Exception(f"resolving {hostname}: {e}"))

def axfr(zone, hostname):
	addrs = resolve(hostname)
	last_exc = None

	for addr in addrs:
		try:
			return dns.zone.from_xfr(dns.query.xfr(addr, zone))
		except Exception as e:
			last_exc = e
	raise (Exception(str(last_exc)))

def print_catalog(zone):
	for name, node in zone.nodes.items():
		for rdata in node.rdatasets:
			if rdata.rdtype == dns.rdatatype.PTR:
				for ptr in rdata:
					print(str(ptr.target).rstrip('.'))


def main():
	if len(sys.argv) != 3:
		print(f"Usage: {sys.argv[0]} <zone> <server>")
		return 1

	zone_name = sys.argv[1]
	server = sys.argv[2]

	try:
		zone = axfr(zone_name, server)
		print_catalog(zone)
	except Exception as e:
		print(f"Zone transfer failed: {e}")
		return 1

	return 0

if __name__ == "__main__":
	sys.exit(main())