#!/bin/env bash

#
# Bash Prompt, Classic Edition
#

# Shorten the pwd in the prompt to show only the n number of characters
# NOTE: Bash 4+ only
PROMPT_DIRTRIM=2

# Ascii color escape codes
# Note that 256 color foreground codes follow the following format:
CSI="\[\e["
RED="${CSI}0;31m\]"
YELLOW="${CSI}0;33m\]"
CYAN="${CSI}0;36m\]"
COLOR_NONE="${CSI}0m\]"
INDIGO="${CSI}38;2;90;86;224m\]"

# Show Git branch in prompt
git_completion=/usr/local/etc/bash_completion.d/git-completion.bash
if [ -f $git_completion ]; then
    export GIT_PS1_SHOWDIRTYSTATE=true
    export GIT_PS1_SHOWUNTRACKEDFILES=true
    export GIT_PS1_SHOWSTASHSTATE=true
fi

# Format the prompt
function prompt_func() {

    # Show Git branch in prompt
    if [ -f $git_completion ]; then
        git_branch=$(__git_ps1 " (%s)")
    fi

    # Python: Show virtualenv in prompt
    if [ -z "$VIRTUAL_ENV" ]; then
        virtualenv=""
    else
        virtualenv="($(basename "$VIRTUAL_ENV")) "
    fi

    if [[ -z $DEMO_PROMPT ]]; then
        PS1="${YELLOW}${virtualenv}${CYAN}\h:${RED}\w${CYAN} \u${YELLOW}${git_branch}${RED} \$${COLOR_NONE} "
    else
        PS1="${INDIGO}>${COLOR_NONE} "
    fi
}

# Format the prompt
PROMPT_COMMAND=prompt_func