# $Id$
# VERSION 1.3 (2005-08-15)
#
# Coloured prompts for bash.
#
# Prompt is user@host:/dir$ where host is highlighted in green if the last 
# command exited 0 or in red (followed by the error code) otherwise.
#
# Override the PROMPT_OK_COLOUR and PROMPT_FAILED_COLOUR environment variables 
# to set different colours.
#
# To use, add a call to __ps1 in your .bash_profile.

export PROMPT_OK_COLOUR=${PROMPT_OK_COLOUR:-32}
export PROMPT_FAILED_COLOUR=${PROMPT_OK_COLOUR:-31}

function __ps1() {
  export PS1='\u@\[\033[1;$(__ps1_col $?)m\]\h\[\033[0m\]$(__ps1_ret $?):\w\$ '
  return 0
}

function __ps1_col() {
  [ $1 -gt 0 ] && echo -n "$PROMPT_FAILED_COLOUR" || echo -n "$PROMPT_OK_COLOUR"
  return $1
}

function __ps1_ret() {
  [ $1 -gt 0 ] && echo -n " ($1)"
  return 0
}