diff --git a/themes/powerline-multiline/powerline-multiline.theme.bash b/themes/powerline-multiline/powerline-multiline.theme.bash index 232a2a8e..6940b9ee 100644 --- a/themes/powerline-multiline/powerline-multiline.theme.bash +++ b/themes/powerline-multiline/powerline-multiline.theme.bash @@ -55,8 +55,7 @@ function set_rgb_color { function powerline_shell_prompt { SHELL_PROMPT_COLOR=${SHELL_THEME_PROMPT_COLOR} - CAN_I_RUN_SUDO=$(sudo -n uptime 2>&1 | grep "load" | wc -l) - if [ ${CAN_I_RUN_SUDO} -gt 0 ]; then + if sudo -n uptime 2>&1 | grep -q "load"; then SHELL_PROMPT_COLOR=${SHELL_THEME_PROMPT_COLOR_SUDO} fi SEGMENT_AT_RIGHT=0 diff --git a/themes/powerline/powerline.theme.bash b/themes/powerline/powerline.theme.bash index 1bc4eca9..944414db 100644 --- a/themes/powerline/powerline.theme.bash +++ b/themes/powerline/powerline.theme.bash @@ -4,7 +4,7 @@ THEME_PROMPT_SEPARATOR="" SHELL_SSH_CHAR=" " SHELL_THEME_PROMPT_COLOR=32 -SHELL_SSH_THEME_PROMPT_COLOR=208 +SHELL_THEME_PROMPT_COLOR_SUDO=202 VIRTUALENV_CHAR="ⓔ " VIRTUALENV_THEME_PROMPT_COLOR=35 @@ -37,13 +37,17 @@ function set_rgb_color { } function powerline_shell_prompt { - if [[ -n "${SSH_CLIENT}" ]]; then - SHELL_PROMPT="${bold_white}$(set_rgb_color - ${SHELL_SSH_THEME_PROMPT_COLOR}) ${SHELL_SSH_CHAR}\u@\h ${normal}" - LAST_THEME_COLOR=${SHELL_SSH_THEME_PROMPT_COLOR} - else - SHELL_PROMPT="${bold_white}$(set_rgb_color - ${SHELL_THEME_PROMPT_COLOR}) \u ${normal}" - LAST_THEME_COLOR=${SHELL_THEME_PROMPT_COLOR} + SHELL_PROMPT_COLOR=${SHELL_THEME_PROMPT_COLOR} + if sudo -n uptime 2>&1 | grep -q "load"; then + SHELL_PROMPT_COLOR=${SHELL_THEME_PROMPT_COLOR_SUDO} fi + if [[ -n "${SSH_CLIENT}" ]]; then + SHELL_PROMPT="${SHELL_SSH_CHAR}\u@\h" + else + SHELL_PROMPT="\u" + fi + SHELL_PROMPT="${bold_white}$(set_rgb_color - ${SHELL_PROMPT_COLOR}) ${SHELL_PROMPT} ${normal}" + LAST_THEME_COLOR=${SHELL_PROMPT_COLOR} } function powerline_virtualenv_prompt {