#!/usr/bin/env python3

import argparse
import sys

from onelogin_aws_cli import OneloginAWS

parser = argparse.ArgumentParser(description="Login to AWS with Onelogin")
parser.add_argument("-c", dest="configure", action="store_true",
                    help="Configure Onelogin and AWS settings")
parser.add_argument("--config_name", default="default",
                    help="Switch configuration name within config file")

args = parser.parse_args()

if args.configure:
    OneloginAWS.generate_config()

config = OneloginAWS.load_config()

if not config or args.config_name not in config:
    sys.exit("Configuration '{}' not defined. "
             "Please run 'onelogin-aws-login -c'".format(args.config_name))

api = OneloginAWS(config[args.config_name])
api.save_credentials()
