prompt.sh

#!/bin/bash

# Source Git Prompt defiinitions
bfg_source "vendor/git-prompt/git-prompt"

# Configure Git Prompt.
export GIT_PS1_SHOWUPSTREAM="verbose"
export GIT_PS1_SHOWCOLORHINTS=
export GIT_PS1_SHOWDIRTYSTATE=1
export GIT_PS1_SHOWSTASHSTATE=
export GIT_PS1_SHOWUNTRACKEDFILES=1
export GIT_PS1_STATESEPARATOR=" "

## Color Variables ##
# These are calculated once at source time to accelerate prompt handlers.
# https://en.wikipedia.org/wiki/ANSI_escape_code#Colors

FG_COLOR_BLACK=$(bfg_escape $'\033[30m')
FG_COLOR_RED=$(bfg_escape $'\033[31m')
FG_COLOR_GREEN=$(bfg_escape $'\033[32m')
FG_COLOR_YELLOW=$(bfg_escape $'\033[33m')
FG_COLOR_BLUE=$(bfg_escape $'\033[34m')
# FG_COLOR_MAGENTA=$(bfg_escape $'\033[35m')
# FG_COLOR_CYAN=$(bfg_escape $'\033[36m')
FG_COLOR_WHITE=$(bfg_escape $'\033[37m')
FG_COLOR_BRIGHT_BLACK=$(bfg_escape $'\033[90m')
FG_COLOR_BRIGHT_GREY=$(bfg_escape $'\033[38;5;250m')
# FG_COLOR_BRIGHT_RED=$(bfg_escape $'\033[91m')
# FG_COLOR_BRIGHT_GREEN=$(bfg_escape $'\033[92m')
# FG_COLOR_BRIGHT_YELLOW=$(bfg_escape $'\033[93m')
# FG_COLOR_BRIGHT_BLUE=$(bfg_escape $'\033[94m')
# FG_COLOR_BRIGHT_MAGENTA=$(bfg_escape $'\033[95m')
# FG_COLOR_BRIGHT_CYAN=$(bfg_escape $'\033[96m')
FG_COLOR_BRIGHT_WHITE=$(bfg_escape $'\033[97m')
FG_COLOR_ORANGE=$(bfg_escape $'\033[38;5;202m')
FG_COLOR_RASPBERRY=$(bfg_escape $'\033[38;5;161m')
FG_COLOR_DARK_GREY=$(bfg_escape $'\033[38;5;239m')

# BG_COLOR_BLACK=$(bfg_escape $'\033[40m')
BG_COLOR_RED=$(bfg_escape $'\033[41m')
BG_COLOR_GREEN=$(bfg_escape $'\033[42m')
BG_COLOR_YELLOW=$(bfg_escape $'\033[43m')
BG_COLOR_BLUE=$(bfg_escape $'\033[44m')
# BG_COLOR_MAGENTA=$(bfg_escape $'\033[45m')
# BG_COLOR_CYAN=$(bfg_escape $'\033[46m')
#BG_COLOR_WHITE=$(bfg_escape $'\033[47m')
BG_COLOR_BRIGHT_BLACK=$(bfg_escape $'\033[100m')
BG_COLOR_BRIGHT_GREY=$(bfg_escape $'\033[48;5;250m')
# BG_COLOR_BRIGHT_RED=$(bfg_escape $'\033[101m')
# BG_COLOR_BRIGHT_GREEN=$(bfg_escape $'\033[102m')
# BG_COLOR_BRIGHT_YELLOW=$(bfg_escape $'\033[103m')
# BG_COLOR_BRIGHT_BLUE=$(bfg_escape $'\033[104m')
# BG_COLOR_BRIGHT_MAGENTA=$(bfg_escape $'\033[105m')
# BG_COLOR_BRIGHT_CYAN=$(bfg_escape $'\033[106m')
# BG_COLOR_BRIGHT_WHITE=$(bfg_escape $'\033[107m')
BG_COLOR_ORANGE=$(bfg_escape $'\033[48;5;202m')
BG_COLOR_RASPBERRY=$(bfg_escape $'\033[48;5;161m')
BG_COLOR_DARK_GREY=$(bfg_escape $'\033[48;5;239m')


# FG_COLOR_RESET=$(bfg_escape $'\033[39m')
BG_COLOR_RESET=$(bfg_escape $'\033[49m')
ALL_COLOR_RESET=$(bfg_escape $'\033[0m')

BOLD_ON=$(bfg_escape $'\033[1m')
BOLD_OFF=$(bfg_escape $'\033[22m')

# XXX This solved a problem at one point, but is causing mangled history on new
# machines. What this does, according to the manual, is "erase to end of line"
# in bash. When pressing the up arrow to reveal a multi-line history entry, this
# erases everything on the same line after the prompt.
#
# Disabling for now, but leaving this here for future reference.
#
# XXX seems like this was important to ZSH, but not bash?
# Adding this back for ZSH in prompt_zsh.sh.
#
# COLOR_ENDLINE=$(bfg_escape $'\033[K')
COLOR_ENDLINE=""

# load minpath
bfg_source "minpath"


## Icon Variables ##
# These are calculated once at source time to accelerate prompt handlers.

# If the shell natively supports unicode, use builtins for speed.
if   [ $'\ue0b0' = "$(perl -CS -E 'print "\x{e0b0}"')" ]; then
    bfg_get_icon() {
        # Correct escape depends on length of code point. See also:
        # https://github.com/ryanoasis/nerd-fonts/wiki/FAQ-and-Troubleshooting
        if [ "${#1}" -eq 5 ]; then
            # need to pass directly for unicode escape to process.
            # shellcheck disable=SC2059
            printf "\U$1"
        else
            # need to pass directly for unicode escape to process.
            # shellcheck disable=SC2059
            printf "\u$1"
        fi
    }
else
    bfg_get_icon() {
        perl -CS -E 'print "\x{'"$1"'}"'
    }
fi

ICON_CHEVRON_RIGHT="$(bfg_get_icon "e0b0")"
ICON_APPLE="$(bfg_get_icon "f179")"
ICON_UBUNTU="$(bfg_get_icon "f31b")"
ICON_RASPI="$(bfg_get_icon "f315")"
ICON_CONSOLE="$(bfg_get_icon "fcb5")"
ICON_SHIELD="$(bfg_get_icon "f49c")"
ICON_LOCK="$(bfg_get_icon "f023")"
ICON_HOME="$(bfg_get_icon "f015")"
ICON_FOLDER="$(bfg_get_icon "f07c")"
ICON_BRANCH="$(bfg_get_icon "f126")"
ICON_NETWORK="$(bfg_get_icon "f0318")"

## Head Segment Detection ##

HEAD_FG_COLOR="$FG_COLOR_WHITE"
HEAD_BG_COLOR="$BG_COLOR_BRIGHT_BLACK"
HEAD_END_COLOR="$FG_COLOR_BRIGHT_BLACK"
HEAD_ICON="$ICON_CONSOLE"


if [ -f /etc/rpi-issue ]; then
    HEAD_FG_COLOR="$FG_COLOR_WHITE"
    HEAD_BG_COLOR="$BG_COLOR_RASPBERRY"
    HEAD_END_COLOR="$FG_COLOR_RASPBERRY"
    HEAD_ICON="$ICON_RASPI"
else
    case "$(uname)" in
        'Darwin')
            HEAD_FG_COLOR="$FG_COLOR_BRIGHT_WHITE"
            HEAD_BG_COLOR="$BG_COLOR_DARK_GREY"
            HEAD_END_COLOR="$FG_COLOR_DARK_GREY"
            HEAD_ICON="$ICON_APPLE"
            ;;
        'Linux')
            HEAD_FG_COLOR="$FG_COLOR_WHITE"
            HEAD_BG_COLOR="$BG_COLOR_ORANGE"
            HEAD_END_COLOR="$FG_COLOR_ORANGE"
            HEAD_ICON="$ICON_UBUNTU"
            ;;
    esac
fi

## Left Prompt Segments ##

bfg_prompt_segment_head() {
    if [ -n "${BFG_DISABLE_ROOT_HEAD+1}" ] || [ "$EUID" -ne 0 ] || [ "$SUDO_UID" ]; then # if effective user ID is NOT root
        PROMPT+="$HEAD_FG_COLOR$HEAD_BG_COLOR"
        PROMPT+=" $HEAD_ICON "
        PROMPT+="$HEAD_END_COLOR"
    else
        PROMPT+="$FG_COLOR_WHITE$BG_COLOR_RED"
        PROMPT+=" $ICON_SHIELD root "
        PROMPT+="$FG_COLOR_RED"
    fi
}

SHOW_SSH_SEGMENT=0
if [ -n "$SSH_CLIENT" ]; then
    SHOW_SSH_SEGMENT=1
fi

bfg_prompt_segment_ssh() {
    # If in a SSH session, show SSH segment.
    if [ "$SHOW_SSH_SEGMENT" -eq 1 ]; then
        PROMPT+="$BG_COLOR_BRIGHT_GREY"
        PROMPT+="$ICON_CHEVRON_RIGHT"
        PROMPT+="$FG_COLOR_BLACK "
        PROMPT+="$ICON_NETWORK "
        if [ -z "$BFG_OVERRIDE_HOSTNAME" ]; then
            PROMPT+=$'\h '
        else
            PROMPT+="$BFG_OVERRIDE_HOSTNAME "
        fi
        PROMPT+="$FG_COLOR_BRIGHT_GREY"
    fi
}

bfg_prompt_segment_directory() {
    PROMPT+="$BG_COLOR_BLUE"
    PROMPT+="$ICON_CHEVRON_RIGHT"
    PROMPT+="$FG_COLOR_WHITE "

    if [ ! -w "$PWD" ]; then
        PROMPT+="$ICON_LOCK"
    elif [ "$PWD" = "$HOME" ]; then
        PROMPT+="$ICON_HOME"
    else
        PROMPT+="$ICON_FOLDER"
    fi

    PROMPT+="$BOLD_ON "
    if [ "${BFG_SHELL_PROMPT_MINPATH:-0}" -eq 1 ]; then
        bfg_minpath
    else
        PROMPT+=$'\w'
    fi
    PROMPT+="$BOLD_OFF "

    PROMPT+="$FG_COLOR_BLUE"
}

bfg_prompt_segment_git() {
    # If git is not available, then skip this segment.
    if ! which git >/dev/null 2>&1; then
        return 0
    fi

    # If the git repository has been configured to be ignored, no segment.
    # Inspired by https://stackoverflow.com/a/29416158.
    #
    # See matching alias
    if [[ $(git config prompt.ignore) ]]; then
        return 0
    fi

    git_status=$(__git_ps1 "%s" | awk 'BEGIN {
        BRANCH_LEN=22
    }{
        printf "%s", substr($1, 1, BRANCH_LEN);
        if (length($1) > BRANCH_LEN) printf "...";
        if (length($2) > 0) printf " %s", $2;
    }')
    if [[ -n "$git_status" ]]; then
        git_color_fg="$FG_COLOR_GREEN"
        git_color_bg="$BG_COLOR_GREEN"
        if [[ "$git_status" =~ [*!%] ]]; then
            git_color_fg="$FG_COLOR_YELLOW"
            git_color_bg="$BG_COLOR_YELLOW"
        fi

        PROMPT+="$git_color_bg"
        PROMPT+="$ICON_CHEVRON_RIGHT"
        PROMPT+="$FG_COLOR_BLACK"
        PROMPT+=" $ICON_BRANCH "
        PROMPT+="$git_status "
        PROMPT+="$git_color_fg"
    fi
}

## Left Prompt Handler ##

bfg_set_prompt() {
    # Clear existing prompt.
    PROMPT=""

    # Add segments.
    bfg_prompt_segment_head
    bfg_prompt_segment_ssh
    bfg_prompt_segment_directory
    bfg_prompt_segment_git

    # Reset colors after prompt, add final chevron.
    PROMPT+="$BG_COLOR_RESET"
    PROMPT+="$ICON_CHEVRON_RIGHT"
    PROMPT+="$ALL_COLOR_RESET$COLOR_ENDLINE"
    PROMPT+=" "
}