# Environment variables:
#
# PS1_SHOWHOSTNAME
#       'auto' (default):       Only show hostname over SSH
#       'yes':                  Always show hostname
#       anything else:          Never show hostname


__show-hostname ()
{
    case "${PS1_SHOWHOSTNAME}" in
        (''|auto)
            [ "${SSH_CONNECTION}" ]
            return $?;;
        (yes)
            return 0;;
        (*)
            return 1;;
    esac
}

__set-title ()
{
    local title
    local git_root=$(git root 2> /dev/null)

    if [ ${git_root} ]
    then
        title=$(basename ${git_root})
    else
        local path=${PWD/~/\~}

        if __show-hostname
        then
            title="${USER}@${HOSTNAME}:${path}"
        else
            title="${USER}:${path}"
        fi
    fi

    echo -ne "\033]2;${title}\007"
}

# In order to know how much space PS1 takes, Bash needs us to delimit
# non-printing characters with \[ and \].
__start-nonprinting ()
{
    if [ ${BUILDING_PS1} ]
    then
        echo -en '\['
    fi
}

__end-nonprinting ()
{
    if [ ${BUILDING_PS1} ]
    then
        echo -en '\]'
    fi
}

__fontify ()
{
    local -A codes
    codes[red]='31'
    codes[green]='32'
    codes[blue]='34'
    codes[white]='37'
    codes[bold]='1'
    codes[dim]='2'
    codes[clear]='0'

    local text=$1
    shift

    output=$(__start-nonprinting)
    output+='\033['

    for ((i=1; i<=$#; i++))
    do
        output+=${codes[${!i}]}

        if ((i<$#))
        then
            output+=';'
        else
            output+='m'
        fi
    done

    output+=$(__end-nonprinting)

    output+="${text}"
    output+=$(__start-nonprinting)
    output+='\033[0m'
    output+=$(__end-nonprinting)

    echo -en "${output}"
}

__set-prompt ()
{
    BUILDING_PS1=t

    local last_status=$1

    PS1=''

    if [ ${last_status} -ne 0 ]
    then
        PS1+=$(__fontify "${last_status} " bold red) 
    fi

    PS1+=$(__fontify '\u' green)

    if __show-hostname
    then
        PS1+=$(__fontify '@\H' dim green)
    fi

    PS1+=$(__fontify : bold white)
    PS1+=$(__fontify '\w' bold blue)
    PS1+=$(__fontify "$(__git_ps1 '(%s)')" red)
    PS1+=$(__fontify '\$' bold white)
    PS1+=' '

    BUILDING_PS1=
}

__draw-rule ()
{
    rule=''
    for ((i=0; i<COLUMNS; i++))
    do
        rule+=—
    done
    __fontify ${rule} dim
}

__refresh-terminal ()
{
    local last_status=$?

    __set-title
    __set-prompt ${last_status}
    __draw-rule
}


. /usr/lib/git-core/git-sh-prompt
GIT_PS1_SHOWDIRTYSTATE=t
GIT_PS1_SHOWUPSTREAM=auto

export PROMPT_COMMAND=__refresh-terminal