prompt.sh

#!/bin/bash
BLUE='\[\033[0;34m\]'
BLUE_BOLD='\[\033[1;34m\]'
RED='\[\033[0;31m\]'
RED_BOLD='\[\033[1;31m\]'
GREEN='\[\033[0;32m\]'
GREEN_BOLD='\[\033[1;32m\]'
YELLOW='\[\033[0;33m\]'
YELLOW_BOLD='\[\033[1;33m\]'
BLUE='\[\033[0;34m\]'
BLUE_BOLD='\[\033[1;34m\]'
MAGENTA='\[\033[0;35m\]'
MAGENTA_BOLD='\[\033[1;35m\]'
CYAN='\[\033[0;36m\]'
CYAN_BOLD='\[\033[1;36m\]'
WHITE='\[\033[0;37m\]'
WHITE_BOLD='\[\033[1;37m\]'
GRAY='\[\033[1;30m\]'
NO_COLOUR='\[\033[0m\]'

MAIN_COLOR=$WHITE
SUB_COLOR=$RED
GIT_BR_COLOR=$YELLOW
GIT_REPO_COLOR=$CYAN

function get_git_reponame {
    echo $(basename $(git rev-parse --show-toplevel))
}


function get_git_branch {
    # Default (non-git) prompt
    venv=$(basename $VIRTUAL_ENV 2> /dev/null)
    if [[ -n $venv ]]; then
        venv_prompt="($venv)"
    else
        venv_prompt=""
    fi

    unstaged_changes=""
    staged_changes=""
    unchecked_changes=""
    semaphore=""
    GIT_PROMPT=""
    CURR_BRANCH=$(git rev-parse --abbrev-ref HEAD 2> /dev/null)
    if test $? -eq 0 ; then
        if [ "$CURR_BRANCH" == 'HEAD' ]; then
            CURR_BRANCH=$(git rev-parse HEAD)  # not on a branch, show commit sha instead
        fi
        rn=`get_git_reponame`
        unstaged_changes_format="${YELLOW}*"
        staged_changes_format="${GREEN}*"
        unchecked_changes_format="${RED}*"
        git diff --no-ext-diff --ignore-submodules --quiet --exit-code || unstaged_changes=$unstaged_changes_format
        git diff-index --cached --quiet --ignore-submodules HEAD 2> /dev/null
        (( $? && $? != 128 )) && staged_changes=$staged_changes_format
        if [[ -z $(git ls-files --other --exclude-standard 2> /dev/null) ]]; then
            unchecked_changes=""
        else
            unchecked_changes=$unchecked_changes_format
        fi

        semaphore=$staged_changes$unstaged_changes$unchecked_changes
        if [ "x"$rn == "x" ]; then
            GIT_PROMPT="${SUB_COLOR}:$GIT_BR_COLOR$CURR_BRANCH$semaphore"
        else
            GIT_PROMPT="${SUB_COLOR}:$GIT_REPO_COLOR$rn$SUB_COLOR:$GIT_BR_COLOR$CURR_BRANCH$semaphore"
        fi
    fi
    PS1="$venv_prompt${SUB_COLOR}[${MAIN_COLOR}\h${SUB_COLOR}${GIT_PROMPT}${SUB_COLOR}:${MAIN_COLOR}\W${SUB_COLOR}]${NO_COLOUR} "
}

export PROMPT_COMMAND=get_git_branch