bashrc function for aws prompt

gistfile1.txt

aws () {
   aws_cmd=`command which aws`
   if [[ $@ =~ 'help' ]]
	then
		command $aws_cmd $@
		return
	fi
   identity=`$aws_cmd sts get-caller-identity` || return
   account=`echo $identity | jq '.Account' | tr -d '"'`
   arn=`echo $identity | jq '.Arn' | tr -d '"'`

   user=`echo $arn | awk -F ":" '{ print $6 }'`
   if [[ $user =~ assumed-role* ]] ; then
     S="_"
   else
     S="/"
   fi
  
   username=`echo $user | awk -F $S '{ print $2 }'`
   export PS1="$username@$account # "
   command $aws_cmd $@
}