X-Git-Url: https://git.madduck.net/etc/zsh.git/blobdiff_plain/2f58f5d883f04e62b6d8869c0c3a14ba785371df..a5062beb39198fd202dec7b8ca7b2a3ed2d81c27:/.zsh/zshrc/85_vcs_prompt?ds=sidebyside

diff --git a/.zsh/zshrc/85_vcs_prompt b/.zsh/zshrc/85_vcs_prompt
index e96b772..0c9db46 100644
--- a/.zsh/zshrc/85_vcs_prompt
+++ b/.zsh/zshrc/85_vcs_prompt
@@ -31,6 +31,38 @@ __git_get_branch()
   echo "${ref#refs/heads/}"
 }
 
+__hg_get_reporoot()
+{
+  hg root
+}
+
+__hg_get_branch()
+{
+  echo "hg:$(hg branch)"
+}
+
+__bzr_get_reporoot()
+{
+  local reporoot
+  reporoot="$(bzr info | sed -rne 's, *branch root: ,,p')"
+  case "$reporoot" in
+    .) echo "$PWD";;
+    *) echo "$reporoot";;
+  esac
+}
+
+__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()
 {
   # return the type of the closest repository in the path hierarchy
@@ -115,6 +147,18 @@ __vcs_set_prompt_variables()
       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) :;;