#!/usr/bin/env zsh

if [[ "$DISPLAY" == ":0.0" ]]
then
  HOSTCOLOR=$YELLOW
else
  HOSTCOLOR=$BLUE
fi

case $UID in
0)
#    PS1="[%h] (%B%m%b:%U%~%u)%# "
    USERCOLOR=$RED
    HISTFILE=/root/.zhistory
    ;;
*)
#    PS1="[%h] (%B%n@%m%b)%U%~%u%# "
    USERCOLOR=$GREEN
    HISTFILE=$HOME/.zhistory
    ;;
esac

PS1="(${USERCOLOR}%n${NORM}@${HOSTCOLOR}%m${NORM})%U%~%u%# "

function precmd
{
    if [ $? -ne 0 ]
    then
        RPROMPT="[${RED}%T${NORM}]"
    else
        RPROMPT="[${GREEN}%T${NORM}]"
    fi
}