#!bash Coloured prompts. # # The prompt comprises multiple parts, some of which may be hidden by unsetting # shell variables or using the ``prompt'' function. # # The first part of the prompt is user@host where host is highlighted in # green if the last command exited 0 or in red otherwise. # This part will be shown only if __ps1_user is 1. By default it is 1. # The success and failure colours can be changed by modifying # $PROMPT_OK_COLOUR and $PROMPT_FAILURE_COLOUR respectively. # # The second part of the prompt is taken from git-completion.bashrc. # It is shown only if __ps1_git is 1. By default it is 0. # # The third part of the prompt is taken from p4-completion.bashrc. # It is shown only if __ps1_p4 is 1. By default it is 0. # # The fourth part of the prompt is taken from svn-completion.bashrc. # It is shown only if __ps1_svn is 1. By default it is 0. # # The fifth part of the prompt is the exit status of the last command. # This part will be shown only if __ps1_user is set and the exit status is # non-zero. # # The final part of the prompt is the (full) working directory and $ string. # # Colouring is performed by the __ps1_col() and __ps1_ret() functions. # We redirect stderr to /dev/null when calling these functions to prevent # bash complaining about not knowing them when you su to another user, # retaining PS1 but not the function definitions. # # Note that $? is passed as an argument to - and is returned from - all # functions. As $? is set following any shell activity it is only guaranteed # to represent the return code of the last command at the beginning of __ps1(). # By passing between subsequent functions we ensure that it is available for # __ps1_ret(). # # Pick a colour based on the terminal capabilities. # OK: dark green. # Failed: dark red. # Git: royal blue. # P4: yellow. # SVN: magenta. case $(tput colors) in 256) PROMPT_OK_COLOUR="1;38;5;34" PROMPT_FAILED_COLOUR="1;38;5;160" GIT_COLOUR="0;38;5;33" SVN_COLOUR="0;38;5;127" P4_COLOUR="0;38;5;142" ;; 88) PROMPT_OK_COLOUR="1;38;5;24" PROMPT_FAILED_COLOUR="1;38;5;48" GIT_COLOUR="0;38;5;23" SVN_COLOUR="0;38;5;49" P4_COLOUR="0;38;5;56" ;; *) PROMPT_OK_COLOUR="1;32" PROMPT_FAILED_COLOUR="1;31" GIT_COLOUR="0;36" SVN_COLOUR="0;35" P4_COLOUR="0;33" ;; esac function __ps1() { # Default __ps1_user to 1. [ -z "$__ps1_user" ] && __ps1_user=1 PS1='$(__ps1_user $? \u@)\[\033[$(__ps1_col $? 2>/dev/null)m\]$(__ps1_user $? \h)\[\033[$(__ps1_colour_escape $? $GIT_COLOUR)m\]$(__ps1_git $? 2>/dev/null)\[\033[0m\]\[\033[$(__ps1_colour_escape $? $P4_COLOUR)m\]$(__ps1_p4 $? 2>/dev/null)\[\033[0m\]\[\033[$(__ps1_colour_escape $? $SVN_COLOUR)m\]$(__ps1_svn $? 2>/dev/null)\[\033[0m\]$(__ps1_ret $? 2>/dev/null):\w\$ ' return 0 } # iTerm doesn't like it if you set bold and colour at the same time. function __ps1_colour_escape() { local ret=$1; shift local bold="${1%%;*}" local colour="${1#*;}" echo -en "${bold}m\033[$colour" return $ret } function __ps1_col() { local ret=$1; shift if [ $ret -gt 0 ]; then __ps1_colour_escape $ret "$PROMPT_FAILED_COLOUR" else __ps1_colour_escape $ret "$PROMPT_OK_COLOUR" fi return $ret } function __ps1_ret() { [ "$__ps1_user" = "1" ] || return $1 [ $1 -gt 0 ] && echo -n " ($1)" return $1 } function __ps1_user() { local ret=$1; shift [ "$__ps1_user" = "1" ] || return $ret echo -n ${1+"$@"} return $ret } function __ps1_git() { [ "$__ps1_git" = "1" ] || return $1 if [ "$__ps1_user" = "1" ]; then __git_ps1 ' %s' else __git_ps1 '%s' fi return $1 } function __ps1_p4() { [ "$__ps1_p4" = "1" ] || return $1 if [ "$__ps1_user" = "1" -o "$__ps1_git" = "1" ]; then __p4_ps1 ' %s' else __p4_ps1 '%s' fi return $1 } function __ps1_svn() { [ "$__ps1_svn" = "1" ] || return $1 if [ "$__ps1_user" = "1" -o "$__ps1_git" = "1" -o "$__ps1_p4" = "1" ]; then __svn_ps1 ' %s' else __svn_ps1 '%s' fi return $1 } function prompt() { local blurb="Usage: prompt hide|show " if [ $# -lt 2 ]; then echo >&2 "$blurb" return 1 fi action="$1" if [ ! "$action" = "hide" -a ! "$action" = "show" ]; then echo >&2 "$blurb" return 1 fi if [ "$action" = "hide" ]; then action=0 else action=1 fi what="$(echo $2 | env LANG= LC_ALL= LC_CTYPE= tr '[:upper:]' '[:lower:]')" eval __ps1_$what=$action } __ps1