#!/usr/lib/rads/venv/bin/python3
# written by ChaseS
import argparse
from python import audit_dns, humanize


def get_args():
    parser = argparse.ArgumentParser(description=__doc__)
    usergroup = parser.add_mutually_exclusive_group()
    usergroup.add_argument(
        '-u',
        '--user',
        action='store',
        dest='User',
        default=False,
        help='Define your User',
    )

    usergroup.add_argument(
        '-s',
        '--server',
        action='store_true',
        dest='Server',
        default=False,
        help='Check DNS for full server rather than one domain',
    )

    usergroup.add_argument(
        '-d',
        '--domain',
        action='store',
        dest='Domain',
        default=False,
        help='Audit dns by domain',
    )

    parser.add_argument(
        '-l',
        '--long',
        action='store_true',
        default=False,
        dest='longview',
        help='Show full rather than parsed results',
    )

    parser.add_argument('--version', action='version', version='%(prog)s 1.0')

    results = parser.parse_args()
    return results


def print_humanized(args):
    print('Please be patient. This takes a while.')
    if args.Server is True:
        humanize.h_check_dns(
            data=audit_dns.server(), external_check=True, longview=args.longview
        )
    elif args.Domain is not False:
        humanize.h_check_dns(
            data=audit_dns.domain(args.Domain),
            external_check=True,
            longview=args.longview,
        )
    elif args.User is not False:
        humanize.h_check_dns(
            data=audit_dns.user(args.User),
            external_check=True,
            longview=args.longview,
        )


def main():
    args = get_args()
    print_humanized(args)


if __name__ == "__main__":
    main()