aws mfa cli tool

aws-mfa.sh

#!/bin/bash

# From a user's MFA code, generate accesskey, secret, session token
#
# ENV inputs:
# AWS_CLI_INPUT = /path/to/token/config.json
# AWS_PROFILE = The Aws profile to use for aws commands

MFA_CODE=$1
if [ -z $MFA_CODE ]; then
  echo "Usage: $0 MFA_CODE"
  exit 1
fi

if [ -z "$AWS_CLI_INPUT"]; then
  AWS_CLI_INPUT=~/.aws/generate_token_nonprod.json
fi
# generate .json with aws sts get-session-token --generate-cli-skeleton > ~/.aws/generate_token_nonprod.json

if [ ! -f $AWS_CLI_INPUT ]; then
   echo "generate .json with aws sts get-session-token --generate-cli-skeleton > $AWS_CLI_INPUT"
   exit 1
fi
AWS_CLI_INPUT=file://${AWS_CLI_INPUT}

if [ ! -z "$AWS_PROFILE" ]; then
  TOKENS=`aws sts get-session-token --profile ${AWS_PROFILE} --cli-input-json ${AWS_CLI_INPUT} --token-code ${MFA_CODE} --output json | jq -r '.Credentials | "\(.AccessKeyId) \(.SecretAccessKey) \(.SessionToken)"'`
else
  TOKENS=`aws sts get-session-token --cli-input-json ${AWS_CLI_INPUT} --token-code ${MFA_CODE} --output json | jq -r '.Credentials | "\(.AccessKeyId) \(.SecretAccessKey) \(.SessionToken)"'`
fi

read AWS_ACCESS_KEY_ID AWS_SECRET_ACCESS_KEY AWS_SESSION_TOKEN <<< ${TOKENS}
export AWS_ACCESS_KEY_ID AWS_SECRET_ACCESS_KEY AWS_SESSION_TOKEN
echo $AWS_ACCESS_KEY_ID $AWS_SECRET_ACCESS_KEY $AWS_SESSION_TOKEN