diff --git a/.zshrc b/.zshrc index 9a3dd59..f0f5970 100644 --- a/.zshrc +++ b/.zshrc @@ -19,5 +19,21 @@ export skip_global_compinit=1 source $ZDOTDIR/modules/completion.zsh # Prompt -fpath=($ZDOTDIR/modules $fpath) -autoload -Uz prompt_init; prompt_init +prompt_builder() { + local EXITCODE=$? + if [ $UID = 0 ] && [ $EXITCODE = 0 ]; then + PROMPT=$'%F{white}%~ %B%F{red}#%F{blue}>%f%b ' + RPROMPT='%F{white}%? %F{red}%n' + elif [ $UID = 0 ]; then + PROMPT=$'%F{white}%~ %B%F{red}#%F{red}>%f%b ' + RPROMPT=$'%F{red}%? %n' + elif [ $UID != 0 ] && [ $EXITCODE = 0 ]; then + PROMPT=$'%F{white}%~ %B%F{blue}>%f%b ' + RPROMPT='%F{white}%? %F{white}%n' + else + PROMPT=$'%F{white}%~ %B%F{red}>%f%b ' + RPROMPT=$'%F{red}%? %F{white}%n' + fi +} +precmd_functions+=(prompt_builder) +