153 lines
4.6 KiB
Bash
153 lines
4.6 KiB
Bash
#!/usr/bin/env bash
|
||
# shellcheck disable=2034,2154
|
||
|
||
# Theme custom glyphs
|
||
SCM_GIT_CHAR_GITLAB=${BARBUK_GITLAB_CHAR:=' '}
|
||
SCM_GIT_CHAR_BITBUCKET=${BARBUK_BITBUCKET_CHAR:=' '}
|
||
SCM_GIT_CHAR_GITHUB=${BARBUK_GITHUB_CHAR:=' '}
|
||
SCM_GIT_CHAR_DEFAULT=${BARBUK_GIT_DEFAULT_CHAR:=' '}
|
||
SCM_GIT_CHAR_ICON_BRANCH=${BARBUK_GIT_BRANCH_ICON:=''}
|
||
SCM_HG_CHAR=${BARBUK_HG_CHAR:='☿ '}
|
||
SCM_SVN_CHAR=${BARBUK_SVN_CHAR:='⑆ '}
|
||
EXIT_CODE_ICON=${BARBUK_EXIT_CODE_ICON:=' '}
|
||
PYTHON_VENV_CHAR=${BARBUK_PYTHON_VENV_CHAR:=' '}
|
||
COMMAND_DURATION_ICON=${BARBUK_COMMAND_DURATION_ICON:=' '}
|
||
|
||
# Ssh user and hostname display
|
||
SSH_INFO=${BARBUK_SSH_INFO:=true}
|
||
HOST_INFO=${BARBUK_HOST_INFO:=long}
|
||
|
||
# Command duration
|
||
COMMAND_DURATION=${BARBUK_COMMAND_DURATION:=true}
|
||
COMMAND_DURATION_FILE="/tmp/bashit_theme_execution_$$"
|
||
|
||
# Bash-it default glyphs customization
|
||
SCM_NONE_CHAR=
|
||
SCM_THEME_PROMPT_DIRTY=" ${bold_red}✗"
|
||
SCM_THEME_PROMPT_CLEAN=" ${bold_green}✓"
|
||
SCM_THEME_PROMPT_PREFIX="|"
|
||
SCM_THEME_PROMPT_SUFFIX="${green}| "
|
||
SCM_GIT_BEHIND_CHAR="${bold_red}↓${normal}"
|
||
SCM_GIT_AHEAD_CHAR="${bold_green}↑${normal}"
|
||
SCM_GIT_UNTRACKED_CHAR="⌀"
|
||
SCM_GIT_UNSTAGED_CHAR="${bold_yellow}•${normal}"
|
||
SCM_GIT_STAGED_CHAR="${bold_green}+${normal}"
|
||
GIT_THEME_PROMPT_DIRTY=" ${bold_red}✗"
|
||
GIT_THEME_PROMPT_CLEAN=" ${bold_green}✓"
|
||
GIT_THEME_PROMPT_PREFIX="${cyan}"
|
||
GIT_THEME_PROMPT_SUFFIX="${cyan}"
|
||
SCM_THEME_BRANCH_TRACK_PREFIX="${normal} ⤏ ${cyan}"
|
||
SCM_THEME_CURRENT_USER_PREFFIX=' '
|
||
SCM_GIT_SHOW_CURRENT_USER=false
|
||
|
||
function _git-uptream-remote-logo {
|
||
[[ "$(_git-upstream)" == "" ]] && SCM_GIT_CHAR="$SCM_GIT_CHAR_DEFAULT"
|
||
|
||
local remote remote_domain
|
||
remote=$(_git-upstream-remote)
|
||
remote_domain=$(git config --get remote."$remote".url | awk -F'[@:.]' '{print $2}')
|
||
|
||
# remove // suffix for https:// url
|
||
remote_domain=${remote_domain//\//}
|
||
|
||
case $remote_domain in
|
||
github ) SCM_GIT_CHAR="$SCM_GIT_CHAR_GITHUB";;
|
||
gitlab ) SCM_GIT_CHAR="$SCM_GIT_CHAR_GITLAB";;
|
||
bitbucket ) SCM_GIT_CHAR="$SCM_GIT_CHAR_BITBUCKET";;
|
||
* ) SCM_GIT_CHAR="$SCM_GIT_CHAR_DEFAULT";;
|
||
esac
|
||
}
|
||
|
||
function git_prompt_info {
|
||
git_prompt_vars
|
||
echo -e " on $SCM_GIT_CHAR_ICON_BRANCH $SCM_PREFIX$SCM_BRANCH$SCM_STATE$SCM_GIT_AHEAD$SCM_GIT_BEHIND$SCM_GIT_STASH$SCM_SUFFIX"
|
||
}
|
||
|
||
function _exit-code {
|
||
if [[ "$1" -ne 0 ]]; then
|
||
exit_code=" ${purple}${EXIT_CODE_ICON}${yellow}${exit_code}${bold_orange}"
|
||
else
|
||
exit_code="${bold_green}"
|
||
fi
|
||
}
|
||
|
||
function _command_duration {
|
||
local command_duration command_start current_time="$1"
|
||
|
||
if [ "$COMMAND_DURATION" != true ]; then
|
||
return
|
||
fi
|
||
|
||
if [[ -f "$COMMAND_DURATION_FILE" ]]; then
|
||
command_start=$(< "$COMMAND_DURATION_FILE")
|
||
command_duration=$(( current_time - command_start ))
|
||
command rm "$COMMAND_DURATION_FILE"
|
||
else
|
||
command_duration=0
|
||
fi
|
||
|
||
if [[ "$command_duration" -gt 0 ]]; then
|
||
timer_m=$(( command_duration / 60 ))
|
||
timer_s=$(( command_duration % 60 ))
|
||
fi
|
||
|
||
if [[ "$timer_m" -gt 0 ]]; then
|
||
echo "${bold_blue}$COMMAND_DURATION_ICON${normal}${timer_m}m ${timer_s}s"
|
||
elif [[ "$timer_s" -gt 0 ]]; then
|
||
echo "${bold_blue}$COMMAND_DURATION_ICON${normal}${timer_s}s"
|
||
fi
|
||
}
|
||
|
||
function delete_temp_file() {
|
||
if [[ -f "$COMMAND_DURATION_FILE" ]]; then
|
||
rm -f "$COMMAND_DURATION_FILE"
|
||
fi
|
||
}
|
||
|
||
function _pre_exec {
|
||
date +%s > "$COMMAND_DURATION_FILE"
|
||
}
|
||
|
||
PS0="\[\$(_pre_exec)\]"
|
||
|
||
function _prompt {
|
||
local exit_code="$?" wrap_char=' ' dir_color=$green ssh_info='' python_venv='' host current_time
|
||
current_time=$(date +%s)
|
||
|
||
trap delete_temp_file EXIT HUP INT TRAP TERM
|
||
|
||
_exit-code exit_code
|
||
_git-uptream-remote-logo
|
||
|
||
history -a
|
||
|
||
# Detect root shell
|
||
if [ "$(whoami)" = root ]; then
|
||
dir_color=$red
|
||
fi
|
||
|
||
# Detect ssh
|
||
if [[ -n "${SSH_CONNECTION}" ]] && [ "$SSH_INFO" = true ]; then
|
||
if [ "$HOST_INFO" = long ]; then
|
||
host="\H"
|
||
else
|
||
host="\h"
|
||
fi
|
||
ssh_info="${bold_blue}\u${bold_orange}@${cyan}$host ${bold_orange}in"
|
||
fi
|
||
|
||
# Detect python venv
|
||
if [[ -n "${CONDA_DEFAULT_ENV}" ]]; then
|
||
python_venv="$PYTHON_VENV_CHAR${CONDA_DEFAULT_ENV} "
|
||
elif [[ -n "${VIRTUAL_ENV}" ]]; then
|
||
python_venv="$PYTHON_VENV_CHAR$(basename "${VIRTUAL_ENV}") "
|
||
fi
|
||
|
||
PS1="\\n${ssh_info} ${purple}$(scm_char)${python_venv}${dir_color}\\w${normal}$(scm_prompt_info)$(_command_duration "$current_time")${exit_code}"
|
||
|
||
[[ ${#PS1} -gt $((COLUMNS*3)) ]] && wrap_char="\\n"
|
||
PS1="${PS1}${wrap_char}❯${normal} "
|
||
}
|
||
|
||
safe_append_prompt_command _prompt
|