bash_prompt

#!/bin/bash
#
# PS1 magic
#
# Mostly copied from YUVAL KOGMAN version, added my own __git_ps1 stuff
# Original: http://gist.github.com/621452
#
# See video demo of this at http://vimeo.com/15789794
#
# To enable save as .bash_prompt in $HOME and add to .bashrc:
#
# . ~/.bash_prompt
#
# Pedro Melo, <melo@simplicidade.org>

_update_prompt () {
  ## Save $? early, we'll need it later
  local exit="$?"  
  
  ## define some colors
  local red="31";
  local green="32";
  local yellow="33";
  local purple="35";
  local cyan="36";
  local white="37";

  local pre="\[\e[";
  local suf="\]";

  local e_green="${pre}0;${green}m$suf";
  local e_purple="${pre}0;${purple}m$suf";
  local e_cyan="${pre}0;${cyan}m$suf";
  local e_white="${pre}0;${white}m$suf";
  local e_bred="$pre$red;1m$suf";
  local e_byellow="$pre$yellow;1m$suf";

  local e_normal="\[\e[0;0m\]"

  ## Initial prompt
  _prompt="[$e_white\h$e_normal:$e_cyan\w$e_normal]";

  ## Color based on exit code
  local lookofd="\340\262\240_\340\262\240"; # look of disapproval
  local lookofa="\312\230\342\200\277\312\230"; # look of approval
  case "$exit" in
    "0" )  ex="$e_green$lookofa$e_normal " ;;
    *   )  ex="$e_bred$lookofd$e_normal "  ;;
  esac

  ## Color current user
  local u;
  local p;
  if [ "$UID" = "0" ]; then
    u="$e_bred\u$e_normal";
    p="$e_bred#$e_normal";
  else
    u="$e_purple\u$e_normal";
    p="\$";
  fi

  ## Color git status if any
  branch=`__git_ps1 " (%s)"`
  if [ -n "$branch" ] ; then
    local is_big_repo=`git config bash.big-repo`
    if [ -z "$is_big_repo" ] ; then
      ## Assumes that untracked files are always listed after modified ones
      ## True for all git versions I could find
      git status --porcelain | perl -ne 'exit(1) if /^ /; exit(2) if /^[?]/'
      case "$?" in
        "0" )  branch="$e_green$branch$e_normal"   ;; 
        "1" )  branch="$e_bred$branch$e_normal"    ;; 
        "2" )  branch="$e_byellow$branch$e_normal" ;; 
        "130") git config bash.big-repo 1           ;;
      esac
    fi
  fi
  
  export PS1="$_prompt $u$branch $ex$p ";
}

if [ -n "$PS1" ] ; then
  PROMPT_COMMAND='_update_prompt'
  export PROMPT_COMMAND
fi

# helper commands to explicitly change the setting:

dumb_prompt () {
  git config bash.big-repo 1
}

smart_prompt () {
  git config --unset bash.big-repo
}