diff --git a/plugins/mercurial/mercurial.plugin.zsh b/plugins/mercurial/mercurial.plugin.zsh index 062baaa34..32060a47d 100644 --- a/plugins/mercurial/mercurial.plugin.zsh +++ b/plugins/mercurial/mercurial.plugin.zsh @@ -17,4 +17,14 @@ function hg_current_branch() { if [ -d .hg ]; then echo $(hg branch) fi +} + +parse_hg_dirty() { + hg status 2> /dev/null \ + | awk '$1 == "?" { unknown = 1 } + $1 != "?" { changed = 1 } + END { + if (changed) printf "$ZSH_THEME_GIT_PROMPT_DIRTY" + else if (unknown) printf "$ZSH_THEME_GIT_PROMPT_CLEAN" + }' } \ No newline at end of file