Como adicionar ramificação git colorida ao meu prompt bash?

Question

Eu encurtei meu prompt bash (usando o código na linha # 120 em este script pastebin ). Agora eu quero adicionar (colored) git branch ao prompt (abreviado), para que quando o branch tiver un-committed changes, ele seja mostrado em vermelho e se o diretório de trabalho estiver limpo (tenha nothingn to commit) então o branch é mostrado em verde.

Obrigado!

    

Answer 1

Este Github Gist por srguiwiz mostra um prompt com cores baseadas no status da ramificação:

# http://henrik.nyh.se/2008/12/git-dirty-prompt
# http://www.simplisticcomplexity.com/2008/03/13/show-your-git-branch-name-in-your-prompt/
#   username@Machine ~/dev/dir [master]$   # clean working directory green
#   username@Machine ~/dev/dir [master*]$  # dirty working directory red*
#
function git_branch {
  git branch --no-color 2> /dev/null | sed -e '/^[^*]/d' -e 's/* \(.*\)//'
}
# http://unix.stackexchange.com/questions/88307/escape-sequences-with-echo-e-in-different-shells
function markup_git_branch {
  if [[ "x$1" = "x" ]]; then
    echo -e "[$1]"
  else
    if [[ $(git status 2> /dev/null | tail -n1) = "nothing to commit, working directory clean" ]]; then
      echo -e '3[1;32m['"$1"']3[0;0m'
    else
      echo -e '3[1;31m['"$1"'*]3[0;0m'
    fi
  fi
}
export PS1='\u@\h \[3[0;34m\]\w\[3[0m\] $(markup_git_branch $(git_branch))$ '

Você só precisa usar $(markup_git_branch $(git_branch)) no seu prompt, onde quer que você queira mostrar o status e a ramificação do commit.

Há uma pequena alteração no código original aqui: o 3[0;30m no final deve ser 3[0;0m para redefinir a cor do prompt. O primeiro define preto, o último redefine a formatação no texto.

O código de srguiwiz parece ter alguns problemas, jcgoble3 fez uma versão melhor :

# Adds the current branch to the bash prompt when the working directory is
# part of a Git repository. Includes color-coding and indicators to quickly
# indicate the status of working directory.
#
# To use: Copy into ~/.bashrc and tweak if desired.
#
# Based upon the following gists:
# <https://gist.github.com/henrik/31631>
# <https://gist.github.com/srguiwiz/de87bf6355717f0eede5>
# Modified by me, using ideas from comments on those gists.
#
# License: MIT, unless the authors of those two gists object :)

git_branch() {
    # -- Finds and outputs the current branch name by parsing the list of
    #    all branches
    # -- Current branch is identified by an asterisk at the beginning
    # -- If not in a Git repository, error message goes to /dev/null and
    #    no output is produced
    git branch --no-color 2>/dev/null | sed -e '/^[^*]/d' -e 's/* \(.*\)//'
}

git_status() {
    # Outputs a series of indicators based on the status of the
    # working directory:
    # + changes are staged and ready to commit
    # ! unstaged changes are present
    # ? untracked files are present
    # S changes have been stashed
    # P local commits need to be pushed to the remote
    local status="$(git status --porcelain 2>/dev/null)"
    local output=''
    [[ -n $(egrep '^[MADRC]' <<<"$status") ]] && output="$output+"
    [[ -n $(egrep '^.[MD]' <<<"$status") ]] && output="$output!"
    [[ -n $(egrep '^\?\?' <<<"$status") ]] && output="$output?"
    [[ -n $(git stash list) ]] && output="${output}S"
    [[ -n $(git log --branches --not --remotes) ]] && output="${output}P"
    [[ -n $output ]] && output="|$output"  # separate from branch name
    echo $output
}

git_color() {
    # Receives output of git_status as argument; produces appropriate color
    # code based on status of working directory:
    # - White if everything is clean
    # - Green if all changes are staged
    # - Red if there are uncommitted changes with nothing staged
    # - Yellow if there are both staged and unstaged changes
    local staged=$([[ $1 =~ \+ ]] && echo yes)
    local dirty=$([[ $1 =~ [!\?] ]] && echo yes)
    if [[ -n $staged ]] && [[ -n $dirty ]]; then
        echo -e '3[1;33m'  # bold yellow
    elif [[ -n $staged ]]; then
        echo -e '3[1;32m'  # bold green
    elif [[ -n $dirty ]]; then
        echo -e '3[1;31m'  # bold red
    else
        echo -e '3[1;37m'  # bold white
    fi
}

git_prompt() {
    # First, get the branch name...
    local branch=$(git_branch)
    # Empty output? Then we're not in a Git repository, so bypass the rest
    # of the function, producing no output
    if [[ -n $branch ]]; then
        local state=$(git_status)
        local color=$(git_color $state)
        # Now output the actual code to insert the branch and status
        echo -e "\x01$color\x02[$branch$state]\x013[00m\x02"  # last bit resets color
    fi
}

# Sample prompt declaration based off of the default Ubuntu 14.04.1 color
# prompt. Tweak as you see fit, or just stick "$(git_prompt)" into your
# favorite prompt.
PS1='$debian_chroot\[3[01;32m\]\u@\h\[3[00m\]:\[3[01;34m\]\w$(git_prompt)\[3[00m\]\$ '