如何實(shí)現(xiàn)在Shell提示符中顯示Git分支名稱

本篇內(nèi)容介紹了“如何實(shí)現(xiàn)在Shell提示符中顯示Git分支名稱”的有關(guān)知識(shí)，在實(shí)際案例的操作過(guò)程中，不少人都會(huì)遇到這樣的困境，接下來(lái)就讓小編帶領(lǐng)大家學(xué)習(xí)一下如何處理這些情況吧！希望大家仔細(xì)閱讀，能夠?qū)W有所成！

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

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

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

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

代碼如下:

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

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

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

代碼如下:

ref: refs/heads/BRANCH-NAME

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

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

代碼如下:

## 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=''
}

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

代碼如下:

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

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

最終效果如下：

UPDATE – 2010/06/23:

如果你安裝了隨 Git 附送的 git-completion.sh 子命令自動(dòng)完成腳本，也可以使用該腳本提供的方法：

代碼如下:

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

“如何實(shí)現(xiàn)在Shell提示符中顯示Git分支名稱”的內(nèi)容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業(yè)相關(guān)的知識(shí)可以關(guān)注億速云網(wǎng)站，小編將為大家輸出更多高質(zhì)量的實(shí)用文章！