mirror of
https://github.com/limosek/zaf.git
synced 2024-11-22 02:19:06 +01:00
460 lines
12 KiB
Bash
Executable File
460 lines
12 KiB
Bash
Executable File
#!/bin/sh
|
|
|
|
if [ -z "$secondstage" ]; then
|
|
ZAF_CFG_FILE="/etc/zaf.conf"
|
|
if [ -f $ZAF_CFG_FILE ]; then
|
|
. $ZAF_CFG_FILE
|
|
else
|
|
echo "Missing config file $ZAF_CFG_FILE! Exiting."
|
|
exit 2
|
|
fi
|
|
|
|
# Read options as config for ZAF
|
|
for pair in "$@"; do
|
|
if echo $pair | grep -qE '^ZAF\_(.*)='; then
|
|
option=$(echo $pair|cut -d '=' -f 1)
|
|
value=$(echo $pair|cut -d '=' -f 2-)
|
|
eval "${option}='$value'"
|
|
export secondstage=1
|
|
else
|
|
params="$params $pair"
|
|
fi
|
|
done
|
|
export $(set |grep ^ZAF_ | cut -d '=' -f 1)
|
|
# If some variables in cmd were stripped, rerun only with right arguments and exported variables
|
|
[ -n "$secondstage" ] && exec $0 $params
|
|
fi
|
|
|
|
if [ -f ./lib/zaf.lib.sh ]; then
|
|
. ./lib/zaf.lib.sh
|
|
. ./lib/plugin.lib.sh
|
|
. ./lib/os.lib.sh
|
|
. ./lib/ctrl.lib.sh
|
|
. ./lib/cache.lib.sh
|
|
. ./lib/zbxapi.lib.sh
|
|
[ -f ./lib/zaf.${ZAF_OS}.sh ] && . ./lib/zaf.${ZAF_OS}.sh
|
|
else
|
|
. ${ZAF_LIB_DIR}/zaf.lib.sh
|
|
. ${ZAF_LIB_DIR}/plugin.lib.sh
|
|
. ${ZAF_LIB_DIR}/os.lib.sh
|
|
. ${ZAF_LIB_DIR}/ctrl.lib.sh
|
|
. ${ZAF_LIB_DIR}/cache.lib.sh
|
|
. ${ZAF_LIB_DIR}/zbxapi.lib.sh
|
|
[ -f ${ZAF_LIB_DIR}/zaf.${ZAF_OS}.sh ] && . ${ZAF_LIB_DIR}/zaf.${ZAF_OS}.sh
|
|
fi
|
|
|
|
[ -z "$ZAF_TMP_BASE" ] && ZAF_TMP_BASE=/tmp/zaf
|
|
ZAF_TMP_DIR="${ZAF_TMP_BASE}-$(zaf_random)"
|
|
[ -z "$ZAF_CACHE_DIR" ] && ZAF_CACHE_DIR=${ZAF_TMP_BASE}c
|
|
|
|
[ -z "$ZAF_DEBUG" ] && ZAF_DEBUG=1
|
|
export ZAF_DEBUG
|
|
export ZAF_LOG_STDERR="-s"
|
|
|
|
# If debug is on, do not remove tmp dir
|
|
if [ "$ZAF_DEBUG" -le 3 ]; then
|
|
trap "rm -rf ${ZAF_TMP_DIR}" EXIT
|
|
else
|
|
trap 'zaf_wrn "Leaving $ZAF_TMP_DIR" contents due to ZAF_DEBUG.' EXIT
|
|
fi
|
|
|
|
! [ -d "${ZAF_TMP_DIR}" ] && mkdir "${ZAF_TMP_DIR}"
|
|
if ! [ -d "${ZAF_CACHE_DIR}" ]; then
|
|
mkdir "${ZAF_CACHE_DIR}"
|
|
fi
|
|
|
|
if zaf_is_root; then
|
|
chgrp $(id -g zabbix) "${ZAF_CACHE_DIR}"
|
|
chmod 770 "${ZAF_CACHE_DIR}"
|
|
fi
|
|
|
|
case $1 in
|
|
|
|
cache-clean)
|
|
zaf_cache_clean
|
|
;;
|
|
cache-list)
|
|
zaf_cache_list
|
|
;;
|
|
|
|
version)
|
|
echo "$ZAF_VERSION<git $ZAF_GITBRANCH>"
|
|
[ $ZAF_DEBUG -gt 1 ] && set |grep -E ^ZAF_
|
|
;;
|
|
|
|
|
|
###### Agent related commands
|
|
userparms)
|
|
for plugin in $(zaf_list_plugins); do
|
|
plugindir=$ZAF_PLUGINS_DIR/$plugin
|
|
control=$plugindir/control.zaf
|
|
zaf_ctrl_generate_cfg "${control}" "${plugin}" "noscripts" \
|
|
| zaf_far '{PLUGINDIR}' "${plugindir}"
|
|
done
|
|
;;
|
|
agent-config)
|
|
zaf_wrn "Generating config files in $ZAF_AGENT_CONFIGD ..."
|
|
for plugin in $(zaf_list_plugins); do
|
|
plugindir=$ZAF_PLUGINS_DIR/$plugin
|
|
control=$plugindir/control.zaf
|
|
zaf_dbg "Generating ${ZAF_AGENT_CONFIGD}/zaf_${plugin}.conf"
|
|
zaf_ctrl_generate_cfg "${control}" "${plugin}" \
|
|
| zaf_far '{PLUGINDIR}' "${plugindir}" >${ZAF_AGENT_CONFIGD}/zaf_${plugin}.conf
|
|
done
|
|
;;
|
|
|
|
###### Plugins related commands
|
|
update)
|
|
zaf_wrn "Updating repository ${ZAF_REPO_GITURL}..."
|
|
zaf_update_repo
|
|
;;
|
|
upgrade)
|
|
ZAF_DEBUG=$ZAF_DEBUG $0 reinstall $(zaf_list_plugins)
|
|
;;
|
|
show)
|
|
shift;
|
|
if [ -z "$1" ]; then
|
|
zaf_list_plugins | while read plugin; do
|
|
zaf_plugin_info $ZAF_PLUGINS_DIR/$plugin/control.zaf
|
|
done
|
|
else
|
|
if echo $1 | grep -vq "\."; then
|
|
if zaf_list_plugins | grep -q "^$1"; then
|
|
zaf_is_plugin $1 && zaf_plugin_info $ZAF_PLUGINS_DIR/$1/control.zaf || zaf_err "Plugin $1 not installed."
|
|
else
|
|
if echo $1 |grep -q ^http; then
|
|
zaf_prepare_plugin "$1" "$ZAF_TMP_DIR/plugin"
|
|
zaf_plugin_info "$ZAF_TMP_DIR/plugin/control.zaf"
|
|
else
|
|
zaf_err "Plugin $1 not installed."
|
|
fi
|
|
fi
|
|
else
|
|
zaf_item_info "$1"
|
|
fi
|
|
fi
|
|
;;
|
|
plugins)
|
|
zaf_list_plugins
|
|
;;
|
|
items)
|
|
shift
|
|
if [ -z "$1" ]; then
|
|
zaf_list_items
|
|
else
|
|
zaf_list_plugin_items "$1"
|
|
fi
|
|
;;
|
|
test)
|
|
[ "$USER" != "zabbix" ] && zaf_wrn "You are not zabbix user. Test will be run with your privileges and sudo access!"
|
|
shift
|
|
if echo $1|grep -q '\.'; then
|
|
zaf_test_item "$1"
|
|
exit
|
|
fi
|
|
if [ -n "$1" ]; then
|
|
plugins="$@"
|
|
else
|
|
plugins="$(zaf_list_plugins)"
|
|
fi
|
|
for p in $plugins; do
|
|
! zaf_is_plugin $p && zaf_err "Unknown plugin $p"
|
|
for i in $(zaf_list_plugin_items $p test); do
|
|
zaf_test_item $i
|
|
echo
|
|
done
|
|
done
|
|
;;
|
|
get)
|
|
shift
|
|
if echo $1|grep -q '\.'; then
|
|
zaf_get_item "$1"
|
|
exit
|
|
fi
|
|
if [ -n "$1" ]; then
|
|
plugins="$@"
|
|
else
|
|
plugins="$(zaf_list_plugins)"
|
|
fi
|
|
for p in $plugins; do
|
|
for i in $(zaf_list_plugin_items $p test); do
|
|
echo $i: $(zaf_get_item $i)
|
|
echo
|
|
done
|
|
done
|
|
;;
|
|
precache)
|
|
shift
|
|
for i in $*; do
|
|
if zaf_is_plugin $i; then
|
|
for j in $(zaf_list_plugin_items $i precache); do
|
|
item=$(echo $j | cut -d '[' -f 1)
|
|
params=$(echo $j | cut -d '[' -f 2 | cut -d ']' -f 1 | tr ',' ' ')
|
|
zaf_precache_item $params >/dev/null
|
|
done
|
|
else
|
|
item=$(echo $i | cut -d '[' -f 1)
|
|
params=$(echo $i | cut -d '[' -f 2 | cut -d ']' -f 1 | tr ',' ' ')
|
|
zaf_precache_item $params >/dev/null
|
|
fi
|
|
done
|
|
;;
|
|
install)
|
|
shift
|
|
[ -z "$1" ] && echo "$0 install plugin [plugin]..."
|
|
for p in $@; do
|
|
if zaf_is_plugin "$(basename $p)"; then
|
|
zaf_wrn "Plugin $(basename $p) already installed. Skipping installation."
|
|
continue
|
|
fi
|
|
zaf_install_plugin "$p"
|
|
installed=1
|
|
done
|
|
[ -n "$installed" ] && zaf_is_root && zaf_restart_agent
|
|
;;
|
|
reinstall)
|
|
shift
|
|
[ -z "$1" ] && echo "$0 reinstall plugin [plugin]..."
|
|
for p in $@; do
|
|
if zaf_is_plugin "$p"; then
|
|
zaf_remove_plugin "$(basename $p)"
|
|
reinstalled=1
|
|
fi
|
|
zaf_install_plugin "$p"
|
|
reinstalled=1
|
|
done
|
|
[ -n "$reinstalled" ] && zaf_is_root && zaf_restart_agent
|
|
;;
|
|
remove)
|
|
shift
|
|
[ -z "$1" ] && echo "$0 remove plugin [plugin]..."
|
|
for p in $@; do
|
|
if zaf_is_plugin "$p"; then
|
|
zaf_remove_plugin "$p"
|
|
removed=1
|
|
fi
|
|
done
|
|
[ -n "$removed" ] && zaf_is_root && zaf_restart_agent
|
|
;;
|
|
|
|
###### Zaf related commands
|
|
self-upgrade)
|
|
shift
|
|
[ -z "$1" ] && auto=auto
|
|
zaf_os_specific zaf_check_deps zaf && zaf_err "Zaf is installed as system package. Cannot self-upgrade."
|
|
if ! which curl >/dev/null;
|
|
then
|
|
zaf_err "Curl not found. Cannot continue. Please install it."
|
|
fi
|
|
zaf_fetch_url $ZAF_RAW_URL/$ZAF_GITBRANCH/install.sh | exec sh -s $auto "$@"
|
|
exit
|
|
;;
|
|
self-remove)
|
|
shift
|
|
zaf_os_specific zaf_check_deps zaf && zaf_err "Zaf is installed as system package. Cannot self-remove."
|
|
. /etc/zaf.conf
|
|
if [ "$1" = "force" ]; then
|
|
rm -rf /etc/zaf.conf ${ZAF_PLUGINS_DIR} ${ZAF_REPO_DIR} ${ZAF_LIB_DIR} \
|
|
${ZAF_BIN_DIR}/zaf ${ZAF_AGENT_CONFIGD}/zaf_*
|
|
else
|
|
echo "This will remove zaf from this computer and erase all configuration."
|
|
echo "This command will be executed:"
|
|
echo "rm -rf /etc/zaf.conf ${ZAF_PLUGINS_DIR} ${ZAF_REPO_DIR} ${ZAF_LIB_DIR} \
|
|
${ZAF_BIN_DIR}/zaf ${ZAF_AGENT_CONFIGD}/zaf_*"
|
|
echo
|
|
echo "To continue, please do $0 self-remove force"
|
|
fi
|
|
;;
|
|
###### API related commands
|
|
api)
|
|
zaf_zbxapi_login
|
|
case $2 in
|
|
get-*-id)
|
|
obj=$(echo $2|cut -d '-' -f 2)
|
|
[ -z "$3" ] && zaf_err "$0 $1 $2 <$obj>"
|
|
eval zaf_zbxapi_get${obj}id "$3"
|
|
;;
|
|
get-byid-*|get-id-*)
|
|
obj=$(echo $2|cut -d '-' -f 3)
|
|
[ -z "$3" ] && zaf_err "$0 $1 $2 <id>"
|
|
eval zaf_zbxapi_get${obj} "$3" "$4"
|
|
;;
|
|
get-host-ids)
|
|
if [ -n "$3" ]; then
|
|
gid=$(zaf_zbxapi_gethostgroupid "$3") || exit 1
|
|
zaf_dbg "Selecting all hosts in group $3($gid)"
|
|
zaf_zbxapi_gethostsingroup $gid
|
|
else
|
|
zaf_dbg "Selecting all hosts in system"
|
|
zaf_zbxapi_gethostids
|
|
fi
|
|
;;
|
|
get-inventory)
|
|
[ -z "$3" ] && zaf_err "$0 $1 field [host]. If host unspecified, use hostname"
|
|
if [ -z "$4" ]; then
|
|
host=$(hostname)
|
|
else
|
|
host="$4"
|
|
fi
|
|
zaf_zbxapi_gethostinventory $host $3
|
|
;;
|
|
get-template-ids)
|
|
if [ -n "$3" ]; then
|
|
gid=$(zaf_zbxapi_gethostgroupid "$3") || exit 1
|
|
zaf_dbg "Selecting all templates in group $3($gid)"
|
|
zaf_zbxapi_gettemplatesingroup $gid
|
|
else
|
|
zaf_dbg "Selecting all templates in system"
|
|
zaf_zbxapi_gettemplateids
|
|
fi
|
|
;;
|
|
get-map-ids)
|
|
zaf_zbxapi_getmapid
|
|
;;
|
|
export-hosts)
|
|
shift; shift
|
|
[ -z "$1" ] && zaf_err "$0 api export-hosts dir [hostgroup]"
|
|
dir="$1"
|
|
shift
|
|
if [ -n "$1" ]; then
|
|
gid=$(zaf_zbxapi_gethostgroupid "$1") || exit 1
|
|
shift
|
|
hosts=$(zaf_zbxapi_gethostsingroup $gid)
|
|
else
|
|
hosts=$(zaf_zbxapi_gethostids)
|
|
fi
|
|
zaf_wrn "Will backup this hosts: $hosts"
|
|
zaf_wrn "Output dir: $dir"
|
|
for h in $hosts; do
|
|
if zaf_bglimit 5; then
|
|
(
|
|
hn=$(zaf_zbxapi_gethost $h name)
|
|
zaf_wrn "Exporting host $hn($h)..."
|
|
zaf_zbxapi_export_host $h >"$dir/$hn.xml"
|
|
) &
|
|
else
|
|
hn=$(zaf_zbxapi_gethost $h name)
|
|
zaf_wrn "Exporting host $hn($h)..."
|
|
zaf_zbxapi_export_host $h >"$dir/$hn.xml"
|
|
fi
|
|
done
|
|
wait
|
|
;;
|
|
export-host)
|
|
shift; shift
|
|
hostid=$(zaf_zbxapi_gethostid "$1") || exit 1
|
|
zaf_wrn "Exporting host $3($hostid)..."
|
|
zaf_zbxapi_export_host $hostid
|
|
;;
|
|
export-template)
|
|
shift; shift
|
|
templateid=$(zaf_zbxapi_gettemplateid "$1") || exit 1
|
|
zaf_wrn "Exporting template $3($hostid)..."
|
|
zaf_zbxapi_export_template $templateid
|
|
;;
|
|
export-templates)
|
|
shift; shift
|
|
[ -z "$1" ] && zaf_err "$0 api export-templates dir"
|
|
dir="$1"
|
|
templates=$(zaf_zbxapi_gettemplateids)
|
|
zaf_wrn "Will backup this templates: $templates"
|
|
zaf_wrn "Output dir: $dir"
|
|
for t in $templates; do
|
|
if zaf_bglimit 5; then
|
|
(
|
|
tn=$(zaf_zbxapi_gettemplate $t name)
|
|
zaf_wrn "Exporting template $tn($t)..."
|
|
zaf_zbxapi_export_template $t >"$dir/$tn.xml"
|
|
) &
|
|
else
|
|
tn=$(zaf_zbxapi_gettemplate $t name)
|
|
zaf_wrn "Exporting template $tn($t)..."
|
|
zaf_zbxapi_export_template $t >"$dir/$tn.xml"
|
|
fi
|
|
done
|
|
wait
|
|
;;
|
|
import-template)
|
|
shift; shift
|
|
if zaf_is_plugin $1; then
|
|
if [ -f "$ZAF_PLUGINS_DIR/$1/template.xml" ]; then
|
|
template="$ZAF_PLUGINS_DIR/$1/template.xml"
|
|
zaf_wrn "Importing template $template"
|
|
zaf_zbxapi_import_config $template || zaf_err "Error importing template"
|
|
else
|
|
url="$(zaf_plugin_option $1 Template)"
|
|
if [ -n "$url" ]; then
|
|
zaf_fetch_url $url | zaf_zbxapi_import_config || zaf_err "Error importing template"
|
|
else
|
|
url="$(zaf_plugin_option $1 Url)"
|
|
zaf_fetch_url $url/template.xml | zaf_zbxapi_import_config || zaf_err "Error importing template"
|
|
fi
|
|
fi
|
|
else
|
|
if [ -f $1 ]; then
|
|
zaf_wrn "Importing template $1"
|
|
zaf_zbxapi_import_config $1 || zaf_err "Error importing template"
|
|
else
|
|
zaf_err "Unknown plugin $1!"
|
|
fi
|
|
fi
|
|
;;
|
|
*)
|
|
echo "$0 api command [parameters]"
|
|
for i in host template map; do
|
|
zaf_hlp "get-${i}-id $i" "Get $i id"
|
|
zaf_hlp "get-byid-${i} id [property]" "Get $i property from id. Leave empty property for JSON"
|
|
done
|
|
zaf_hlp "get-inventory host [fields]" "Get inventory fields [or all fields]"
|
|
zaf_hlp "get-host-ids [hg]" "Get all hostids or hostids in group hg"
|
|
zaf_hlp "get-template-ids [hg]" "Get all templateids or templateids in group hg"
|
|
zaf_hlp "get-map-ids" "Get all mapids"
|
|
zaf_hlp "export-hosts dir [hg]" "Backup all hosts [in group hg] (get their config from zabbix and save to dir/hostname.xml)"
|
|
zaf_hlp "export-host host" "Backup host (get config from zabbix to stdout)"
|
|
zaf_hlp "import-template {plugin|file}" "Import template for plugin or from file"
|
|
zaf_hlp "export-template name" "Export template to stdout"
|
|
zaf_hlp "export-templates dir" "Export all template to dir"
|
|
echo
|
|
[ -n "$2" ] && zaf_err "Bad API command '$2'!"
|
|
;;
|
|
esac
|
|
;;
|
|
|
|
*)
|
|
echo "$0 Version ${ZAF_VERSION}. Please use some of this commands:"
|
|
echo "$0 Cmd [ZAF_OPTION=value] [ZAF_CTRL_Option=value] [ZAF_CTRLI_Item_Option=value] ..."
|
|
echo "Plugin manipulation commands:"
|
|
zaf_hlp "$0 update" "To update repo (not plugins, similar to apt-get update)"
|
|
zaf_hlp "$0 upgrade" "To upgrade installed plugins from repo"
|
|
zaf_hlp "$0 install plugin" "To install plugin"
|
|
zaf_hlp "$0 remove plugin" "To remove plugin"
|
|
echo
|
|
echo "Plugin info commands:"
|
|
zaf_hlp "$0 plugins" "To list installed plugins"
|
|
zaf_hlp "$0 show [plugin]" "To show installed plugins or plugin info"
|
|
zaf_hlp "$0 items [plugin]" "To list all suported items [for plugin]"
|
|
echo
|
|
echo "Plugin diagnostic commands:"
|
|
zaf_hlp "$0 test [plugin[.item]]" "To test [all] suported items by zabbix_agentd [for plugin]"
|
|
zaf_hlp "$0 get [plugin[.item]]" "To test [all] suported items by zabbix_get [for plugin]"
|
|
zaf_hlp "$0 precache [plugin[.item]]" "To precache [all] suported items"
|
|
echo
|
|
echo "Zabbix API commands:"
|
|
zaf_hlp "$0 api" "To zabbix API functions. See $0 api for more info."
|
|
echo
|
|
echo "Agent config info commands:"
|
|
zaf_hlp "$0 userparms" "See userparms generated from zaf on stdout"
|
|
zaf_hlp "$0 agent-config" "Reconfigure zabbix userparms in $ZAF_AGENT_CONFIGD"
|
|
echo
|
|
echo "Zaf related commands:"
|
|
zaf_hlp "$0 self-upgrade" "To self-upgrade zaf"
|
|
zaf_hlp "$0 self-remove" "To self-remove zaf and its config"
|
|
zaf_hlp "$0 cache-clean" "To remove all entries from cache"
|
|
zaf_hlp "$0 cache-list" "To show all entries in cache"
|
|
echo
|
|
[ -n "$1" ] && zaf_err "Bad command '$1'!"
|
|
;;
|
|
|
|
esac
|
|
|