2022-03-01 17:32:57 +00:00
|
|
|
#!/bin/bash
|
|
|
|
# shebang for syntax highlightin purposes
|
|
|
|
|
2022-03-01 20:11:44 +00:00
|
|
|
__debug_trap() {
|
2022-03-02 15:58:27 +00:00
|
|
|
# Set necessary pre-command variables (PROMPT_COMMAND is a
|
|
|
|
# command so its excluded here)
|
2022-03-07 03:12:14 +00:00
|
|
|
if [[ "$BASH_COMMAND" != "$PROMPT_COMMAND"
|
|
|
|
&& "$LAST_BASH_COMMAND" == "$PROMPT_COMMAND" ]]; then
|
|
|
|
LAST_RT="$EPOCHREALTIME"
|
|
|
|
INC_TIME=1
|
2022-03-01 20:11:44 +00:00
|
|
|
fi
|
2022-03-02 06:29:04 +00:00
|
|
|
LAST_BASH_COMMAND="$BASH_COMMAND"
|
2022-03-01 20:11:44 +00:00
|
|
|
}
|
2022-03-01 17:32:57 +00:00
|
|
|
|
2022-03-01 20:11:44 +00:00
|
|
|
trap '__debug_trap' DEBUG
|
2022-03-01 17:32:57 +00:00
|
|
|
|
|
|
|
__get_cmd_time() {
|
2022-03-01 20:11:44 +00:00
|
|
|
# Set hours minutes seconds and remove preceding zeros
|
2022-03-06 14:11:27 +00:00
|
|
|
local YEAR=$((10#0$(($(TZ=UTC printf '%(%Y)T' $CMD_TIME)))-1970))
|
2022-03-07 03:12:14 +00:00
|
|
|
local DAYS=$((10#0$(TZ=UTC printf '%(%j)T' $CMD_TIME)))
|
|
|
|
local HOUR=$((10#0$(TZ=UTC printf '%(%H)T' $CMD_TIME)))
|
|
|
|
local MINS=$((10#0$(TZ=UTC printf '%(%M)T' $CMD_TIME)))
|
|
|
|
local SECS=$((10#0$(TZ=UTC printf '%(%S)T' $CMD_TIME)))
|
2022-03-01 20:11:44 +00:00
|
|
|
|
|
|
|
# Choose whether or not to print hours minutes and seconds
|
2022-03-06 14:11:27 +00:00
|
|
|
[[ $CMD_TIME -ge 31536000 ]] && printf '%sy ' ${YEAR:-0}
|
|
|
|
[[ $CMD_TIME -ge 86400 ]] && printf '%sd ' ${DAYS:-0}
|
|
|
|
[[ $CMD_TIME -ge 3600 ]] && printf '%sh ' ${HOUR:-0}
|
|
|
|
[[ $CMD_TIME -ge 60 ]] && printf '%sm ' ${MINS:-0}
|
|
|
|
[[ $CMD_TIME -ge 1 ]] && printf '%ss ' ${SECS:-0}
|
|
|
|
printf '%sμs' ${CMD_US:-0}
|
2022-03-01 17:32:57 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
__sig() {
|
2022-03-01 20:11:44 +00:00
|
|
|
# Giant switch case for getting the name of the signal (`kill -l`)
|
2022-03-06 14:11:27 +00:00
|
|
|
j=0
|
|
|
|
for i in $@; do
|
|
|
|
if [[ $j != 0 ]]; then
|
2022-04-18 17:48:29 +00:00
|
|
|
printf '%s|%s' "$_WHT" "$_RED"
|
2022-03-06 14:11:27 +00:00
|
|
|
fi
|
|
|
|
j=$((j+1))
|
|
|
|
case $i in
|
|
|
|
126) printf ACCES ;;
|
|
|
|
127) printf NOENT ;;
|
|
|
|
129) printf HUP ;;
|
|
|
|
130) printf INT ;;
|
|
|
|
131) printf QUIT ;;
|
|
|
|
132) printf ILL ;;
|
|
|
|
133) printf TRAP ;;
|
|
|
|
134) printf ABRT ;;
|
|
|
|
135) printf BUS ;;
|
|
|
|
136) printf FPE ;;
|
|
|
|
137) printf KILL ;;
|
|
|
|
138) printf USR1 ;;
|
|
|
|
139) printf SEGV ;;
|
|
|
|
140) printf USR2 ;;
|
|
|
|
141) printf PIPE ;;
|
|
|
|
142) printf ALRM ;;
|
|
|
|
143) printf TERM ;;
|
|
|
|
144) printf STKFLT ;;
|
|
|
|
145) printf CHLD ;;
|
|
|
|
146) printf CONT ;;
|
|
|
|
147) printf STOP ;;
|
|
|
|
148) printf TSTP ;;
|
|
|
|
149) printf TTIN ;;
|
|
|
|
150) printf TTOU ;;
|
|
|
|
151) printf URG ;;
|
|
|
|
152) printf XCPU ;;
|
|
|
|
153) printf XFSZ ;;
|
|
|
|
154) printf VTALRM ;;
|
|
|
|
155) printf PROF ;;
|
|
|
|
156) printf WINCH ;;
|
|
|
|
157) printf IO ;;
|
|
|
|
158) printf PWR ;;
|
|
|
|
159) printf SYS ;;
|
2022-03-10 19:47:07 +00:00
|
|
|
16[3-9]|1[7-8][0-9]|19[0-2]) printf RT$(($i-128)) ;; # Savagery
|
2022-03-06 14:11:27 +00:00
|
|
|
*) printf $i ;; # Print exit code if not in list
|
|
|
|
esac
|
|
|
|
done
|
2022-03-01 17:32:57 +00:00
|
|
|
}
|
|
|
|
|
2022-04-18 17:48:29 +00:00
|
|
|
__ssh() {
|
|
|
|
local CON=($SSH_CONNECTION)
|
|
|
|
local SRV_IP="${CON[2]}"
|
|
|
|
[[ -z "$SRV_IP" ]] && return
|
|
|
|
local SRV_PORT="${CON[3]}"
|
|
|
|
# 4 chars from startand 4 chars from end
|
|
|
|
local SRV_IP_CUT="${_CYN}${SRV_IP}"
|
|
|
|
[[ ${#SRV_IP} -gt 8 ]] && local SRV_IP_CUT="${_CYN}${SRV_IP:0:4}${_WHT}*${_CYN}${SRV_IP: -4}"
|
|
|
|
local RET="${_GRN}${_BLU}[${SRV_IP_CUT}${_PRP}${_BLD}:${_NON}${_CYN}${SRV_PORT}${_BLU}]${_NON}"
|
|
|
|
printf '%s' "$RET"
|
|
|
|
}
|
|
|
|
|
2022-03-01 17:32:57 +00:00
|
|
|
__prompt() {
|
2022-03-01 20:11:44 +00:00
|
|
|
# Get exit code
|
2022-03-06 14:11:27 +00:00
|
|
|
local PLC=(${PIPESTATUS[@]})
|
2022-03-02 15:58:27 +00:00
|
|
|
|
2022-03-06 14:11:27 +00:00
|
|
|
# *_RT may not be set
|
2022-03-02 17:56:48 +00:00
|
|
|
LAST_RT="${LAST_RT:-$EPOCHREALTIME}"
|
2022-03-06 14:11:27 +00:00
|
|
|
PROMPT_RT="${PROMPT_RT:-$EPOCHREALTIME}"
|
2022-03-02 17:56:48 +00:00
|
|
|
|
2022-03-01 20:11:44 +00:00
|
|
|
# Reset time when prompt was first displayed after command
|
2022-03-06 14:11:27 +00:00
|
|
|
if [[ ${INC_TIME:-1} != 0 ]]; then
|
2022-03-02 15:58:27 +00:00
|
|
|
PROMPT_RT="$EPOCHREALTIME"
|
2022-03-01 20:23:03 +00:00
|
|
|
INC_TIME=0
|
2022-03-01 17:32:57 +00:00
|
|
|
fi
|
2022-03-01 20:11:44 +00:00
|
|
|
|
|
|
|
# Get relative times
|
2022-03-02 06:29:04 +00:00
|
|
|
|
2022-03-08 19:15:01 +00:00
|
|
|
# Remove decimal point, simulating multiplying by 1 million
|
|
|
|
PROMPT_RT1M="${PROMPT_RT%.*}${PROMPT_RT#*.}"
|
|
|
|
LAST_RT1M="${LAST_RT%.*}${LAST_RT#*.}"
|
|
|
|
|
2022-03-02 06:29:04 +00:00
|
|
|
# Subtract and divide by 1 million to get a more accurate difference
|
|
|
|
CMD_TIME="$(((PROMPT_RT1M-LAST_RT1M)/1000000))"
|
2022-03-02 15:58:27 +00:00
|
|
|
CMD_TIME="${CMD_TIME:-0}"
|
2022-03-02 06:29:04 +00:00
|
|
|
|
2022-04-18 14:04:27 +00:00
|
|
|
CMD_US="$(($PROMPT_RT1M-$LAST_RT1M))"
|
2022-03-19 04:02:50 +00:00
|
|
|
[[ ${#CMD_US} -lt 6 ]] || CMD_US="${CMD_US: -6}"
|
2022-04-18 14:04:27 +00:00
|
|
|
CMD_US="$((10#0$CMD_US))"
|
2022-03-01 20:11:44 +00:00
|
|
|
|
2022-03-02 04:33:34 +00:00
|
|
|
# Set prompt sections
|
|
|
|
|
2022-03-02 04:55:56 +00:00
|
|
|
# Colors
|
2022-04-18 17:48:29 +00:00
|
|
|
_NON="\[\033[0m\]"
|
|
|
|
_BLD="\[\033[1m\]" # YEP BALD
|
|
|
|
_BLK="\[\033[30m\]"
|
|
|
|
_RED="\[\033[31m\]"
|
|
|
|
_GRN="\[\033[32m\]"
|
|
|
|
_YLW="\[\033[33m\]"
|
|
|
|
_BLU="\[\033[34m\]"
|
|
|
|
_PRP="\[\033[35m\]"
|
|
|
|
_CYN="\[\033[36m\]"
|
|
|
|
_WHT="\[\033[37m\]"
|
2022-03-02 04:55:56 +00:00
|
|
|
|
|
|
|
# Text
|
|
|
|
|
2022-04-18 17:48:29 +00:00
|
|
|
# ssh detection and indicator
|
|
|
|
[[ "$SSH_CONNECTION" ]] && local SSH="$(__ssh) "
|
|
|
|
|
2022-03-02 04:33:34 +00:00
|
|
|
# [INT], [4], etc.
|
2022-03-06 14:11:27 +00:00
|
|
|
for i in ${PLC[@]}; do
|
|
|
|
if [[ $i > 0 ]]; then
|
2022-04-18 17:48:29 +00:00
|
|
|
local SIG="$(printf '%s[%s%s%s] ' "$_BLU" "$_RED" "$(__sig ${PLC[@]})" "$_BLU")"
|
2022-03-06 14:11:27 +00:00
|
|
|
break
|
|
|
|
fi
|
|
|
|
done
|
2022-03-02 04:55:56 +00:00
|
|
|
|
2022-03-02 04:33:34 +00:00
|
|
|
# [user@homeserver:~]
|
2022-04-18 17:48:29 +00:00
|
|
|
local COL="$([[ $UID == 0 ]] && printf '%s' "$_RED" || printf '%s' "$_YLW")"
|
|
|
|
local UHD="${_BLD}${_BLU}[${COL}\u${_PRP}@${_CYN}\h${_PRP}:${_GRN}\w${_BLU}]"
|
2022-03-02 04:55:56 +00:00
|
|
|
|
2022-03-02 04:33:34 +00:00
|
|
|
# 2y 351d 12m 43s 382969μs
|
2022-04-18 17:48:29 +00:00
|
|
|
local TIME="${_YLW}$(__get_cmd_time)"
|
2022-03-02 04:55:56 +00:00
|
|
|
|
2022-03-02 04:33:34 +00:00
|
|
|
# Random colored $ or #
|
2022-03-02 04:55:56 +00:00
|
|
|
local RAND256="\[\033[38;2;$RANDOM;$RANDOM;${RANDOM}m\]"
|
2022-04-18 14:04:27 +00:00
|
|
|
local IND="${RAND256}\\$"
|
2022-03-02 04:33:34 +00:00
|
|
|
|
2022-03-01 20:11:44 +00:00
|
|
|
# Set the prompt
|
2022-04-18 17:48:29 +00:00
|
|
|
PS1="${_NON}${SSH}${SIG}${UHD} ${TIME} ${IND} ${_NON}"
|
2022-03-01 17:32:57 +00:00
|
|
|
}
|
2022-03-02 04:33:34 +00:00
|
|
|
|
2022-03-01 20:23:03 +00:00
|
|
|
PROMPT_COMMAND=__prompt
|