workspace prompt for bash

gistfile1.sh

BLUE="34;1m"
ORANGE="33m"
GREEN="32m"
RED="31m"
ROOT_COLOR="31;4m"
ENV_COLOR="$GREEN" # set from puppet or chef, should be RED for prod

function _colorify() {
	echo -en "\[\e[$1\]$2\[\e[0m\]"
}

function _user_color() {
	if [[ "$UID" -eq 0 ]]; then
		echo -en "$(_colorify $ROOT_COLOR "$1")"
	else
		echo -n "$1"
	fi
}

function _prompt_command() {
        ret=$?
        PS1="${_PS1}"
        if [[ $ret -ne 0 ]]; then
                PS1="$(_colorify "$ORANGE" "[$ret]"):$PS1"
        fi
}

_PS1="$(_user_color "\u")@$(_colorify $ENV_COLOR "\h"):$(_colorify "$BLUE" "\w")$ "
PROMPT_COMMAND="_prompt_command"