diff --git a/plugins/mercurial/mercurial.plugin.zsh b/plugins/mercurial/mercurial.plugin.zsh index 8cda356af..ce419b052 100644 --- a/plugins/mercurial/mercurial.plugin.zsh +++ b/plugins/mercurial/mercurial.plugin.zsh @@ -18,6 +18,12 @@ alias hgca='hg commit --amend' # list unresolved files (since hg does not list unmerged files in the status command) alias hgun='hg resolve --list' +# Configuration + +# Comment this line if you don't want to show if the branch is dirty or not. +# This can save a lot of time of very large repositories. +SHOW_DIRTY_BRANCH=1 + function hg_get_dir() { # Defines path as current directory local current_dir=$PWD @@ -38,13 +44,23 @@ function in_hg() { fi } +function hg_get_branch_id() { + local hg_id_tip=`hg log -T "{node|short}" -l 1 -b .` + local hg_id=`hg id -i -r .` + + if [[ $hg_id != $hg_id_tip ]]; then + echo "@${hg_id}" + fi +} + function hg_get_branch_name() { local hg_dir=$(hg_get_dir) if [[ $hg_dir != "" ]]; then + local hg_id=$(hg_get_branch_id) if [[ -f "${hg_dir}/branch" ]]; then - echo $(<"${hg_dir}/branch") + echo $(<"${hg_dir}/branch")$hg_id else - echo "default" + echo "default${hg_id}" fi fi } @@ -52,10 +68,11 @@ function hg_get_branch_name() { function hg_prompt_info { local branch=$(hg_get_branch_name) if [[ $branch != "" ]]; then - echo "$ZSH_PROMPT_BASE_COLOR$ZSH_THEME_HG_PROMPT_PREFIX\ -$ZSH_THEME_REPO_NAME_COLOR${display}$ZSH_PROMPT_BASE_COLOR\ -$ZSH_PROMPT_BASE_COLOR$(hg_dirty)$ZSH_THEME_HG_PROMPT_SUFFIX\ -$ZSH_PROMPT_BASE_COLOR" + if [[ $SHOW_DIRTY_BRANCH == 1 ]]; then + branch="${branch}$(hg_dirty)" + fi + echo "$ZSH_THEME_HG_PROMPT_PREFIX${branch}$ZSH_THEME_HG_PROMPT_SUFFIX" + fi } function hg_dirty_choose {