#!/bin/bash

# 自动部署脚本
set -e

# 下载并解压
wget https://dl.9j6.cn/categraf/categraf-v0.4.13-linux-amd64.tar.gz
tar zxvf categraf-v0.4.13-linux-amd64.tar.gz
mv categraf-v0.4.13-linux-amd64 /usr/local/categraf-v0.4.13

# 安装服务
/usr/local/categraf-v0.4.13/categraf --install

# 获取主机 IP 地址
HOST_IP=$(hostname -I | awk '{print $1}')

# 获取主机名称
HOST_NAME=$(hostname)

# 初始化目标 URL
DEFAULT_URL="https://jk.9j6.cn"
TARGET_URL=$DEFAULT_URL

# 根据 IP 网段调整目标 URL
if [[ $HOST_IP =~ ^10\.34\.44\.* ]]; then
  TARGET_URL="http://10.34.44.30:17000"
elif [[ $HOST_IP =~ ^10\.34\.45\.* ]]; then
  TARGET_URL="http://10.34.45.30:17000"
elif [[ $HOST_IP =~ ^172\.168\.100\.* ]]; then
  TARGET_URL="http://172.168.100.30:17000"
elif [[ $HOST_IP =~ ^172\.168\.200\.* ]]; then
  TARGET_URL="http://172.168.200.30:17000"
else
  echo "IP地址网段未匹配，使用默认URL：$DEFAULT_URL"
fi

# 更新配置文件中的 URL
sed -i "s|http://127.0.0.1:17000|${TARGET_URL}|g" /usr/local/categraf-v0.4.13/conf/config.toml

# 更新配置文件中的 hostname
NEW_HOSTNAME="${HOST_NAME}-${HOST_IP}"
sed -i "s|hostname = \"\"|hostname = \"${NEW_HOSTNAME}\"|g" /usr/local/categraf-v0.4.13/conf/config.toml

# 重启服务
systemctl restart categraf

# 清理下载文件
rm -rf categraf-v0.4.13-linux-amd64.tar.gz

# 清空历史记录
history -c

echo "Categraf 部署完成，配置文件已更新："
echo "目标 URL: ${TARGET_URL}"
echo "主机名: ${NEW_HOSTNAME}"

