X-Git-Url: https://git.madduck.net/etc/zsh.git/blobdiff_plain/2b684e2957334aa808204f559ed6714bef03ae36..133e70749425901f59a1b8e15855e647848919b5:/.zsh/zshrc/85_vcs_prompt diff --git a/.zsh/zshrc/85_vcs_prompt b/.zsh/zshrc/85_vcs_prompt index c6b3c79..a99ca18 100644 --- a/.zsh/zshrc/85_vcs_prompt +++ b/.zsh/zshrc/85_vcs_prompt @@ -10,8 +10,9 @@ # Shamelessly based on http://glandium.org/blog/?p=170 # -__git_get_repo_root() +__git_get_reporoot() { + # return the full path to the root of the current git repository local relroot relroot="$(git rev-parse --show-cdup 2>/dev/null)" || return 1 if [ -n "$relroot" ]; then @@ -23,59 +24,143 @@ __git_get_repo_root() __git_get_branch() { + # return the name of the git branch we're on local ref ref=$(git symbolic-ref -q HEAD 2>/dev/null \ - || git-name-rev --name-only HEAD 2>/dev/null) + || git-name-rev --name-only HEAD 2>/dev/null) || return 1 echo "${ref#refs/heads/}" } -__get_root_offsets() +__hg_get_reporoot() { - local pwda reporoot loc - pwda=(${(s:/:)PWD}) - reporoot=(${(s:/:)1}) - echo $((1 - $#reporoot)) $(($#pwda - $#reporoot)) + hg root } -__get_prompt_path_components() +__hg_get_branch() +{ + echo "hg:$(hg branch)" +} + +__bzr_get_reporoot() { local reporoot - reporoot="$1" + reporoot="$(bzr info | sed -rne 's, *branch root: ,,p')" + case "$reporoot" in + .) echo "$PWD";; + *) echo "$reporoot";; + esac +} - set -- $(__get_root_offsets "$reporoot") - if [ "$2" -le 0 ]; then - echo %~ - else - echo "%${1}~" "%${2}~" - fi +__bzr_get_branch() +{ + local branch revno + bzr version-info | while read i j; do + case "$i" in + revno:) revno="$j";; + branch-nick:) branch="$j";; + esac + done + echo "bzr:${branch}@$revno" } __vcs_get_repo_type() { - if __git_get_repo_root >/dev/null; then - echo git - else - echo NONE + # return the type of the closest repository in the path hierarchy + local dir + while true; do + [ -d ${dir}.git ] && echo git && break + [ -d ${dir}.bzr ] && echo bzr && break + [ -d ${dir}.hg ] && echo hg && break + [ "$(readlink -f ${dir:-.})" = / ] && echo NONE && break + dir="../$dir" + done +} + +__vcs_get_prompt_path_components() +{ + # return formatted path components (prefix branch postfix) given + # the repository root and the branch. + + # shortcut: if there are no arguments, return a default prompt + if [ -z "${1:-}" ]; then + pwdnamed="%${_PROMPT_PATH_MAXLEN}<..<%~%<<" + pwdnamed="${(%)pwdnamed}" + echo "$pwdnamed" + return fi + + local reporoot branch + reporoot="${1%%/}" + branch="$2" + + # replace named directories in the PWD, we need thi for the proper component + # count later + local pwdnamed="%~" + pwdnamed="${(%)pwdnamed}" + + # store paths in arrays for component count calculation + typeset -la apwd apwdnamed areporoot + apwd=(${(s:/:)PWD}) + apwdnamed=(${(s:/:)pwdnamed}) + areporoot=(${(s:/:)reporoot}) + + # get the number of leading and trailing path components. Since we're using + # %~ later and then /home/madduck suddenly becomes ~, which is 1, not + # 2 components, we calculate the leading component count by using the named + # path and the number of post components + local precomps postcomps + postcomps=$(($#apwd - $#areporoot)) + precomps=$(($#apwdnamed - $postcomps)) + + local postfix + if (( $postcomps > 0 )); then + postfix="%${postcomps}~" + postfix="${(%)postfix}" + fi + + # we don't want the prompt to get too long, so keep the total prompt length + # under $_PROMPT_PATH_MAXLEN (25), but ensure that the prefix is not shorter + # than $_PROMPT_PATH_MINLEN (10), no matter what + local prelen minlen prefix + prelen=$((${_PROMPT_PATH_MAXLEN:-25} - $#branch - $#postfix)) + minlen=${_PROMPT_PATH_MINLEN:-10} + (( $prelen < $minlen )) && prelen=$minlen + prefix="%${prelen}<..<%-${precomps}~%<<" + prefix="${(%)prefix}" + + echo "$prefix" "$branch" "$postfix" } __vcs_set_prompt_variables() { - local pre branch post - local MAXLEN=25 + # set psvar[1..3] depending on repo type, or just psvar[1] if no repo found + local reporoot branch repotype + repotype="${1:-$(__vcs_get_repo_type)}" - case "${1:-$(__vcs_get_repo_type)}" in + case "$repotype" in git) - local reporoot="$(__git_get_repo_root)" - set -- $(__get_prompt_path_components "$reporoot") - branch="$(__git_get_branch)" - post="${(%)2}" - local prelen="$((MAXLEN - $#post - $#branch))" - [ $prelen -lt 10 ] && prelen=10 - pre="%${prelen}<..<${1}%<<" - pre="${(%)pre}" + reporoot="$(__git_get_reporoot)" || + { error "could not determine git repository root"; return 1 } + branch="$(__git_get_branch)" || + { error "could not determine git branch"; return 1 } + ;; + hg) + reporoot="$(__hg_get_reporoot)" || + { error "could not determine hg repository root"; return 1 } + branch="$(__hg_get_branch)" || + { error "could not determine hg branch"; return 1 } + ;; + bzr) + reporoot="$(__bzr_get_reporoot)" || + { error "could not determine bzr repository root"; return 1 } + branch="$(__bzr_get_branch)" || + { error "could not determine bzr branch"; return 1 } ;; *) + case "$repotype" in + NONE) :;; + *) warn "$repotype repositories not (yet) supported in the prompt";; + esac local p="%${MAXLEN}<..<%~%<<" #TODO find a better way so we don't have to nuke $psvar, but since the # %(nv.true.false) check for prompts checks element count, not @@ -84,9 +169,10 @@ __vcs_set_prompt_variables() return esac - psvar[1]="$pre" - psvar[2]="$branch" - psvar[3]="$post" + set -- $(__vcs_get_prompt_path_components "$reporoot" "$branch") + psvar[1]="$1" + psvar[2]="$2" + psvar[3]="$3" } if ! is_root; then @@ -96,10 +182,10 @@ if ! is_root; then local vcs="$(__vcs_get_repo_type)" case "$(history $(($HISTCMD - 1)))" in # $vcs appeared in last command, so be sure to update - *${vcs}*) __vcs_set_prompt_variables + *${vcs}*) __vcs_set_prompt_variables "$vcs" esac } - precmd_functions+=_update_vcs prompt_vars_if_vcs_ran + precmd_functions+=_update_vcs_prompt_vars_if_vcs_ran _update_vcs_prompt_vars() { __vcs_set_prompt_variables