Show username and host if in sudo

.oh-my-zsh/themes/oke.zsh-theme

@@ -3,13 +3,10 @@ PROMPT=""
 
 PROMPT+="%(?:%{$fg_bold[green]%}➜ :%{$fg_bold[red]%}➜ )"
 
-if [[ "$(env | grep SSH_)" != "" ]]; then
-    PROMPT+="%{$fg[grey]%}%m:"
+if [[ "$(env | grep SSH_)" != "" || "$(env | grep SUDO_)" != "" ]]; then
+    PROMPT+="%{$FG[007]%}%n@%m:"
 fi
 
-
-
-
 PROMPT+='%{$fg[cyan]%}%c%{$reset_color%}$(git_prompt_info) '
 
 ZSH_THEME_GIT_PROMPT_PREFIX="%{$fg_bold[blue]%}(%{$fg[red]%}"