Update nezha.sh

nezha.sh

@@ -14,7 +14,7 @@ err() {
 }
 
 warn() {
-    printf "${yellow}%s${plain}\n" "$*"
+    printf "${red}%s${plain}\n" "$*"
 }
 
 success() {
@@ -38,7 +38,7 @@ mustn() {
 }
 
 deps_check() {
-    deps="curl wget unzip grep openssl"
+    deps="curl wget unzip grep"
     for dep in $deps; do
         if ! command -v "$dep" >/dev/null 2>&1; then
             err "未找到依赖 $dep，请先安装。"
@@ -47,6 +47,21 @@ deps_check() {
     done
 }
 
+geo_check() {
+    api_list="https://blog.cloudflare.com/cdn-cgi/trace https://developers.cloudflare.com/cdn-cgi/trace"
+    ua="Mozilla/5.0 (X11; Linux x86_64; rv:60.0) Gecko/20100101 Firefox/81.0"
+    for url in $api_list; do
+        text="$(curl -A "$ua" -m 10 -s "$url")"
+        endpoint="$(echo "$text" | sed -n 's/.*h=\([^ ]*\).*/\1/p')"
+        if echo "$text" | grep -qw 'CN'; then
+            isCN=true
+            break
+        elif echo "$url" | grep -q "$endpoint"; then
+            break
+        fi
+    done
+}
+
 env_check() {
     uname=$(uname -m)
     case "$uname" in
@@ -80,7 +95,8 @@ init() {
     env_check
     ## 检测是否为国内IP
     if [ -z "$CN" ]; then
-        if curl -s https://ipinfo.io/country | grep -q 'CN'; then
+        geo_check
+        if [ -n "$isCN" ]; then
             CN=true
             GITHUB_RAW_URL="gitee.com/naibahq/scripts/raw/main"
             GITHUB_URL="gitee.com"
@@ -99,7 +115,7 @@ install() {
     echo "> 安装"
 
     # 创建哪吒监控目录
-    mkdir -p "$NZ_DASHBOARD_PATH/data"
+    mkdir -p "$NZ_DASHBOARD_PATH"
 
     modify_config
 
@@ -109,35 +125,26 @@ install() {
 modify_config() {
     echo "> 修改配置"
 
+    # 下载配置文件
+    echo "正在下载配置文件"
+
+    wget -t 2 -T 60 -O /tmp/nezha-config.yaml "https://${GITHUB_RAW_URL}/extras/config.yaml"
+
     # 设置配置参数，从环境变量读取，如果没有则使用默认值
     nz_site_title=${NZ_SITE_TITLE:-"哪吒监控"}
-    nz_dashboard_port=${NZ_DASHBOARD_PORT:-"7860"}
-    nz_agent_port=${NZ_AGENT_PORT:-"5555"}
+    nz_port=${NZ_PORT:-"7860"}  # Hugging Face Spaces 默认使用 7860 端口
+    nz_hostport=${NZ_HOSTPORT:-"example.com:443"}
     nz_tls=${NZ_TLS:-"true"}
     nz_lang=${NZ_LANG:-"zh_CN"}
-    nz_space_name=${NZ_SPACE_NAME:-"your-space-name"}
-
-    # 生成随机的 csrf token
-    csrf_token=$(openssl rand -hex 16)
-
-    # 写入配置文件
-    cat <<EOF > ${NZ_DASHBOARD_PATH}/data/config.yaml
-site:
-  brand: "${nz_site_title}"
-  url: "https://${nz_space_name}.hf.space"
-  language: "${nz_lang}"
-http:
-  listen: "0.0.0.0:${nz_dashboard_port}"
-  root: "./static"
-grpc:
-  listen: "0.0.0.0:${nz_agent_port}"
-  tls:
-    enable: ${nz_tls}
-    cert: "./server.crt"
-    key: "./server.key"
-security:
-  csrf: "${csrf_token}"
-EOF
+
+    sed -i "s/nz_port/${nz_port}/g" /tmp/nezha-config.yaml
+    sed -i "s/nz_language/${nz_lang}/" /tmp/nezha-config.yaml
+    sed -i "s/nz_site_title/${nz_site_title}/" /tmp/nezha-config.yaml
+    sed -i "s/nz_hostport/${nz_hostport}/" /tmp/nezha-config.yaml
+    sed -i "s/nz_tls/${nz_tls}/" /tmp/nezha-config.yaml
+
+    mkdir -p "$NZ_DASHBOARD_PATH/data"
+    mv -f /tmp/nezha-config.yaml "${NZ_DASHBOARD_PATH}/data/config.yaml"
 
     # 确保数据目录具有写权限
     chmod -R 777 "${NZ_DASHBOARD_PATH}/data"