# This file will help to serve as a starting point for using the rest of the tools
# Things we want to figure out
# 1) Is your key active?
# 2) If active, can you read monitoring configs, can you write?
# 3) Okay, you can read monitoring configs. We recommend things to avoid. Want to go further? Use write access to disable (if applicable)
# 4) Don't want to do anything with monitoring? That's fine, let's guide you through figuring out what your access looks like
# 5) Help with a printout of options from this point forward

import boto3
import argparse
import os
from config import AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY
from botocore.exceptions import ClientError
from modules import *
import sys

os.environ['AWS_SHARED_CREDENTIALS_FILE'] = '.env'

sys.path.append("modules")
for module in all_modules:
    exec("from %s import *"%module)


parser = argparse.ArgumentParser()
parser.add_argument("-s", "--step", help="list the step you would like to run",
action="store", type=str, required=True)
parser.add_argument("-t", "--target", help="Give your target a name so we can track results",
action="store", type=str, required=True)
parser.add_argument("-a", "--arguments", help="Provide a list of arguments, comma separated. Ex: arg1,arg2,arg3",
action="store", type=str, required=False)
parser.add_argument("-l", "--list", help="list steps", action="store_true")
parser.add_argument("-v", "--verbosity", help="increase output verbosity",
action="store_true")
args = parser.parse_args()

def perform_credential_check():
    try:
        #client = boto3.client("sts", aws_access_key_id=AWS_ACCESS_KEY_ID, aws_secret_access_key=AWS_SECRET_ACCESS_KEY)
        client = boto3.client("sts")
        account_id = client.get_caller_identity()["Account"]
    except botocore.exceptions.NoCredentialsError as e:
        print("Error: Unable to locate credentials")
        sys.exit("fix your credentials file -exiting...")
    except ClientError as e:
        print("The AWS Access Keys are not valid/active")
        sys.exit(1)

def step_recon():
    print("!!!")

def method_create():
    try:
        arg = globals()["step_" + args.step]
        return arg
    except KeyError:
        print("That step does not exist")
        exit(1)

# Need to figure out if we have keys in the ENV or not
if AWS_ACCESS_KEY_ID and AWS_SECRET_ACCESS_KEY:
    perform_credential_check()
else:
    print("Please supply keys as outlined in our README.md file")
    #exit(1)

if (args.list):
    pass


# arg_list has to be defined otherwise will cause an exception
arg_list = None

if (args.arguments):
    arg_list = args.arguments.split(',')

# We need the user to tell us the step they want to proceed on
if (args.step):
    arg = method_create()
    if callable(arg):
        if arg_list:
            arg(arg_list)
        else:
             arg()


# Allow the user to specify verbosity for debugging
if (args.verbosity):
    print("Verbosity is enabled")