1 files changed, 90 insertions, 0 deletions
diff --git a/segments/vcs_branch.sh b/segments/vcs_branch.sh
new file mode 100755
index 0000000..f8d973d
--- /dev/null
+++ b/segments/vcs_branch.sh
@@ -0,0 +1,90 @@
+# Prints current branch in a VCS directory if it could be detected.
+
+# Source lib to get the function get_tmux_pwd
+source "${TMUX_POWERLINE_DIR_LIB}/tmux_adapter.sh"
+
+branch_symbol="⭠"
+git_colour="5"
+svn_colour="220"
+hg_colour="45"
+
+
+run_segment() {
+	tmux_path=$(get_tmux_cwd)
+	cd "$tmux_path"
+	branch=""
+	if [ -n "${git_branch=$(__parse_git_branch)}" ]; then
+		branch="$git_branch"
+	elif [ -n "${svn_branch=$(__parse_svn_branch)}" ]; then
+		branch="$svn_branch"
+	elif [ -n "${hg_branch=$(__parse_hg_branch)}" ]; then
+		branch="$hg_branch"
+	fi
+
+	if [ -n "$branch" ]; then
+		echo "${branch}"
+	fi
+	return 0
+}
+
+
+# Show git banch.
+__parse_git_branch() {
+	type git >/dev/null 2>&1
+	if [ "$?" -ne 0 ]; then
+		return
+	fi
+
+	#git branch 2> /dev/null | sed -e '/^[^*]/d' -e 's/* \(.*\)/ \[\1\]/'
+
+	# Quit if this is not a Git repo.
+	branch=$(git symbolic-ref HEAD 2> /dev/null)
+	if [[ -z $branch ]] ; then
+		# attempt to get short-sha-name
+		branch=":$(git rev-parse --short HEAD 2> /dev/null)"
+	fi
+	if [ "$?" -ne 0 ]; then
+		# this must not be a git repo
+		return
+	fi
+
+	# Clean off unnecessary information.
+	branch=${branch##*/}
+
+	echo  -n "#[fg=colour${git_colour}]${branch_symbol} #[fg=colour${TMUX_POWERLINE_CUR_SEGMENT_FG}]${branch}"
+}
+
+# Show SVN branch.
+__parse_svn_branch() {
+	type svn >/dev/null 2>&1
+	if [ "$?" -ne 0 ]; then
+		return
+	fi
+
+	local svn_info=$(svn info 2>/dev/null)
+	if [ -z "${svn_info}" ]; then
+		return
+	fi
+
+
+	local svn_root=$(echo "${svn_info}" | sed -ne 's#^Repository Root: ##p')
+	local svn_url=$(echo "${svn_info}" | sed -ne 's#^URL: ##p')
+
+	local branch=$(echo "${svn_url}" | egrep -o '[^/]+$')
+	echo "#[fg=colour${svn_colour}]${branch_symbol} #[fg=colour${TMUX_POWERLINE_CUR_SEGMENT_FG}]${branch}"
+}
+
+__parse_hg_branch() {
+	type hg >/dev/null 2>&1
+	if [ "$?" -ne 0 ]; then
+		return
+	fi
+
+	summary=$(hg summary)
+	if [ "$?" -ne 0 ]; then
+		return
+	fi
+
+	local branch=$(echo "$summary" | grep 'branch:' | cut -d ' ' -f2)
+	echo  "#[fg=colour${hg_colour}]${branch_symbol} #[fg=colour${TMUX_POWERLINE_CUR_SEGMENT_FG}]${branch}"
+}