git-prompt.sh

# don't set prompt if this is not interactive shell
# it is better if this test is done before git-prompt.sh is source-ed for performance reasons. 
[[ $- != *i* ]]  &&  return

###################################################################   CONFIG

	#####  read config file if any.

	unset 	dir_color rc_color root_id_color init_vcs_color clean_vcs_color modified_vcs_color added_vcs_color mixed_vcs_color untracked_vcs_color op_vcs_color detached_vcs_color

	conf=git-prompt.conf; 			[[ -r $conf ]]  && . $conf
	conf=/etc/git-prompt.conf; 		[[ -r $conf ]]  && . $conf
	conf=~/.git-prompt.conf;  		[[ -r $conf ]]  && . $conf
	unset conf

	#####  set defaults if not set

	git_module=${git_module:-on}
	svn_module=${svn_module:-off}
	hg_module=${hg_module:-on}
	vim_module=${vim_module:-on}
	error_bell=${error_bell:-off}


	#### dir, rc, root color 
	if [ 0`tput colors` -ge 8 ];  then				#  if terminal supports colors
		dir_color=${dir_color:-CYAN}
		rc_color=${rc_color:-red}
		root_id_color=${root_id_color:-magenta}
	else											#  only B/W
		dir_color=${dir_color:-bw_bold}
		rc_color=${rc_color:-bw_bold}
	fi

	#### vcs state colors
		 init_vcs_color=${init_vcs_color:-WHITE}		# initial
		clean_vcs_color=${clean_vcs_color:-blue}		# nothing to commit (working directory clean)
	     modified_vcs_color=${modified_vcs_color:-red}	# Changed but not updated:
		added_vcs_color=${added_vcs_color:-green}	# Changes to be committed:
		mixed_vcs_color=${mixed_vcs_color:-yellow}
	    untracked_vcs_color=${untracked_vcs_color:-BLUE}	# Untracked files:
		   op_vcs_color=${op_vcs_color:-MAGENTA}
	     detached_vcs_color=${detached_vcs_color:-RED}

	max_file_list_length=${max_file_list_length:-100}

   truncate_pwd=${truncate_pwd:-off}
   max_pwd_length=${max_pwd_length:-30}
   min_chars_per_pwd=${min_chars_per_pwd:-1}

   upcase_hostname=${upcase_hostname:-on}


#####################################################################  post config

	################# make PARSE_VCS_STATUS

	[[ $git_module = "on" ]]   &&   type git >&/dev/null   &&   PARSE_VCS_STATUS="parse_git_status"
	[[ $svn_module = "on" ]]   &&   type svn >&/dev/null   &&   PARSE_VCS_STATUS+="||parse_svn_status"
	[[ $hg_module  = "on" ]]   &&   type hg  >&/dev/null   &&   PARSE_VCS_STATUS+="||parse_hg_status"
	                                                            PARSE_VCS_STATUS+="||return"
	################# terminfo colors-16
	#
	#  	black?    0 8			  
	#	red       1 9
	#	green     2 10
	#	yellow    3 11
	#	blue      4 12
	#	magenta   5 13
	#	cyan      6 14
	#	white     7 15
	#
	#	terminfo setaf/setab - sets ansi foreground/background
	#	terminfo sgr0 - resets all attributes
	#	terminfo colors - number of colors
	#
	#################  Colors-256
	#  To use foreground and background colors from the extension, you only
	#  have to remember two escape codes:
	#       Set the foreground color to index N:    \033[38;5;${N}m
	#       Set the background color to index M:    \033[48;5;${M}m
	# To make vim aware of a present 256 color extension, you can either set
	# the $TERM environment variable to xterm-256color or use vim's -T option
	# to set the terminal. I'm using an alias in my bashrc to do this. At the
	# moment I only know of two color schemes which is made for multi-color
	# terminals like urxvt (88 colors) or xterm: inkpot and desert256, 

	### if term support colors,  then use color prompt, else bold

              black='\['`tput sgr0; tput setaf 0`'\]'
                red='\['`tput sgr0; tput setaf 1`'\]'
              green='\['`tput sgr0; tput setaf 2`'\]'
             yellow='\['`tput sgr0; tput setaf 3`'\]'
               blue='\['`tput sgr0; tput setaf 4`'\]'
            magenta='\['`tput sgr0; tput setaf 5`'\]'
               cyan='\['`tput sgr0; tput setaf 6`'\]'
              white='\['`tput sgr0; tput setaf 7`'\]'

              BLACK='\['`tput setaf 0; tput bold`'\]'
                RED='\['`tput setaf 1; tput bold`'\]'
              GREEN='\['`tput setaf 2; tput bold`'\]'
             YELLOW='\['`tput setaf 3; tput bold`'\]'
               BLUE='\['`tput setaf 4; tput bold`'\]'
            MAGENTA='\['`tput setaf 5; tput bold`'\]'
               CYAN='\['`tput setaf 6; tput bold`'\]'  # why 14 doesn't work?
              WHITE='\['`tput setaf 7; tput bold`'\]'

            bw_bold='\['`tput bold`'\]'

	on=''
	off=': '

	bell="\[`eval ${!error_bell} tput bel`\]"

       colors_reset='\['`tput sgr0`'\]'

	# Workaround for UTF readline(?) bug. Disable bell when UTF
	#locale |grep -qi UTF && bell=''	


	# replace symbolic colors names to raw treminfo strings
		 init_vcs_color=${!init_vcs_color}
	     modified_vcs_color=${!modified_vcs_color}
	    untracked_vcs_color=${!untracked_vcs_color}
		clean_vcs_color=${!clean_vcs_color}
		added_vcs_color=${!added_vcs_color}
		   op_vcs_color=${!op_vcs_color}
		mixed_vcs_color=${!mixed_vcs_color}
	     detached_vcs_color=${!detached_vcs_color}

	##################################################################### 
	# if label non empty, append 1 space
	label=${1:+$1 }


	unset PROMPT_COMMAND

	#######  work around for MC bug
	if [ -z "$TERM" -o "$TERM" = "dumb" -o -n "$MC_SID" ]; then
		unset PROMPT_COMMAND
		PS1='\w> '
		return 0
	fi

	export who_where

truncate_working_directory() {
  pwd=`echo $PWD | sed "s:^${HOME}:~:"`
  [[ $truncate_pwd != "on" ]] && return
  chars_per_dir=5
  while [[ $((chars_per_dir--)) -gt $((min_chars_per_pwd)) && `echo ${pwd} | wc -m` -gt $((max_pwd_length)) ]]; do
     pwd=`echo ${pwd} | sed "s:[^\/~]*\(/.\{${chars_per_dir}\}\):\1:g"`
  done
  unset chars_per_dir
}

set_shell_title() { 

	xterm_title() { echo  -n "]2;${@}" ; }   # FIXME: replace hardcodes with terminfo codes

	screen_title() { 
		# FIXME: run this only if screen is in xterm (how to test for this?)
		xterm_title  "sCRn  $label$plain_who_where $@" 

		# FIXME $STY not inherited though "su -"
		[ "$STY" ] && screen -S $STY -X title "$*"
	}

	case $TERM in

		screen*)                                                    
			screen_title "$*"
			;;

		xterm* | rxvt* | gnome-terminal | konsole | eterm | wterm )       
			# is there a capability which we can to test 
			# for "set term title-bar" and its escapes?
			#echo -n "]2;$label$plain_who_where $1"
			xterm_title  "$label$plain_who_where $@"
			;;

		*)                                                     
			;;
	esac
 }

    export -f set_shell_title

###################################################### ID (user name)
	id=`id -un`
	id=${id#$default_user}

########################################################### TTY
	tty=`tty`
	tty=`echo $tty | sed "s:/dev/pts/:p:; s:/dev/tty::" `		# RH tty devs	
	tty=`echo $tty | sed "s:/dev/vc/:vc:" `				# gentoo tty devs

	if [[ "$TERM" = "screen" ]] ;  then

		#	[ "$WINDOW" = "" ] && WINDOW="?"
		#	
		#		# if under screen then make tty name look like s1-p2
		#		# tty="${WINDOW:+s}$WINDOW${WINDOW:+-}$tty"
		#	tty="${WINDOW:+s}$WINDOW"  # replace tty name with screen number
		tty="$WINDOW"  # replace tty name with screen number
	fi

	# we don't need tty name under X11
	case $TERM in
		xterm* | rxvt* | gnome-terminal | konsole | eterm | wterm )  unset tty ;;
		*);;
	esac

	dir_color=${!dir_color}
	rc_color=${!rc_color}
	root_id_color=${!root_id_color}

	########################################################### HOST
	### we don't display home host/domain  $SSH_* set by SSHD or keychain

	# How to find out if session is local or remote? Working with "su -", ssh-agent, and so on ? 
	## is sshd our parent?
	# if 	{ for ((pid=$$; $pid != 1 ; pid=`ps h -o pid --ppid $pid`)); do ps h -o command -p $pid; done | grep -q sshd && echo == REMOTE ==; }
	#then 
	host=${HOSTNAME}
	#host=`hostname --short`
    
    
	[[ $upcase_hostname = "on" ]] && host=`echo ${host%$default_host} | tr a-z A-Z`
	#host=`echo ${host} | tr a-z A-Z`
	
    host_color=${host}_host_color
    host_color=${!host_color}
        
    if [[ -z $host_color && -x /usr/bin/cksum ]] ;  then 
		cksum_color_no=`echo $host | cksum  | awk '{print $1%7}'`
		color_index=(green yellow blue magenta cyan white)		# FIXME:  bw,  color-256
		host_color=${color_index[cksum_color_no]}
	fi

	host_color=${!host_color}

	# we already should have short host name, but just in case
	host=${host%.$localdomain}
	host=${host%.$default_domain}


#################################################################### WHO_WHERE 
	#  [[user@]host[-tty]]
	
	if [[ -n $id  || -n $host ]] ;   then 
		[[ -n $id  &&  -n $host ]]  &&  at='@'  || at=''
		color_who_where="${id:+$host_color$id}$at${host:+$host_color$host}${tty:+ $tty}"
		plain_who_where="${id}$at$host"

		# add trailing " "
		color_who_where="$color_who_where "
		plain_who_where="$plain_who_where "
		
		# if root then make it root_color
		if	[ "$id" == "root" ]  ; then 
		    color_who_where="$root_id_color$color_who_where$colors_reset"
		fi
	else
		    color_who_where=''
	fi

parse_svn_status() {

        [[   -d .svn  ]] || return 1

        vcs=svn

        ### get rev
        eval `
            svn info |
                sed -n "                                                                                                                          
                    s@^URL[^/]*//@repo_dir=@p
                    s/^Revision: /rev=/p
                "
        `
        ### get status

        unset status modified added clean init added mixed untracked op detached
        eval `svn status 2>/dev/null |
                sed -n '
                    s/^A      \([^.].*\)/modified=modified;		modified_files[${#modified_files[@]}+1]=\"\1\";/p
                    s/^M      \([^.].*\)/modified=modified;		modified_files[${#modified_files[@]}+1]=\"\1\";/p
                    s/^\?      \([^.].*\)/untracked=untracked;	untracked_files[${#untracked_files[@]}+1]=\"\1\";/p
                ' 
        `
        # TODO branch detection if standard repo layout

        [[  -z $modified ]]  && [[ -z $untracked ]] && clean=clean
        vcs_info=svn:r$rev
 }
        
parse_hg_status() {

  [[  -d ./.hg/ ]]  ||  return  1

  vcs=hg

  ### get status
  unset status modified added clean init added mixed untracked op detached

  eval `hg status 2>/dev/null |
    sed -n '
      s/^M \([^.].*\)/modified=modified; modified_files[${#modified_files[@]}+1]=\"\1\";/p
      s/^A \([^.].*\)/added=added; added_files[${#added_files[@]}+1]=\"\1\";/p
      s/^R \([^.].*\)/added=added;/p
      s/^! \([^.].*\)/modified=modified;/p
      s/^? \([^.].*\)/untracked=untracked; untracked_files[${#untracked_files[@]}+1]=\\"\1\\";/p
    '`

  branch=`hg branch 2> /dev/null`

  [[ -z $modified ]] && [[ -z $untracked ]] && [[ -z $added ]] && clean=clean
  vcs_info=${branch/default/D}
 }

parse_git_status() {

        git_dir=`[[ $git_module = "on" ]]  &&  git rev-parse --git-dir 2> /dev/null`
        #git_dir=`eval \$$git_module  git rev-parse --git-dir 2> /dev/null`
        #git_dir=` git rev-parse --git-dir 2> /dev/null`

        [[  -n ${git_dir/./} ]]   ||   return  1

        vcs=git

        ##########################################################   GIT STATUS
        unset status modified added clean init added mixed untracked op detached
        eval `
                git status 2>/dev/null |
                    sed -n '
                        s/^# On branch /branch=/p
                        s/^nothing to commit (working directory clean)/clean=clean/p
                        s/^# Initial commit/init=init/p

			s/^#	\.\./: SKIP/

                        /^# Untracked files:/,/^[^#]/{
                            s/^# Untracked files:/untracked=untracked;/p
                            s/^#	\(.*\)/untracked_files[${#untracked_files[@]}+1]=\\"\1\\"/p   
                        }

                        /^# Changed but not updated:/,/^# [A-Z]/ {
                            s/^# Changed but not updated:/modified=modified;/p
                            s/^#	modified:   \(.*\)/modified_files[${#modified_files[@]}+1]=\"\1\"/p
                            s/^#	unmerged:   \(.*\)/modified_files[${#modified_files[@]}+1]=\"\1\"/p
                        }

                        /^# Changes to be committed:/,/^# [A-Z]/ {
                            s/^# Changes to be committed:/added=added;/p
                            s/^#	modified:   \(.*\)/added_files[${#added_files[@]}+1]=\"\1\"/p
                            s/^#	new file:   \(.*\)/added_files[${#added_files[@]}+1]=\"\1\"/p
                            s/^#	renamed:[^>]*> \(.*\)/added_files[${#added_files[@]}+1]=\"\1\"/p
                            s/^#	copied:[^>]*> \(.*\)/added_files[${#added_files[@]}+1]=\"\1\"/p
                        }
                    ' 
        `

        if  ! grep -q "^ref:" $git_dir/HEAD  2>/dev/null;   then 
            detached=detached
        fi


        #################  GET GIT OP 

        unset op
        
        if [[ -d "$git_dir/.dotest" ]] ;  then

            if [[ -f "$git_dir/.dotest/rebasing" ]] ;  then
                op="rebase"

            elif [[ -f "$git_dir/.dotest/applying" ]] ; then
                op="am"

            else
                op="am/rebase"

            fi

        elif  [[ -f "$git_dir/.dotest-merge/interactive" ]] ;  then
            op="rebase -i"
            # ??? branch="$(cat "$git_dir/.dotest-merge/head-name")"

        elif  [[ -d "$git_dir/.dotest-merge" ]] ;  then
            op="rebase -m"
            # ??? branch="$(cat "$git_dir/.dotest-merge/head-name")"
            
        # lvv: not always works. Should  ./.dotest  be used instead?
        elif  [[ -f "$git_dir/MERGE_HEAD" ]] ;  then
            op="merge"
            # ??? branch="$(git symbolic-ref HEAD 2>/dev/null)"
            
        else
            [[  -f "$git_dir/BISECT_LOG"  ]]   &&  op="bisect"
            # ??? branch="$(git symbolic-ref HEAD 2>/dev/null)" || \
            #    branch="$(git describe --exact-match HEAD 2>/dev/null)" || \
            #    branch="$(cut -c1-7 "$git_dir/HEAD")..."
        fi


        ####  GET GIT HEX-REVISION

        rawhex=`git rev-parse HEAD 2>/dev/null`
        rawhex=${rawhex/HEAD/}
        rawhex=${rawhex:0:6}
        
        ### branch

        branch=${branch/master/M}

                        # another method of above:
                        # branch=$(git symbolic-ref -q HEAD || { echo -n "detached:" ; git name-rev --name-only HEAD 2>/dev/null; } )
                        # branch=${branch#refs/heads/}

        ### compose vcs_info

        if [[ $init ]];  then 
            vcs_info=M$white=init

        else
            if [[ "$detached" ]] ;  then     
                branch="<detached:`git name-rev --name-only HEAD 2>/dev/null`"


            elif   [[ "$op" ]];  then
                    branch="$op:$branch"
                    if [[ "$op" == "merge" ]] ;  then     
                        branch+="<--$(git name-rev --name-only $(<$git_dir/MERGE_HEAD))"
                    fi
                    #branch="<$branch>"
            fi
            vcs_info="$branch$white=$rawhex"

        fi
 }


parse_vcs_status() {

        unset   file_list modified_files untracked_files added_files 
        unset   vcs vcs_info
        unset   status modified untracked added init detached
        unset   file_list modified_files untracked_files added_files 

        eval '[[ $HOME != $PWD ]]&&'  $PARSE_VCS_STATUS

     
        ### status:  choose primary (for branch color)
        unset status
        status=${op:+op}
        status=${status:-$detached}
        status=${status:-$clean}
        status=${status:-$modified}
        status=${status:-$added}
        status=${status:-$untracked}
        status=${status:-$init}
                                # at least one should be set
                                : ${status?prompt internal error: git status}
        eval vcs_color="\${${status}_vcs_color}"
                                # no def:  vcs_color=${vcs_color:-$WHITE}    # default 


        ### VIM  ( not yet works for multiple files )
        
	if  [[ $vim_module = "on" ]] ;  then
		unset vim_glob vim_file vim_files
		old_nullglob=`shopt -p nullglob`
		    shopt -s nullglob
		    vim_glob=`echo .*.swp`
		eval $old_nullglob

		if [[ $vim_glob ]];  then  
		    vim_file=${vim_glob#.}
		    vim_file=${vim_file%.swp}
		    # if swap is newer,  then unsaved vim session
		    [[ .${vim_file}.swp -nt $vim_file ]]  && vim_files=$vim_file
		fi
	fi


        ### file list
        unset file_list
        [[ ${added_files[1]}     ]]  &&  file_list+=" "$added_vcs_color${added_files[@]}
        [[ ${modified_files[1]}  ]]  &&  file_list+=" "$modified_vcs_color${modified_files[@]}
        [[ ${untracked_files[1]} ]]  &&  file_list+=" "$untracked_vcs_color${untracked_files[@]}
        [[ ${vim_files}          ]]  &&  file_list+=" "${RED}VIM:${vim_files}
        file_list=${file_list:+:$file_list}

	if [[ ${#file_list} -gt $max_file_list_length ]]  ;  then
		file_list=${file_list:0:$max_file_list_length} 	
		if [[ $max_file_list_length -gt 0 ]]  ;  then
			file_list="${file_list% *} ..."
		fi
	fi


        head_local="(${vcs_info}$vcs_color${file_list}$vcs_color)"

        ### fringes (added depended on location)
        head_local="${head_local+$vcs_color$head_local }"
        #above_local="${head_local+$vcs_color$head_local\n}"
        #tail_local="${tail_local+$vcs_color $tail_local}${dir_color}"
 }


 trap - DEBUG  >& /dev/null
 trap '[[ $BASH_COMMAND != prompt_command_function ]] && set_shell_title $BASH_COMMAND' DEBUG  >& /dev/null

###################################################################### PROMPT_COMMAND

prompt_command_function() {
	rc="$?"

	if [[ "$rc" == "0" ]]; then 
		rc=""
	else
		rc="$rc_color$rc$colors_reset$bell "
	fi

	set_shell_title "$PWD/" 
	parse_vcs_status
	truncate_working_directory
	PS1="$colors_reset$rc$head_local$label$color_who_where$dir_color$pwd$tail_local$dir_color$colors_reset\$ $colors_reset"
	
	unset head_local tail_local pwd
 }
    
    PROMPT_COMMAND=prompt_command_function

    unset rc id tty modified_files file_list  

# vim: set syntax=sh: