diff --git a/midgard.zsh-theme b/midgard.zsh-theme index 85c0bc6..6933467 100644 --- a/midgard.zsh-theme +++ b/midgard.zsh-theme @@ -8,7 +8,17 @@ function my_pwd { :end') } -PROMPT='%{$fg[yellow]%}$(my_pwd)$(git_prompt_info)$(git_prompt_status) %{$reset_color%} ' +function my_retval { + RETVAL=$? + + if [[ $RETVAL -eq 0 ]]; then + echo "$fg[yellow]" + else + echo "$fg[red]$RETVAL " + fi +} + +PROMPT='%{$(my_retval)%}$(my_pwd)$(git_prompt_info)$(git_prompt_status) %{$reset_color%} ' ZSH_THEME_GIT_PROMPT_PREFIX=" %{$fg[green]%}  " ZSH_THEME_GIT_PROMPT_SUFFIX=""