#!/bin/bash
# Git status in git projects....
# in your .bashrc file:
#   if [ -f ~/.prompt ]; then
#      . ~/.prompt
#   fi
        RED="\[\033[0;31m\]"
     YELLOW="\[\033[0;33m\]"
  GREEN="\[\033[0;32m\]"
       BLUE="\[\033[0;34m\]"
  LIGHT_RED="\[\033[1;31m\]"
LIGHT_GREEN="\[\033[1;32m\]"
      WHITE="\[\033[1;37m\]"
 LIGHT_GRAY="\[\033[0;37m\]"
 COLOR_NONE="\[\e[0m\]"

DIRTY_BRANCH="⚡"
RETURN="➔"
AHEAD_ORIGIN="↑"
BEHIND_ORIGIN="↓"
DIVERGE_ORIGIN="↕"

function parse_git_branch () {
  git rev-parse --git-dir &> /dev/null
  git_status="$(git status 2> /dev/null)"
  branch_pattern="^# On branch ([^${IFS}]*)"
  remote_pattern="# Your branch is (.*) of"
  diverge_pattern="# Your branch and (.*) have diverged"
  if [[ ! ${git_status}} =~ "working directory clean" ]]; then
    state="${RED}${DIRTY_BRANCH}"
  fi
  # add an else if or two here if you want to get more specific
  if [[ ${git_status} =~ ${remote_pattern} ]]; then
    if [[ ${BASH_REMATCH[1]} == "ahead" ]]; then
      remote="${YELLOW}${AHEAD_ORIGIN}"
    else
      remote="${YELLOW}${BEHIND_ORIGIN}"
    fi
  fi
  if [[ ${git_status} =~ ${diverge_pattern} ]]; then
    remote="${YELLOW}${DIVERGE_ORIGIN}"
  fi
  if [[ ${git_status} =~ ${branch_pattern} ]]; then
    branch=${BASH_REMATCH[1]}
    echo " (${branch})${remote}${state}"
  fi
}
function prompt_func () {
    case $TERM in
     xterm*|rxvt*)
         local TITLEBAR='\[\033]0;\u@\h: \w\007\]'
          ;;
     *)
         local TITLEBAR=""
          ;;
    esac
    previous_return_value=$?;
    prompt="${LIGHT_GRAY}\u@\h: [${BLUE}\W ${GREEN}$(parse_git_branch)${LIGHT_GRAY}]${COLOR_NONE} "
    if test $previous_return_value -eq 0
    then
        PS1="$TITLEBAR${prompt}${RETURN} "
    else
        PS1="$TITLEBAR${prompt}${RED}${RETURN}${COLOR_NONE} "
    fi
}
PROMPT_COMMAND=prompt_func