#!/usr/bin/env bash
# Usage: bash <(curl -L -s https://git.io/fxxlb) setup
####### 安装使用原理 本脚本实现自动安装################
# apt-get install vnstat
# ip addr 查看网卡名称是否是 eth0，如果是网卡名是 ens3 或者 venet0
# 编辑/etc/vnstat.conf 替换，重启vnstat服务(本脚本自动能自动修改网卡名)
# sed -i "s/eth0/ens3/g" /etc/vnstat.conf
# systemctl restart vnstat
# crontab -e 修改定时任务, 添加运行脚本
# # netlog.sh 定时执行转html脚本，每小时一次，为了修改方便和多台机器用，直接到github更新
# 59 * * * * wget -qO- git.io/fxxlb | bash
######################################################
# 输出网络流量日志到html
output_html(){
# html 写文件位置
if [ ! -e '/etc/redhat-release' ]; then
INDEX_HTML=/var/www/html/index.html
mkdir -p /var/www/html/
else
INDEX_HTML=/usr/share/nginx/html/index.html
fi
echo '<!DOCTYPE html><meta charset=utf-8><pre>' > ${INDEX_HTML}
top -b | head -5 >> ${INDEX_HTML}
vnstat -u
vnstat -m >> ${INDEX_HTML}
vnstat -d >> ${INDEX_HTML}
vnstat -h >> ${INDEX_HTML}
echo ' ' >> ${INDEX_HTML}
echo ' Usage: bash <(curl -L -s https://git.io/fxxlb) setup ' >> ${INDEX_HTML}
}
# 安装 vnstat 添加定期运行
vnstat_install(){
# 判断系统 安装软件
if [ ! -e '/etc/redhat-release' ]; then
# debian 系安装
apt -y install vnstat nginx
else
# centos 系安装 vnstat nginx，如果web没法访问，需要关防火墙
yum -y install vnstat nginx
systemctl enable nginx
systemctl restart nginx
fi
# vps网卡如果不是eth0，修改成实际网卡
ni=$(ls /sys/class/net | awk {print} | grep -e eth. -e ens. -e venet.)
if [ $ni != "eth0" ]; then
sed -i "s/eth0/${ni}/g" /etc/vnstat.conf
fi
systemctl restart vnstat
# 设置定时运行脚本
crontab -l >> crontab.txt
echo "59 * * * * wget -qO- git.io/fxxlb | bash" >> crontab.txt
crontab crontab.txt
sleep 2
if [ ! -e '/etc/redhat-release' ]; then
systemctl restart cron
else
systemctl restart crond
fi
rm -f crontab.txt
echo "vnstat conf @ /etc/vnstat.conf"
}
if [[ $# > 0 ]]; then
key="$1"
case $key in
setup)
vnstat_install
;;
esac
else
# 输出网络流量信息到html文件
output_html
fi