centos ssh终端下高亮显示git分支名

ssh终端显示git分支名：

#set git branch
green=$'e[0;32m'
magenta=$'e[0;36m'
normal_colours=$'e[m'

function find_git_branch {
    local dir=. head
    until [ "$dir" -ef / ]; do
        if [ -f "$dir/.git/HEAD" ]; then
            head=$(< "$dir/.git/HEAD")
            if [[ $head == ref: refs/heads/* ]]; then
                git_branch="(${head#*/*/})#"
            elif [[ $head != '' ]]; then
                git_branch='(detached)'
            else
                git_branch='(unknown)'
            fi
            return
        fi
        dir="../$dir"
    done
    git_branch='#'
}

PROMPT_COMMAND="find_git_branch; $PROMPT_COMMAND"
PS1="[$green][u@h W][$magenta]$git_branch[$green]\[$normal_colours] "

将上面这段，追加 /etc/profile 或 ~/.bash_profie 或 ~/.bashrc 任何一个的最后，然后source /etc/profile (或source ~/.bash_profile 或 source ~/.bashrc)，让其生效即可。

最终效果：