浏览代码

feat(mlh): add separate prompt symbol for root user (#11451)

Bartek Pacia 2 年之前
父节点
当前提交
7de55844b2
共有 1 个文件被更改,包括 9 次插入1 次删除
  1. 9 1
      themes/mlh.zsh-theme

+ 9 - 1
themes/mlh.zsh-theme

@@ -47,6 +47,10 @@ if [ -z "$MLH_SHELL_SYMBOL" ]; then
   MLH_SHELL_SYMBOL="$ "
   MLH_SHELL_SYMBOL="$ "
 fi
 fi
 
 
+if [ -z "$MLH_SHELL_SYMBOL_ROOT" ]; then
+  MLH_SHELL_SYMBOL_ROOT="# "
+fi
+
 # colors
 # colors
 USER_COLOR="%F{001}"
 USER_COLOR="%F{001}"
 DEVICE_COLOR="%F{033}"
 DEVICE_COLOR="%F{033}"
@@ -83,7 +87,11 @@ exit_code() {
 }
 }
 
 
 prompt_end() {
 prompt_end() {
-  printf "\n$MLH_SHELL_SYMBOL"
+  if [ "$UID" -eq 0 ]; then
+    printf "\n$MLH_SHELL_SYMBOL_ROOT"
+  else
+    printf "\n$MLH_SHELL_SYMBOL"
+  fi
 }
 }
 
 
 # Set git_prompt_info text
 # Set git_prompt_info text