p4prompt.sh #2

#!/bin/bash

# Copyright (c) Perforce Software, Inc., 1997-2009. All rights reserved  
# Redistribution and use in source and binary forms, with or without  
# modification, are permitted provided that the following conditions are  
# met:
#   1. Redistributions of source code must retain the above copyright  
# notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright  
# notice, this list of conditions and the following disclaimer in the  
# documentation and/or other materials provided with the distribution.
#   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS  
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT  
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR  
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL PERFORCE  
# SOFTWARE, INC. BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,  
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT  
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,  
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY  
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT  
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE  
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

 
# User contributed content on the Perforce Public Depot is not supported  
# by Perforce, although it may be supported by its author. This applies  
# to all contributions even those submitted by Perforce employees.


function get_p4set
{
    echo `p4 set $1 | sed -e "s/$1=//" -e 's/ (.*)$//' 2>&1`;
    return 0;
}

function p4prompt
{
    local NOCOLOR="\[\033[0m\]"
    local  YELLOW="\e[0;33m"
    local    GRAY="\[\033[0;37m\]"
    local    CYAN="\[\033[0;36m\]"

    nP4CLIENT=`get_p4set P4CLIENT`
    nP4PORT=`get_p4set P4PORT`

    if [ ${#nP4CLIENT} -eq 0 ]
    then
	nP4CLIENT="__client_unset__"
    fi

    if [ ${#nP4PORT} -eq 0 ]
    then
	# "p4" actually defaults to using "perforce:1666", so the
	# P4PORT is never really "unset".
	nP4PORT="__server_unset__"
    fi

    PS1="$GRAY[$CYAN\${nP4CLIENT}$GRAY@$YELLOW\${nP4PORT}$GRAY] \
\${PWD} $NOCOLOR->   "
}

PROMPT_COMMAND=p4prompt