Update nezha.sh

nezha.sh

@@ -38,7 +38,7 @@ mustn() {
 }
 
 deps_check() {
-    deps="curl wget unzip grep"
+    deps="curl wget unzip grep openssl"
     for dep in $deps; do
         if ! command -v "$dep" >/dev/null 2>&1; then
             err "未找到依赖 $dep，请先安装。"
@@ -47,21 +47,6 @@ 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
@@ -115,7 +100,7 @@ install() {
     echo "> 安装"
 
     # 创建哪吒监控目录
-    mkdir -p "$NZ_DASHBOARD_PATH"
+    mkdir -p "$NZ_DASHBOARD_PATH/data"
 
     modify_config
 
@@ -125,26 +110,34 @@ 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_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"}
-
-    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"
+    nz_tls=${NZ_TLS:-"true"}
+    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_port}"
+  root: "./static"
+grpc:
+  listen: "0.0.0.0:${nz_port}"
+  tls:
+    enable: ${nz_tls}
+    cert: "./server.crt"
+    key: "./server.key"
+security:
+  csrf: "${csrf_token}"
+EOF
 
     # 确保数据目录具有写权限
     chmod -R 777 "${NZ_DASHBOARD_PATH}/data"