EdgeOne 获取用户真实 IP 完整解决方案（Nginx 适配版）

#!/bin/bash
# ==============================================
# EdgeOne 海外节点 IP 段自动更新脚本（Nginx 适配版）
# 功能：接口获取IP段 → 替换配置 → 验证重启 → 备份回滚
# 适配：CentOS + 宝塔面板 + 海外业务
# ==============================================

# ---------------------- 核心参数（需根据实际调整）----------------------
NGINX_CONF="/www/server/panel/vhost/nginx/python_music.conf"  # 你的站点Nginx配置路径
IP_API="https://api.edgeone.ai/ips?version=v4&area=overseas"  # 海外IPv4专属接口
TIMEOUT=30  # 接口超时时间（秒）
RETRY=3     # 接口失败重试次数
# ---------------------------------------------------------------------------

# 临时文件（规避Sed长字符串语法问题）
TMP_IP_FILE=$(mktemp)
TMP_NGINX_CONF=$(mktemp)
BACKUP_CONF="${NGINX_CONF}.bak.$(date +%Y%m%d)"  # 带日期的配置备份文件

# 1. 接口获取IP段（适配海外网络，带重试+证书兼容）
echo "[$(date +'%Y-%m-%d %H:%M:%S')] 从官方接口获取EdgeOne海外IP段（最多重试$RETRY次）..."
for ((i=1; i<=$RETRY; i++)); do
    # curl参数说明：跳过证书校验、设置超时、失败重试
    curl -s --connect-timeout 10 --max-time $TIMEOUT \
         --retry $RETRY --retry-delay 3 \
         --insecure \
         "$IP_API" > "$TMP_IP_FILE"

    # 验证获取结果：非空且包含标准IPv4网段格式
    if [ -s "$TMP_IP_FILE" ] && grep -qE '^([0-9]{1,3}\.){3}[0-9]{1,3}/[0-9]{1,2}$' "$TMP_IP_FILE"; then
        echo "[$(date +'%Y-%m-%d %H:%M:%S')] ✅ 第$i次尝试成功，获取到有效IP段"
        break
    else
        echo "[$(date +'%Y-%m-%d %H:%M:%S')] ❌ 第$i次尝试失败，正在重试..."
        if [ $i -eq $RETRY ]; then
            echo "[$(date +'%Y-%m-%d %H:%M:%S')] ❌ 所有重试失败，无法获取IP段，退出脚本"
            rm -f "$TMP_IP_FILE" "$TMP_NGINX_CONF"
            exit 1
        fi
    fi
done

# 2. 过滤无效IP段并生成Nginx格式配置
echo "[$(date +'%Y-%m-%d %H:%M:%S')] 过滤无效IP段并生成配置..."
# 仅保留标准IPv4网段，剔除异常内容
grep -E '^([0-9]{1,3}\.){3}[0-9]{1,3}/[0-9]{1,2}$' "$TMP_IP_FILE" > "$TMP_IP_FILE.tmp"
mv "$TMP_IP_FILE.tmp" "$TMP_IP_FILE"

# 检查过滤后是否有有效IP段
if [ ! -s "$TMP_IP_FILE" ]; then
    echo "[$(date +'%Y-%m-%d %H:%M:%S')] ❌ 过滤后无有效IP段，退出脚本"
    rm -f "$TMP_IP_FILE" "$TMP_NGINX_CONF"
    exit 1
fi

# 生成带标记的完整IP配置（写入临时文件，避免Sed拼接过长）
IP_CONFIG_FILE=$(mktemp)
echo "# ====== EDGEONE_IP_START 自动更新标记（请勿删除）======" > "$IP_CONFIG_FILE"
echo "# 自动更新时间：$(date +'%Y-%m-%d %H:%M:%S')" >> "$IP_CONFIG_FILE"
echo "# EdgeOne 海外节点IP段（从官方接口获取）" >> "$IP_CONFIG_FILE"
# 将IP段转换为Nginx的set_real_ip_from格式
awk '{print "    set_real_ip_from " $0 ";"}' "$TMP_IP_FILE" >> "$IP_CONFIG_FILE"
echo "# ====== EDGEONE_IP_END 自动更新标记（请勿删除）======" >> "$IP_CONFIG_FILE"

# 3. 备份原Nginx配置（避免替换失败导致服务异常）
cp "$NGINX_CONF" "$BACKUP_CONF"
echo "[$(date +'%Y-%m-%d %H:%M:%S')] ✅ 原配置已备份至：$BACKUP_CONF"

# 4. 替换配置文件中的IP段（文件分段拼接，兼容长内容）
echo "[$(date +'%Y-%m-%d %H:%M:%S')] 开始替换Nginx配置中的IP段..."
# 步骤1：提取标记前的内容
sed -n '/# ====== EDGEONE_IP_START 自动更新标记（请勿删除）======/q;p' "$NGINX_CONF" > "$TMP_NGINX_CONF"
# 步骤2：插入新生成的IP配置
cat "$IP_CONFIG_FILE" >> "$TMP_NGINX_CONF"
# 步骤3：提取标记后的内容（排除结束标记本身，避免重复）
sed -n '/# ====== EDGEONE_IP_END 自动更新标记（请勿删除）======/,$!d;/# ====== EDGEONE_IP_END 自动更新标记（请勿删除）======/!p' "$NGINX_CONF" >> "$TMP_NGINX_CONF"

# 覆盖原配置并保持文件权限（宝塔默认Nginx配置权限为644）
cp "$TMP_NGINX_CONF" "$NGINX_CONF"
chmod 644 "$NGINX_CONF"

# 5. 验证配置替换结果
if grep -q "set_real_ip_from 43.174.0.0/15;" "$NGINX_CONF"; then
    echo "[$(date +'%Y-%m-%d %H:%M:%S')] ✅ IP段成功写入Nginx配置"
else
    echo "[$(date +'%Y-%m-%d %H:%M:%S')] ❌ IP段写入失败，正在恢复原配置..."
    cp "$BACKUP_CONF" "$NGINX_CONF"
    rm -f "$TMP_IP_FILE" "$TMP_NGINX_CONF" "$IP_CONFIG_FILE"
    exit 1
fi

# 6. 测试Nginx配置并重启服务
echo "[$(date +'%Y-%m-%d %H:%M:%S')] 测试Nginx配置语法有效性..."
if /www/server/nginx/sbin/nginx -t; then
    echo "[$(date +'%Y-%m-%d %H:%M:%S')] ✅ 配置语法正确，重启Nginx服务..."
    /www/server/nginx/sbin/nginx -s reload
    IP_COUNT=$(wc -l < "$TMP_IP_FILE")
    echo "[$(date +'%Y-%m-%d %H:%M:%S')] 🎉 全部操作完成！本次共更新$IP_COUNT个海外IP段"
else
    echo "[$(date +'%Y-%m-%d %H:%M:%S')] ❌ 配置语法错误，正在恢复原配置..."
    cp "$BACKUP_CONF" "$NGINX_CONF"
    /www/server/nginx/sbin/nginx -s reload
    rm -f "$TMP_IP_FILE" "$TMP_NGINX_CONF" "$IP_CONFIG_FILE"
    exit 1
fi

# 清理临时文件
rm -f "$TMP_IP_FILE" "$TMP_NGINX_CONF" "$IP_CONFIG_FILE"

# 1. 给脚本添加可执行权限（仅需执行一次）
chmod +x /root/update_edgeone_ips.sh

# 2. 手动执行脚本，验证流程是否正常
/root/update_edgeone_ips.sh