#!/bin/bash

AWS_USERNAME=$( aws iam get-user --query 'User.[UserName]' --output text )

AWS_MFA_DEVICE=$( aws iam list-mfa-devices --user-name "$AWS_USERNAME" --query 'MFADevices[*].SerialNumber' --output text ) 

read -p "Enter your MFA code now: " AWS_MFA_CODE

AWS_TOKEN=$( aws sts get-session-token --serial-number "$AWS_MFA_DEVICE" --query 'Credentials.[AccessKeyId,SecretAccessKey,SessionToken]' --output text --token-code "$AWS_MFA_CODE" )

export AWS_ACCESS_KEY_ID=$( echo $AWS_TOKEN | cut -f 1 -d ' ' )
export AWS_SECRET_ACCESS_KEY=$( echo $AWS_TOKEN | cut -f 2 -d ' ' )
export AWS_SESSION_TOKEN=$( echo $AWS_TOKEN | cut -f 3 -d ' ' )