Assume AWS role

sts

role=$1
if ! [ $role ]; then
    echo "error: [sts] missing required argument ROLE"
    exit 1
fi

account_id=$(aws sts get-caller-identity | jq .Account | sed s/\"//g)
if ! [ $account_id ]; then
     echo "error: unable to get AWS account ID"
     exit 1
fi

aws sts assume-role \
        --role-arn arn:aws:iam::${account_id}:role/${role} \
        --role-session-name AWSCLI-Session \
    | jq '.Credentials|[.AccessKeyId, .SecretAccessKey, .SessionToken]' \
    | sed 's/,//g' \
    | sed 's/\[//g' \
    | sed 's/\]//g' \
    | sed 's/\"//g' \
    | awk 'NF' \
    | awk 'BEGIN { \
        names[0]="AWS_ACCESS_KEY_ID"; \
        names[1]="AWS_SECRET_ACCESS_KEY"; \
        names[2]="AWS_SESSION_TOKEN"; \
    } { print names[NR-1]"="$1 }'