zsh prompt

git_branch() {
  local branch=$(git rev-parse --abbrev-ref HEAD 2>/dev/null)
  if [[ -n "$branch" ]]; then
    echo "${git_color}${branch}${reset} "
  fi
}

git_color=$(tput setaf 2) # green
dir_color=$(tput setaf 6) # cyan
reset=$(tput sgr0)

if [[ $UID -eq 0 ]]; then
  PROMPT=' %n %1~ $(git_branch)💀 ' # root user
else
  PROMPT=' %n ${dir_color}%1~${reset} $(git_branch)👺 ' # normal user
fi