321 lines
7.6 KiB
Bash
321 lines
7.6 KiB
Bash
# shellcheck shell=bash
|
|
|
|
# Atomic Bash Prompt for Bash-it
|
|
# By lfelipe base on the theme brainy of MunifTanjim
|
|
|
|
############
|
|
## Colors ##
|
|
############
|
|
IRed="\e[1;49;31m"
|
|
IGreen="\e[1;49;32m"
|
|
IYellow="\e[1;49;33m"
|
|
IWhite="\e[1;49;37m"
|
|
BIWhite="\e[1;49;37m"
|
|
BICyan="\e[1;49;36m"
|
|
|
|
#############
|
|
## Symbols ##
|
|
#############
|
|
Line="\342\224\200"
|
|
LineA="\342\224\214\342\224\200"
|
|
SX="\342\234\227"
|
|
LineB="\342\224\224\342\224\200\342\224\200"
|
|
Circle="\342\227\217"
|
|
Face="\342\230\273"
|
|
|
|
#############
|
|
## Parsers ##
|
|
#############
|
|
|
|
____atomic_top_left_parse() {
|
|
ifs_old="${IFS}"
|
|
IFS="|"
|
|
read -r -a args <<< "$@"
|
|
IFS="${ifs_old}"
|
|
if [ -n "${args[3]}" ]; then
|
|
_TOP_LEFT+="${args[2]}${args[3]}"
|
|
fi
|
|
_TOP_LEFT+="${args[0]}${args[1]}"
|
|
if [ -n "${args[4]}" ]; then
|
|
_TOP_LEFT+="${args[2]}${args[4]}"
|
|
fi
|
|
_TOP_LEFT+=""
|
|
}
|
|
|
|
____atomic_top_right_parse() {
|
|
ifs_old="${IFS}"
|
|
IFS="|"
|
|
read -r -a args <<< "$@"
|
|
IFS="${ifs_old}"
|
|
_TOP_RIGHT+=" "
|
|
if [ -n "${args[3]}" ]; then
|
|
_TOP_RIGHT+="${args[2]}${args[3]}"
|
|
fi
|
|
_TOP_RIGHT+="${args[0]}${args[1]}"
|
|
if [ -n "${args[4]}" ]; then
|
|
_TOP_RIGHT+="${args[2]}${args[4]}"
|
|
fi
|
|
__TOP_RIGHT_LEN=$((__TOP_RIGHT_LEN + ${#args[1]} + ${#args[3]} + ${#args[4]} + 1))
|
|
((__SEG_AT_RIGHT += 1))
|
|
}
|
|
|
|
____atomic_bottom_parse() {
|
|
ifs_old="${IFS}"
|
|
IFS="|"
|
|
read -r -a args <<< "$@"
|
|
IFS="${ifs_old}"
|
|
_BOTTOM+="${args[0]}${args[1]}"
|
|
[ ${#args[1]} -gt 0 ] && _BOTTOM+=" "
|
|
}
|
|
|
|
____atomic_top() {
|
|
_TOP_LEFT=""
|
|
_TOP_RIGHT=""
|
|
__TOP_RIGHT_LEN=0
|
|
__SEG_AT_RIGHT=0
|
|
|
|
for seg in ${___ATOMIC_TOP_LEFT}; do
|
|
info="$(___atomic_prompt_"${seg}")"
|
|
[ -n "${info}" ] && ____atomic_top_left_parse "${info}"
|
|
done
|
|
|
|
___cursor_right="\e[500C"
|
|
_TOP_LEFT+="${___cursor_right}"
|
|
|
|
for seg in ${___ATOMIC_TOP_RIGHT}; do
|
|
info="$(___atomic_prompt_"${seg}")"
|
|
[ -n "${info}" ] && ____atomic_top_right_parse "${info}"
|
|
done
|
|
|
|
[ $__TOP_RIGHT_LEN -gt 0 ] && __TOP_RIGHT_LEN=$((__TOP_RIGHT_LEN - 0))
|
|
___cursor_adjust="\e[${__TOP_RIGHT_LEN}D"
|
|
_TOP_LEFT+="${___cursor_adjust}"
|
|
|
|
printf "%s%s" "${_TOP_LEFT}" "${_TOP_RIGHT}"
|
|
}
|
|
|
|
____atomic_bottom() {
|
|
_BOTTOM=""
|
|
for seg in $___ATOMIC_BOTTOM; do
|
|
info="$(___atomic_prompt_"${seg}")"
|
|
[ -n "${info}" ] && ____atomic_bottom_parse "${info}"
|
|
done
|
|
printf "\n%s" "${_BOTTOM}"
|
|
}
|
|
|
|
##############
|
|
## Segments ##
|
|
##############
|
|
|
|
___atomic_prompt_user_info() {
|
|
color=$white
|
|
box="${normal}${LineA}\$([[ \$? != 0 ]] && echo \"${BIWhite}[${IRed}${SX}${BIWhite}]${normal}${Line}\")${Line}${BIWhite}[|${BIWhite}]${normal}${Line}"
|
|
info="${IYellow}\u${IRed}@${IGreen}\h"
|
|
|
|
printf "%s|%s|%s|%s" "${color}" "${info}" "${white}" "${box}"
|
|
}
|
|
|
|
___atomic_prompt_dir() {
|
|
color=${IRed}
|
|
box="[|]${normal}"
|
|
info="\w"
|
|
printf "%s|%s|%s|%s" "${color}" "${info}" "${bold_white}" "${box}"
|
|
}
|
|
|
|
___atomic_prompt_scm() {
|
|
[ "${THEME_SHOW_SCM}" != "true" ] && return
|
|
color=$bold_green
|
|
box="${Line}[${IWhite}$(scm_char)] "
|
|
info="$(scm_prompt_info)"
|
|
printf "%s|%s|%s|%s" "${color}" "${info}" "${bold_white}" "${box}"
|
|
}
|
|
|
|
___atomic_prompt_python() {
|
|
[ "${THEME_SHOW_PYTHON}" != "true" ] && return
|
|
color=$bold_yellow
|
|
box="[|]"
|
|
info="$(python_version_prompt)"
|
|
printf "%s|%s|%s|%s" "${color}" "${info}" "${bold_blue}" "${box}"
|
|
}
|
|
|
|
___atomic_prompt_ruby() {
|
|
[ "${THEME_SHOW_RUBY}" != "true" ] && return
|
|
color=$bold_white
|
|
box="[|]"
|
|
info="rb-$(ruby_version_prompt)"
|
|
printf "%s|%s|%s|%s" "${color}" "${info}" "${bold_red}" "${box}"
|
|
}
|
|
|
|
___atomic_prompt_todo() {
|
|
[ "${THEME_SHOW_TODO}" != "true" ] \
|
|
|| [ -z "$(which todo.sh)" ] && return
|
|
color=$bold_white
|
|
box="[|]"
|
|
info="t:$(todo.sh ls | grep -E "TODO: [0-9]+ of ([0-9]+)" | awk '{ print $4 }')"
|
|
printf "%s|%s|%s|%s" "${color}" "${info}" "${bold_green}" "${box}"
|
|
}
|
|
|
|
___atomic_prompt_clock() {
|
|
[ "${THEME_SHOW_CLOCK}" != "true" ] && return
|
|
color=$THEME_CLOCK_COLOR
|
|
box="[|]"
|
|
info="$(date +"${THEME_CLOCK_FORMAT}")"
|
|
printf "%s|%s|%s|%s" "${color}" "${info}" "${bold_white}" "${box}"
|
|
}
|
|
|
|
___atomic_prompt_battery() {
|
|
! _command_exists battery_percentage \
|
|
|| [ "${THEME_SHOW_BATTERY}" != "true" ] \
|
|
|| [ "$(battery_percentage)" = "no" ] && return
|
|
|
|
batp=$(battery_percentage)
|
|
if [ "$batp" -eq 50 ] || [ "$batp" -gt 50 ]; then
|
|
color=$bold_green
|
|
elif [ "$batp" -lt 50 ] && [ "$batp" -gt 25 ]; then
|
|
color=$bold_yellow
|
|
elif [ "$batp" -eq 25 ] || [ "$batp" -lt 25 ]; then
|
|
color=$IRed
|
|
fi
|
|
box="[|]"
|
|
ac_adapter_connected && info="+"
|
|
ac_adapter_disconnected && info="-"
|
|
info+=$batp
|
|
[ "$batp" -eq 100 ] || [ "$batp" -gt 100 ] && info="AC"
|
|
printf "%s|%s|%s|%s" "${color}" "${info}" "${bold_white}" "${box}"
|
|
}
|
|
|
|
___atomic_prompt_exitcode() {
|
|
[ "${THEME_SHOW_EXITCODE}" != "true" ] && return
|
|
color=$bold_purple
|
|
[ "$exitcode" -ne 0 ] && printf "%s|%s" "${color}" "${exitcode}"
|
|
}
|
|
|
|
___atomic_prompt_char() {
|
|
color=$white
|
|
prompt_char="${__ATOMIC_PROMPT_CHAR_PS1}"
|
|
if [ "${THEME_SHOW_SUDO}" == "true" ]; then
|
|
if sudo -vn 1> /dev/null 2>&1; then
|
|
prompt_char="${__ATOMIC_PROMPT_CHAR_PS1_SUDO}"
|
|
fi
|
|
fi
|
|
printf "%s|%s" "${color}" "${prompt_char}"
|
|
}
|
|
|
|
#########
|
|
## cli ##
|
|
#########
|
|
|
|
__atomic_show() {
|
|
typeset _seg=${1:-}
|
|
shift
|
|
export "THEME_SHOW_${_seg}"=true
|
|
}
|
|
|
|
__atomic_hide() {
|
|
typeset _seg=${1:-}
|
|
shift
|
|
export "THEME_SHOW_${_seg}"=false
|
|
}
|
|
|
|
_atomic_completion() {
|
|
local cur _action actions segments
|
|
COMPREPLY=()
|
|
cur="${COMP_WORDS[COMP_CWORD]}"
|
|
_action="${COMP_WORDS[1]}"
|
|
actions="show hide"
|
|
segments="battery clock exitcode python ruby scm sudo todo"
|
|
case "${_action}" in
|
|
show | hide)
|
|
# shellcheck disable=SC2207
|
|
COMPREPLY=($(compgen -W "${segments}" -- "${cur}"))
|
|
return 0
|
|
;;
|
|
esac
|
|
|
|
# shellcheck disable=SC2207
|
|
COMPREPLY=($(compgen -W "${actions}" -- "${cur}"))
|
|
return 0
|
|
}
|
|
|
|
atomic() {
|
|
typeset action=${1:-}
|
|
shift
|
|
typeset segs=${*:-}
|
|
typeset func
|
|
case $action in
|
|
show)
|
|
func=__atomic_show
|
|
;;
|
|
hide)
|
|
func=__atomic_hide
|
|
;;
|
|
esac
|
|
for seg in ${segs}; do
|
|
seg=$(printf "%s" "${seg}" | tr '[:lower:]' '[:upper:]')
|
|
$func "${seg}"
|
|
done
|
|
}
|
|
|
|
complete -F _atomic_completion atomic
|
|
|
|
###############
|
|
## Variables ##
|
|
###############
|
|
|
|
export SCM_THEME_PROMPT_PREFIX=""
|
|
export SCM_THEME_PROMPT_SUFFIX=""
|
|
|
|
export RBENV_THEME_PROMPT_PREFIX=""
|
|
export RBENV_THEME_PROMPT_SUFFIX=""
|
|
export RBFU_THEME_PROMPT_PREFIX=""
|
|
export RBFU_THEME_PROMPT_SUFFIX=""
|
|
export RVM_THEME_PROMPT_PREFIX=""
|
|
export RVM_THEME_PROMPT_SUFFIX=""
|
|
|
|
export SCM_THEME_PROMPT_DIRTY=" ${bold_red}✗${normal}"
|
|
export SCM_THEME_PROMPT_CLEAN=" ${bold_green}✓${normal}"
|
|
|
|
THEME_SHOW_SUDO=${THEME_SHOW_SUDO:-"true"}
|
|
THEME_SHOW_SCM=${THEME_SHOW_SCM:-"true"}
|
|
THEME_SHOW_RUBY=${THEME_SHOW_RUBY:-"false"}
|
|
THEME_SHOW_PYTHON=${THEME_SHOW_PYTHON:-"false"}
|
|
THEME_SHOW_CLOCK=${THEME_SHOW_CLOCK:-"true"}
|
|
THEME_SHOW_TODO=${THEME_SHOW_TODO:-"false"}
|
|
THEME_SHOW_BATTERY=${THEME_SHOW_BATTERY:-"true"}
|
|
THEME_SHOW_EXITCODE=${THEME_SHOW_EXITCODE:-"false"}
|
|
|
|
THEME_CLOCK_COLOR=${THEME_CLOCK_COLOR:-"${BICyan}"}
|
|
THEME_CLOCK_FORMAT=${THEME_CLOCK_FORMAT:-"%a %b %d - %H:%M"}
|
|
|
|
__ATOMIC_PROMPT_CHAR_PS1=${THEME_PROMPT_CHAR_PS1:-"${normal}${LineB}${bold_white}${Circle}"}
|
|
__ATOMIC_PROMPT_CHAR_PS2=${THEME_PROMPT_CHAR_PS2:-"${normal}${LineB}${bold_white}${Circle}"}
|
|
|
|
__ATOMIC_PROMPT_CHAR_PS1_SUDO=${THEME_PROMPT_CHAR_PS1_SUDO:-"${normal}${LineB}${bold_red}${Face}"}
|
|
__ATOMIC_PROMPT_CHAR_PS2_SUDO=${THEME_PROMPT_CHAR_PS2_SUDO:-"${normal}${LineB}${bold_red}${Face}"}
|
|
|
|
___ATOMIC_TOP_LEFT=${___ATOMIC_TOP_LEFT:-"user_info dir scm"}
|
|
___ATOMIC_TOP_RIGHT=${___ATOMIC_TOP_RIGHT:-"exitcode python ruby todo clock battery"}
|
|
___ATOMIC_BOTTOM=${___ATOMIC_BOTTOM:-"char"}
|
|
|
|
############
|
|
## Prompt ##
|
|
############
|
|
|
|
__atomic_ps1() {
|
|
printf "%s%s%s" "$(____atomic_top)" "$(____atomic_bottom)" "${normal}"
|
|
}
|
|
|
|
__atomic_ps2() {
|
|
color=$bold_white
|
|
printf "%s%s%s" "${color}" "${__ATOMIC_PROMPT_CHAR_PS2} " "${normal}"
|
|
}
|
|
|
|
_atomic_prompt() {
|
|
exitcode="$?"
|
|
|
|
PS1="$(__atomic_ps1)"
|
|
PS2="$(__atomic_ps2)"
|
|
}
|
|
|
|
safe_append_prompt_command _atomic_prompt
|