325 lines
8.1 KiB
Bash
325 lines
8.1 KiB
Bash
# shellcheck shell=bash
|
|
# shellcheck disable=SC2034 # Expected behavior for themes.
|
|
|
|
# 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 ##
|
|
#############
|
|
|
|
function ____atomic_top_left_parse() {
|
|
local ifs_old="${IFS}"
|
|
local 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+=""
|
|
}
|
|
|
|
function ____atomic_top_right_parse() {
|
|
local ifs_old="${IFS}"
|
|
local 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))
|
|
}
|
|
|
|
function ____atomic_bottom_parse() {
|
|
local ifs_old="${IFS}"
|
|
local IFS="|"
|
|
read -r -a args <<< "$@"
|
|
IFS="${ifs_old}"
|
|
_BOTTOM+="${args[0]?}${args[1]?${FUNCNAME[0]}}"
|
|
[[ ${#args[1]} -gt 0 ]] && _BOTTOM+=" "
|
|
}
|
|
|
|
function ____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}"
|
|
}
|
|
|
|
function ____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 ##
|
|
##############
|
|
|
|
function ___atomic_prompt_user_info() {
|
|
local color="${white?}" box
|
|
local info="${IYellow}\u${IRed}@${IGreen}\h"
|
|
box="${normal?}${LineA?}\$([[ \$? != 0 ]] && echo \"${BIWhite?}[${IRed?}${SX?}${BIWhite?}]${normal?}${Line?}\")${Line?}${BIWhite?}[|${BIWhite?}]${normal?}${Line?}"
|
|
|
|
printf "%s|%s|%s|%s" "${color}" "${info}" "${white?}" "${box}"
|
|
}
|
|
|
|
function ___atomic_prompt_dir() {
|
|
local color="${IRed?}"
|
|
local box="[|]${normal}"
|
|
local info="\w"
|
|
printf "%s|%s|%s|%s" "${color}" "${info}" "${bold_white?}" "${box}"
|
|
}
|
|
|
|
function ___atomic_prompt_scm() {
|
|
[[ "${THEME_SHOW_SCM:-}" != "true" ]] && return
|
|
local color="${bold_green?}" box info
|
|
box="${Line?}[${IWhite?}$(scm_char)] "
|
|
info="$(scm_prompt_info)"
|
|
printf "%s|%s|%s|%s" "${color}" "${info}" "${bold_white?}" "${box}"
|
|
}
|
|
|
|
function ___atomic_prompt_python() {
|
|
[[ "${THEME_SHOW_PYTHON:-}" != "true" ]] && return
|
|
local color="${bold_yellow?}"
|
|
local box="[|]" info
|
|
info="$(python_version_prompt)"
|
|
printf "%s|%s|%s|%s" "${color}" "${info}" "${bold_blue?}" "${box}"
|
|
}
|
|
|
|
function ___atomic_prompt_ruby() {
|
|
[[ "${THEME_SHOW_RUBY:-}" != "true" ]] && return
|
|
local color="${bold_white?}"
|
|
local box="[|]" info
|
|
info="rb-$(ruby_version_prompt)"
|
|
printf "%s|%s|%s|%s" "${color}" "${info}" "${bold_red?}" "${box}"
|
|
}
|
|
|
|
function ___atomic_prompt_todo() {
|
|
[[ "${THEME_SHOW_TODO:-}" != "true" ||
|
|
-z "$(which todo.sh)" ]] && return
|
|
local color="${bold_white?}"
|
|
local box="[|]" info
|
|
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}"
|
|
}
|
|
|
|
function ___atomic_prompt_clock() {
|
|
[[ "${THEME_SHOW_CLOCK:-}" != "true" ]] && return
|
|
local color="${THEME_CLOCK_COLOR:-}"
|
|
local box="[|]" info
|
|
info="$(date +"${THEME_CLOCK_FORMAT}")"
|
|
printf "%s|%s|%s|%s" "${color}" "${info}" "${bold_white?}" "${box}"
|
|
}
|
|
|
|
function ___atomic_prompt_battery() {
|
|
local batp box info
|
|
! _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}"
|
|
}
|
|
|
|
function ___atomic_prompt_exitcode() {
|
|
[[ "${THEME_SHOW_EXITCODE:-}" != "true" ]] && return
|
|
local color="${bold_purple?}"
|
|
[[ "${exitcode?}" -ne 0 ]] && printf "%s|%s" "${color}" "${exitcode}"
|
|
}
|
|
|
|
function ___atomic_prompt_char() {
|
|
local color="${white?}"
|
|
local 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 ##
|
|
#########
|
|
|
|
function __atomic_show() {
|
|
local _seg="${1?}"
|
|
export "THEME_SHOW_${_seg}"=true
|
|
}
|
|
|
|
function __atomic_hide() {
|
|
local _seg="${1?}"
|
|
export "THEME_SHOW_${_seg}"=false
|
|
}
|
|
|
|
function _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
|
|
}
|
|
|
|
function atomic() {
|
|
local action="${1?}"
|
|
shift
|
|
local segs=("${@?}")
|
|
local func
|
|
case "${action}" in
|
|
show)
|
|
func=__atomic_show
|
|
;;
|
|
hide)
|
|
func=__atomic_hide
|
|
;;
|
|
*)
|
|
_log_error "${FUNCNAME[0]}: unknown action '${action}'"
|
|
return 1
|
|
;;
|
|
esac
|
|
for seg in "${segs[@]}"; do
|
|
seg="$(printf "%s" "${seg}" | tr '[:lower:]' '[:upper:]')"
|
|
"${func}" "${seg}"
|
|
done
|
|
}
|
|
|
|
complete -F _atomic_completion atomic
|
|
|
|
###############
|
|
## Variables ##
|
|
###############
|
|
|
|
SCM_THEME_PROMPT_PREFIX=""
|
|
SCM_THEME_PROMPT_SUFFIX=""
|
|
|
|
RBENV_THEME_PROMPT_PREFIX=""
|
|
RBENV_THEME_PROMPT_SUFFIX=""
|
|
RBFU_THEME_PROMPT_PREFIX=""
|
|
RBFU_THEME_PROMPT_SUFFIX=""
|
|
RVM_THEME_PROMPT_PREFIX=""
|
|
RVM_THEME_PROMPT_SUFFIX=""
|
|
|
|
SCM_THEME_PROMPT_DIRTY=" ${bold_red}✗${normal}"
|
|
SCM_THEME_PROMPT_CLEAN=" ${bold_green}✓${normal}"
|
|
|
|
: "${THEME_SHOW_SUDO:="true"}"
|
|
: "${THEME_SHOW_SCM:="true"}"
|
|
: "${THEME_SHOW_RUBY:="false"}"
|
|
: "${THEME_SHOW_PYTHON:="false"}"
|
|
: "${THEME_SHOW_CLOCK:="true"}"
|
|
: "${THEME_SHOW_TODO:="false"}"
|
|
: "${THEME_SHOW_BATTERY:="true"}"
|
|
: "${THEME_SHOW_EXITCODE:="false"}"
|
|
|
|
: "${THEME_CLOCK_COLOR:=${BICyan?}}"
|
|
: "${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:="user_info dir scm"}"
|
|
: "${___ATOMIC_TOP_RIGHT:="exitcode python ruby todo clock battery"}"
|
|
: "${___ATOMIC_BOTTOM:="char"}"
|
|
|
|
############
|
|
## Prompt ##
|
|
############
|
|
|
|
function __atomic_ps1() {
|
|
printf "%s%s%s" "$(____atomic_top)" "$(____atomic_bottom)" "${normal?}"
|
|
}
|
|
|
|
function __atomic_ps2() {
|
|
color="${bold_white?}"
|
|
printf "%s%s%s" "${color}" "${__ATOMIC_PROMPT_CHAR_PS2?} " "${normal?}"
|
|
}
|
|
|
|
function _atomic_prompt() {
|
|
exitcode="$?"
|
|
|
|
PS1="$(__atomic_ps1)"
|
|
PS2="$(__atomic_ps2)"
|
|
}
|
|
|
|
safe_append_prompt_command _atomic_prompt
|