【TrafficCop大更新】支持TG推送！

话说预设配置大家可以帮忙pr啊，总不能全让我一个人写吧家人们

可以多个小鸡用一个bot 还是每个小鸡都要申请？

CDT那个配置文件我看写的限制是300 是为啥

@jeez #1 代码逻辑上，只要你每个小鸡都装脚本，可以共用bot，但是目前我还没做机器名区分的代码，之后慢慢弄吧，先跑起来再说

@rshinesun #2 脑子抽了没注意到，我改一下

@bacon159 #3 等大佬升级 一个bot能区分服务器

我敲 我刚把bark推送加上
你竟然更新了tg

@芷理 #6 那你速速pr哈哈哈哈，最好是单独脚本，不要干扰主脚本哈哈，现在已经又臭又长了，一点都不敢动了

前排支持。感谢，加鸡腿

@-- #8 谢谢支持！可以测试使用下

@bacon159 #7
哈哈 我就自己用的 很粗糙
只是在主脚本加了两句话 限速或者关机的时候先curl了一下

# 推送URL
PUSH_URL=""

...
# 修改 check_and_limit_traffic 函数
check_and_limit_traffic() {
    local current_usage=$(get_traffic_usage)
    local limit_threshold=$(echo "$TRAFFIC_LIMIT - $TRAFFIC_TOLERANCE" | bc)
    
    echo "$(date '+%Y-%m-%d %H:%M:%S') 当前使用流量: $current_usage GB，限制流量: $limit_threshold GB" | tee -a "$LOG_FILE"
    
    if (( $(echo "$current_usage > $limit_threshold" | bc -l) )); then
        echo "$(date '+%Y-%m-%d %H:%M:%S') 流量超出限制" | tee -a "$LOG_FILE"
        if [ "$LIMIT_MODE" = "tc" ]; then
            echo "$(date '+%Y-%m-%d %H:%M:%S') 使用 TC 模式限速" | tee -a "$LOG_FILE"
            if [ -n "$PUSH_URL" ]; then
                curl -X GET "$PUSH_URL/Aliyun%E6%9C%8D%E5%8A%A1%E5%99%A8%E6%B5%81%E9%87%8F%E8%AD%A6%E5%91%8A%2F%E5%BD%93%E5%89%8D%E4%BD%BF%E7%94%A8%E6%B5%81%E9%87%8F%3A%20$current_usage%20GB%0A%E9%99%90%E5%88%B6%E6%B5%81%E9%87%8F%3A%20$limit_threshold%20GB%0A%E5%B7%B2%E9%99%90%E9%80%9F"
            fi
            tc qdisc add dev $MAIN_INTERFACE root tbf rate ${LIMIT_SPEED}kbit burst 32kbit latency 400ms
        elif [ "$LIMIT_MODE" = "shutdown" ]; then
            echo "$(date '+%Y-%m-%d %H:%M:%S') 流量超出限制，系统将在 1 分钟后关机" | tee -a "$LOG_FILE"
            if [ -n "$PUSH_URL" ]; then
                curl -X GET "$PUSH_URL/Aliyun%E6%9C%8D%E5%8A%A1%E5%99%A8%E6%B5%81%E9%87%8F%E8%AD%A6%E5%91%8A%2F%E6%B5%81%E9%87%8F%E8%B6%85%E5%87%BA%E9%99%90%E5%88%B6%EF%BC%8C%E7%B3%BB%E7%BB%9F%E5%B0%86%E5%9C%A8%201%20%E5%88%86%E9%92%9F%E5%90%8E%E5%85%B3%E6%9C%BA"
            fi
            shutdown -h +1 "流量超出限制，系统将在 1 分钟后关机"
        fi
    else
        echo "$(date '+%Y-%m-%d %H:%M:%S') 流量正常，清除所有限制" | tee -a "$LOG_FILE"
        tc qdisc del dev $MAIN_INTERFACE root 2>/dev/null
        shutdown -c 2>/dev/null  # 取消可能存在的关机计划
    fi
}