"""Defines how to parse CLI arguments for the DomainTools API"""

parser = argparse.ArgumentParser(description='The DomainTools Python API Wrapper provides an interface to work with our cybersecurity and related data tools provided by our Iris Investigate™, Iris Enrich™, and Iris Enrich™ products. See the included README file, the examples folder and API documentation (https://app.swaggerhub.com/apis-docs/DomainToolsLLC/DomainTools_APIs/1.0#) for more info',

formatter_class=argparse.RawTextHelpFormatter)

parser.add_argument('-u', '--username', dest='user', default='', help='API Username')

parser.add_argument('-k', '--key', dest='key', default='', help='API Key')

parser.add_argument('-c', '--credfile', dest='credentials', default=os.path.expanduser('~/.dtapi'),

help='Optional file with API username and API key, one per line.')

parser.add_argument('-l', '--rate-limit', dest='rate_limit', action='store_true', default=False,

help='Rate limit API calls against the API based on per minute limits.')

parser.add_argument('-f', '--format', dest='format', choices=['list', 'json', 'xml', 'html'], default='json')

parser.add_argument('-o', '--outfile', dest='out_file', type=argparse.FileType('wbU'), default=sys.stdout,

help='Output file (defaults to stdout)')

parser.add_argument('-v', '--version', action='version', version='DomainTools CLI API Client {0}'.format(version))

parser.add_argument('--no-verify-ssl', dest='verify_ssl', action='store_false', default=True,

help='Skip verification of SSL certificate when making HTTPs API calls')

subparsers = parser.add_subparsers(help='The name of the API call you wish to perform (`whois` for example)',

dest='api_call')

subparsers.required = True

for api_call in API_CALLS:

api_method = getattr(API, api_call)

subparser = subparsers.add_parser(api_call, formatter_class=argparse.RawTextHelpFormatter, help=inspect.getdoc(api_method), description=inspect.getdoc(api_method))

spec = inspect.getfullargspec(api_method)

for argument_name, default in reversed(list(zip_longest(reversed(spec.args or []),

reversed(spec.defaults or []), fillvalue='EMPTY'))):

if argument_name == 'self':

continue

elif default == 'EMPTY':

subparser.add_argument(argument_name)

else:

subparser.add_argument('--{0}'.format(argument_name.replace('_', '-')), dest=argument_name,

default=default, nargs='*')

arguments = vars(parser.parse_args(args) if args else parser.parse_args())

if not arguments.get('user') or not arguments.get('key'):

try:

with open(arguments.pop('credentials')) as credentials:

arguments['user'], arguments['key'] = credentials.readline().strip(), credentials.readline().strip()

except Exception:

pass

for key, value in arguments.items():

if value in ('-', ['-']):

arguments[key] = (line.strip() for line in sys.stdin.readlines())

elif value == []:

arguments[key] = True

elif isinstance(value, list) and len(value) == 1:

arguments[key] = value[0]

"""Defines how to start the CLI for the DomainTools API"""

out_file, out_format, arguments = parse()

user, key = arguments.pop('user', None), arguments.pop('key', None)

if not user or not key:

sys.stderr.write('Credentials are required to perform API calls.\n')

sys.exit(1)

api = API(user, key, app_name="python_wrapper_cli", verify_ssl=arguments.pop('verify_ssl'), rate_limit=arguments.pop('rate_limit'))

api_call = arguments.pop('api_call')

response = getattr(api, api_call)(**arguments)

if api_call in ["available_api_calls"]:

output = '\n'.join(response)

else:

output = str(getattr(response, out_format) if out_format != 'list' else response.as_list())