From cc8e82941800a8ebb8a59ab22987a41c5aaa9c3e Mon Sep 17 00:00:00 2001
From: Yaroslav de la Peña Smirnov <yps@yaroslavps.com>
Date: Fri, 7 Apr 2023 20:27:28 +0300
Subject: Use zsh config with root user as well

---
 dotfiles/.config/zsh/.zshrc | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

(limited to 'dotfiles')

diff --git a/dotfiles/.config/zsh/.zshrc b/dotfiles/.config/zsh/.zshrc
index aa1bc69..99ab1d1 100644
--- a/dotfiles/.config/zsh/.zshrc
+++ b/dotfiles/.config/zsh/.zshrc
@@ -6,8 +6,14 @@ precmd_vcs_info() { vcs_info }
 precmd_functions+=( precmd_vcs_info )
 setopt prompt_subst
 
-PROMPT=$'\n''%{%}%B%~ ${reset_color}%F{yellow}${vcs_info_msg_0_}%f'$'\n''%F{green}→%f '
-RPROMPT='%{%} %n%f@%F{red}%m%f'
+if [ "$USER" = "root" ]; then
+	# Make it so that we display root's user@host on the left and colored red so
+	# that it is quite clear that we are under root.
+	PROMPT=$'\n''%{%}%B%~ ${reset_color}%F{yellow}${vcs_info_msg_0_}%f'$'\n''%F{red}%n%f@%F{red}%m%f %F{green}→%f '
+else
+	PROMPT=$'\n''%{%}%B%~ ${reset_color}%F{yellow}${vcs_info_msg_0_}%f'$'\n''%F{green}→%f '
+	RPROMPT='%{%} %n%f@%F{red}%m%f'
+fi
 zstyle ':vcs_info:git:*' formats ' %b'
 
 # Basic auto/tab complete:
-- 
cgit v1.2.3