tmux/dot-config/tmux-powerline/segments/vcs_branch.sh

124 lines
3 KiB
Bash
Executable file

# Prints current branch in a VCS directory if it could be detected.
# This is a patched copy of the vcs_branch segment that ships with
# tmux-powerline. It adds __normalize_color calls so that the segment works
# with true-colour themes.
# Source lib to get the function get_tmux_pwd
source "${TMUX_POWERLINE_DIR_LIB}/tmux_adapter.sh"
TMUX_POWERLINE_SEG_VCS_BRANCH_MAX_LEN_DEFAULT=24
branch_symbol=""
git_colour="5"
svn_colour="220"
hg_colour="45"
generate_segmentrc() {
read -d '' rccontents << EORC
# Max length of the branch name.
export TMUX_POWERLINE_SEG_VCS_BRANCH_MAX_LEN="${TMUX_POWERLINE_SEG_VCS_BRANCH_MAX_LEN_DEFAULT}"
EORC
echo "$rccontents"
}
run_segment() {
__process_settings
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#refs\/heads\/}
branch=$(__truncate_branch_name $branch)
echo -n "#[fg=colour${git_colour}]${branch_symbol} #[fg=$(__normalize_color "$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}" | grep -E -o '[^/]+$')
branch=$(__truncate_branch_name $branch)
echo "#[fg=colour${svn_colour}]${branch_symbol} #[fg=$(__normalize_color "$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)
branch=$(__truncate_branch_name $branch)
echo "#[fg=colour${hg_colour}]${branch_symbol} #[fg=$(__normalize_color "$TMUX_POWERLINE_CUR_SEGMENT_FG")]${branch}"
}
__truncate_branch_name() {
trunc_symbol="…"
branch=$(echo $1 | sed "s/\(.\{$TMUX_POWERLINE_SEG_VCS_BRANCH_MAX_LEN\}\).*/\1$trunc_symbol/")
echo -n $branch
}
__process_settings() {
if [ -z "$TMUX_POWERLINE_SEG_VCS_BRANCH_MAX_LEN" ]; then
export TMUX_POWERLINE_SEG_VCS_BRANCH_MAX_LEN="${TMUX_POWERLINE_SEG_VCS_BRANCH_MAX_LEN_DEFAULT}"
fi
}