#!/usr/bin/env python3

import argparse
import dnsmgr
import sys

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=dnsmgr.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 = dnsmgr.DNSManager(cfgfile=args.config)
    except RuntimeError as e:
        dnsmgr.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:
        dnsmgr.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:
            dnsmgr.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 dnsmgr.RECORD_TYPES:
                    continue
                for value in rdataset:
                    records.append({
                        'name': name.to_text(),
                        'name_unicode': name.to_unicode(),
                        'ttl': str(rdataset.ttl),
                        'type': str(rdataset.rdtype.to_text(rdataset.rdtype)),
                        'value': value.to_text(origin=zone.origin, relativize=False)})

        records.sort(key=lambda r: f'{r["name_unicode"]}{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]:
                name = record['name_unicode'] if args.decode else record['name']
                row = [name, record['ttl'], record['type'], record['value']]
                rows.append(row)

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


if __name__ == '__main__':
    main()