helpers.sh

#################################
#            Functions          #
#################################

function aws {
	/usr/local/bin/aws "$@" | ppj
}

function ppj {
	pygmentize -f terminal256 -l json -O style=monokai
}

function gen_keys {
	if [ -z "$1" ]
	then
		read -p "Enter AWS account number: "
	else
		acct=$1
	fi

	if [ -z "$2" ]
	then
		read -p "Enter AWS UserName: " username
	else
		username=$2
	fi

	if [ -z "$3" ]
	then
		read -p "Enter MFA Token: " token
	else
		token=$3
	fi

	creds=$(/usr/local/bin/aws sts get-session-token --serial-number arn:aws:iam::$acct:mfa/$username --token-code $token --profile init --duration-seconds=43200 --output text | awk '{print $2 " " $4 " " $5}' )
	read access secret session <<<$creds

	export AWS_ACCESS_KEY_ID=$access
	export AWS_SECRET_ACCESS_KEY=$secret
	export AWS_SESSION_TOKEN=$session
}