#!/bin/bash

red="\033[1;91m"
green="\e[1;32m"
yellow="\e[1;33m"
plain="\033[0m"
 
NZ_BASE_PATH="/opt/nezha"
NZ_DASHBOARD_PATH="${NZ_BASE_PATH}/dashboard"
NZ_AGENT_PATH="${NZ_BASE_PATH}/agent"
NZ_AGENT_SERVICE="/etc/systemd/system/nezha-agent.service"
NZ_AGENT_SERVICERC="/etc/init.d/nezha-agent"
NZ_DASHBOARD_SERVICE="/etc/systemd/system/nezha-dashboard.service"
NZ_DASHBOARD_SERVICERC="/etc/init.d/nezha-dashboard"
export VERSION=${VERSION:-'17.5'}
export PATH=$PATH:/usr/local/bin
NZ_VERSION="v0.$VERSION"

os_arch=""
[ -e /etc/os-release ] && cat /etc/os-release | grep -i "PRETTY_NAME" | grep -qi "alpine" && os_alpine='1'

pre_check() {

    [ "$os_alpine" != 1 ] && ! command -v systemctl >/dev/null 2>&1 && echo "不支持此系统：未找到 systemctl 命令" && exit 1

    # check root
    [[ $EUID -ne 0 ]] && echo -e "${red}错误: 请使用root用户运行此脚本！${plain}\n" && exit 1

    ## os_arch
    if uname -m | grep -q 'x86_64'; then
        os_arch="amd64"
    elif uname -m | grep -q 'i386\|i686'; then
        os_arch="386"
    elif uname -m | grep -q 'aarch64\|armv8b\|armv8l'; then
        os_arch="arm64"
    elif uname -m | grep -q 'arm'; then
        os_arch="arm"
    elif uname -m | grep -q 's390x'; then
        os_arch="s390x"
    elif uname -m | grep -q 'riscv64'; then
        os_arch="riscv64"
    fi
    
    # Debug information
    echo -e "${yellow}检测到的系统架构: $(uname -m) -> ${os_arch}${plain}"

    ## China_IP
    if [ -z "$CN" ]; then
        if [ ! -z "$isCN" ]; then
            echo "根据geoip api提供的信息，当前IP可能在中国"
            printf "是否选用中国镜像完成安装? [Y/n] (自定义镜像输入 3):"
            read -r input
            case $input in
            [yY][eE][sS] | [yY])
                echo "使用中国镜像"
                CN=true
                ;;

            [nN][oO] | [nN])
                echo "不使用中国镜像"
                ;;

            [3])
                echo "使用自定义镜像"
                printf "请输入自定义镜像 (例如:dn-dao-github-mirror.daocloud.io),留空为不使用: "
                read -r input
                case $input in
                *)
                    CUSTOM_MIRROR=$input
                    ;;
                esac

                ;;
            *)
                echo "使用中国镜像"
                CN=true
                ;;
            esac
        fi
    fi

    if [ -n "$CUSTOM_MIRROR" ]; then
        GITHUB_RAW_URL="gitee.com/naibahq/nezha/raw/master"
        GITHUB_URL=$CUSTOM_MIRROR
        Get_Docker_URL="get.docker.com"
        Get_Docker_Argu=" -s docker --mirror Aliyun"
        Docker_IMG="registry.cn-shanghai.aliyuncs.com\/naibahq\/nezha-dashboard"
        GITHUB_RELEASE_URL="https://gitee.com/naibahq/nezha/releases/download/${NZ_VERSION}"
    else
        if [ -z "$CN" ]; then
            GITHUB_RAW_URL="raw.githubusercontent.com/naiba/nezha/master"
            GITHUB_URL="github.com"
            Get_Docker_URL="get.docker.com"
            Get_Docker_Argu=" "
            Docker_IMG="ghcr.io\/naiba\/nezha-dashboard"
            GITHUB_RELEASE_URL="https://github.com/naiba/nezha/releases/download/${NZ_VERSION}"
        else
            GITHUB_RAW_URL="gitee.com/naibahq/nezha/raw/master"
            GITHUB_URL="gitee.com"
            Get_Docker_URL="get.docker.com"
            Get_Docker_Argu=" -s docker --mirror Aliyun"
            Docker_IMG="registry.cn-shanghai.aliyuncs.com\/naibahq\/nezha-dashboard"
            GITHUB_RELEASE_URL="https://gitee.com/naibahq/nezha/releases/download/${NZ_VERSION}" 
        fi
    fi
}

confirm() {
    if [[ $# > 1 ]]; then
        echo && read -e -p "$1 [默认$2]: " temp
        if [[ x"${temp}" == x"" ]]; then
            temp=$2
        fi
    else
        read -e -p "$1 [y/n]: " temp
    fi
    if [[ x"${temp}" == x"y" || x"${temp}" == x"Y" ]]; then
        return 0
    else
        return 1
    fi
}

update_script() {
    echo -e "> 更新脚本"

    curl -sL https://${GITHUB_RAW_URL}/script/install.sh -o /tmp/nezha.sh
    new_version=$(cat /tmp/nezha.sh | grep "VERSION" | head -n 1 | awk -F "=" '{print $2}' | sed 's/\"//g;s/,//g;s/ //g')
    if [ ! -n "$new_version" ]; then
        echo -e "脚本获取失败，请检查本机能否链接 https://${GITHUB_RAW_URL}/script/install.sh"
        return 1
    fi
    echo -e "当前最新版本为: ${new_version}"
    mv -f /tmp/nezha.sh ./nezha.sh && chmod a+x ./nezha.sh

    echo -e "3s后执行新脚本"
    sleep 3s
    clear
    exec ./nezha.sh
    exit 0
}

before_show_menu() {
    echo && echo -n -e "${yellow}* 按回车返回主菜单 *${plain}" && read temp
    show_menu
}

install_base() {
    (command -v git >/dev/null 2>&1 && command -v curl >/dev/null 2>&1 && command -v wget >/dev/null 2>&1 && command -v unzip >/dev/null 2>&1 && command -v getenforce >/dev/null 2>&1) ||
    (install_soft curl wget git unzip)
    
    # Additional checks for Alpine Linux
    if [ "$os_alpine" = 1 ]; then
        # Ensure required packages are installed for Alpine
        if ! command -v openrc >/dev/null 2>&1; then
            echo -e "${yellow}安装 OpenRC 服务管理器...${plain}"
            apk add --no-cache openrc
        fi
        
        # Ensure the binary directory exists and has proper permissions
        mkdir -p /opt/nezha/agent
        chmod 755 /opt/nezha/agent
    fi
}

install_arch(){
    echo -e "${green}提示: ${plain} Arch安装libselinux需添加nezha-agent用户，安装完会自动删除，建议手动检查一次\n"
    read -e -r -p "是否安装libselinux? [Y/n] " input
    case $input in
        [yY][eE][sS] | [yY])
            useradd -m nezha-agent
            sed -i "$ a\nezha-agent ALL=(ALL ) NOPASSWD:ALL" /etc/sudoers
            sudo -iu nezha-agent bash -c 'gpg --keyserver keys.gnupg.net --recv-keys BE22091E3EF62275;
                                        cd /tmp; git clone https://aur.archlinux.org/libsepol.git; cd libsepol; makepkg -si --noconfirm --asdeps; cd ..;
                                        git clone https://aur.archlinux.org/libselinux.git; cd libselinux; makepkg -si --noconfirm; cd ..;
                                        rm -rf libsepol libselinux'
            sed -i '/nezha-agent/d'  /etc/sudoers && sleep 30s && killall -u nezha-agent&&userdel nezha-agent
            echo -e "${red}提示: ${plain}已删除用户nezha-agent，请务必手动核查一遍！\n"
        ;;
        [nN][oO] | [nN])
            echo "不安装libselinux"
        ;;
        *)
            echo "不安装libselinux"
            exit 0
        ;;
    esac
}

install_soft() {
    (command -v yum >/dev/null 2>&1 && yum makecache && yum install $* selinux-policy -y) ||
    (command -v apt >/dev/null 2>&1 && apt update && apt install $* selinux-utils -y) ||
    (command -v pacman >/dev/null 2>&1 && pacman -Syu $* base-devel --noconfirm && install_arch)  ||
    (command -v apt-get >/dev/null 2>&1 && apt-get update && apt-get install $* selinux-utils -y) ||
    (command -v apk >/dev/null 2>&1 && apk update && apk add $* -f)
}

install_dashboard() {
    install_base

    echo -e "> 安装面板"

    # 哪吒监控文件夹
    if [ ! -d "${NZ_DASHBOARD_PATH}/docker-compose.yaml" ]; then
        mkdir -p $NZ_DASHBOARD_PATH
    else
        echo "您可能已经安装过面板端，重复安装会覆盖数据，请注意备份。"
        read -e -r -p "是否退出安装? [Y/n] " input
        case $input in
            [yY][eE][sS] | [yY])
                echo "退出安装"
                exit 0
            ;;
            [nN][oO] | [nN])
                echo "继续安装"
            ;;
            *)
                echo "退出安装"
                exit 0
            ;;
        esac
    fi

    chmod 777 -R $NZ_DASHBOARD_PATH

    command -v docker >/dev/null 2>&1
    if [[ $? != 0 ]]; then
        echo -e "正在安装 Docker"
        bash <(curl -sL https://${Get_Docker_URL}) ${Get_Docker_Argu} >/dev/null 2>&1
        if [[ $? != 0 ]]; then
            echo -e "${red}下载脚本失败，请检查本机能否连接 ${Get_Docker_URL}${plain}"
            return 0
        fi
        systemctl enable docker.service
        systemctl start docker.service
        echo -e "${green}Docker 安装成功${plain}"
    fi

    modify_dashboard_config 0

    if [[ $# == 0 ]]; then
        before_show_menu
    fi
}

install_dashboard_standalone() {
    install_base
    
    echo -e "> 安装面板"
    
    # 哪吒监控文件夹
    if [ ! -d "${NZ_DASHBOARD_PATH}/app" ]; then
        mkdir -p $NZ_DASHBOARD_PATH
    else
        echo "您可能已经安装过面板端，重复安装会覆盖数据，请注意备份。"
        read -e -r -p "是否退出安装? [Y/n] " input
        case $input in
            [yY][eE][sS] | [yY])
                echo "退出安装"
                exit 0
            ;;
            [nN][oO] | [nN])
                echo "继续安装"
            ;;
            *)
                echo "退出安装"
                exit 0
            ;;
        esac
    fi
    
    if [ ! -d "${NZ_DASHBOARD_PATH}/resource/template/theme-custom" ] || [ ! -d "${NZ_DASHBOARD_PATH}/resource/static/custom" ]; then
        mkdir -p "${NZ_DASHBOARD_PATH}/resource/template/theme-custom" "${NZ_DASHBOARD_PATH}/resource/static/custom" >/dev/null 2>&1
    fi

    chmod 777 -R $NZ_DASHBOARD_PATH
    
    modify_dashboard_config_standalone 0
    
    if [[ $# == 0 ]]; then
        before_show_menu
    fi
}

selinux(){
    #判断当前的状态
    if [ "$os_alpine" != 1 ];then
        getenforce | grep '[Ee]nfor'
        if [ $? -eq 0 ];then
            echo -e "SELinux是开启状态，正在关闭！"
            setenforce 0 &>/dev/null
            find_key="SELINUX="
            sed -ri "/^$find_key/c${find_key}disabled" /etc/selinux/config
        fi
    fi
}

install_agent() {
    install_base
    selinux

    echo -e "${green}> 安装监控Agent指定版本${VERSION}${plain}"

    mkdir -p $NZ_AGENT_PATH
    chmod 777 -R $NZ_AGENT_PATH

    echo -e "${yellow}正在下载监控端${plain}"

    if [ -z "$CN" ]; then
        NZ_AGENT_URL="https://${GITHUB_URL}/nezhahq/agent/releases/download/${NZ_VERSION}/nezha-agent_linux_${os_arch}.zip"
    else
        NZ_AGENT_URL="https://${GITHUB_URL}/naibahq/agent/releases/download/${NZ_VERSION}/nezha-agent_linux_${os_arch}.zip"
    fi

    wget -t 2 -T 60 -O nezha-agent_linux_${os_arch}.zip $NZ_AGENT_URL >/dev/null 2>&1

    if [[ $? != 0 ]]; then                            
        echo -e "${red}Release 下载失败，请检查本机能否连接 ${GITHUB_URL}${plain}"
        return 0
    fi

    unzip -qo nezha-agent_linux_${os_arch}.zip &&
    mv nezha-agent $NZ_AGENT_PATH &&
    rm -rf nezha-agent_linux_${os_arch}.zip README.md &&
    chmod +x $NZ_AGENT_PATH/nezha-agent
    
    # Additional checks for Alpine Linux
    if [ "$os_alpine" = 1 ]; then
        # Verify the binary is executable and compatible
        if [ ! -x "$NZ_AGENT_PATH/nezha-agent" ]; then
            echo -e "${red}错误: nezha-agent 二进制文件不可执行${plain}"
            echo -e "${yellow}尝试修复权限...${plain}"
            chmod +x "$NZ_AGENT_PATH/nezha-agent"
        fi
        
        # Check file type and architecture
        echo -e "${yellow}检查二进制文件信息...${plain}"
        if command -v file >/dev/null 2>&1; then
            file_info=$(file "$NZ_AGENT_PATH/nezha-agent" 2>/dev/null)
            echo -e "${yellow}文件信息: ${file_info}${plain}"
        else
            echo -e "${yellow}file命令不可用，跳过文件类型检查${plain}"
            file_info=""
        fi
        
        # Test if the binary can run (basic compatibility check)
        echo -e "${yellow}测试二进制文件兼容性...${plain}"
        
        # First check if binary can be executed at all
        if ! timeout 5 "$NZ_AGENT_PATH/nezha-agent" --version >/dev/null 2>&1; then
            echo -e "${red}二进制文件无法执行，可能是架构不匹配${plain}"
            echo -e "${yellow}尝试检查二进制文件架构...${plain}"
            
            # Check if we can get more info about the binary
            if command -v ldd >/dev/null 2>&1; then
                echo -e "${yellow}依赖库检查:${plain}"
                ldd "$NZ_AGENT_PATH/nezha-agent" 2>&1 || echo "无法检查依赖库"
            fi
            
            # Check if it's a valid ELF file
            if command -v readelf >/dev/null 2>&1; then
                echo -e "${yellow}ELF文件信息:${plain}"
                readelf -h "$NZ_AGENT_PATH/nezha-agent" 2>&1 || echo "不是有效的ELF文件"
            fi
        else
            echo -e "${green}二进制文件可以执行${plain}"
        fi
        
        # Try --help command
        if timeout 10 "$NZ_AGENT_PATH/nezha-agent" --help >/dev/null 2>&1; then
            echo -e "${green}二进制文件兼容性检查通过${plain}"
        else
            echo -e "${red}错误: nezha-agent 二进制文件不兼容当前系统架构${plain}"
            echo -e "${yellow}检测到的架构: ${os_arch}${plain}"
            echo -e "${yellow}系统架构: $(uname -m)${plain}"
            echo -e "${yellow}文件信息: ${file_info}${plain}"
            
            # Try to download the correct architecture
            echo -e "${yellow}尝试重新下载正确的架构...${plain}"
            rm -f "$NZ_AGENT_PATH/nezha-agent"
            wget -t 2 -T 60 -O nezha-agent_linux_${os_arch}.zip $NZ_AGENT_URL >/dev/null 2>&1
            if [[ $? == 0 ]]; then
                unzip -qo nezha-agent_linux_${os_arch}.zip &&
                mv nezha-agent $NZ_AGENT_PATH &&
                rm -rf nezha-agent_linux_${os_arch}.zip README.md &&
                chmod +x $NZ_AGENT_PATH/nezha-agent
                
                # Test again
                if timeout 10 "$NZ_AGENT_PATH/nezha-agent" --help >/dev/null 2>&1; then
                    echo -e "${green}重新下载后二进制文件兼容性检查通过${plain}"
                else
                    echo -e "${red}重新下载后仍然不兼容，请检查发布页面是否有对应架构的版本${plain}"
                    return 1
                fi
            else
                echo -e "${red}重新下载失败${plain}"
                return 1
            fi
        fi
    fi

    if [ $# -ge 3 ]; then
        modify_agent_config "$@"
    else
        modify_agent_config 0
    fi

    if [[ $# == 0 ]]; then
        before_show_menu
    fi
}

modify_agent_config() {
    echo -e "> 修改Agent配置"
    if [ $# -lt 3 ]; then
        echo "请先在管理面板上添加Agent，记录下密钥" &&
        read -ep "请输入一个解析到面板所在IP的域名（不可套CDN）: " nz_grpc_host &&
        read -ep "请输入面板RPC端口 (默认值 5555): " nz_grpc_port &&
        read -ep "请输入Agent 密钥: " nz_client_secret &&
        read -ep "是否启用针对 gRPC 端口的 SSL/TLS加密 (--tls)，需要请按 [y]，默认是不需要，不理解用户可回车跳过: " nz_grpc_proxy
        grep -qiw 'Y' <<< "${nz_grpc_proxy}" && args='--tls'
        if [[ -z "${nz_grpc_host}" || -z "${nz_client_secret}" ]]; then
            echo -e "${red}所有选项都不能为空${plain}"
            before_show_menu
            return 1
        fi
        if [[ -z "${nz_grpc_port}" ]]; then
            nz_grpc_port=5555
        fi
    else
        nz_grpc_host=$1
        nz_grpc_port=$2
        nz_client_secret=$3
        shift 3
        if [ $# -gt 0 ]; then
            args=" $*"
        fi
    fi

    if [ "$os_alpine" != 1 ];then
        cat > $NZ_AGENT_SERVICE << EOF

[Unit]
Description=哪吒探针监控端
ConditionFileIsExecutable=/opt/nezha/agent/nezha-agent


[Service]
StartLimitInterval=5
StartLimitBurst=10
ExecStart=/opt/nezha/agent/nezha-agent "-s" "${nz_grpc_host}:${nz_grpc_port}" "-p" "${nz_client_secret}" --disable-auto-update

WorkingDirectory=/root


Restart=always

RestartSec=120
EnvironmentFile=-/etc/sysconfig/nezha-agent

[Install]
WantedBy=multi-user.target
EOF

        chmod +x $NZ_AGENT_SERVICE
        systemctl daemon-reload
        systemctl enable nezha-agent
        systemctl restart nezha-agent

    else
        cat > $NZ_AGENT_SERVICERC << EOF
#!/sbin/openrc-run

description="哪吒探针监控端"

command="/opt/nezha/agent/nezha-agent"
command_args="-s ${nz_grpc_host}:${nz_grpc_port} -p ${nz_client_secret} --disable-auto-update"
command_background=true
pidfile="/var/run/nezha-agent.pid"

start() {
    ebegin "Starting nezha-agent"
    start-stop-daemon --start --background --make-pidfile --pidfile /var/run/nezha-agent.pid --exec /opt/nezha/agent/nezha-agent -- -s ${nz_grpc_host}:${nz_grpc_port} -p ${nz_client_secret} --disable-auto-update
    eend \$?
}

stop() {
    ebegin "Stopping nezha-agent"
    start-stop-daemon --stop --pidfile /var/run/nezha-agent.pid
    eend \$?
}
EOF

        chmod +x $NZ_AGENT_SERVICERC
        
        # Verify the service file was created correctly
        if [ ! -f "$NZ_AGENT_SERVICERC" ] || [ ! -x "$NZ_AGENT_SERVICERC" ]; then
            echo -e "${red}错误: 服务文件创建失败或权限不正确${plain}"
            return 1
        fi
        
        rc-update add nezha-agent default
        service nezha-agent restart
    fi

if [ "$os_alpine" != 1 ]; then
  if [ "$(systemctl is-active nezha-agent)" = "active" ]; then
    echo -e "Agent配置 ${green}修改成功，等待重启生效${plain}"
  else
    echo -e "Agent配置 ${green}修改失败，启动失败${plain}"
  fi
else
  if service nezha-agent status | grep -q "started"; then
    echo -e "Agent配置 ${green}修改成功，等待重启生效${plain}"
  else
    echo -e "Agent配置 ${green}修改失败，启动失败${plain}"
  fi
fi
}

modify_dashboard_config() {
    echo -e "> 修改面板配置"

    echo -e "正在下载 Docker 脚本"
    wget -t 2 -T 10 -O /tmp/nezha-docker-compose.yaml https://${GITHUB_RAW_URL}/script/docker-compose.yaml >/dev/null 2>&1
    if [[ $? != 0 ]]; then
        echo -e "${red}下载脚本失败，请检查本机能否连接 ${GITHUB_RAW_URL}${plain}"
        return 0
    fi

    wget -t 2 -T 10 -O /tmp/nezha-config.yaml https://${GITHUB_RAW_URL}/script/config.yaml >/dev/null 2>&1
    if [[ $? != 0 ]]; then
        echo -e "${red}下载脚本失败，请检查本机能否连接 ${GITHUB_RAW_URL}${plain}"
        return 0
    fi

    echo "关于 GitHub Oauth2 应用：在 https://github.com/settings/developers 创建，无需审核，Callback 填 http(s)://域名或IP/oauth2/callback" &&
    echo "关于 Gitee Oauth2 应用：在 https://gitee.com/oauth/applications 创建，无需审核，Callback 填 http(s)://域名或IP/oauth2/callback" &&
    read -ep "请输入 OAuth2 提供商(github/gitlab/jihulab/gitee，默认 github): " nz_oauth2_type &&
    read -ep "请输入 Oauth2 应用的 Client ID: " nz_github_oauth_client_id &&
    read -ep "请输入 Oauth2 应用的 Client Secret: " nz_github_oauth_client_secret &&
    read -ep "请输入 GitHub/Gitee 登录名作为管理员，多个以逗号隔开: " nz_admin_logins &&
    read -ep "请输入站点标题: " nz_site_title &&
    read -ep "请输入站点访问端口: (默认 8008)" nz_site_port &&
    read -ep "请输入用于 Agent 接入的 RPC 端口: (默认 5555)" nz_grpc_port

    if [[ -z "${nz_admin_logins}" || -z "${nz_github_oauth_client_id}" || -z "${nz_github_oauth_client_secret}" || -z "${nz_site_title}" ]]; then
        echo -e "${red}所有选项都不能为空${plain}"
        before_show_menu
        return 1
    fi

    if [[ -z "${nz_site_port}" ]]; then
        nz_site_port=8008
    fi
    if [[ -z "${nz_grpc_port}" ]]; then
        nz_grpc_port=5555
    fi
    if [[ -z "${nz_oauth2_type}" ]]; then
        nz_oauth2_type=github
    fi

    sed -i "s/nz_oauth2_type/${nz_oauth2_type}/" /tmp/nezha-config.yaml
    sed -i "s/nz_admin_logins/${nz_admin_logins}/" /tmp/nezha-config.yaml
    sed -i "s/nz_grpc_port/${nz_grpc_port}/" /tmp/nezha-config.yaml
    sed -i "s/nz_github_oauth_client_id/${nz_github_oauth_client_id}/" /tmp/nezha-config.yaml
    sed -i "s/nz_github_oauth_client_secret/${nz_github_oauth_client_secret}/" /tmp/nezha-config.yaml
    sed -i "s/nz_language/zh-CN/" /tmp/nezha-config.yaml
    sed -i "s/nz_site_title/${nz_site_title}/" /tmp/nezha-config.yaml
    sed -i "s/nz_site_port/${nz_site_port}/" /tmp/nezha-docker-compose.yaml
    sed -i "s/nz_grpc_port/${nz_grpc_port}/g" /tmp/nezha-docker-compose.yaml
    sed -i "s/nz_image_url/${Docker_IMG}/" /tmp/nezha-docker-compose.yaml

    mkdir -p $NZ_DASHBOARD_PATH/data
    mv -f /tmp/nezha-config.yaml ${NZ_DASHBOARD_PATH}/data/config.yaml
    mv -f /tmp/nezha-docker-compose.yaml ${NZ_DASHBOARD_PATH}/docker-compose.yaml

    echo -e "面板配置 ${green}修改成功，请稍等重启生效${plain}"

    restart_and_update

    if [[ $# == 0 ]]; then
        before_show_menu
    fi
}

modify_dashboard_config_standalone() {
    echo -e "> 修改面板配置"
    
    echo -e "正在下载配置模板"
    
    wget -t 2 -T 10 -O /tmp/nezha-config.yaml https://${GITHUB_RAW_URL}/script/config.yaml >/dev/null 2>&1
    if [[ $? != 0 ]]; then
        echo -e "${red}下载文件失败，请检查本机能否连接 ${GITHUB_RAW_URL}${plain}"
        return 0
    fi
    
    echo "关于 GitHub Oauth2 应用：在 https://github.com/settings/developers 创建，无需审核，Callback 填 http(s)://域名或IP/oauth2/callback" &&
    echo "关于 Gitee Oauth2 应用：在 https://gitee.com/oauth/applications 创建，无需审核，Callback 填 http(s)://域名或IP/oauth2/callback" &&
    read -ep "请输入 OAuth2 提供商(github/gitlab/jihulab/gitee，默认 github): " nz_oauth2_type &&
    read -ep "请输入 Oauth2 应用的 Client ID: " nz_github_oauth_client_id &&
    read -ep "请输入 Oauth2 应用的 Client Secret: " nz_github_oauth_client_secret &&
    read -ep "请输入 GitHub/Gitee 登录名作为管理员，多个以逗号隔开: " nz_admin_logins &&
    read -ep "请输入站点标题: " nz_site_title &&
    read -ep "请输入站点访问端口: (默认 8008)" nz_site_port &&
    read -ep "请输入用于 Agent 接入的 RPC 端口: (默认 5555)" nz_grpc_port
    
    if [[ -z "${nz_admin_logins}" || -z "${nz_github_oauth_client_id}" || -z "${nz_github_oauth_client_secret}" || -z "${nz_site_title}" ]]; then
        echo -e "${red}所有选项都不能为空${plain}"
        before_show_menu
        return 1
    fi
    
    if [[ -z "${nz_site_port}" ]]; then
        nz_site_port=8008
    fi
    if [[ -z "${nz_grpc_port}" ]]; then
        nz_grpc_port=5555
    fi
    if [[ -z "${nz_oauth2_type}" ]]; then
        nz_oauth2_type=github
    fi
    
    sed -i "s/nz_oauth2_type/${nz_oauth2_type}/" /tmp/nezha-config.yaml
    sed -i "s/nz_admin_logins/${nz_admin_logins}/" /tmp/nezha-config.yaml
    sed -i "s/nz_grpc_port/${nz_grpc_port}/" /tmp/nezha-config.yaml
    sed -i "s/nz_github_oauth_client_id/${nz_github_oauth_client_id}/" /tmp/nezha-config.yaml
    sed -i "s/nz_github_oauth_client_secret/${nz_github_oauth_client_secret}/" /tmp/nezha-config.yaml
    sed -i "s/nz_language/zh-CN/" /tmp/nezha-config.yaml
    sed -i "s/nz_site_title/${nz_site_title}/" /tmp/nezha-config.yaml
    sed -i "s/80/${nz_site_port}/" /tmp/nezha-config.yaml
    
    mkdir -p $NZ_DASHBOARD_PATH/data
    mv -f /tmp/nezha-config.yaml ${NZ_DASHBOARD_PATH}/data/config.yaml

    echo -e "正在下载服务文件"

    if [ "$os_alpine" != 1 ];then
        wget -t 2 -T 10 -O $NZ_DASHBOARD_SERVICE https://${GITHUB_RAW_URL}/script/nezha-dashboard.service >/dev/null 2>&1
        else
        wget -t 2 -T 10 -O $NZ_DASHBOARD_SERVICERC https://${GITHUB_RAW_URL}/script/nezha-dashboard >/dev/null 2>&1
        chmod +x $NZ_DASHBOARD_SERVICERC
        if [[ $? != 0 ]]; then
            echo -e "${red}文件下载失败，请检查本机能否连接 ${GITHUB_RAW_URL}${plain}"
            return 0
        fi
    fi

    
    echo -e "面板配置 ${green}修改成功，请稍等重启生效${plain}"
    
    restart_and_update_standalone
    
    if [[ $# == 0 ]]; then
        before_show_menu
    fi
}

restart_and_update() {
    echo -e "${green}> 重启并降级面板至$VERSION${plain}"

    cd $NZ_DASHBOARD_PATH

    sed -i "s|image: ghcr.io/naiba/nezha-dashboard.*|image: ghcr.io/naiba/nezha-dashboard:${NZ_VERSION}|" /opt/nezha/dashboard/docker-compose.yaml
    
    docker compose version
    if [[ $? == 0 ]]; then
        docker compose down
        docker pull ghcr.io/naiba/nezha-dashboard:$NZ_VERSION
        docker compose up -d
    else
        docker-compose down
        docker pull ghcr.io/naiba/nezha-dashboard:$NZ_VERSION
        docker-compose up -d
    fi

    if [[ $? == 0 ]]; then
        echo -e "${green}哪吒面板重启并降级成功${plain}"
        echo -e "${yellow}默认管理面板地址：域名:站点访问端口${plain}"
    else
        echo -e "${red}重启失败，可能是因为启动时间超过了两秒，请稍后查看日志信息${plain}"
    fi

    if [[ $# == 0 ]]; then
        before_show_menu
    fi
}

restart_and_update_standalone() {
    echo -e "> 重启并降级面板"
    
    cd $NZ_DASHBOARD_PATH

    if [ "$os_alpine" != 1 ]; then
        systemctl stop nezha-dashboard
    else
        rc-service nezha-dashboard stop
    fi

    wget -qO app.zip $GITHUB_RELEASE_URL/dashboard-linux-$os_arch.zip >/dev/null 2>&1 && unzip -qq app.zip && mv dist/dashboard-linux-$os_arch app && rm -r app.zip dist

    if [ "$os_alpine" != 1 ];then
        systemctl daemon-reload
        systemctl enable nezha-dashboard
        systemctl restart nezha-dashboard
    else
        rc-update add nezha-dashboard
        rc-service nezha-dashboard restart
    fi
    
    if [[ $? == 0 ]]; then
        echo -e "${green}哪吒面板重启并降级成功${plain}"
        echo -e "${yellow}默认管理面板地址：域名:站点访问端口${plain}"
    else
        echo -e "${red}重启失败，可能是因为启动时间超过了两秒，请稍后查看日志信息${plain}"
    fi
    
    if [[ $# == 0 ]]; then
        before_show_menu
    fi
}

start_dashboard() {
    echo -e "> 启动面板"

    docker compose version
    if [[ $? == 0 ]]; then
        cd $NZ_DASHBOARD_PATH && docker compose up -d
    else
        cd $NZ_DASHBOARD_PATH && docker-compose up -d
    fi

    if [[ $? == 0 ]]; then
        echo -e "${green}哪吒监控 启动成功${plain}"
    else
        echo -e "${red}启动失败，请稍后查看日志信息${plain}"
    fi

    if [[ $# == 0 ]]; then
        before_show_menu
    fi
}

start_dashboard_standalone() {
    echo -e "> 启动面板"
    
    if [ "$os_alpine" != 1 ]; then
        systemctl start nezha-dashboard
    else
        rc-service nezha-dashboard start
    fi
    
    if [[ $? == 0 ]]; then
        echo -e "${green}哪吒监控 启动成功${plain}"
    else
        echo -e "${red}启动失败，请稍后查看日志信息${plain}"
    fi
    
    if [[ $# == 0 ]]; then
        before_show_menu
    fi
}

stop_dashboard() {
    echo -e "> 停止面板"

    docker compose version
    if [[ $? == 0 ]]; then
        cd $NZ_DASHBOARD_PATH && docker compose down
    else
        cd $NZ_DASHBOARD_PATH && docker-compose down
    fi

    if [[ $? == 0 ]]; then
        echo -e "${green}哪吒监控 停止成功${plain}"
    else
        echo -e "${red}停止失败，请稍后查看日志信息${plain}"
    fi

    if [[ $# == 0 ]]; then
        before_show_menu
    fi
}

stop_dashboard_standalone() {
    echo -e "> 停止面板"
    
    if [ "$os_alpine" != 1 ]; then
        systemctl stop nezha-dashboard
    else
        rc-service nezha-dashboard stop
    fi
    
    if [[ $? == 0 ]]; then
        echo -e "${green}哪吒监控 停止成功${plain}"
    else
        echo -e "${red}停止失败，请稍后查看日志信息${plain}"
    fi
    
    if [[ $# == 0 ]]; then
        before_show_menu
    fi
}

show_dashboard_log() {
    echo -e "> 获取面板日志"

    docker compose version
    if [[ $? == 0 ]]; then
        cd $NZ_DASHBOARD_PATH && docker compose logs -f
    else
        cd $NZ_DASHBOARD_PATH && docker-compose logs -f
    fi

    if [[ $# == 0 ]]; then
        before_show_menu
    fi
}

show_dashboard_log_standalone() {
    echo -e "> 获取面板日志"
    
    if [ "$os_alpine" != 1 ]; then
        journalctl -xf -u nezha-dashboard.service
    else
        echo "很抱歉，OpenRC 无此功能"
    fi

    if [[ $# == 0 ]]; then
        before_show_menu
    fi
}

uninstall_dashboard() {
    echo -e "> 卸载管理面板"

    docker compose version
    if [[ $? == 0 ]]; then
        cd $NZ_DASHBOARD_PATH && docker compose down
    else
        cd $NZ_DASHBOARD_PATH && docker-compose down
    fi

    rm -rf $NZ_DASHBOARD_PATH
    docker rmi -f ghcr.io/naiba/nezha-dashboard > /dev/null 2>&1
    docker rmi -f registry.cn-shanghai.aliyuncs.com/naibahq/nezha-dashboard > /dev/null 2>&1
    clean_all

    if [[ $# == 0 ]]; then
        before_show_menu
    fi
}

uninstall_dashboard_standalone() {
    echo -e "> 卸载管理面板"
    
    rm -rf $NZ_DASHBOARD_PATH

    if [ "$os_alpine" != 1 ]; then
        systemctl stop nezha-dashboard
    else
        rc-service nezha-dashboard stop
    fi

    if [ "$os_alpine" != 1 ]; then
        rm $NZ_DASHBOARD_SERVICE
    else
        rm $NZ_DASHBOARD_SERVICERC
    fi

    clean_all
    
    if [[ $# == 0 ]]; then
        before_show_menu
    fi
}

show_agent_log() {
    echo -e "> 获取Agent日志"

    if [ "$os_alpine" != 1 ]; then
        journalctl -xf -u nezha-agent.service
    else
        echo "很抱歉，OpenRC 无此功能"
    fi

    if [[ $# == 0 ]]; then
        before_show_menu
    fi
}

uninstall_agent() {
    echo -e "> 卸载Agent"

    if [ "$os_alpine" != 1 ];then
        systemctl disable nezha-agent.service
        systemctl stop nezha-agent.service
        rm -rf $NZ_AGENT_SERVICE
        systemctl daemon-reload
    else
        rc-update del nezha-agent
        rc-service nezha-agent stop
        rm -rf $NZ_AGENT_SERVICERC
    fi

    rm -rf $NZ_AGENT_PATH
    clean_all

    if [[ $# == 0 ]]; then
        before_show_menu
    fi
}

restart_agent() {
    echo -e "> 重启Agent"

    if [ "$os_alpine" != 1 ];then
        systemctl restart nezha-agent.service
    else
        rc-service nezha-agent restart
    fi

    if [[ $# == 0 ]]; then
        before_show_menu
    fi
}

redownload_agent() {
    echo -e "> 重新下载Agent二进制文件"
    
    if [ "$os_alpine" != 1 ]; then
        echo -e "${red}此功能仅适用于Alpine Linux系统${plain}"
        return 1
    fi
    
    echo -e "${yellow}当前系统架构: $(uname -m)${plain}"
    echo -e "${yellow}检测到的架构: ${os_arch}${plain}"
    
    # Remove existing binary
    if [ -f "/opt/nezha/agent/nezha-agent" ]; then
        echo -e "${yellow}删除现有二进制文件...${plain}"
        rm -f /opt/nezha/agent/nezha-agent
    fi
    
    # Download URL
    if [ -z "$CN" ]; then
        NZ_AGENT_URL="https://${GITHUB_URL}/nezhahq/agent/releases/download/${NZ_VERSION}/nezha-agent_linux_${os_arch}.zip"
    else
        NZ_AGENT_URL="https://${GITHUB_URL}/naibahq/agent/releases/download/${NZ_VERSION}/nezha-agent_linux_${os_arch}.zip"
    fi
    
    echo -e "${yellow}下载URL: ${NZ_AGENT_URL}${plain}"
    echo -e "${yellow}正在下载...${plain}"
    
    wget -t 2 -T 60 -O nezha-agent_linux_${os_arch}.zip $NZ_AGENT_URL
    
    if [[ $? != 0 ]]; then
        echo -e "${red}下载失败，请检查网络连接${plain}"
        return 1
    fi
    
    echo -e "${yellow}解压文件...${plain}"
    unzip -qo nezha-agent_linux_${os_arch}.zip &&
    mv nezha-agent /opt/nezha/agent/ &&
    rm -rf nezha-agent_linux_${os_arch}.zip README.md &&
    chmod +x /opt/nezha/agent/nezha-agent
    
    echo -e "${yellow}测试新下载的二进制文件...${plain}"
    if timeout 10 /opt/nezha/agent/nezha-agent --version >/dev/null 2>&1; then
        echo -e "${green}重新下载成功，二进制文件可以正常运行${plain}"
    else
        echo -e "${red}重新下载后仍然无法运行${plain}"
        echo -e "${yellow}文件信息: $(file /opt/nezha/agent/nezha-agent)${plain}"
        return 1
    fi
    
    if [[ $# == 0 ]]; then
        before_show_menu
    fi
}

clean_all() {
    if [ -z "$(ls -A ${NZ_BASE_PATH})" ]; then
        rm -rf ${NZ_BASE_PATH}
    fi
}

select_version() {
    if command -v docker compose >/dev/null 2>&1; then
        if docker compose ls | grep -qw "$NZ_DASHBOARD_PATH/docker-compose.yaml" >/dev/null 2>&1; then
            IS_DOCKER_NEZHA=1
        fi
    fi
    if [[ -f $NZ_DASHBOARD_PATH/app ]]; then
        IS_DOCKER_NEZHA=0
    else
        echo -e "${yellow}请自行选择您的安装方式（如果你是安装Agent，输入哪个都是一样的）：\n1. Docker\n2. 独立安装${plain}"
        while true; do
            read -e -r -p "请输入数字 [1-2]：" option
            case "${option}" in
                1)
                    IS_DOCKER_NEZHA=1
                    break
                ;;
                2)
                    IS_DOCKER_NEZHA=0
                    break
                ;;
                *)
                    echo "输入有误，请重新输入"
                ;;
            esac
        done
    fi
}

show_usage() {
    echo "哪吒监控 管理脚本使用方法: "
    echo "--------------------------------------------------------"
    echo "./nezha.sh                            - 显示管理菜单"
    echo "./nezha.sh install_dashboard          - 安装面板端"
    echo "./nezha.sh modify_dashboard_config    - 修改面板配置"
    echo "./nezha.sh start_dashboard            - 启动面板"
    echo "./nezha.sh stop_dashboard             - 停止面板"
    echo "./nezha.sh restart_and_update         - 重启并降级面板"
    echo "./nezha.sh show_dashboard_log         - 查看面板日志"
    echo "./nezha.sh uninstall_dashboard        - 卸载管理面板"
    echo "--------------------------------------------------------"
    echo "./nezha.sh install_agent              - 安装监控Agent"
    echo "./nezha.sh modify_agent_config        - 修改Agent配置"
    echo "./nezha.sh show_agent_log             - 查看Agent日志"
    echo "./nezha.sh uninstall_agent            - 卸载Agen"
    echo "./nezha.sh restart_agent              - 重启Agent"
    echo "./nezha.sh redownload_agent           - 重新下载Agent二进制文件"
    echo "./nezha.sh update_script              - 更新脚本"
    echo "--------------------------------------------------------"
}

show_menu() {
    echo -e "
    ${green}哪吒监控管理脚本${plain} ${red}${VERSION}${plain}
    --- https://github.com/eooce ---
    ${green}1.  安装面板端${plain}
    ${green}2.  修改面板配置${plain}
    ${green}3.  启动面板${plain}
    ${green}4.  停止面板${plain} 
    ${yellow}5.  重启并降级面板${plain}
    ${green}6.  查看面板日志${plain} 
    ${green}7.  卸载管理面板${plain}
    ————————————————-
    ${green}8.  安装监控Agent指定版本${plain}
    ${green}9.  修改Agent配置${plain}
    ${green}10. 查看Agent日志${plain}
    ${green}11. 卸载Agent${plain}
    ${green}12. 重启Agent${plain}
    ${green}13. 重新下载Agent文件${plain}
    ————————————————-
    ${green}14. 更新脚本${plain}
    ————————————————-
    ${green}0.  退出脚本${plain}
    "
    echo && read -ep "请输入选择 [0-14]: " num
    if [[ $IS_DOCKER_NEZHA == 1 ]]; then
        case "${num}" in
            0)
                exit 0
            ;;
            1)
                install_dashboard
            ;;
            2)
                modify_dashboard_config
            ;;
            3)
                start_dashboard
            ;;
            4)
                stop_dashboard
            ;;
            5)
                restart_and_update
            ;;
            6)
                show_dashboard_log
            ;;
            7)
                uninstall_dashboard
            ;;
            8)
                install_agent
            ;;
            9)
                modify_agent_config
            ;;
            10)
                show_agent_log
            ;;
            11)
                uninstall_agent
            ;;
            12)
                restart_agent
            ;;
            13)
                redownload_agent
            ;;
            14)
                update_script
            ;;
            *)
                echo -e "${red}请输入正确的数字 [0-14]${plain}"
            ;;
        esac
    elif [[ $IS_DOCKER_NEZHA == 0 ]]; then
        case "${num}" in
            0)
                exit 0
            ;;
            1)
                install_dashboard_standalone
            ;;
            2)
                modify_dashboard_config_standalone
            ;;
            3)
                start_dashboard_standalone
            ;;
            4)
                stop_dashboard_standalone
            ;;
            5)
                restart_and_update_standalone
            ;;
            6)
                show_dashboard_log_standalone
            ;;
            7)
                uninstall_dashboard_standalone
            ;;
            8)
                install_agent
            ;;
            9)
                modify_agent_config
            ;;
            10)
                show_agent_log
            ;;
            11)
                uninstall_agent
            ;;
            12)
                restart_agent
            ;;
            13)
                redownload_agent
            ;;
            14)
                update_script
            ;;
            *)
                echo -e "${red}请输入正确的数字 [0-14]${plain}"
            ;;
        esac
    else
        select_version
    fi
}

pre_check

if [[ $# > 0 ]]; then
    if [[ $IS_DOCKER_NEZHA == 1 ]]; then
        case $1 in
            "install_dashboard")
                install_dashboard 0
            ;;
            "modify_dashboard_config")
                modify_dashboard_config 0
            ;;
            "start_dashboard")
                start_dashboard 0
            ;;
            "stop_dashboard")
                stop_dashboard 0
            ;;
            "restart_and_update")
                restart_and_update 0
            ;;
            "show_dashboard_log")
                show_dashboard_log 0
            ;;
            "uninstall_dashboard")
                uninstall_dashboard 0
            ;;
            "install_agent")
                shift
                if [ $# -ge 3 ]; then
                    install_agent "$@"
                else
                    install_agent 0
                fi
            ;;
            "modify_agent_config")
                modify_agent_config 0
            ;;
            "show_agent_log")
                show_agent_log 0
            ;;
            "uninstall_agent")
                uninstall_agent 0
            ;;
            "restart_agent")
                restart_agent 0
            ;;
            "redownload_agent")
                redownload_agent 0
            ;;
            "update_script")
                update_script 0
            ;;
            *) show_usage ;;
        esac
    elif [[ $IS_DOCKER_NEZHA == 0 ]]; then
        case $1 in
            "install_dashboard")
                install_dashboard_standalone 0
            ;;
            "modify_dashboard_config")
                modify_dashboard_config_standalone 0
            ;;
            "start_dashboard")
                start_dashboard_standalone 0
            ;;
            "stop_dashboard")
                stop_dashboard_standalone 0
            ;;
            "restart_and_update")
                restart_and_update_standalone 0
            ;;
            "show_dashboard_log")
                show_dashboard_log_standalone 0
            ;;
            "uninstall_dashboard")
                uninstall_dashboard_standalone 0
            ;;
            "install_agent")
                shift
                if [ $# -ge 3 ]; then
                    install_agent "$@"
                else
                    install_agent 0
                fi
            ;;
            "modify_agent_config")
                modify_agent_config 0
            ;;
            "show_agent_log")
                show_agent_log 0
            ;;
            "uninstall_agent")
                uninstall_agent 0
            ;;
            "restart_agent")
                restart_agent 0
            ;;
            "redownload_agent")
                redownload_agent 0
            ;;
            "update_script")
                update_script 0
            ;;
            *) show_usage ;;
        esac
    else
        case $1 in
            "install_agent")
                shift
                if [ $# -ge 3 ]; then
                    install_agent "$@"
                else
                    install_agent 0
                fi
            ;;
            "modify_agent_config")
                modify_agent_config 0
            ;;
            "show_agent_log")
                show_agent_log 0
            ;;
            "uninstall_agent")
                uninstall_agent 0
            ;;
            "restart_agent")
                restart_agent 0
            ;;
            "redownload_agent")
                redownload_agent 0
            ;;
            "update_script")
                update_script 0
            ;;
            *) show_usage ;;
        esac
    fi
else
    select_version
    show_menu
fi
