import argparse
import re
import sys

from . import NAMED_DEFAULT_VIEW, RECORD_TYPES, DEFAULT_CFGFILE, DNSManager, printe, prettytable
from dns.reversename import ipv4_reverse_domain, ipv6_reverse_domain
from json import dumps


def main():
    preparser = argparse.ArgumentParser(add_help=False)
    preparser.add_argument('-b', '--batch', action='store_true')
    preargs, args = preparser.parse_known_args()
    nargs = None if preargs.batch else '?'

    parser = argparse.ArgumentParser(description='Show DNS zone records.')
    parser.add_argument('-a', '--all-zones', help='do not ignore zones that are not managed', action='store_true')
    parser.add_argument('-A', '--all-records', help='do not ignore unsupported record types', action='store_true')
    parser.add_argument('-b', '--batch', help='run in batch mode (no user input)', action='store_true')
    parser.add_argument('-c', '--config', help='path to config file', default=DEFAULT_CFGFILE)
    parser.add_argument('-d', '--decode', help='decode internationalized domain names (IDN)', action='store_true')
    parser.add_argument('zone', metavar='ZONE[@VIEWS]', nargs=nargs, help='DNS zone name and optional list of views (comma separated or asterisk to select all views)', default=None)
    output = parser.add_mutually_exclusive_group()
    output.add_argument('-j', '--json', help='print json format', action='store_true')
    output.add_argument('-J', '--json-pretty', help='print pretty json format', action='store_true')
    output.add_argument('-r', '--raw', help='print raw format', action='store_true')
    args = parser.parse_args()

    try:
        manager = DNSManager(cfgfile=args.config)
    except RuntimeError as e:
        printe(f'config: {e}')
        sys.exit(100)

    try:
        if args.zone is None:
            zones = manager.select_zones(args.all_zones)
        else:
            zones = manager.get_zones(args.zone, args.all_zones)

    except RuntimeError as e:
        printe(e)
        sys.exit(150)
    except KeyboardInterrupt:
        sys.exit(0)

    zones.sort(key=lambda zone: zone.view)

    zone_records = {}
    for zone in zones:
        try:
            manager.get_zone_content(zone)
        except RuntimeError as e:
            printe(f"zone transfer of '{zone.origin.to_text(True)}@{zone.view}': {e}")
            sys.exit(160)

        records = []
        for name, node in zone.items():
            for rdataset in node:
                if not args.all_records and rdataset.rdtype not in RECORD_TYPES:
                    continue
                for value in rdataset:
                    records.append({
                        'name': name.to_unicode() if args.decode else name.to_text(),
                        'ttl': str(rdataset.ttl),
                        'type': str(rdataset.rdtype.to_text(rdataset.rdtype)),
                        'value': value.to_text(origin=zone.origin, relativize=False)})

        if zone.origin.is_subdomain(ipv4_reverse_domain):
            records.sort(key=lambda r: int(re.findall(r'\d+|$', r['name'])[0] or 0))
        elif zone.origin.is_subdomain(ipv6_reverse_domain):
            records.sort(key=lambda r: int(''.join(re.findall(r'\d+', r['name'])) or 0))
        else:
            records.sort(key=lambda r: f'{r["name"]}{r["type"]}')
        zone_records[zone.view] = records

    views = sorted(zone_records.keys())

    if args.raw:
        for view in views:
            print(f';\n; View: {view}\n;\n')
            for record in zone_records[view]:
                print('\t'.join([record['name'], record['ttl'], record['type'], record['value']]))
            print(f'\n;\n; End of view: {view}\n;\n')
    elif args.json or args.json_pretty:
        json_output = []
        for view in views:
            json_output.append({'view': view, 'records': zone_records[view]})
        if args.json_pretty:
            print(dumps(json_output, indent=2))
        else:
            print(dumps(json_output))

    else:
        field_names = ['Name', 'TTL', 'Type', 'Value']
        for view in views:
            rows = []
            for record in zone_records[view]:
                row = [record['name'], record['ttl'], record['type'], record['value']]
                rows.append(row)

            if len(views) > 1 or view != NAMED_DEFAULT_VIEW:
                print(f'View: {view}')
            print(prettytable(field_names, rows, truncate=True))
            print()


if __name__ == '__main__':
    main()