#!/bin/bash
# 此脚本主要针对CentOS 6 or 7 版本进行zabbix-agent添加

# 以root用户运行脚本
if [ $(id -u) != "0" ]; then
    echo "You must be root to run this script ..."
    exit 1
fi

if [ $# != "1" ]; then
    echo "Give me ZABBIX server IP ..."
    exit 2
fi

# selinux 会阻止服务开启，所以关闭
sed -i ‘/SELINUX/s/enforcing/disabled/‘ /etc/selinux/config
setenforce 0

if [ -f /etc/redhat-release ];then
        Edition=`grep -o ‘[[:digit:]]‘ /etc/redhat-release |head -1`
else
        Edition=`uname`
fi

case "$Edition" in
      7)
        echo -e "\n\033[34mStart installation……\033[0m\n"
        rpm -ivh http://repo.zabbix.com/zabbix/3.0/rhel/7/x86_64/zabbix-release-3.0-1.el7.noarch.rpm
        yum -y install zabbix-agent zabbix-sender
        sed -i  "/Server/s/Server=127.0.0.1/Server=$1/" /etc/zabbix/zabbix_agentd.conf
        sed -i  "/Server/s/ServerActive=127.0.0.1/ServerActive=$1/" /etc/zabbix/zabbix_agentd.conf
        systemctl start zabbix-agent
        systemctl enable zabbix-agent
        ;;
      6)
        echo -e "\n\033[34mStart installation……\033[0m\n"
        rpm -ivh http://repo.zabbix.com/zabbix/3.0/rhel/6/x86_64/zabbix-release-3.0-1.el6.noarch.rpm
        yum -y install zabbix-agent zabbix-sender
        sed -i  "/Server/s/Server=127.0.0.1/Server=$1/" /etc/zabbix/zabbix_agentd.conf
        sed -i  "/Server/s/ServerActive=127.0.0.1/ServerActive=$1/" /etc/zabbix/zabbix_agentd.conf
        /etc/init.d/zabbix-agent start
        chkconfig zabbix-agent on
        ;;
    *)
        echo "========================================================================"
        echo "This system is not CentOS 6 or 7, script execution failed!!"
        echo "========================================================================"
        exit 3
esac

if ss -ntl|grep 10050 &>/dev/null; then
    echo -e "\n\033[32mCongratulations, ZABBIX client installed successfully\033[0m\n"
else
    echo -e "\n\033[31mSorry, please rerun the command or contact the administrator\033[0m\n"
fi

zabbix-client-install