#!/usr/bin/env python3

import argparse
import dns.rdataclass
import dns.rdataset
import dnsmgr
import sys


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 '?'
    nvalueargs = '+' if preargs.batch else '*'

    parser = argparse.ArgumentParser(description='Add DNS records.')
    parser.add_argument('-a', '--all-zones', help='allow zones that are not managed', action='store_true')
    parser.add_argument('-A', '--all-types', help='allow 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('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)
    parser.add_argument('name', metavar='NAME', nargs=nargs, help='DNS record name', default=None)
    parser.add_argument('ttl', metavar='TTL', nargs=nargs, help='DNS record TTL in seconds', type=int, default=None)
    parser.add_argument('type', metavar='TYPE', nargs=nargs, help='DNS record type', default=None)
    parser.add_argument('value', metavar='VALUE', nargs=nvalueargs, help='DNS record value, multiple values are choined by a space character', default=None)
    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)

        origin = zones[0].origin

        if args.name is None:
            name = dnsmgr.input_name(origin, prompt='Record name')
        else:
            name = dnsmgr.name_from_text(args.name, origin)

        if args.ttl is None:
            ttl = dnsmgr.input_ttl()
        else:
            ttl = dnsmgr.ttl_from_text(args.ttl)

        if args.type is None:
            rdtype = dnsmgr.select_type(args.all_types)
        else:
            rdtype = dnsmgr.type_from_text(args.type, args.all_types)

        if args.value is None:
            rdata = dnsmgr.input_rdata(rdtype, origin)
        else:
            rdata = dnsmgr.rdata_from_text(rdtype, ' '.join(args.value), origin)

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

    rdataset = dns.rdataset.Rdataset(dns.rdataclass.IN, rdtype, ttl=ttl)
    rdataset.add(rdata)

    if not args.batch:
        for zone in zones:
            text = rdataset.to_text(origin=zone.origin, relativize=False)
            print(f'View: {zone.view}')
            print(f'\033[32m+ {name} {text}\033[0m\n')

        if not dnsmgr.input_yes_no():
            sys.exit(0)

    for zone in zones:
        origin = zone.origin.to_text(omit_final_dot=True)
        if len(zones) > 1 or zone.view != dnsmgr.NAMED_DEFAULT_VIEW:
            origin = f'{origin}@{zone.view}'
        print(f"Sending DDNS updates for '{origin}'... ", end='')

        try:
            manager.add_zone_record(zone, name, rdataset)
            print('OK')
        except RuntimeError as e:
            dnsmgr.printe(e)
            sys.exit(160)


if __name__ == '__main__':
    main()