快捷導(dǎo)航

在 Shell 提示符中顯示 Git 分支名稱的方法

更新時(shí)間：2015年04月03日 09:48:25 投稿：junjie

這篇文章主要介紹了在 Shell 提示符中顯示 Git 分支名稱的方法,本文詳細(xì)的講解了解決這個(gè)需求的過程,需要的朋友可以參考下

Git 的好處之一就是把代碼的分支管理變成了一件極其便捷的事情，分支只保留差異，不用復(fù)制任何文件，不用連接網(wǎng)絡(luò)，快速創(chuàng)建，用完即刪。Git 分支與項(xiàng)目的復(fù)雜程度無關(guān)，不管你的項(xiàng)目多么復(fù)雜，創(chuàng)建 Git 分支永遠(yuǎn)都是瞬間的事情。同時(shí)，因?yàn)楸Ａ袅烁割惙种У男畔?，所以分支的合并也變得異常簡單?/p>

當(dāng)在一個(gè)項(xiàng)目中頻繁使用多個(gè)分支時(shí)，可以使用 git status 命令查詢自己現(xiàn)在正工作在哪個(gè)分支下面，不過難免有腦子發(fā)昏的時(shí)候，忘記自己在哪個(gè)分支下面，因而發(fā)生誤操作之類的杯具。

那么把分支顯示在 Shell 提示符中無疑方便了很多，再也不需要頻繁的使用 git status 命令了…

實(shí)現(xiàn)原理很簡單，大體就是查詢當(dāng)前目錄下面的 Git 分支名稱，然后嵌入到 PS1 變量中。那么，Git 分支名稱可以通過下面的腳本輕松的獲得：

復(fù)制代碼代碼如下:

git branch --no-color 2> /dev/null | sed -e '/^[^*]/d' -e 's/* $.*$/(\1)/'

把上面的腳本封裝到函數(shù)中，修改 PS1 變量，嵌入函數(shù)… 大體是這樣。但是這樣也意味著一個(gè)問題，就是每次 shell 活動（比如切換目錄，甚至只是敲下回車）都會執(zhí)行一次 git … sed 命令，這樣每次都啟動2個(gè)進(jìn)程，實(shí)在是有些不爽。

好在，可以使用另外一種方式來獲取 Git 分支名稱，在每個(gè) Git 項(xiàng)目中，都有一個(gè) .git 目錄，這個(gè)目錄下面有個(gè)叫做 HEAD 的文件，里面包含的當(dāng)前分支的路徑信息：

復(fù)制代碼代碼如下:

ref: refs/heads/BRANCH-NAME

我們只要讀取這個(gè)文件，然后再和對應(yīng)的路徑互相匹配一下就知道正確地分支名稱了。不要只是簡單的從 HEAD 內(nèi)容中拆分出最后的 BRANCH-NAME，因?yàn)樗灰欢ㄊ钦_地。

以下是 Aaron Crane 的實(shí)現(xiàn)方式：

復(fù)制代碼代碼如下:

## Parses out the branch name from .git/HEAD:

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=" → (unknow)"

            fi

            return

        fi

        dir="../$dir"

    done

    git_branch=''

}

接下來，將這個(gè)函數(shù)加入到 PROMPT_COMMAND 中，保證 Bash 在創(chuàng)建 prompt 之前調(diào)用這個(gè)函數(shù)取得分支名稱：

復(fù)制代碼代碼如下:

PROMPT_COMMAND="find_git_branch; $PROMPT_COMMAND"

最后只要重新定義 PS1 變量即可：

復(fù)制代碼代碼如下:

# 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[$magenta\u$white@$green\h$white:$cyan\w$yellow\$git_branch$white]\$ $normal"

以上的代碼你可以放在 ~/.profile 或者 ~/.bash_profile 等文件中即可，我的系統(tǒng)是 Snow Leopard，PS1 定義在 /etc/bashrc 中，所以我直接修改的這個(gè)文件。

最終效果如下：