Shell stuff for putting AWS context in my prompt (make sure jq is on your path)

aws_filter.jq

if .Account == "9999999" then "test" elif .Account == "111111" then "prod" else "" end

aws_prompt_info.sh

aws_prompt_info () {
	local PID_FILE=~/.aws-login/default.pid 
	local IDENTITY_FILE=~/.aws-login/identity.txt 
	local FILTER_FILE=~/.oh-my-zsh/custom/aws_filter.jq 
	local PROF="" 
	if [[ -f "$PID_FILE" ]]
	then
		PROF=$( cat "$IDENTITY_FILE" 2>/dev/null || aws sts get-caller-identity | jq -r -f "$FILTER_FILE" | tee "$IDENTITY_FILE") 
		[[ -n "$PROF" ]] && echo "[$PROF]"
	else
		[[ -f "$IDENTITY_FILE" ]] && rm "$IDENTITY_FILE"
	fi
}

zsh.theme

# Depending on what you're using for a shell this will probably be different. In my shell I have my name instead of dude in case I forget it again
HOST_PROMPT_='%{$fg_bold[red]%}@dude$(aws_prompt_info) %{$fg_bold[cyan]%}%c '