mirror of
https://github.com/ohmyzsh/ohmyzsh.git
synced 2024-11-28 03:22:53 +08:00
139 lines
3.3 KiB
Bash
139 lines
3.3 KiB
Bash
# Prompt
|
|
#
|
|
# Below are the color init strings for the basic file types. A color init
|
|
# string consists of one or more of the following numeric codes:
|
|
# Attribute codes:
|
|
# 00=none 01=bold 04=underscore 05=blink 07=reverse 08=concealed
|
|
# Text color codes:
|
|
# 30=black 31=red 32=green 33=yellow 34=blue 35=magenta 36=cyan 37=white
|
|
# Background color codes:
|
|
# 40=black 41=red 42=green 43=yellow 44=blue 45=magenta 46=cyan 47=white
|
|
function precmd {
|
|
|
|
local TERMWIDTH
|
|
(( TERMWIDTH = ${COLUMNS} - 1 ))
|
|
|
|
|
|
###
|
|
# Truncate the path if it's too long.
|
|
|
|
PR_FILLBAR=""
|
|
PR_PWDLEN=""
|
|
|
|
local promptsize=${#${(%):---(%n@%M:%l)---()}}
|
|
local pwdsize=${#${(%):-%~}}
|
|
local gitbranch="$(git_prompt_info)"
|
|
local rvmprompt="$(rvm_prompt_info)"
|
|
local gitbranchsize=${#${gitbranch:-''}}
|
|
local rvmpromptsize=${#${rvmprompt:-''}}
|
|
|
|
if [[ "$promptsize + $pwdsize + $rvmpromptsize + $gitbranchsize" -gt $TERMWIDTH ]]; then
|
|
((PR_PWDLEN=$TERMWIDTH - $promptsize))
|
|
else
|
|
PR_FILLBAR="\${(l.(($TERMWIDTH - ($promptsize + $pwdsize + $rvmpromptsize + $gitbranchsize)))..${PR_SPACE}.)}"
|
|
fi
|
|
}
|
|
|
|
|
|
setopt extended_glob
|
|
|
|
preexec () {
|
|
if [[ "$TERM" == "screen" ]]; then
|
|
local CMD=${1[(wr)^(*=*|sudo|-*)]}
|
|
echo -n "\ek$CMD\e\\"
|
|
fi
|
|
|
|
if [[ "$TERM" == "xterm" ]]; then
|
|
print -Pn "\e]0;$1\a"
|
|
fi
|
|
|
|
if [[ "$TERM" == "rxvt" ]]; then
|
|
print -Pn "\e]0;$1\a"
|
|
fi
|
|
|
|
}
|
|
|
|
setprompt () {
|
|
###
|
|
# Need this so the prompt will work.
|
|
|
|
setopt prompt_subst
|
|
|
|
|
|
###
|
|
# See if we can use colors.
|
|
|
|
autoload zsh/terminfo
|
|
for color in RED GREEN YELLOW BLUE MAGENTA CYAN WHITE; do
|
|
eval PR_$color='%{$terminfo[bold]$fg[${(L)color}]%}'
|
|
eval PR_LIGHT_$color='%{$fg[${(L)color}]%}'
|
|
(( count = $count + 1 ))
|
|
done
|
|
PR_NO_COLOUR="%{$terminfo[sgr0]%}"
|
|
|
|
|
|
###
|
|
# See if we can use extended characters to look nicer.
|
|
|
|
typeset -A altchar
|
|
# set -A altchar "${(s..)terminfo[acsc]}"
|
|
PR_SET_CHARSET="%{$terminfo[enacs]%}"
|
|
PR_HBAR=${altchar[q]:--}
|
|
PR_ULCORNER=${altchar[l]:--}
|
|
PR_LLCORNER=${altchar[m]:--}
|
|
PR_LRCORNER=${altchar[j]:--}
|
|
PR_URCORNER=${altchar[k]:--}
|
|
|
|
###
|
|
# Modify Git prompt
|
|
ZSH_THEME_GIT_PROMPT_PREFIX=" ["
|
|
ZSH_THEME_GIT_PROMPT_SUFFIX="]"
|
|
###
|
|
# Modify RVM prompt
|
|
ZSH_THEME_RVM_PROMPT_PREFIX=" ["
|
|
ZSH_THEME_RVM_PROMPT_SUFFIX="]"
|
|
|
|
|
|
###
|
|
# Decide if we need to set titlebar text.
|
|
|
|
case $TERM in
|
|
xterm*|*rxvt*)
|
|
PR_TITLEBAR=$'%{\e]0;%(!.-=*[ROOT]*=- | .)%n@%M:%~ $(git_prompt_info) $(rvm_prompt_info) | ${COLUMNS}x${LINES} | %y\a%}'
|
|
;;
|
|
screen)
|
|
PR_TITLEBAR=$'%{\e_screen \005 (\005t) | %(!.-=[ROOT]=- | .)%n@%m:%~ | ${COLUMNS}x${LINES} | %y\e\\%}'
|
|
;;
|
|
*)
|
|
PR_TITLEBAR=''
|
|
;;
|
|
esac
|
|
|
|
|
|
###
|
|
# Decide whether to set a screen title
|
|
if [[ "$TERM" == "screen" ]]; then
|
|
PR_STITLE=$'%{\ekzsh\e\\%}'
|
|
else
|
|
PR_STITLE=''
|
|
fi
|
|
|
|
###
|
|
# Finally, the prompt.
|
|
#
|
|
PROMPT='$PR_SET_CHARSET$PR_STITLE${(e)PR_TITLEBAR}\
|
|
$PR_RED$PR_HBAR<\
|
|
$PR_BLUE%(!.$PR_RED%SROOT%s.%n)$PR_GREEN@$PR_BLUE%M:$PR_GREEN%$PR_PWDLEN<...<%~$PR_CYAN$(git_prompt_info)$(rvm_prompt_info)\
|
|
$PR_RED>$PR_HBAR$PR_SPACE${(e)PR_FILLBAR}\
|
|
$PR_RED$PR_HBAR<\
|
|
$PR_GREEN%l$PR_RED>$PR_HBAR\
|
|
|
|
$PR_RED$PR_HBAR<\
|
|
%(?..$PR_LIGHT_RED%?$PR_BLUE:)\
|
|
$PR_LIGHT_BLUE%(!.$PR_RED.$PR_WHITE)%#$PR_RED>$PR_HBAR\
|
|
$PR_NO_COLOUR '
|
|
|
|
}
|
|
|
|
setprompt
|