| # 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/* \(.*\)/\1/' | |
| } | |
| 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 | |
| # - Blue if there are unpushed commits | |
| local staged=$([[ $1 =~ \+ ]] && echo yes) | |
| local dirty=$([[ $1 =~ [!\?] ]] && echo yes) | |
| local needs_push=$([[ $1 =~ P ]] && echo yes) | |
| if [[ -n $staged ]] && [[ -n $dirty ]]; then | |
| echo -e '\033[1;33m' # bold yellow | |
| elif [[ -n $staged ]]; then | |
| echo -e '\033[1;32m' # bold green | |
| elif [[ -n $dirty ]]; then | |
| echo -e '\033[1;31m' # bold red | |
| elif [[ -n $needs_push ]]; then | |
| echo -e '\033[1;34m' # bold blue | |
| else | |
| echo -e '\033[1;37m' # bold white | |
| fi | |
| } | |
| virtualenv_info(){ | |
| # Get Virtual Env | |
| if [[ -n "$VIRTUAL_ENV" ]]; then | |
| # Strip out the path and just leave the env name | |
| venv=`basename $VIRTUAL_ENV` | |
| else | |
| # In case you don't have one activated | |
| venv='' | |
| fi | |
| [[ -n "$venv" ]] && echo "($venv) " | |
| } | |
| 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]\x01\033[00m\x02" # last bit resets color | |
| fi | |
| } | |
| # Sample prompt declaration. Tweak as you see fit, or just stick | |
| # "$(git_prompt)" into your favorite prompt. | |
| PS1='\033[31m$(virtualenv_info)\[\033[00m\u@\h:$(git_prompt)\[\033[00m\]\$ ' |
Place the file .gitprompt in your user directory. In your .base_profile add the line source ~/.gitprompt.
To see it in action immediately, then source your profile with . ~/.bash_profile.
You will now see your current, active virtualenv if any, and the current branch and status of your current git repo, if any.