Create nezha.sh

nezha.sh

@@ -0,0 +1,183 @@
+#!/bin/sh
+
+NZ_BASE_PATH="/opt/nezha"
+NZ_DASHBOARD_PATH="${NZ_BASE_PATH}/dashboard"
+
+red='\033[0;31m'
+green='\033[0;32m'
+yellow='\033[0;33m'
+plain='\033[0m'
+
+err() {
+    printf "${red}%s${plain}\n" "$*" >&2
+}
+
+warn() {
+    printf "${red}%s${plain}\n" "$*"
+}
+
+success() {
+    printf "${green}%s${plain}\n" "$*"
+}
+
+info() {
+    printf "${yellow}%s${plain}\n" "$*"
+}
+
+println() {
+    printf "$*\n"
+}
+
+mustn() {
+    set -- "$@"
+    if ! "$@" >/dev/null 2>&1; then
+        err "运行 '$*' 失败。"
+        exit 1
+    fi
+}
+
+deps_check() {
+    deps="curl wget unzip grep"
+    for dep in $deps; do
+        if ! command -v "$dep" >/dev/null 2>&1; then
+            err "未找到依赖 $dep，请先安装。"
+            exit 1
+        fi
+    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
+        amd64|x86_64)
+            os_arch="amd64"
+            ;;
+        i386|i686)
+            os_arch="386"
+            ;;
+        aarch64|arm64)
+            os_arch="arm64"
+            ;;
+        *arm*)
+            os_arch="arm"
+            ;;
+        s390x)
+            os_arch="s390x"
+            ;;
+        riscv64)
+            os_arch="riscv64"
+            ;;
+        *)
+            err "未知架构：$uname"
+            exit 1
+            ;;
+    esac
+}
+
+init() {
+    deps_check
+    env_check
+    ## 检测是否为国内IP
+    if [ -z "$CN" ]; then
+        geo_check
+        if [ -n "$isCN" ]; then
+            CN=true
+            GITHUB_RAW_URL="gitee.com/naibahq/scripts/raw/main"
+            GITHUB_URL="gitee.com"
+            Docker_IMG="registry.cn-shanghai.aliyuncs.com\/naibahq\/nezha"
+        else
+            GITHUB_RAW_URL="raw.githubusercontent.com/nezhahq/scripts/main"
+            GITHUB_URL="github.com"
+            Docker_IMG="ghcr.io\/nezhahq\/nezha"
+        fi
+    fi
+
+    IS_DOCKER_NEZHA=0
+}
+
+install() {
+    echo "> 安装"
+
+    # 创建哪吒监控目录
+    mkdir -p $NZ_DASHBOARD_PATH
+
+    modify_config
+
+    restart_and_update
+}
+
+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:-8008}
+    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}/" /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
+
+    success "Dashboard 配置修改成功"
+}
+
+restart_and_update() {
+    echo "> 重启并更新"
+
+    restart_and_update_standalone
+
+    success "哪吒监控更新成功"
+}
+
+restart_and_update_standalone() {
+    _version=$(curl -m 10 -sL "https://api.github.com/repos/nezhahq/nezha/releases/latest" | grep "tag_name" | head -n 1 | awk -F ":" '{print $2}' | sed 's/\"//g;s/,//g;s/ //g')
+    if [ -z "$_version" ]; then
+        err "获取 Dashboard 版本号失败，请检查网络连接"
+        exit 1
+    else
+        echo "当前最新版本为： ${_version}"
+    fi
+
+    if [ -z "$CN" ]; then
+        NZ_DASHBOARD_URL="https://${GITHUB_URL}/nezhahq/nezha/releases/download/${_version}/dashboard-linux-${os_arch}.zip"
+    else
+        NZ_DASHBOARD_URL="https://${GITHUB_URL}/naibahq/nezha/releases/download/${_version}/dashboard-linux-${os_arch}.zip"
+    fi
+
+    wget -qO $NZ_DASHBOARD_PATH/app.zip "$NZ_DASHBOARD_URL" && \
+    unzip -qq -o $NZ_DASHBOARD_PATH/app.zip -d $NZ_DASHBOARD_PATH && \
+    mv $NZ_DASHBOARD_PATH/dashboard-linux-$os_arch $NZ_DASHBOARD_PATH/app && \
+    rm $NZ_DASHBOARD_PATH/app.zip
+    chmod +x $NZ_DASHBOARD_PATH/app
+}
+
+init
+
+install