1 files changed, 33 insertions, 0 deletions
diff --git a/dot-oh-my-zsh/custom/themes/spaceship-prompt/sections/hg_branch.zsh b/dot-oh-my-zsh/custom/themes/spaceship-prompt/sections/hg_branch.zsh
new file mode 100644
index 00000000..941337fe
--- /dev/null
+++ b/dot-oh-my-zsh/custom/themes/spaceship-prompt/sections/hg_branch.zsh
@@ -0,0 +1,33 @@
+#
+# Mercurial (hg) branch
+#
+# Show current Mercurial branch
+
+# ------------------------------------------------------------------------------
+# Configuration
+# ------------------------------------------------------------------------------
+
+SPACESHIP_HG_BRANCH_SHOW="${SPACESHIP_HG_BRANCH_SHOW=true}"
+SPACESHIP_HG_BRANCH_PREFIX="${SPACESHIP_HG_BRANCH_PREFIX="$SPACESHIP_HG_SYMBOL"}"
+SPACESHIP_HG_BRANCH_SUFFIX="${SPACESHIP_HG_BRANCH_SUFFIX=""}"
+SPACESHIP_HG_BRANCH_COLOR="${SPACESHIP_HG_BRANCH_COLOR="magenta"}"
+
+# ------------------------------------------------------------------------------
+# Section
+# ------------------------------------------------------------------------------
+
+spaceship_hg_branch() {
+  [[ $SPACESHIP_HG_BRANCH_SHOW == false ]] && return
+
+  spaceship::is_hg || return
+
+  local hg_info=$(hg log -r . --template '{activebookmark}')
+
+  if [[ -z $hg_info ]]; then
+    hg_info=$(hg branch)
+  fi
+
+  spaceship::section \
+    "$SPACESHIP_HG_BRANCH_COLOR" \
+    "$SPACESHIP_HG_BRANCH_PREFIX"$hg_info"$SPACESHIP_HG_BRANCH_SUFFIX"
+}