#!/usr/bin/env bash
#  vim:ts=4:sts=4:sw=4:et
#
#  Author: Hari Sekhon
#  Date: circa 2006 - 2012 (forked from .bashrc)
#
#  https://github.com/HariSekhon/DevOps-Bash-tools
#
#  License: see accompanying Hari Sekhon LICENSE file
#
#  If you're using my code you're welcome to connect with me on LinkedIn and optionally send me feedback to help steer this or other code I publish
#
#  https://www.linkedin.com/in/HariSekhon
#

# ============================================================================ #
#                            P r o m p t   M a g i c
# ============================================================================ #

# XXX: Warning: This must be perfect - edit at your own peril as imperfect PS1 prompt codes cause terminal wrap around to the same line

[ -n "${GOOGLE_CLOUD_SHELL:-}" ] && return
#[ "${USER:-}" = vagrant ] && return

# \[\033k\033\\\] is required for Screen auto title feature to detect prompt
# replace \033 with \e as it's directly supported in PS1

# need 'Defaults env_keep=STY' for this to not trigger on sudo su
# shellcheck disable=SC1117
SCREEN_ESCAPE="\[\ek\e\\\\\]"

PS1=""

# if inside Screen, set the screen escape inside PS1
[ -n "$STY" ] && PS1="$SCREEN_ESCAPE"

# defined in adjacent colors.sh
# shellcheck disable=SC2154
PS1_COLOUR="$txtgrn"
# shellcheck disable=SC2154
PS1_USER_COLOUR="$txtcyn"

if [ $EUID -eq 0 ]; then
    # shellcheck disable=SC2154
    PS1_COLOUR="$bldred"
    PS1_USER_COLOUR="$bldred"
fi

# XXX: important that every single escape sequence is enclosed in \[ \] to make sure it isn't included in the line wrapping calculcation otherwise the lines wrap back on to themselves
#   \W      basename of cwd
#   \w      full path of cwd
#   \h      host
# shellcheck disable=SC2154
# export PS1+="\[$PS1_COLOUR\]\t \[$bldblu\]\w \[$PS1_COLOUR\]> \[$txtrst\]"
# shellcheck disable=SC1117
export PS1+="\[$PS1_COLOUR\]\t \[$PS1_USER_COLOUR\]\u\[$txtwht\]@\[$bldgrn\]\h:\[$bldpur\]\$(git branch 2>/dev/null | awk '/^\\*/{print \$2\" \"}')\[$bldblu\]\w \[$PS1_COLOUR\]> \[$txtrst\]"

#if type kube_ps1 &>/dev/null; then
#    PS1='$(kube_ps1)'" $PS1"
#fi


# Screen relies on prompt having a dollar sign to detect the next word to set the screen title dynamically - .screenrc needs the following setting
#
# shelltitle ' > |'

# TODO: make screen auto title detect # for root


# For passing PS1 around:
#
#   base64 <<< "$PS1"
#
# and then pass the result through base64 --decode
#
#   base64 <<< "$PS1" | base64 --decode
#
# is a noop to demonstrate

if [ -z "$BASH_TIMING" ]; then
    export PS4="--> "
fi

# Bash performance profiling, can be heavy on performance, &>/tmp/bash_perf.out then use a profiling script
# combine with DEBUG=1 or set -x
bash_timing(){
    export BASH_TIMING=1
    export PS4='$(date "+%s.%N ($LINENO) + ") --> '
}

debug_bashrc(){
    PS4='+ $BASH_SOURCE:$LINENO:' bash -xic '' 2>&1 | less
}

debug_bash_profile(){
    PS4='+ $BASH_SOURCE:$LINENO:' bash -xlic '' 2>&1 | less
}

if is_mac; then
    # turn this off on mac thing - it heavily pollutes $DEBUG output and does nothing because we have a custom prompt
    if [[ "${PROMPT_COMMAND:-}" =~ update_terminal_cwd ]]; then
        # this unsets direnv's hook
        #unset PROMPT_COMMAND
        PROMPT_COMMAND="${PROMPT_COMMAND//update_terminal_cwd;/}"
    fi
    # stripping update_terminal_cwd can leave some weird broken PROMPT_COMMAND start due to interaction with direnv, so strip it without forking to sed/perl
    while [[ "${PROMPT_COMMAND:-}" =~ ^[[:space:]]|^\; ]]; do
        PROMPT_COMMAND="${PROMPT_COMMAND##[[:space:]]}"
        PROMPT_COMMAND="${PROMPT_COMMAND##;}"
    done
    export PROMPT_COMMAND

    export SHELL_SESSION_HISTORY=0
fi