在 Shell 提示符中显示 Git 分支名称的方法

git 的好处之一就是把代码的分支管理变成了一件极其便捷的事情，分支只保留差异，不用复制任何文件，不用连接网络，快速创建，用完即删。Git 分支与项目的复杂程度无关，不管你的项目多么复杂，创建 Git 分支永远都是瞬间的事情。同时，因为保留了父类分支的信息，所以分支的合并也变得异常简单。

当在一个项目中频繁使用多个分支时，可以使用 git status 命令查询自己现在正工作在哪个分支下面，不过难免有脑子发昏的时候，忘记自己在哪个分支下面，因而发生误操作之类的杯具。

那么把分支显示在 shell 提示符中无疑方便了很多，再也不需要频繁的使用 git status 命令了…

实现原理很简单，大体就是查询当前目录下面的 Git 分支名称，然后嵌入到 PS1 变量中。那么，Git 分支名称可以通过下面的脚本轻松的获得：

git branch --no-color 2> /dev/null | sed -e '/^[^*]/d' -e 's*/}"

            elif [[ $head != '' ]]; then

                git_branch=" → (detached)"

            else

                git_branch=" → (unknow)"

            fi

            return

        fi

        dir="../$dir"

    done

    git_branch=''

}

接下来，将这个函数加入到 PROMPT_COMMAND 中，保证 Bash 在创建 prompt 之前调用这个函数取得分支名称：

PROMPT_COMMAND="find_git_branch; $PROMPT_COMMAND"

最后只要重新定义 PS1 变量即可：

# Here is bash color codes you can use

  black=$'[e[1;30m]'

    red=$'[e[1;31m]'

  green=$'[e[1;32m]'

 yellow=$'[e[1;33m]'

   blue=$'[e[1;34m]'

magenta=$'[e[1;35m]'

   cyan=$'[e[1;36m]'

  white=$'[e[1;37m]'

 nORMal=$'[e[m]'

 

PS1="$white[$magentau$white@$greenh$white:$cyanw$yellow$git_branch$white]$ $normal"

以上的代码你可以放在 ~/.profile 或者 ~/.bash_profile 等文件中即可，我的系统是 Snow Leopard，PS1 定义在 /etc/bashrc 中，所以我直接修改的这个文件。

最终效果如下：

UPDATE ? 2010/06/23:

如果你安装了随 Git 附送的 git-completion.sh 子命令自动完成脚本，也可以使用该脚本提供的方法：

export PS1="[u@h W"'$(__git_ps1 " (%s)")'"]$ "

ubuntu 系统，请参考： /etc/bash_completion.d/git