From 84259876ebf817174de97330c9771e68e485ce8b Mon Sep 17 00:00:00 2001 From: "martin f. krafft" Date: Mon, 5 May 2008 23:56:04 +0100 Subject: [PATCH] do the git prompt thing proper --- .zsh/zshrc/85_git_prompt | 89 ++++++++++++++++++++++++++++++++++------ .zsh/zshrc/90_prompt | 4 +- 2 files changed, 78 insertions(+), 15 deletions(-) diff --git a/.zsh/zshrc/85_git_prompt b/.zsh/zshrc/85_git_prompt index 626b4c3..32d0e4f 100644 --- a/.zsh/zshrc/85_git_prompt +++ b/.zsh/zshrc/85_git_prompt @@ -7,36 +7,99 @@ # # Source repository: http://git.madduck.net/v/etc/zsh.git # -# Shamelessly adapted from http://www.jukie.net/~bart/conf/zsh.d/S55_git +# Shamelessly based on http://glandium.org/blog/?p=170 # -_get_git_cur_branch() { - git branch --no-color | sed -rne 's,^\* ,,p' +__git_get_repo_root() +{ + local relroot + relroot="$(git rev-parse --show-cdup 2>/dev/null)" || return 1 + if [ -n "$relroot" ]; then + readlink -f "$relroot" + else + echo $PWD + fi +} + +__git_get_branch() +{ + local ref + ref=$(git symbolic-ref -q HEAD 2>/dev/null \ + || git-name-rev --name-only HEAD 2>/dev/null) + echo "${ref#refs/heads/}" } -_is_git_repo() { - [ "$(git rev-parse --is-inside-work-tree 2>/dev/null)" = true ] +__get_root_offsets() +{ + local pwda reporoot loc + pwda=(${(s:/:)PWD}) + reporoot=(${(s:/:)1}) + echo $((1 - $#reporoot)) $(($#pwda - $#reporoot)) +} + +__get_prompt_path_components() +{ + local reporoot + reporoot="$1" + + set -- $(__get_root_offsets "$reporoot") + if [ "$2" -le 0 ]; then + echo %~ + else + echo "%${1}~" "%${2}~" + fi } -_set_git_psvar() { - if _is_git_repo; then - psvar[1]="$(_get_git_cur_branch)" +__vcs_get_repo_type() +{ + if __git_get_repo_root >/dev/null; then + echo git else - [ -n "$psvar[1]" ] && unset "psvar[1]" + echo NONE fi } +__vcs_set_prompt_variables() +{ + local pre branch post + local MAXLEN=25 + + case "${1:-$(__vcs_get_repo_type)}" 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}" + ;; + *) + 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 + # content, that's all we get for now + psvar=("${(%)p}") + return + esac + + psvar[1]="$pre" + psvar[2]="$branch" + psvar[3]="$post" +} + update_git_vars_if_git_ran() { - # if the last command included the word git, the branch may have changed. - # checkout should be alright, but you never know... + local vcs="$(__vcs_get_repo_type)" case "$(history $(($HISTCMD - 1)))" in - *git*) _set_git_psvar;; + # $vcs appeared in last command, so be sure to update + *${vcs}*) __vcs_set_prompt_variables esac } precmd_functions+=update_git_vars_if_git_ran update_git_vars() { - _set_git_psvar + __vcs_set_prompt_variables } chpwd_functions+=update_git_vars diff --git a/.zsh/zshrc/90_prompt b/.zsh/zshrc/90_prompt index 6d5f635..fcb97f4 100644 --- a/.zsh/zshrc/90_prompt +++ b/.zsh/zshrc/90_prompt @@ -9,8 +9,8 @@ # is_root && PS1_HL=U -PS1="%${PS1_HL:=B}%m%${(L)PS1_HL}:%25<..<%~%# " +PS1="%${PS1_HL:=B}%m%${(L)PS1_HL}:%1v%(2v.|%B%2v%b|.)%(3v.%3v.)%# " unset PS1_HL -RPS1="#%(0?..%?,)%(1v.%1v,.)%!" +RPS1="#%(0?..%?,)%!" # vim:ft=zsh -- 2.39.5