#!/bin/bash
# 已在 10.0.1.47 执行。工具请求经 Squid 访问 http://172.17.0.1:8777。
# 再跑一次只会改这一行环境变量，并重建两个代理容器。
# 不重建 dify-web-1、api、vLLM、New-API。
set -eu

python3 - <<'PY'
from pathlib import Path
path = Path("/usr/cluster/dify/src/docker/.env")
lines = path.read_text(encoding="utf-8").splitlines()
for index, line in enumerate(lines):
    if line.startswith("SSRF_PROXY_ALLOW_PRIVATE_IPS="):
        lines[index] = "SSRF_PROXY_ALLOW_PRIVATE_IPS=172.17.0.0/16,10.0.0.0/8"
        break
else:
    raise SystemExit("找不到 SSRF_PROXY_ALLOW_PRIVATE_IPS")
path.write_text("\n".join(lines) + "\n", encoding="utf-8")
print("SSRF_PROXY_ALLOW_PRIVATE_IPS=172.17.0.0/16,10.0.0.0/8")
PY

cd /usr/cluster/dify/src/docker
docker compose -p dify up -d --force-recreate --no-deps ssrf_proxy agent_ssrf_proxy