aynic.os
1 year ago
parent
743e874c59
commit
97cd91a020
@ -1,218 +1,283 @@ |
||||
# shellcheck shell=sh |
||||
## rc_function.sh defines customs shell functions |
||||
# author: Yann "aya" Autissier |
||||
# license: MIT |
||||
# updated: 2021/03/04 |
||||
|
||||
## force() runs a command sine die |
||||
force() { |
||||
if [ $# -gt 0 ]; then |
||||
while true; do |
||||
"$@" |
||||
sleep 1 |
||||
done |
||||
fi |
||||
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 |
||||
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); } |
||||
# first field matched |
||||
$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)}' |
||||
} |
||||
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; |
||||
}' |
||||
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 |
||||
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\]" |
||||
;; |
||||
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_RESET="\[\033[0m\]" |
||||
;; |
||||
*) |
||||
;; |
||||
esac |
||||
|
||||
PS1_STATUS="\$?" |
||||
PS1_COUNT="${COLOR_DGRAY}[\` |
||||
case \"$PS1_STATUS\" in |
||||
0) |
||||
printf \"${COLOR_BLUE}${PS1_STATUS}\";; |
||||
1) |
||||
printf \"${COLOR_YELLOW}${PS1_STATUS}\";; |
||||
*) |
||||
;; |
||||
printf \"${COLOR_RED}${PS1_STATUS}\";; |
||||
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}" |
||||
type process_count >/dev/null 2>&1 && printf\ |
||||
\"${COLOR_DGRAY}|${COLOR_BLUE}%s\"\ |
||||
\"\$(process_count 2>/dev/null)\" |
||||
type user_count >/dev/null 2>&1 && printf\ |
||||
\"${PS1_COUNT}${COLOR_DGRAY}|${COLOR_BLUE}%s\"\ |
||||
\"\$(user_count 2>/dev/null)\" |
||||
type load_average >/dev/null 2>&1 && printf\ |
||||
\"${PS1_COUNT}${COLOR_DGRAY}|${COLOR_BLUE}%s\"\ |
||||
\"\$(load_average 2>/dev/null)\" |
||||
\`${COLOR_DGRAY}]${COLOR_RESET}" |
||||
PS1_END="${COLOR_DGRAY}\$( |
||||
if [ \"\$(id -u)\" = 0 ]; then |
||||
printf \"#\"; |
||||
else |
||||
printf \"\$\"; |
||||
fi |
||||
)${COLOR_RESET}" |
||||
PS1_GIT="\$( |
||||
if type __git_ps1 >/dev/null 2>&1; then |
||||
PS1_GIT="\$(__git_ps1 2>/dev/null \" (%s)\")" |
||||
printf \"\$(__git_ps1 2>/dev/null \" (%s)\")\" |
||||
else |
||||
PS1_GIT="\$(BRANCH=\$(git rev-parse --abbrev-ref HEAD 2>/dev/null); [ -n \"\${BRANCH}\" ] && printf \" (\${BRANCH})\")" |
||||
printf \"\$(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 |
||||
)" |
||||
PS1_GIT="${COLOR_CYAN}${PS1_GIT}${COLOR_RESET}" |
||||
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_RESET}" |
||||
PS1_USER_COLOR="\$( |
||||
if [ \"\$(id -u)\" = 0 ]; then |
||||
printf \"${COLOR_RED}\"; |
||||
else |
||||
printf \"${COLOR_BROWN}\"; |
||||
fi |
||||
)" |
||||
PS1_USER="${PS1_USER_COLOR}\$(id -nu):\$(id -u)${COLOR_RESET}" |
||||
PS1_WORKDIR="${COLOR_GRAY}\$( |
||||
pwd |sed 's|^'\${HOME}'\(/.*\)*$|~\1|' |
||||
)${COLOR_RESET}" |
||||
PS1="${PS1_COUNT}${PS1_USER}${COLOR_DGRAY}@${PS1_HOSTNAME}" |
||||
PS1="${PS1}${COLOR_DGRAY}:${PS1_WORKDIR}${PS1_GIT}${PS1_END} " |
||||
export 'PS1' |
||||
unset PS1_COUNT PS1_END PS1_GIT PS1_HOSTNAME PS1_HOSTNAME_COLOR\ |
||||
PS1_USER PS1_USER_COLOR PS1_STATUS 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 |
||||
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 |
||||
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 |
||||
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 |
||||
# 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 |
||||
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 |
||||
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 |
||||
# 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 |
||||
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 |
||||
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 |
||||
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; |
||||
}' |
||||
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; |
||||
}' |
||||
} |
||||
|
||||
# vim:ts=2:sw=2:sts=2:et |
||||
|
||||
@ -1,124 +1,283 @@ |
||||
# force a command to run and restart it when it exits |
||||
force () { |
||||
PS_X_FIELD=1 |
||||
if [ $# -gt 0 ]; then |
||||
# awk expression to match $@ |
||||
while true; do |
||||
[ $(ps wwx -o args |awk ' |
||||
BEGIN {nargs=split("'"$*"'",args)} |
||||
$field == args[1] { |
||||
matched=1; |
||||
for (i=1;i<=NF-field;i++) { |
||||
if ($(i+field) == args[i+1]) {matched++} |
||||
} |
||||
if (matched == nargs) {found++} |
||||
} |
||||
END {print found+0}' field="${PS_X_FIELD}") -eq 0 ] \ |
||||
&& "$@" || sleep 1; |
||||
done |
||||
fi |
||||
} |
||||
# shellcheck shell=sh |
||||
## rc_function.sh defines customs shell functions |
||||
# author: Yann "aya" Autissier |
||||
# license: MIT |
||||
# updated: 2021/03/04 |
||||
|
||||
# start an ssh agent and add any private key in ~/.ssh |
||||
ssh_agent () { |
||||
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/\..*//')" |
||||
[ -z "${SSH_AUTH_SOCK}" ] \ |
||||
&& { [ -d "${SSH_AGENT_DIR}" ] || { mkdir "${SSH_AGENT_DIR}" 2>/dev/null && chmod 0700 "${SSH_AGENT_DIR}"; } } \ |
||||
&& [ $(ps wwx -o args |awk '$1 == "ssh-agent" && $3 == "'"${SSH_AGENT_SOCK}"'"' |wc -l) -eq 0 ] \ |
||||
&& rm -f "${SSH_AGENT_SOCK}" \ |
||||
&& ssh-agent -a "${SSH_AGENT_SOCK}" >/dev/null 2>&1 |
||||
export SSH_AUTH_SOCK="${SSH_AUTH_SOCK:-${SSH_AGENT_SOCK}}" |
||||
(echo "${HOME}"/.ssh/id_rsa; grep -l 'PRIVATE KEY' "${HOME}"/.ssh/* |grep -vE "^${HOME}/.ssh/id_rsa$") |while read -r file; do |
||||
[ -r "${file}" ] && [ -z "$(ssh-add -l |awk '{print $3}' |grep -E "^${file}$")" ] && ssh-add "${file}" |
||||
## force() runs a command sine die |
||||
force() { |
||||
if [ $# -gt 0 ]; then |
||||
while true; do |
||||
"$@" |
||||
sleep 1 |
||||
done |
||||
unset SSH_AGENT_DIR SSH_AGENT_SOCK |
||||
fi |
||||
} |
||||
|
||||
# attach an existing screen or create a new one |
||||
attach_screen () { |
||||
command -v screen >/dev/null 2>&1 || return |
||||
if [ -z "${STY}" ]; then |
||||
# attach screen in tmux window 0 |
||||
[ -n "${TMUX}" ] && [ "$(tmux list-window 2>/dev/null |awk '$NF == "(active)" {print $1}' |sed 's/:$//')" != "0" ] && return |
||||
/bin/echo -n 'Attaching screen.' && sleep 1 && /bin/echo -n '.' && sleep 1 && /bin/echo -n '.' && sleep 1 && screen -xRR -S "$(id -nu)" 2>/dev/null |
||||
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); } |
||||
# first field matched |
||||
$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 |
||||
} |
||||
|
||||
# attach an existing tmux or create a new one |
||||
attach_tmux () { |
||||
command -v tmux >/dev/null 2>&1 || return |
||||
SESSION_NAME="$(id -nu)@$(hostname |sed 's/\..*//')" |
||||
if [ -z "${TMUX}" ]; then |
||||
/bin/echo -n 'Attaching tmux.' && sleep 1 && /bin/echo -n '.' && sleep 1 && /bin/echo -n '.' && sleep 1 && tmux -L"${SESSION_NAME}" -q has-session >/dev/null 2>&1 && tmux -L"${SESSION_NAME}" attach-session -d || tmux -L"${SESSION_NAME}" new-session -s"${SESSION_NAME}" |
||||
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)}' |
||||
} |
||||
|
||||
# echo the "number of running processes"/"total number of processes"/"number of processes in D-state" |
||||
process_count () { |
||||
ps ax -o stat 2>/dev/null |awk '$1 ~ /R/ {process_running++}; $1 ~ /D/ {process_dstate++}; END {print process_running+0"/"NR-1"/"process_dstate+0}' |
||||
## 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; }' |
||||
} |
||||
|
||||
# echo the "number of distinct logged in users"/"number of distinct users running processes"/"number of logged in users" |
||||
user_count () { |
||||
ps ax -o user,tty,comm 2>/dev/null |awk '$2 !~ /^\?/ && $3 !~ /getty$/ {logged[$1]++; tty[$2]++}; {user[$1]++}; END {for (ul in logged) {sl = sl" "ul;}; users_logged=split(sl,a," ")-1; for (uu in user) {su = su" "uu;}; users_user=split(su,a," ")-1; for (ut in tty) {st = st" "ut;}; users_tty=split(st,a," ")-1; print users_logged"/"users_user"/"users_tty}' |
||||
## 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 |
||||
} |
||||
|
||||
# echo the load average |
||||
load_average () { |
||||
awk '{print $1}' /proc/loadavg 2>/dev/null || uptime 2>/dev/null |awk '{print $(NF-2)}' |
||||
} |
||||
## 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_RESET="\[\033[0m\]" |
||||
;; |
||||
*) |
||||
;; |
||||
esac |
||||
|
||||
# export PS1 |
||||
custom_ps1 () { |
||||
case "$0" in |
||||
*ash) |
||||
local DGRAY="\[\033[1;30m\]" |
||||
local RED="\[\033[01;31m\]" |
||||
local GREEN="\[\033[01;32m\]" |
||||
local BROWN="\[\033[0;33m\]" |
||||
local YELLOW="\[\033[01;33m\]" |
||||
local BLUE="\[\033[01;34m\]" |
||||
local CYAN="\[\033[0;36m\]" |
||||
local GRAY="\[\033[0;37m\]" |
||||
local NC="\[\033[0m\]" |
||||
;; |
||||
PS1_STATUS="\$?" |
||||
PS1_COUNT="${COLOR_DGRAY}[\` |
||||
case \"$PS1_STATUS\" in |
||||
0) |
||||
printf \"${COLOR_BLUE}${PS1_STATUS}\";; |
||||
1) |
||||
printf \"${COLOR_YELLOW}${PS1_STATUS}\";; |
||||
*) |
||||
;; |
||||
printf \"${COLOR_RED}${PS1_STATUS}\";; |
||||
esac |
||||
type process_count >/dev/null 2>&1 && printf\ |
||||
\"${COLOR_DGRAY}|${COLOR_BLUE}%s\"\ |
||||
\"\$(process_count 2>/dev/null)\" |
||||
type user_count >/dev/null 2>&1 && printf\ |
||||
\"${PS1_COUNT}${COLOR_DGRAY}|${COLOR_BLUE}%s\"\ |
||||
\"\$(user_count 2>/dev/null)\" |
||||
type load_average >/dev/null 2>&1 && printf\ |
||||
\"${PS1_COUNT}${COLOR_DGRAY}|${COLOR_BLUE}%s\"\ |
||||
\"\$(load_average 2>/dev/null)\" |
||||
\`${COLOR_DGRAY}]${COLOR_RESET}" |
||||
PS1_END="${COLOR_DGRAY}\$( |
||||
if [ \"\$(id -u)\" = 0 ]; then |
||||
printf \"#\"; |
||||
else |
||||
printf \"\$\"; |
||||
fi |
||||
)${COLOR_RESET}" |
||||
PS1_GIT="\$( |
||||
if type __git_ps1 >/dev/null 2>&1; then |
||||
printf \"\$(__git_ps1 2>/dev/null \" (%s)\")\" |
||||
else |
||||
printf \"\$(BRANCH=\$(git rev-parse --abbrev-ref HEAD 2>/dev/null);\ |
||||
[ -n \"\${BRANCH}\" ] && printf \" (\${BRANCH})\")\" |
||||
fi |
||||
)" |
||||
PS1_GIT="${COLOR_CYAN}${PS1_GIT}${COLOR_RESET}" |
||||
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_RESET}" |
||||
PS1_USER_COLOR="\$( |
||||
if [ \"\$(id -u)\" = 0 ]; then |
||||
printf \"${COLOR_RED}\"; |
||||
else |
||||
printf \"${COLOR_BROWN}\"; |
||||
fi |
||||
)" |
||||
PS1_USER="${PS1_USER_COLOR}\$(id -nu):\$(id -u)${COLOR_RESET}" |
||||
PS1_WORKDIR="${COLOR_GRAY}\$( |
||||
pwd |sed 's|^'\${HOME}'\(/.*\)*$|~\1|' |
||||
)${COLOR_RESET}" |
||||
PS1="${PS1_COUNT}${PS1_USER}${COLOR_DGRAY}@${PS1_HOSTNAME}" |
||||
PS1="${PS1}${COLOR_DGRAY}:${PS1_WORKDIR}${PS1_GIT}${PS1_END} " |
||||
export 'PS1' |
||||
unset PS1_COUNT PS1_END PS1_GIT PS1_HOSTNAME PS1_HOSTNAME_COLOR\ |
||||
PS1_USER PS1_USER_COLOR PS1_STATUS 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 |
||||
} |
||||
|
||||
local COLOR="\$([ \"\$(id -u)\" = 0 ] && echo \"${RED}\" || echo \"${BROWN}\")" |
||||
local COUNT="${DGRAY}[${BLUE}\$(user_count 2>/dev/null)${DGRAY}|${BLUE}\$(process_count 2>/dev/null)${DGRAY}|${BLUE}\$(load_average 2>/dev/null)${DGRAY}]" |
||||
local END="\$([ \"\$(id -u)\" = 0 ] && echo \"#\" || echo \"\$\")" |
||||
local HOSTNAME="\$(hostname |sed 's/\..*//')" |
||||
## 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 |
||||
} |
||||
|
||||
type __git_ps1 >/dev/null 2>&1 \ |
||||
&& local GIT="\$(__git_ps1 2>/dev/null \" (%s)\")" \ |
||||
|| local GIT="\$(BRANCH=\$(git rev-parse --abbrev-ref HEAD 2>/dev/null); [ -n \"\${BRANCH}\" ] && echo \" (\${BRANCH})\")" |
||||
## 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 |
||||
} |
||||
|
||||
local USER="\$(id -nu)" |
||||
local WORKDIR="\$(pwd |sed 's|^'\${HOME}'\(/.*\)*$|~\1|')" |
||||
## 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 |
||||
} |
||||
|
||||
export PS1="${COUNT}${COLOR}${USER}${DGRAY}@${CYAN}${HOSTNAME}${DGRAY}:${GRAY}${WORKDIR}${CYAN}${GIT}${DGRAY}${END}${NC} " |
||||
## tmux_detach() detaches current tmux session |
||||
tmux_detach() { |
||||
tmux detach |
||||
} |
||||
|
||||
# export PROMPT_COMMAND |
||||
custom_prompt () { |
||||
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 |
||||
[ -n "${STY}" ] \ |
||||
&& export PROMPT_COMMAND='printf "${ESCAPE_CODE_DCS:-\033]0;}%s${ESCAPE_CODE_ST:-\007}" "${PWD##*/}"' \ |
||||
|| export PROMPT_COMMAND='printf "${ESCAPE_CODE_DCS:-\033]0;}%s@%s:%s${ESCAPE_CODE_ST:-\007}" "${USER}" "${HOSTNAME%%.*}" "${PWD##*/}"' |
||||
## 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; |
||||
}' |
||||
} |
||||
|
||||
# vim:ts=2:sw=2:sts=2:et |
||||
|
||||