Panabit Shell脚本实现定时更新IP群组地址列表

#!/bin/bash

# 定义变量
json_file='latest.json'          # JSON文件名
timing_exec='24'                 # 定时执行的小时数（24小时制）
http_url='https://res.mail.qq.com/zh_CN/wework_ip/latest.json'  # HTTP URL
testing_project='json ping curl'  # 测试项目列表
IP="$(pwd)/ipaddress.txt"       # IP地址文件路径

# 定义测试任务函数
function testing_task() {
    case "$1" in 
        json)
            # 检查jq工具是否存在
            if command -v jq > /dev/null; then
                return 0
            else
                return 127
            fi
        ;;
        ping)
            # 执行ping测试
            if ping -c 4 223.5.5.5 > /dev/null; then
                return 0
            else
                return 126
            fi
        ;;
        curl)
            # 执行curl HTTP请求测试
            if curl -o /dev/null -s -w "%{http_code}" -k "$http_url" | grep -qE "200|301|302"; then
                return 0
            else
                return 125
            fi
        ;;
        *)
            return 255
        ;;
    esac
}

# 定义测试执行函数
function testing_exec() {
    successes=0  # 初始化成功测试计数器
    echo "Running time $(date '+%Y-%m-%d %H:%M:%S')" >> "$(pwd)/run.log"  # 记录运行开始时间
    
    for i in $testing_project; do
        testing_task "$i"  # 执行测试任务
        status=$?  # 获取测试状态码
        
        if [ "$status" -eq 0 ]; then
            # 如果测试成功，记录到日志
            case "$i" in
                json)
                    echo 'JSON test is successful' >> "$(pwd)/run.log"
                    ((successes++))
                    ;;
                ping)
                    echo 'Ping is successful' >> "$(pwd)/run.log"
                    ((successes++))
                    ;;
                curl)
                    echo 'HTTP request successful' >> "$(pwd)/run.log"
                    ((successes++))
                    ;;
            esac
        else
            # 如果测试失败，记录错误到日志并退出
            case "$status" in
                127)
                    echo "Error: jq is not installed or not in the PATH" >> "$(pwd)/run.log"
                    ;;
                126)
                    echo "Error: Ping failed" >> "$(pwd)/run.log"
                    ;;
                125)
                    echo "Error: HTTP request failed" >> "$(pwd)/run.log"
                    ;;
                *)
                    echo "An unknown error occurred with status $status" >> "$(pwd)/run.log"
                    exit 255
                    ;;
            esac
        fi
    done
}

# 定义筛选IP的函数
function screening() {
    # 使用wget下载JSON文件
    wget -O "$json_file" --no-check-certificate "$http_url"
    # 解析JSON文件并提取IP地址
    jq '.[] | select(.Section == "Section4") | .IPList' $json_file | grep -Eo '((25[0-5]|2[0-4][0-9]|1[0-9]{2}|[1-9][0-9]|[0-9])\.){3}(25[0-5]|2[0-4][0-9]|1[0-9]{2}|[1-9][0-9]|[0-9])' > "$IP"
}

# 定义主运行函数
function run_main() {
    while true; do
        # 检查是否到达设定的执行时间
        if [ "$(date +%H)" -eq "$timing_exec" ]; then
            testing_exec  # 执行测试
            screening     # 执行IP筛选
            # 如果所有测试成功，则执行floweye命令
            if [ "$successes" -eq 3 ]; then
                floweye table loadfile tid=1 file="$IP" clear=1 # tid为IP群组ID号，需要先创建IP群组。
            fi
        fi
        sleep 3600  # 休眠一小时
    done
}

# 运行主函数
run_main