helm/shell/zsh/prompt.zsh
2017-12-06 01:39:10 -05:00

74 lines
2.2 KiB
Bash

# Loosely based off Pure <https://github.com/sindresorhus/pure>
_strlen() { echo ${#${(S%%)1//$~%([BSUbfksu]|([FB]|){*})/}}; }
# fastest possible way to check if repo is dirty
prompt_git_dirty() {
_is_callable git || return
# check if we're in a git repo
[[ "$(command git rev-parse --is-inside-work-tree 2>/dev/null)" == "true" ]] || return
# check if it's dirty
command test -n "$(git status --porcelain --ignore-submodules -unormal)" || return
local r=$(command git rev-list --right-only --count HEAD...@'{u}' 2>/dev/null)
local l=$(command git rev-list --left-only --count HEAD...@'{u}' 2>/dev/null)
(( ${r:-0} > 0 )) && echo -n " %F{red}${r}-"
(( ${l:-0} > 0 )) && echo -n " %F{green}${l}+"
echo -n '%f'
}
## Hooks ###############################
prompt_hook_precmd() {
vcs_info # get git info
# Newline before prompt, except on init
[[ $PROMPT_DONE ]] && print ""; PROMPT_DONE=1
}
## Initialization ######################
prompt_init() {
# prevent the extra space in the rprompt
[[ $EMACS ]] || ZLE_RPROMPT_INDENT=0
# prevent percentage showing up
# if output doesn't end with a newline
export PROMPT_EOL_MARK=
# prompt_opts=(cr subst percent)
setopt promptsubst
autoload -Uz add-zsh-hook
autoload -Uz vcs_info
add-zsh-hook precmd prompt_hook_precmd
# Updates cursor shape and prompt symbol based on vim mode
zle-keymap-select() {
case $KEYMAP in
vicmd) PROMPT_SYMBOL="%F{magenta}« " ;;
main|viins) PROMPT_SYMBOL="%(?.%F{green}.%F{red})── " ;;
esac
zle reset-prompt
zle -R
}
zle -N zle-keymap-select
zle -A zle-keymap-select zle-line-init
zstyle ':vcs_info:*' enable git
zstyle ':vcs_info:*' use-simple true
zstyle ':vcs_info:*' max-exports 2
zstyle ':vcs_info:git*' formats ' %b'
zstyle ':vcs_info:git*' actionformats ' %b (%a)'
# show username@host if logged in through SSH
if [[ $SSH_CONNECTION ]]; then
prompt_username='%m '
if [[ $(whoami) != hlissner ]]; then
prompt_username="%n.$prompt_username"
fi
fi
RPROMPT='%F{blue}%~%F{magenta}${vcs_info_msg_0_}$(prompt_git_dirty)%f'
PROMPT='%F{blue}${prompt_username}%f${PROMPT_SYMBOL:-$ }- '
}
prompt_init "$@"