82 lines
2.1 KiB
Bash
82 lines
2.1 KiB
Bash
#!/usr/bin/bash
|
|
#
|
|
|
|
_get_wrappers_names() {
|
|
grep "alerts_wrapper" -Rs /etc/nagios/ | grep -v -E "^\s*#" | awk '{ for (i=1 ; i<=NF; i++) { if ($i ~ /^(-n|--name)$/) { print $(i+1); break } } }' | tr ',' '\n' | sort | uniq
|
|
}
|
|
|
|
_monitoringctl_completion() {
|
|
local cur=${COMP_WORDS[COMP_CWORD]};
|
|
local prev=${COMP_WORDS[COMP_CWORD-1]};
|
|
|
|
local action=""
|
|
for w in "${COMP_WORDS[@]}"; do
|
|
case "$w" in
|
|
status|check|enable|disable)
|
|
action="${w}"
|
|
;;
|
|
esac
|
|
done
|
|
|
|
local words="--help"
|
|
case "${action}" in
|
|
status)
|
|
words="${words}"
|
|
;;
|
|
check)
|
|
local checks="$(_get_wrappers_names)"
|
|
local check=""
|
|
for w in "${COMP_WORDS[@]}"; do
|
|
for c in ${checks}; do
|
|
if [ "${c}" == "${w}" ]; then
|
|
check="${w}"
|
|
break
|
|
fi
|
|
done
|
|
done
|
|
if [ -z "${check}" ]; then
|
|
words="--bypass-nrpe ${checks} ${words}"
|
|
else
|
|
words="--bypass-nrpe ${words}"
|
|
fi
|
|
;;
|
|
enable)
|
|
if [ "${prev}" == "enable" ]; then
|
|
words="all $(_get_wrappers_names)"
|
|
else
|
|
words="--comment ${words}"
|
|
fi
|
|
;;
|
|
disable)
|
|
if [ "${prev}" == "disable" ]; then
|
|
words="all $(_get_wrappers_names)"
|
|
elif [ "${prev}" == "-d" ] || [ "${prev}" == "--during" ]; then
|
|
words="1d 1d12h 1h 1h30m 1m 1m30s 30s"
|
|
else
|
|
words="--during --comment ${words}"
|
|
fi
|
|
;;
|
|
*)
|
|
words="status check enable disable ${words}"
|
|
;;
|
|
esac
|
|
|
|
# Avoid double
|
|
opts=();
|
|
for i in ${words}; do
|
|
for j in "${COMP_WORDS[@]}"; do
|
|
if [[ "$i" == "$j" ]]; then
|
|
continue 2
|
|
fi
|
|
done
|
|
opts+=("$i")
|
|
done
|
|
|
|
COMPREPLY=($(compgen -W "${opts[*]}" -- "${cur}"))
|
|
return 0
|
|
|
|
}
|
|
|
|
complete -F _monitoringctl_completion monitoringctl.sh
|
|
|