From fac90ae03d346e2255fc0fc04491e30fac1108fc Mon Sep 17 00:00:00 2001
From: "xiaobing.wang" <xiaobing.wang@chaitin.com>
Date: Fri, 14 Jun 2024 11:30:56 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E8=87=AA=E5=8A=A8=E6=A3=80=E6=B5=8B?=
 =?UTF-8?q?=E5=9B=BD=E5=AE=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 release/latest/setup.sh   | 8 ++++++++
 release/latest/upgrade.sh | 7 +++++++
 2 files changed, 15 insertions(+)

diff --git a/release/latest/setup.sh b/release/latest/setup.sh
index c0fab9e..6378d72 100755
--- a/release/latest/setup.sh
+++ b/release/latest/setup.sh
@@ -292,6 +292,14 @@ echo "MGT_PORT=9443" >> .env
 echo "POSTGRES_PASSWORD=$(LC_ALL=C tr -dc A-Za-z0-9 </dev/urandom | head -c 32)" >> .env
 echo "SUBNET_PREFIX=$SUBNET_PREFIX" >> .env
 
+if [ -z "$CDN" ]; then
+    if [[ $(curl -s ipinfo.io/country) == "CN" ]]; then
+        CDN=1
+    else
+        CDN=0
+    fi
+fi
+
 if [ $CDN -eq 0 ]; then
     echo "IMAGE_PREFIX=chaitin" >>".env"
 else
diff --git a/release/latest/upgrade.sh b/release/latest/upgrade.sh
index 035a8ca..6477804 100755
--- a/release/latest/upgrade.sh
+++ b/release/latest/upgrade.sh
@@ -272,6 +272,13 @@ grep "MGT_PORT" ".env" >/dev/null || echo "MGT_PORT=9443" >>".env"
 grep "POSTGRES_PASSWORD" ".env" >/dev/null || echo "POSTGRES_PASSWORD=$(LC_ALL=C tr -dc A-Za-z0-9 </dev/urandom | head -c 32)" >>".env"
 grep "SUBNET_PREFIX" ".env" >/dev/null || echo "SUBNET_PREFIX=172.22.222" >>".env"
 
+if [ -z "$CDN" ]; then
+    if [[ $(curl -s ipinfo.io/country) == "CN" ]]; then
+        CDN=1
+    else
+        CDN=0
+    fi
+fi
 
 if [ $CDN -eq 0 ]; then
     sed -i "s/IMAGE_PREFIX=.*/IMAGE_PREFIX=chaitin/g" ".env"