219 lines
8.0 KiB
Bash
219 lines
8.0 KiB
Bash
# shellcheck shell=sh
|
|
|
|
## force() runs a command sine die
|
|
force() {
|
|
if [ $# -gt 0 ]; then
|
|
while true; do
|
|
"$@"
|
|
sleep 1
|
|
done
|
|
fi
|
|
}
|
|
|
|
## force8() runs a command sine die if not already running
|
|
force8() {
|
|
if [ $# -gt 0 ]; then
|
|
while true; do
|
|
# awk expression to match $@
|
|
[ "$(ps wwx -o args 2>/dev/null |awk -v field="${PS_X_FIELD:-1}" '
|
|
BEGIN {nargs=split("'"$*"'",args)}
|
|
# if match first field
|
|
$field == args[1] {
|
|
matched=1;
|
|
# match following fields
|
|
for (i=1;i<=NF-field;i++) {
|
|
if ($(i+field) == args[i+1]) {matched++}
|
|
}
|
|
# all fields matched
|
|
if (matched == nargs) {found++}
|
|
}
|
|
END {print found+0}'
|
|
)" = 0 ] && "$@"
|
|
sleep 1
|
|
done
|
|
fi
|
|
}
|
|
|
|
## load_average() prints the current load average
|
|
load_average() {
|
|
awk '{printf "%.1f\n" $1}' /proc/loadavg 2>/dev/null \
|
|
|| uptime 2>/dev/null |awk '{printf "%.1f\n", $(NF-2)}'
|
|
}
|
|
|
|
## process_count() prints number of "processes"/"running processes"/"D-state"
|
|
process_count() {
|
|
ps ax -o stat 2>/dev/null |awk '
|
|
$1 ~ /R/ {process_running++};
|
|
$1 ~ /D/ {process_dstate++};
|
|
END {
|
|
print NR-1"/"process_running+0"/"process_dstate+0;
|
|
}'
|
|
}
|
|
|
|
## prompt_set() exports custom PROMPT_COMMAND
|
|
prompt_set() {
|
|
case "${TERM}" in
|
|
screen*)
|
|
ESCAPE_CODE_DCS="\033k"
|
|
ESCAPE_CODE_ST="\033\\"
|
|
;;
|
|
linux*|xterm*|rxvt*)
|
|
ESCAPE_CODE_DCS="\033]0;"
|
|
ESCAPE_CODE_ST="\007"
|
|
;;
|
|
*)
|
|
;;
|
|
esac
|
|
# in a screen
|
|
if [ -n "${STY}" ]; then
|
|
export PROMPT_COMMAND='printf "${ESCAPE_CODE_DCS:-\033]0;}%s${ESCAPE_CODE_ST:-\007}" "${PWD##*/}"'
|
|
else
|
|
export PROMPT_COMMAND='printf "${ESCAPE_CODE_DCS:-\033]0;}%s@%s:%s${ESCAPE_CODE_ST:-\007}" "${USER}" "${HOSTNAME%%.*}" "${PWD##*/}"'
|
|
fi
|
|
unset ESCAPE_CODE_DCS ESCAPE_CODE_ST
|
|
}
|
|
|
|
## ps1_set() exports custom PS1
|
|
ps1_set() {
|
|
case "$0" in
|
|
*sh)
|
|
COLOR_DGRAY="\[\033[1;30m\]"
|
|
COLOR_RED="\[\033[01;31m\]"
|
|
COLOR_GREEN="\[\033[01;32m\]"
|
|
COLOR_BROWN="\[\033[0;33m\]"
|
|
COLOR_YELLOW="\[\033[01;33m\]"
|
|
COLOR_BLUE="\[\033[01;34m\]"
|
|
COLOR_CYAN="\[\033[0;36m\]"
|
|
COLOR_GRAY="\[\033[0;37m\]"
|
|
COLOR_NC="\[\033[0m\]"
|
|
;;
|
|
*)
|
|
;;
|
|
esac
|
|
|
|
PS1_COUNT="${COLOR_DGRAY}[${COLOR_BLUE}\$(process_count 2>/dev/null)${COLOR_DGRAY}|${COLOR_BLUE}\$(user_count 2>/dev/null)${COLOR_DGRAY}|${COLOR_BLUE}\$(load_average 2>/dev/null)${COLOR_DGRAY}]${COLOR_NC}"
|
|
PS1_END="${COLOR_DGRAY}\$(if [ \"\$(id -u)\" = 0 ]; then printf \"#\"; else printf \"\$\"; fi)${COLOR_NC}"
|
|
if type __git_ps1 >/dev/null 2>&1; then
|
|
PS1_GIT="\$(__git_ps1 2>/dev/null \" (%s)\")"
|
|
else
|
|
PS1_GIT="\$(BRANCH=\$(git rev-parse --abbrev-ref HEAD 2>/dev/null); [ -n \"\${BRANCH}\" ] && printf \" (\${BRANCH})\")"
|
|
fi
|
|
PS1_GIT="${COLOR_CYAN}${PS1_GIT}${COLOR_NC}"
|
|
PS1_HOSTNAME_COLOR="\`case \"\${ENV}\" in [Pp][Rr][0Oo][Dd]*) printf \"${COLOR_RED}\";; *) if [ -n \"\${ENV}\" ]; then printf \"${COLOR_YELLOW}\"; else printf \"${COLOR_GREEN}\"; fi;; esac\`"
|
|
PS1_HOSTNAME="${PS1_HOSTNAME_COLOR}\$(hostname |sed 's/\..*//')${COLOR_NC}"
|
|
PS1_USER_COLOR="\$(if [ \"\$(id -u)\" = 0 ]; then printf \"${COLOR_RED}\"; else printf \"${COLOR_BROWN}\"; fi)"
|
|
PS1_USER="${PS1_USER_COLOR}\$(id -nu)${COLOR_NC}"
|
|
PS1_WORKDIR="${COLOR_GRAY}\$(pwd |sed 's|^'\${HOME}'\(/.*\)*$|~\1|')${COLOR_NC}"
|
|
|
|
export PS1="${PS1_COUNT}${PS1_USER}${COLOR_DGRAY}@${PS1_HOSTNAME}${COLOR_DGRAY}:${PS1_WORKDIR}${PS1_GIT}${PS1_END} "
|
|
unset PS1_COUNT PS1_END PS1_GIT PS1_HOSTNAME PS1_HOSTNAME_COLOR PS1_USER PS1_USER_COLOR PS1_WORKDIR
|
|
}
|
|
|
|
## screen_attach() attaches existing screen session or creates a new one
|
|
screen_attach() {
|
|
command -v screen >/dev/null 2>&1 || return
|
|
SCREEN_SESSION="$(id -nu)@$(hostname |sed 's/\..*//')"
|
|
if [ -z "${STY}" ]; then
|
|
# attach screen in tmux window 0 only ;)
|
|
[ -n "${TMUX}" ] && [ "$(tmux list-window 2>/dev/null |awk '$NF == "(active)" {print $1}' |sed 's/:$//')" != "0" ] && return
|
|
printf 'Attaching screen.' && sleep 1 && printf '.' && sleep 1 && printf '.' && sleep 1
|
|
exec screen -xRR -S "${SCREEN_SESSION}"
|
|
fi
|
|
unset SCREEN_SESSION
|
|
}
|
|
|
|
## screen_detach() detaches current screen session
|
|
screen_detach() {
|
|
screen -d
|
|
}
|
|
|
|
|
|
## ssh_add() loads all private keys in ~/.ssh/ to ssh agent
|
|
ssh_add() {
|
|
command -v ssh-agent >/dev/null 2>&1 && command -v ssh-add >/dev/null 2>&1 || return
|
|
SSH_AGENT_DIR="/tmp/ssh-$(id -u)"
|
|
SSH_AGENT_SOCK="${SSH_AGENT_DIR}/agent@$(hostname |sed 's/\..*//')"
|
|
# launch a new agent
|
|
if [ -z "${SSH_AUTH_SOCK}" ]; then
|
|
[ ! -d "${SSH_AGENT_DIR}" ] && mkdir -p "${SSH_AGENT_DIR}" 2>/dev/null && chmod 0700 "${SSH_AGENT_DIR}"
|
|
# search for an already running agent
|
|
if ps wwx -o args |awk '$1 ~ "ssh-agent$" && $3 == "'"${SSH_AGENT_SOCK}"'"' |wc -l |grep -q 0; then
|
|
rm -f "${SSH_AGENT_SOCK}"
|
|
ssh-agent -a "${SSH_AGENT_SOCK}" >/dev/null 2>&1
|
|
fi
|
|
fi
|
|
# attach to agent
|
|
export SSH_AUTH_SOCK="${SSH_AUTH_SOCK:-${SSH_AGENT_SOCK}}"
|
|
# list private keys to add
|
|
# shellcheck disable=SC2068
|
|
for dir in ${@:-${HOME}/.ssh}; do
|
|
if [ "${SSH_ADD_RECURSIVE:-}" = true ]; then
|
|
GREP_RECURSIVE_FLAG="r"
|
|
else
|
|
GREP_RECURSIVE_CHAR="*"
|
|
fi
|
|
SSH_PRIVATE_KEYS="${SSH_PRIVATE_KEYS:-} ${dir}/id_rsa $(grep -l${GREP_RECURSIVE_FLAG:-} 'PRIVATE KEY' "${dir}"/"${GREP_RECURSIVE_CHAR:-}" 2>/dev/null |grep -vw "${dir}"/id_rsa)"
|
|
done
|
|
# shellcheck disable=SC2086
|
|
printf '%s\n' ${SSH_PRIVATE_KEYS} |while read -r file; do
|
|
[ -r "${file}" ] || continue
|
|
# add private key to agent
|
|
ssh-add -l |grep -q "$(ssh-keygen -lf "${file}" 2>/dev/null |awk '{print $2}')" 2>/dev/null || ssh-add "${file}"
|
|
done
|
|
unset GREP_RECURSIVE_CHAR GREP_RECURSIVE_FLAG SSH_AGENT_DIR SSH_AGENT_SOCK SSH_PRIVATE_KEYS
|
|
}
|
|
|
|
## ssh_del() removes all private keys in ~/.ssh/ from ssh agent
|
|
ssh_del() {
|
|
command -v ssh-add >/dev/null 2>&1 || return
|
|
# attach to agent
|
|
if [ -z "${SSH_AUTH_SOCK}" ]; then
|
|
return
|
|
fi
|
|
# list private keys to del
|
|
# shellcheck disable=SC2068
|
|
for dir in ${@:-${HOME}/.ssh}; do
|
|
if [ "${SSH_DEL_RECURSIVE:-}" = true ]; then
|
|
GREP_RECURSIVE_FLAG="r"
|
|
else
|
|
GREP_RECURSIVE_CHAR="*"
|
|
fi
|
|
SSH_PRIVATE_KEYS="${SSH_PRIVATE_KEYS:-} ${dir}/id_rsa $(grep -l${GREP_RECURSIVE_FLAG:-} 'PRIVATE KEY' "${dir}"/"${GREP_RECURSIVE_CHAR:-}" 2>/dev/null |grep -vw "${dir}"/id_rsa)"
|
|
done
|
|
# shellcheck disable=SC2086
|
|
printf '%s\n' ${SSH_PRIVATE_KEYS} |while read -r file; do
|
|
[ -r "${file}" ] || continue
|
|
# remove private key from agent
|
|
ssh-add -l |grep -q "$(ssh-keygen -lf "${file}" 2>/dev/null |awk '{print $2}')" 2>/dev/null && ssh-add -d "${file}"
|
|
done
|
|
unset GREP_RECURSIVE_CHAR GREP_RECURSIVE_FLAG SSH_PRIVATE_KEYS
|
|
}
|
|
|
|
## tmux_attach() attaches existing tmux session or creates a new one
|
|
tmux_attach() {
|
|
command -v tmux >/dev/null 2>&1 || return
|
|
TMUX_SESSION="$(id -nu)@$(hostname |sed 's/\..*//')"
|
|
if [ -z "${TMUX}" ]; then
|
|
printf 'Attaching tmux.' && sleep 1 && printf '.' && sleep 1 && printf '.' && sleep 1
|
|
exec tmux -L"${TMUX_SESSION}" new-session -A -s"${TMUX_SESSION}"
|
|
fi
|
|
unset TMUX_SESSION
|
|
}
|
|
|
|
## tmux_detach() detaches current tmux session
|
|
tmux_detach() {
|
|
tmux detach
|
|
}
|
|
|
|
## user_count() prints number of "users sessions"/"users"/"logged users"
|
|
user_count() {
|
|
ps ax -o user,tty 2>/dev/null |awk '
|
|
$2 ~ /^(pts|tty)/ {users_session++; logged[$1]++;};
|
|
{count[$1]++;}
|
|
END {
|
|
for (uc in count) {c = c" "uc;}; users_count=split(c,v," ")-1;
|
|
for (ul in logged) {l = l" "ul;}; users_logged=split(l,v," ")-1;
|
|
print users_session+0"/"users_count"/"users_logged;
|
|
}'
|
|
}
|