Git branch with status color in Bash prompt

.bashrc

# Git branch status
# Modified from https://coderwall.com/p/pn8f0g/show-your-git-status-and-branch\-in-color-at-the-command-prompt
function git_color {
    local color_red="\e[31m"
    local color_green="\e[1;32m"
    local color_yellow="\e[1;33m"
    local color_ochre="\e[38;5;95m"
    local git_status="$(git status 2> /dev/null)"
    if [[ ! $git_status =~ "working tree clean" ]]; then
        echo -e $color_red
    elif [[ $git_status =~ "Your branch is ahead of" ]]; then
        echo -e $color_yellow
    elif [[ $git_status =~ "nothing to commit" ]]; then
        echo -e $color_green
    else
        echo -e $color_ochre
    fi
}
# Add colored branch name
PS1+='\[$(git_color)\]$(__git_ps1 "[%s] ")'
# Break line
PS1+='\n'
# Color '$' and reset color
PS1+='\[\e[37m\]$ \[\e[m\]'