commit 182fa1c5f1e40bec4ad8ee700bc9ad5f5ba8f9d3 Author: uubulb Date: Mon Oct 21 16:12:19 2024 +0800 create project diff --git a/LICENSE b/LICENSE new file mode 100644 index 0000000..7f600ab --- /dev/null +++ b/LICENSE @@ -0,0 +1,13 @@ +Copyright 2024 nezhahq + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + +http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. diff --git a/cmd/scriptgen/main.go b/cmd/scriptgen/main.go new file mode 100644 index 0000000..525471e --- /dev/null +++ b/cmd/scriptgen/main.go @@ -0,0 +1,74 @@ +package main + +import ( + "encoding/json" + "fmt" + "os" + "text/template" +) + +const ( + zh_CN = "zh_CN" + en_US = "en_US" +) + +var localeMap map[string]map[string]interface{} + +func main() { + if len(os.Args) != 3 { + printUsage() + exitWithError("Error: Need exactly 2 arguments") + } + + if err := readLocalesFromFile(os.Args[1]); err != nil { + exitWithError(fmt.Sprintf("readLocalesFromFile: %v", err)) + } + + if err := parseTemplate(os.Args[2]); err != nil { + exitWithError(fmt.Sprintf("parseTemplate: %v", err)) + } +} + +func printUsage() { + fmt.Printf("usage: %s [locale-file] [locale]\n", os.Args[0]) +} + +func exitWithError(message string) { + fmt.Fprintln(os.Stderr, message) + os.Exit(1) +} + +func readLocalesFromFile(file string) error { + b, err := os.ReadFile(file) + if err != nil { + return err + } + + return json.Unmarshal(b, &localeMap) +} + +func parseTemplate(lang string) error { + tmpl, err := template.ParseFiles("template.sh") + if err != nil { + return err + } + + switch lang { + case zh_CN: + file, err := os.Create("install.sh") + if err != nil { + return err + } + defer file.Close() + return tmpl.Execute(file, localeMap[zh_CN]) + case en_US: + file, err := os.Create("install_en.sh") + if err != nil { + return err + } + defer file.Close() + return tmpl.Execute(file, localeMap[en_US]) + default: + return fmt.Errorf("unsupported locale: %s", lang) + } +} diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..9bea57e --- /dev/null +++ b/go.mod @@ -0,0 +1,5 @@ +module github.com/nezhahq/scripts + +go 1.21 + +toolchain go1.23.2 diff --git a/install.sh b/install.sh new file mode 100644 index 0000000..6979879 --- /dev/null +++ b/install.sh @@ -0,0 +1,943 @@ +#!/bin/sh +# Generated by nezhahq/scriptgen. DO NOT EDIT + +NZ_BASE_PATH="/opt/nezha" +NZ_DASHBOARD_PATH="${NZ_BASE_PATH}/dashboard" +NZ_AGENT_PATH="${NZ_BASE_PATH}/agent" +NZ_DASHBOARD_SERVICE="/etc/systemd/system/nezha-dashboard.service" +NZ_DASHBOARD_SERVICERC="/etc/init.d/nezha-dashboard" + +red='\033[0;31m' +green='\033[0;32m' +yellow='\033[0;33m' +plain='\033[0m' +export PATH="$PATH:/usr/local/bin" + +os_arch="" +[ -e /etc/os-release ] && grep -i "PRETTY_NAME" /etc/os-release | grep -qi "alpine" && os_alpine='1' + +sudo() { + myEUID=$(id -ru) + if [ "$myEUID" -ne 0 ]; then + if command -v sudo > /dev/null 2>&1; then + command sudo "$@" + else + err "错误: 您的系统未安装 sudo，因此无法进行该项操作。" + exit 1 + fi + else + "$@" + fi +} + +check_systemd() { + if [ "$os_alpine" != 1 ] && ! command -v systemctl >/dev/null 2>&1; then + echo "System not supported: systemctl not found" + exit 1 + fi +} + +err() { + printf "${red}%s${plain}\n" "$*" >&2 +} + +success() { + printf "${green}%s${plain}\n" "$*" +} + +info() { + printf "${yellow}%s${plain}\n" "$*" +} + +geo_check() { + api_list="https://blog.cloudflare.com/cdn-cgi/trace https://dash.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" + set -- "$api_list" + 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 +} + +pre_check() { + umask 077 + + ## 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 + + ## China_IP + if [ -z "$CN" ]; then + geo_check + if [ -n "$isCN" ]; then + echo "According to the information provided by various geoip api, the current IP may be in China" + printf "Will the installation be done with a Chinese Mirror? [Y/n] (Custom Mirror Input 3): " + read -r input + case $input in + [yY][eE][sS] | [yY]) + echo "Use Chinese Mirror" + CN=true + ;; + + [nN][oO] | [nN]) + echo "Do Not Use Chinese Mirror" + ;; + + [3]) + echo "Use Custom Mirror" + printf "Please enter a custom image (e.g. :dn-dao-github-mirror.daocloud.io). If left blank, it won't be used: " + read -r input + case $input in + *) + CUSTOM_MIRROR=$input + ;; + esac + + ;; + *) + echo "Do Not Use Chinese Mirror" + ;; + 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" + 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" + 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" + fi + fi +} + +installation_check() { + if docker compose version >/dev/null 2>&1; then + DOCKER_COMPOSE_COMMAND="docker compose" + if sudo $DOCKER_COMPOSE_COMMAND ls | grep -qw "$NZ_DASHBOARD_PATH/docker-compose.yaml" >/dev/null 2>&1; then + NEZHA_IMAGES=$(sudo docker images --format ":" | grep -w "nezha-dashboard") + if [ -n "$NEZHA_IMAGES" ]; then + echo "Docker image with nezha-dashboard repository exists:" + echo "$NEZHA_IMAGES" + IS_DOCKER_NEZHA=1 + FRESH_INSTALL=0 + return + else + echo "No Docker images with the nezha-dashboard repository were found." + fi + fi + elif command -v docker-compose >/dev/null 2>&1; then + DOCKER_COMPOSE_COMMAND="docker-compose" + if sudo $DOCKER_COMPOSE_COMMAND -f "$NZ_DASHBOARD_PATH/docker-compose.yaml" config >/dev/null 2>&1; then + NEZHA_IMAGES=$(sudo docker images --format ":" | grep -w "nezha-dashboard") + if [ -n "$NEZHA_IMAGES" ]; then + echo "Docker image with nezha-dashboard repository exists:" + echo "$NEZHA_IMAGES" + IS_DOCKER_NEZHA=1 + FRESH_INSTALL=0 + return + else + echo "No Docker images with the nezha-dashboard repository were found." + fi + fi + fi + + if [ -f "$NZ_DASHBOARD_PATH/app" ]; then + IS_DOCKER_NEZHA=0 + FRESH_INSTALL=0 + fi +} + +select_version() { + if [ -z "$IS_DOCKER_NEZHA" ]; then + info "Select your installation method(Input anything is ok if you are installing agent):" + info "1. Docker" + info "2. Standalone" + while true; do + printf "Please enter [1-2]: " + read -r option + case "${option}" in + 1) + IS_DOCKER_NEZHA=1 + break + ;; + 2) + IS_DOCKER_NEZHA=0 + break + ;; + *) + err "Please enter the correct number [1-2]" + ;; + esac + done + fi +} + +update_script() { + echo "> Update Script" + + curl -sL https://${GITHUB_RAW_URL}/script/install_en.sh -o /tmp/nezha.sh + new_version=$(grep "NZ_VERSION" /tmp/nezha.sh | head -n 1 | awk -F "=" '{print $2}' | sed 's/\"//g;s/,//g;s/ //g') + if [ -z "$new_version" ]; then + echo "Script failed to get, please check if the network can link https://${GITHUB_RAW_URL}/script/install.sh" + return 1 + fi + echo "The current latest version is: ${new_version}" + mv -f /tmp/nezha.sh ./nezha.sh && chmod a+x ./nezha.sh + + echo "Execute new script after 3s" + sleep 3s + clear + exec ./nezha.sh + exit 0 +} + +before_show_menu() { + echo && info "* Press Enter to return to the main menu *" && read temp + show_menu +} + +install_base() { + (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 unzip) +} + +install_arch() { + info "Archlinux needs to add nezha-agent user to install libselinux. It will be deleted automatically after installation. It is recommended to check manually" + read -r -p "Do you need to install 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 4695881C254508D1; + 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 -r nezha-agent + echo -e "${red}Info: ${plain}user nezha-agent has been deleted, Be sure to check it manually!\n" + ;; + [nN][oO] | [nN]) + echo "Libselinux will not be installed" + ;; + *) + echo "Libselinux will not be installed" + exit 0 + ;; + esac +} + +install_soft() { + (command -v yum >/dev/null 2>&1 && sudo yum makecache && sudo yum install "$@" selinux-policy -y) || + (command -v apt >/dev/null 2>&1 && sudo apt update && sudo apt install "$@" selinux-utils -y) || + (command -v pacman >/dev/null 2>&1 && sudo pacman -Syu "$@" base-devel --noconfirm && install_arch) || + (command -v apt-get >/dev/null 2>&1 && sudo apt-get update && sudo apt-get install "$@" selinux-utils -y) || + (command -v apk >/dev/null 2>&1 && sudo apk update && sudo apk add "$@" -f) +} + +install_dashboard() { + check_systemd + install_base + + echo "> Install Dashboard" + + # Nezha Monitoring Folder + if [ ! "$FRESH_INSTALL" = 0 ]; then + sudo mkdir -p $NZ_DASHBOARD_PATH + else + echo "You may have already installed the dashboard, repeated installation will overwrite the data, please pay attention to backup." + printf "Exit the installation? [Y/n] " + read -r input + case $input in + [yY][eE][sS] | [yY]) + echo "Exit the installation." + exit 0 + ;; + [nN][oO] | [nN]) + echo "Continue." + ;; + *) + echo "Exit the installation." + exit 0 + ;; + esac + fi + + if [ "$IS_DOCKER_NEZHA" = 1 ]; then + install_dashboard_docker + elif [ "$IS_DOCKER_NEZHA" = 0 ]; then + install_dashboard_standalone + fi + + modify_dashboard_config 0 + + if [ $# = 0 ]; then + before_show_menu + fi +} + +install_dashboard_docker() { + if [ ! "$FRESH_INSTALL" = 0 ]; then + if ! command -v docker >/dev/null 2>&1; then + echo "Installing Docker" + if [ "$os_alpine" != 1 ]; then + if ! curl -sL https://${Get_Docker_URL} | sudo bash -s "${Get_Docker_Argu}"; then + err "Script failed to get, please check if the network can link ${Get_Docker_URL}" + return 0 + fi + sudo systemctl enable docker.service + sudo systemctl start docker.service + else + sudo apk add docker docker-compose + sudo rc-update add docker + sudo rc-service docker start + fi + success "Docker installed successfully" + installation_check + fi + fi +} + +install_dashboard_standalone() { + if [ ! -d "${NZ_DASHBOARD_PATH}/resource/template/theme-custom" ] || [ ! -d "${NZ_DASHBOARD_PATH}/resource/static/custom" ]; then + sudo mkdir -p "${NZ_DASHBOARD_PATH}/resource/template/theme-custom" "${NZ_DASHBOARD_PATH}/resource/static/custom" >/dev/null 2>&1 + fi +} + +selinux() { + #Check SELinux + if command -v getenforce >/dev/null 2>&1; then + if getenforce | grep '[Ee]nfor'; then + echo "SELinux running, closing now!" + sudo setenforce 0 >/dev/null 2>&1 + find_key="SELINUX=" + sudo sed -ri "/^$find_key/c${find_key}disabled" /etc/selinux/config + fi + fi +} + +install_agent() { + install_base + selinux + + echo "> Install Agent" + + echo "Obtaining Agent version number" + + + _version=$(curl -m 10 -sL "https://api.github.com/repos/nezhahq/agent/releases/latest" | grep "tag_name" | head -n 1 | awk -F ":" '{print $2}' | sed 's/\"//g;s/,//g;s/ //g') + if [ -z "$_version" ]; then + _version=$(curl -m 10 -sL "https://gitee.com/api/v5/repos/naibahq/agent/releases/latest" | awk -F '"' '{for(i=1;i<=NF;i++){if($i=="tag_name"){print $(i+2)}}}') + fi + if [ -z "$_version" ]; then + _version=$(curl -m 10 -sL "https://fastly.jsdelivr.net/gh/nezhahq/agent/" | grep "option\.value" | awk -F "'" '{print $2}' | sed 's/nezhahq\/agent@/v/g') + fi + if [ -z "$_version" ]; then + _version=$(curl -m 10 -sL "https://gcore.jsdelivr.net/gh/nezhahq/agent/" | grep "option\.value" | awk -F "'" '{print $2}' | sed 's/nezhahq\/agent@/v/g') + fi + + if [ -z "$_version" ]; then + err "Fail to obtain agent version, please check if the network can link https://api.github.com/repos/nezhahq/agent/releases/latest" + return 1 + else + echo "The current latest version is: ${_version}" + fi + + # Nezha Monitoring Folder + sudo mkdir -p $NZ_AGENT_PATH + + echo "Downloading Agent" + if [ -z "$CN" ]; then + NZ_AGENT_URL="https://${GITHUB_URL}/nezhahq/agent/releases/download/${_version}/nezha-agent_linux_${os_arch}.zip" + else + NZ_AGENT_URL="https://${GITHUB_URL}/naibahq/agent/releases/download/${_version}/nezha-agent_linux_${os_arch}.zip" + fi + + _cmd="wget -t 2 -T 60 -O nezha-agent_linux_${os_arch}.zip $NZ_AGENT_URL >/dev/null 2>&1" + if ! eval "$_cmd"; then + err "Fail to download agent, please check if the network can link ${GITHUB_URL}" + return 1 + fi + + sudo unzip -qo nezha-agent_linux_${os_arch}.zip && + sudo mv nezha-agent $NZ_AGENT_PATH && + sudo rm -rf nezha-agent_linux_${os_arch}.zip README.md + + if [ $# -ge 3 ]; then + modify_agent_config "$@" + else + modify_agent_config 0 + fi + + if [ $# = 0 ]; then + before_show_menu + fi +} + +modify_agent_config() { + echo "> Modify Agent Configuration" + + if [ $# -lt 3 ]; then + echo "Please add Agent in the Dashboard first, record the secret" + printf "Please enter a domain that resolves to the IP where Dashboard is located (no CDN): " + read -r nz_grpc_host + printf "Please enter Dashboard RPC port (default 5555): " + read -r nz_grpc_port + printf "Please enter the Agent secret: " + read -r nz_client_secret + printf "Do you want to enable SSL/TLS encryption for the gRPC port (--tls)? Press [y] if yes, the default is not required, and users can press Enter to skip if you don't understand: " + read -r nz_grpc_proxy + echo "${nz_grpc_proxy}" | grep -qiw 'Y' && args='--tls' + if [ -z "$nz_grpc_host" ] || [ -z "$nz_client_secret" ]; then + err "All options cannot be empty" + 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 + + _cmd="sudo ${NZ_AGENT_PATH}/nezha-agent service install -s $nz_grpc_host:$nz_grpc_port -p $nz_client_secret $args >/dev/null 2>&1" + + if ! eval "$_cmd"; then + sudo "${NZ_AGENT_PATH}"/nezha-agent service uninstall >/dev/null 2>&1 + sudo "${NZ_AGENT_PATH}"/nezha-agent service install -s "$nz_grpc_host:$nz_grpc_port" -p "$nz_client_secret" "$args" >/dev/null 2>&1 + fi + + success "Agent configuration modified successfully, please wait for agent self-restart to take effect" + + #if [[ $# == 0 ]]; then + # before_show_menu + #fi +} + +modify_dashboard_config() { + echo "> Modify Dashboard Configuration" + + if [ "$IS_DOCKER_NEZHA" = 1 ]; then + if [ -n "$DOCKER_COMPOSE_COMMAND" ]; then + echo "Download Docker Script" + _cmd="wget -t 2 -T 60 -O /tmp/nezha-docker-compose.yaml https://${GITHUB_RAW_URL}/script/docker-compose.yaml >/dev/null 2>&1" + if ! eval "$_cmd"; then + err "Script failed to get, please check if the network can link ${GITHUB_RAW_URL}" + return 0 + fi + else + err "Please install docker-compose manually. https://docs.docker.com/compose/install/linux/" + before_show_menu + fi + fi + + _cmd="wget -t 2 -T 60 -O /tmp/nezha-config.yaml https://${GITHUB_RAW_URL}/script/config.yaml >/dev/null 2>&1" + if ! eval "$_cmd"; then + err "Script failed to get, please check if the network can link ${GITHUB_RAW_URL}" + return 0 + fi + + echo "About the GitHub Oauth2 application: create it at https://github.com/settings/developers, no review required, and fill in the http(s)://domain_or_IP/oauth2/callback" + echo "(Not recommended) About the Gitee Oauth2 application: create it at https://gitee.com/oauth/applications, no auditing required, and fill in the http(s)://domain_or_IP/oauth2/callback" + printf "Please enter the OAuth2 provider (github/gitlab/jihulab/gitee, default github): " + read -r nz_oauth2_type + printf "Please enter the Client ID of the Oauth2 application: " + read -r nz_github_oauth_client_id + printf "Please enter the Client Secret of the Oauth2 application: " + read -r nz_github_oauth_client_secret + printf "Please enter your GitHub/Gitee login name as the administrator, separated by commas: " + read -r nz_admin_logins + printf "Please enter the site title: " + read -r nz_site_title + printf "Please enter the site access port: (default 8008)" + read -r nz_site_port + printf "Please enter the RPC port to be used for Agent access: (default 5555)" + read -r 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 + err "All options cannot be empty" + 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 + if [ "$IS_DOCKER_NEZHA" = 1 ]; then + 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 + elif [ "$IS_DOCKER_NEZHA" = 0 ]; then + sed -i "s/80/${nz_site_port}/" /tmp/nezha-config.yaml + fi + + sudo mkdir -p $NZ_DASHBOARD_PATH/data + sudo mv -f /tmp/nezha-config.yaml ${NZ_DASHBOARD_PATH}/data/config.yaml + if [ "$IS_DOCKER_NEZHA" = 1 ]; then + sudo mv -f /tmp/nezha-docker-compose.yaml ${NZ_DASHBOARD_PATH}/docker-compose.yaml + fi + + if [ "$IS_DOCKER_NEZHA" = 0 ]; then + echo "Downloading service file" + if [ "$os_alpine" != 1 ]; then + _download="sudo wget -t 2 -T 60 -O $NZ_DASHBOARD_SERVICE https://${GITHUB_RAW_URL}/script/nezha-dashboard.service >/dev/null 2>&1" + else + _download="sudo wget -t 2 -T 60 -O $NZ_DASHBOARD_SERVICERC https://${GITHUB_RAW_URL}/script/nezha-dashboard >/dev/null 2>&1" + if ! eval "$_download"; then + err "File failed to get, please check if the network can link ${GITHUB_RAW_URL}" + return 0 + fi + sudo chmod +x $NZ_DASHBOARD_SERVICERC + fi + fi + + success "Dashboard configuration modified successfully, please wait for Dashboard self-restart to take effect" + + restart_and_update + + if [ $# = 0 ]; then + before_show_menu + fi +} + +restart_and_update() { + echo "> Restart and Update Dashboard" + + if [ "$IS_DOCKER_NEZHA" = 1 ]; then + _cmd="restart_and_update_docker" + elif [ "$IS_DOCKER_NEZHA" = 0 ]; then + _cmd="restart_and_update_standalone" + fi + + if eval "$_cmd"; then + success "Nezha Monitoring Restart Successful" + info "Default Dashboard address: domain:site_access_port" + else + err "The restart failed, probably because the boot time exceeded two seconds, please check the log information later" + fi + + if [ $# = 0 ]; then + before_show_menu + fi +} + +restart_and_update_docker() { + sudo $DOCKER_COMPOSE_COMMAND -f ${NZ_DASHBOARD_PATH}/docker-compose.yaml pull + sudo $DOCKER_COMPOSE_COMMAND -f ${NZ_DASHBOARD_PATH}/docker-compose.yaml down + sudo $DOCKER_COMPOSE_COMMAND -f ${NZ_DASHBOARD_PATH}/docker-compose.yaml up -d +} + +restart_and_update_standalone() { + _version=$(curl -m 10 -sL "https://api.github.com/repos/naiba/nezha/releases/latest" | grep "tag_name" | head -n 1 | awk -F ":" '{print $2}' | sed 's/\"//g;s/,//g;s/ //g') + if [ -z "$_version" ]; then + _version=$(curl -m 10 -sL "https://gitee.com/api/v5/repos/naibahq/nezha/releases/latest" | awk -F '"' '{for(i=1;i<=NF;i++){if($i=="tag_name"){print $(i+2)}}}') + fi + if [ -z "$_version" ]; then + _version=$(curl -m 10 -sL "https://fastly.jsdelivr.net/gh/naiba/nezha/" | grep "option\.value" | awk -F "'" '{print $2}' | sed 's/naiba\/nezha@/v/g') + fi + if [ -z "$_version" ]; then + _version=$(curl -m 10 -sL "https://gcore.jsdelivr.net/gh/naiba/nezha/" | grep "option\.value" | awk -F "'" '{print $2}' | sed 's/naiba\/nezha@/v/g') + fi + + if [ -z "$_version" ]; then + err "Fail to obtain agent version, please check if the network can link https://api.github.com/repos/nezhahq/agent/releases/latest" + return 1 + else + echo "The current latest version is: ${_version}" + fi + + if [ "$os_alpine" != 1 ]; then + sudo systemctl daemon-reload + sudo systemctl stop nezha-dashboard + else + sudo rc-service nezha-dashboard stop + fi + + if [ -z "$CN" ]; then + NZ_DASHBOARD_URL="https://${GITHUB_URL}/naiba/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 + + sudo wget -qO $NZ_DASHBOARD_PATH/app.zip "$NZ_DASHBOARD_URL" >/dev/null 2>&1 && sudo unzip -qq -o $NZ_DASHBOARD_PATH/app.zip -d $NZ_DASHBOARD_PATH && sudo mv $NZ_DASHBOARD_PATH/dashboard-linux-$os_arch $NZ_DASHBOARD_PATH/app && sudo rm $NZ_DASHBOARD_PATH/app.zip + + if [ "$os_alpine" != 1 ]; then + sudo systemctl enable nezha-dashboard + sudo systemctl restart nezha-dashboard + else + sudo rc-update add nezha-dashboard + sudo rc-service nezha-dashboard restart + fi +} + +start_dashboard() { + echo "> Start Dashboard" + + if [ "$IS_DOCKER_NEZHA" = 1 ]; then + _cmd="start_dashboard_docker" + elif [ "$IS_DOCKER_NEZHA" = 0 ]; then + _cmd="start_dashboard_standalone" + fi + + if eval "$_cmd"; then + success "Nezha Monitoring Start Successful" + else + err "Failed to start, please check the log message later" + fi + + if [ $# = 0 ]; then + before_show_menu + fi +} + +start_dashboard_docker() { + sudo $DOCKER_COMPOSE_COMMAND -f ${NZ_DASHBOARD_PATH}/docker-compose.yaml up -d +} + +start_dashboard_standalone() { + if [ "$os_alpine" != 1 ]; then + sudo systemctl start nezha-dashboard + else + sudo rc-service nezha-dashboard start + fi +} + +stop_dashboard() { + echo "> Stop Dashboard" + + if [ "$IS_DOCKER_NEZHA" = 1 ]; then + _cmd="stop_dashboard_docker" + elif [ "$IS_DOCKER_NEZHA" = 0 ]; then + _cmd="stop_dashboard_standalone" + fi + + if eval "$_cmd"; then + success "Nezha Monitoring Stop Successful" + else + err "Failed to stop, please check the log message later" + fi + + if [ $# = 0 ]; then + before_show_menu + fi +} + +stop_dashboard_docker() { + sudo $DOCKER_COMPOSE_COMMAND -f ${NZ_DASHBOARD_PATH}/docker-compose.yaml down +} + +stop_dashboard_standalone() { + if [ "$os_alpine" != 1 ]; then + sudo systemctl stop nezha-dashboard + else + sudo rc-service nezha-dashboard stop + fi +} + +show_dashboard_log() { + echo "> View Dashboard Log" + + if [ "$IS_DOCKER_NEZHA" = 1 ]; then + show_dashboard_log_docker + elif [ "$IS_DOCKER_NEZHA" = 0 ]; then + show_dashboard_log_standalone + fi + + if [ $# = 0 ]; then + before_show_menu + fi +} + +show_dashboard_log_docker() { + sudo $DOCKER_COMPOSE_COMMAND -f ${NZ_DASHBOARD_PATH}/docker-compose.yaml logs -f +} + +show_dashboard_log_standalone() { + if [ "$os_alpine" != 1 ]; then + sudo journalctl -xf -u nezha-dashboard.service + else + sudo tail -n 10 /var/log/nezha-dashboard.err + fi +} + +uninstall_dashboard() { + echo "> Uninstall Dashboard" + + if [ "$IS_DOCKER_NEZHA" = 1 ]; then + uninstall_dashboard_docker + elif [ "$IS_DOCKER_NEZHA" = 0 ]; then + uninstall_dashboard_standalone + fi + + clean_all + + if [ $# = 0 ]; then + before_show_menu + fi +} + +uninstall_dashboard_docker() { + sudo $DOCKER_COMPOSE_COMMAND -f ${NZ_DASHBOARD_PATH}/docker-compose.yaml down + sudo rm -rf $NZ_DASHBOARD_PATH + sudo docker rmi -f ghcr.io/naiba/nezha-dashboard >/dev/null 2>&1 + sudo docker rmi -f registry.cn-shanghai.aliyuncs.com/naibahq/nezha-dashboard >/dev/null 2>&1 +} + +uninstall_dashboard_standalone() { + sudo rm -rf $NZ_DASHBOARD_PATH + + if [ "$os_alpine" != 1 ]; then + sudo systemctl disable nezha-dashboard + sudo systemctl stop nezha-dashboard + else + sudo rc-update del nezha-dashboard + sudo rc-service nezha-dashboard stop + fi + + if [ "$os_alpine" != 1 ]; then + sudo rm $NZ_DASHBOARD_SERVICE + else + sudo rm $NZ_DASHBOARD_SERVICERC + fi +} + +show_agent_log() { + echo "> View Agent Log" + + if [ "$os_alpine" != 1 ]; then + sudo journalctl -xf -u nezha-agent.service + else + sudo tail -n 10 /var/log/nezha-agent.err + fi + + if [ $# = 0 ]; then + before_show_menu + fi +} + +uninstall_agent() { + echo "> Uninstall Agent" + + sudo ${NZ_AGENT_PATH}/nezha-agent service uninstall + + sudo rm -rf $NZ_AGENT_PATH + clean_all + + if [ $# = 0 ]; then + before_show_menu + fi +} + +restart_agent() { + echo "> Restart Agent" + + sudo ${NZ_AGENT_PATH}/nezha-agent service restart + + if [ $# = 0 ]; then + before_show_menu + fi +} + +clean_all() { + if [ -z "$(ls -A ${NZ_BASE_PATH})" ]; then + sudo rm -rf ${NZ_BASE_PATH} + fi +} + +show_usage() { + echo "Nezha Monitor Management Script Usage: " + echo "--------------------------------------------------------" + echo "./nezha.sh - Show Menu" + echo "./nezha.sh install_dashboard - Install Dashboard" + echo "./nezha.sh modify_dashboard_config - Modify Dashboard Configuration" + echo "./nezha.sh start_dashboard - Start Dashboard" + echo "./nezha.sh stop_dashboard - Stop Dashboard" + echo "./nezha.sh restart_and_update - Restart and Update the Dashboard" + echo "./nezha.sh show_dashboard_log - View Dashboard Log" + echo "./nezha.sh uninstall_dashboard - Uninstall Dashboard" + echo "--------------------------------------------------------" + echo "./nezha.sh install_agent - Install Agent" + echo "./nezha.sh modify_agent_config - Modify Agent Configuration" + echo "./nezha.sh show_agent_log - View Agent Log" + echo "./nezha.sh uninstall_agent - Uninstall Agent" + echo "./nezha.sh restart_agent - Restart Agent" + echo "./nezha.sh update_script - Update Script" + echo "--------------------------------------------------------" +} + +show_menu() { + printf " + ${green}Nezha Monitor Management Script${plain} ${red}${NZ_VERSION}${plain} + --- https://github.com/naiba/nezha --- + ${green}1.${plain} Install Dashboard + ${green}2.${plain} Modify Dashbaord Configuration + ${green}3.${plain} Start Dashboard + ${green}4.${plain} Stop Dashboard + ${green}5.${plain} Restart and Update Dashboard + ${green}6.${plain} View Dashboard Log + ${green}7.${plain} Uninstall Dashboard + ————————————————- + ${green}8.${plain} Install Agent + ${green}9.${plain} Modify Agent Configuration + ${green}10.${plain} View Agent Log + ${green}11.${plain} Uninstall Agent + ${green}12.${plain} Restart Agent + ————————————————- + ${green}13.${plain} Update Script + ————————————————- + ${green}0.${plain} Exit Script + " + echo && printf "Please enter [0-13]: " && read -r num + 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) + update_script + ;; + *) + err "Please enter the correct number [0-13]" + ;; + esac +} + +pre_check +installation_check + +if [ $# -gt 0 ]; 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 + ;; + "update_script") + update_script 0 + ;; + *) show_usage ;; + esac +else + select_version + show_menu +fi diff --git a/locale.json b/locale.json new file mode 100644 index 0000000..e269b3f --- /dev/null +++ b/locale.json @@ -0,0 +1,58 @@ +{ + "zh_CN": { + "locale": "zh_CN", + "script": "install.sh", + "ErrorFetchScript": "Script failed to get, please check if the network can link", + "DoNotEdit": "# Generated by nezhahq/scriptgen. DO NOT EDIT", + "SudoError": "错误: 您的系统未安装 sudo，因此无法进行该项操作。", + "SystemdCheckError": "不支持此系统：未找到 systemctl 命令" + }, + "en_US": { + "locale": "en_US", + "script": "install_en.sh", + "ErrorFetchScript": "Script failed to get, please check if the network can link", + "ErrorObtainVersion": "Fail to obtain %s version, please check if the network can link", + "DoNotEdit": "# Generated by nezhahq/scriptgen. DO NOT EDIT", + "SudoError": "ERROR: sudo is not installed on the system, the action cannot be proceeded.", + "SystemdCheckError": "System not supported: systemctl not found", + "GeoInfo": "According to the information provided by various geoip api, the current IP may be in China", + "PromptMirror": "Will the installation be done with a Chinese Mirror? [Y/n] (Custom Mirror Input 3): ", + "UseCustomMirror": "Use Custom Mirror", + "PromptCustomImage": "Please enter a custom image (e.g. :dn-dao-github-mirror.daocloud.io). If left blank, it won't be used: ", + "UseChineseMirror": "Use Chinese Mirror", + "NoUseChineseMirror": "Do Not Use Chinese Mirror", + "DockerImageFound": "Docker image with nezha-dashboard repository exists:", + "DockerImageNotFound": "No Docker images with the nezha-dashboard repository were found.", + "SelectInstallMethod": "Select your installation method(Input anything is ok if you are installing agent):", + "PromptSelectVersion": "Please enter [1-2]: ", + "StandaloneInstall": "Standalone", + "ErrorSelectVersion": "Please enter the correct number [1-2]", + "UpdateScript": "Update Script", + "UpdateScriptInfo": "The current latest version is:", + "ExecuteScriptInfo": "Execute new script after 3s", + "BeforeShowMenu": "Press Enter to return to the main menu", + "InstallDashboard": "Install Dashboard", + "InstallDashboardInfo": "You may have already installed the dashboard, repeated installation will overwrite the data, please pay attention to backup.", + "PromptInstallDashboard": "Exit the installation? [Y/n]", + "ContinueInstallation": "Continue.", + "ExitInstallation": "Exit the installation.", + "InstallDockerInfo": "Installing Docker", + "InstallDockerSuccess": "Docker installed successfully", + "SELinuxInfo": "SELinux running, closing now!", + "InstallAgent": "Install Agent", + "ObtainAgentVersion": "Obtaining Agent version number", + "CurrentVersionInfo": "The current latest version is:", + "DownloadAgent": "Downloading Agent", + "ErrorDownloadAgent": "Fail to download agent, please check if the network can link", + "ModifyConfiguration": "Modify %s Configuration", + "SuccessModifyConfig": "%s configuration modified successfully, please wait for agent self-restart to take effect", + "ModifyAgentConfInfo": "Please add Agent in the Dashboard first, record the secret", + "PromptModifyAgent_1": "Please enter a domain that resolves to the IP where Dashboard is located (no CDN): ", + "PromptModifyAgent_2": "Please enter Dashboard RPC port (default 5555): ", + "PromptModifyAgent_3": "Please enter the Agent secret: ", + "PromptModifyAgent_4": "Do you want to enable SSL/TLS encryption for the gRPC port (--tls)? Press [y] if yes, the default is not required, and users can press Enter to skip if you don't understand: ", + "ErrorModifyAgent": "All options cannot be empty", + "DownloadDockerScript": "Download Docker Script", + "ErrorCheckDocker": "Please install docker-compose manually." + } +} \ No newline at end of file diff --git a/template.sh b/template.sh new file mode 100644 index 0000000..282ab5e --- /dev/null +++ b/template.sh @@ -0,0 +1,942 @@ +#!/bin/sh +{{.DoNotEdit}} + +NZ_BASE_PATH="/opt/nezha" +NZ_DASHBOARD_PATH="${NZ_BASE_PATH}/dashboard" +NZ_AGENT_PATH="${NZ_BASE_PATH}/agent" +NZ_DASHBOARD_SERVICE="/etc/systemd/system/nezha-dashboard.service" +NZ_DASHBOARD_SERVICERC="/etc/init.d/nezha-dashboard" + +red='\033[0;31m' +green='\033[0;32m' +yellow='\033[0;33m' +plain='\033[0m' +export PATH="$PATH:/usr/local/bin" + +os_arch="" +[ -e /etc/os-release ] && grep -i "PRETTY_NAME" /etc/os-release | grep -qi "alpine" && os_alpine='1' + +sudo() { + myEUID=$(id -ru) + if [ "$myEUID" -ne 0 ]; then + if command -v sudo > /dev/null 2>&1; then + command sudo "$@" + else + err "{{.SudoError}}" + exit 1 + fi + else + "$@" + fi +} + +check_systemd() { + if [ "$os_alpine" != 1 ] && ! command -v systemctl >/dev/null 2>&1; then + echo "{{.SystemdCheckError}}" + exit 1 + fi +} + +err() { + printf "${red}%s${plain}\n" "$*" >&2 +} + +success() { + printf "${green}%s${plain}\n" "$*" +} + +info() { + printf "${yellow}%s${plain}\n" "$*" +} + +geo_check() { + api_list="https://blog.cloudflare.com/cdn-cgi/trace https://dash.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" + set -- "$api_list" + 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 +} + +pre_check() { + umask 077 + + ## 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 + + ## China_IP + if [ -z "$CN" ]; then + geo_check + if [ -n "$isCN" ]; then + echo "{{.GeoInfo}}" + printf "{{.PromptMirror}}" + read -r input + case $input in + [yY][eE][sS] | [yY]) + echo "{{.UseChineseMirror}}" + CN=true + ;; + + [nN][oO] | [nN]) + echo "{{.NoUseChineseMirror}}" + ;; + + [3]) + echo "{{.UseCustomMirror}}" + printf "{{.PromptCustomImage}}" + read -r input + case $input in + *) + CUSTOM_MIRROR=$input + ;; + esac + ;; + *) + echo "{{.NoUseChineseMirror}}" + ;; + 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" + 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" + 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" + fi + fi +} + +installation_check() { + if docker compose version >/dev/null 2>&1; then + DOCKER_COMPOSE_COMMAND="docker compose" + if sudo $DOCKER_COMPOSE_COMMAND ls | grep -qw "$NZ_DASHBOARD_PATH/docker-compose.yaml" >/dev/null 2>&1; then + NEZHA_IMAGES=$(sudo docker images --format "{{.Repository}}:{{.Tag}}" | grep -w "nezha-dashboard") + if [ -n "$NEZHA_IMAGES" ]; then + echo "{{.DockerImageFound}}" + echo "$NEZHA_IMAGES" + IS_DOCKER_NEZHA=1 + FRESH_INSTALL=0 + return + else + echo "{{.DockerImageNotFound}}" + fi + fi + elif command -v docker-compose >/dev/null 2>&1; then + DOCKER_COMPOSE_COMMAND="docker-compose" + if sudo $DOCKER_COMPOSE_COMMAND -f "$NZ_DASHBOARD_PATH/docker-compose.yaml" config >/dev/null 2>&1; then + NEZHA_IMAGES=$(sudo docker images --format "{{.Repository}}:{{.Tag}}" | grep -w "nezha-dashboard") + if [ -n "$NEZHA_IMAGES" ]; then + echo "{{.DockerImageFound}}" + echo "$NEZHA_IMAGES" + IS_DOCKER_NEZHA=1 + FRESH_INSTALL=0 + return + else + echo "{{.DockerImageNotFound}}" + fi + fi + fi + + if [ -f "$NZ_DASHBOARD_PATH/app" ]; then + IS_DOCKER_NEZHA=0 + FRESH_INSTALL=0 + fi +} + +select_version() { + if [ -z "$IS_DOCKER_NEZHA" ]; then + info "{{.SelectInstallMethod}}" + info "1. Docker" + info "2. {{.StandaloneInstall}}" + while true; do + printf "{{.PromptSelectVersion}}" + read -r option + case "${option}" in + 1) + IS_DOCKER_NEZHA=1 + break + ;; + 2) + IS_DOCKER_NEZHA=0 + break + ;; + *) + err "{{.ErrorSelectVersion}}" + ;; + esac + done + fi +} + +update_script() { + echo "> {{.UpdateScript}}" + + curl -sL https://${GITHUB_RAW_URL}/script/{{.script}} -o /tmp/nezha.sh + new_version=$(grep "NZ_VERSION" /tmp/nezha.sh | head -n 1 | awk -F "=" '{print $2}' | sed 's/\"//g;s/,//g;s/ //g') + if [ -z "$new_version" ]; then + echo "{{.ErrorFetchScript}} https://${GITHUB_RAW_URL}/script/install.sh" + return 1 + fi + echo "{{.UpdateScriptInfo}} ${new_version}" + mv -f /tmp/nezha.sh ./nezha.sh && chmod a+x ./nezha.sh + + echo "{{.ExecuteScriptInfo}}" + sleep 3s + clear + exec ./nezha.sh + exit 0 +} + +before_show_menu() { + echo && info "* {{.BeforeShowMenu}} *" && read temp + show_menu +} + +install_base() { + (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 unzip) +} + +install_arch() { + info "Archlinux needs to add nezha-agent user to install libselinux. It will be deleted automatically after installation. It is recommended to check manually" + read -r -p "Do you need to install 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 4695881C254508D1; + 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 -r nezha-agent + echo -e "${red}Info: ${plain}user nezha-agent has been deleted, Be sure to check it manually!\n" + ;; + [nN][oO] | [nN]) + echo "Libselinux will not be installed" + ;; + *) + echo "Libselinux will not be installed" + exit 0 + ;; + esac +} + +install_soft() { + (command -v yum >/dev/null 2>&1 && sudo yum makecache && sudo yum install "$@" selinux-policy -y) || + (command -v apt >/dev/null 2>&1 && sudo apt update && sudo apt install "$@" selinux-utils -y) || + (command -v pacman >/dev/null 2>&1 && sudo pacman -Syu "$@" base-devel --noconfirm && install_arch) || + (command -v apt-get >/dev/null 2>&1 && sudo apt-get update && sudo apt-get install "$@" selinux-utils -y) || + (command -v apk >/dev/null 2>&1 && sudo apk update && sudo apk add "$@" -f) +} + +install_dashboard() { + check_systemd + install_base + + echo "> {{.InstallDashboard}}" + + # Nezha Monitoring Folder + if [ ! "$FRESH_INSTALL" = 0 ]; then + sudo mkdir -p $NZ_DASHBOARD_PATH + else + echo "{{.InstallDashboardInfo}}" + printf "{{.PromptInstallDashboard}}" + read -r input + case $input in + [yY][eE][sS] | [yY]) + echo "{{.ExitInstallation}}" + exit 0 + ;; + [nN][oO] | [nN]) + echo "{{.ContinueInstallation}}" + ;; + *) + echo "{{.ExitInstallation}}" + exit 0 + ;; + esac + fi + + if [ "$IS_DOCKER_NEZHA" = 1 ]; then + install_dashboard_docker + elif [ "$IS_DOCKER_NEZHA" = 0 ]; then + install_dashboard_standalone + fi + + modify_dashboard_config 0 + + if [ $# = 0 ]; then + before_show_menu + fi +} + +install_dashboard_docker() { + if [ ! "$FRESH_INSTALL" = 0 ]; then + if ! command -v docker >/dev/null 2>&1; then + echo "{{.InstallDockerInfo}}" + if [ "$os_alpine" != 1 ]; then + if ! curl -sL https://${Get_Docker_URL} | sudo bash -s "${Get_Docker_Argu}"; then + err "{{.ErrorFetchScript}} ${Get_Docker_URL}" + return 0 + fi + sudo systemctl enable docker.service + sudo systemctl start docker.service + else + sudo apk add docker docker-compose + sudo rc-update add docker + sudo rc-service docker start + fi + success "{{.InstallDockerSuccess}}" + installation_check + fi + fi +} + +install_dashboard_standalone() { + if [ ! -d "${NZ_DASHBOARD_PATH}/resource/template/theme-custom" ] || [ ! -d "${NZ_DASHBOARD_PATH}/resource/static/custom" ]; then + sudo mkdir -p "${NZ_DASHBOARD_PATH}/resource/template/theme-custom" "${NZ_DASHBOARD_PATH}/resource/static/custom" >/dev/null 2>&1 + fi +} + +selinux() { + #Check SELinux + if command -v getenforce >/dev/null 2>&1; then + if getenforce | grep '[Ee]nfor'; then + echo "{{.SELinuxInfo}}" + sudo setenforce 0 >/dev/null 2>&1 + find_key="SELINUX=" + sudo sed -ri "/^$find_key/c${find_key}disabled" /etc/selinux/config + fi + fi +} + +install_agent() { + install_base + selinux + + echo "> {{.InstallAgent}}" + + echo "{{.ObtainAgentVersion}}" + + + _version=$(curl -m 10 -sL "https://api.github.com/repos/nezhahq/agent/releases/latest" | grep "tag_name" | head -n 1 | awk -F ":" '{print $2}' | sed 's/\"//g;s/,//g;s/ //g') + if [ -z "$_version" ]; then + _version=$(curl -m 10 -sL "https://gitee.com/api/v5/repos/naibahq/agent/releases/latest" | awk -F '"' '{for(i=1;i<=NF;i++){if($i=="tag_name"){print $(i+2)}}}') + fi + if [ -z "$_version" ]; then + _version=$(curl -m 10 -sL "https://fastly.jsdelivr.net/gh/nezhahq/agent/" | grep "option\.value" | awk -F "'" '{print $2}' | sed 's/nezhahq\/agent@/v/g') + fi + if [ -z "$_version" ]; then + _version=$(curl -m 10 -sL "https://gcore.jsdelivr.net/gh/nezhahq/agent/" | grep "option\.value" | awk -F "'" '{print $2}' | sed 's/nezhahq\/agent@/v/g') + fi + + if [ -z "$_version" ]; then + err "{{printf .ErrorObtainVersion "Agent"}} https://api.github.com/repos/nezhahq/agent/releases/latest" + return 1 + else + echo "{{.CurrentVersionInfo}} ${_version}" + fi + + # Nezha Monitoring Folder + sudo mkdir -p $NZ_AGENT_PATH + + echo "{{.DownloadAgent}}" + if [ -z "$CN" ]; then + NZ_AGENT_URL="https://${GITHUB_URL}/nezhahq/agent/releases/download/${_version}/nezha-agent_linux_${os_arch}.zip" + else + NZ_AGENT_URL="https://${GITHUB_URL}/naibahq/agent/releases/download/${_version}/nezha-agent_linux_${os_arch}.zip" + fi + + _cmd="wget -t 2 -T 60 -O nezha-agent_linux_${os_arch}.zip $NZ_AGENT_URL >/dev/null 2>&1" + if ! eval "$_cmd"; then + err "{{.ErrorDownloadAgent}} ${GITHUB_URL}" + return 1 + fi + + sudo unzip -qo nezha-agent_linux_${os_arch}.zip && + sudo mv nezha-agent $NZ_AGENT_PATH && + sudo rm -rf nezha-agent_linux_${os_arch}.zip README.md + + if [ $# -ge 3 ]; then + modify_agent_config "$@" + else + modify_agent_config 0 + fi + + if [ $# = 0 ]; then + before_show_menu + fi +} + +modify_agent_config() { + echo "> {{printf .ModifyConfiguration "Agent"}}" + + if [ $# -lt 3 ]; then + echo "{{.ModifyAgentConfInfo}}" + printf "Please enter a domain that resolves to the IP where Dashboard is located (no CDN): " + read -r nz_grpc_host + printf "Please enter Dashboard RPC port (default 5555): " + read -r nz_grpc_port + printf "Please enter the Agent secret: " + read -r nz_client_secret + printf "Do you want to enable SSL/TLS encryption for the gRPC port (--tls)? Press [y] if yes, the default is not required, and users can press Enter to skip if you don't understand: " + read -r nz_grpc_proxy + echo "${nz_grpc_proxy}" | grep -qiw 'Y' && args='--tls' + if [ -z "$nz_grpc_host" ] || [ -z "$nz_client_secret" ]; then + err "All options cannot be empty" + 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 + + _cmd="sudo ${NZ_AGENT_PATH}/nezha-agent service install -s $nz_grpc_host:$nz_grpc_port -p $nz_client_secret $args >/dev/null 2>&1" + + if ! eval "$_cmd"; then + sudo "${NZ_AGENT_PATH}"/nezha-agent service uninstall >/dev/null 2>&1 + sudo "${NZ_AGENT_PATH}"/nezha-agent service install -s "$nz_grpc_host:$nz_grpc_port" -p "$nz_client_secret" "$args" >/dev/null 2>&1 + fi + + success "{{printf .SuccessModifyConfig "Agent"}}" + + #if [[ $# == 0 ]]; then + # before_show_menu + #fi +} + +modify_dashboard_config() { + echo "> {{printf .ModifyConfiguration "Dashboard"}}" + + if [ "$IS_DOCKER_NEZHA" = 1 ]; then + if [ -n "$DOCKER_COMPOSE_COMMAND" ]; then + echo "{{.DownloadDockerScript}}" + _cmd="wget -t 2 -T 60 -O /tmp/nezha-docker-compose.yaml https://${GITHUB_RAW_URL}/script/docker-compose.yaml >/dev/null 2>&1" + if ! eval "$_cmd"; then + err "{{.ErrorFetchScript}} ${GITHUB_RAW_URL}" + return 0 + fi + else + err "{{.ErrorCheckDocker}} https://docs.docker.com/compose/install/linux/" + before_show_menu + fi + fi + + _cmd="wget -t 2 -T 60 -O /tmp/nezha-config.yaml https://${GITHUB_RAW_URL}/script/config.yaml >/dev/null 2>&1" + if ! eval "$_cmd"; then + err "{{.ErrorFetchScript}} ${GITHUB_RAW_URL}" + return 0 + fi + + echo "About the GitHub Oauth2 application: create it at https://github.com/settings/developers, no review required, and fill in the http(s)://domain_or_IP/oauth2/callback" + echo "(Not recommended) About the Gitee Oauth2 application: create it at https://gitee.com/oauth/applications, no auditing required, and fill in the http(s)://domain_or_IP/oauth2/callback" + printf "Please enter the OAuth2 provider (github/gitlab/jihulab/gitee, default github): " + read -r nz_oauth2_type + printf "Please enter the Client ID of the Oauth2 application: " + read -r nz_github_oauth_client_id + printf "Please enter the Client Secret of the Oauth2 application: " + read -r nz_github_oauth_client_secret + printf "Please enter your GitHub/Gitee login name as the administrator, separated by commas: " + read -r nz_admin_logins + printf "Please enter the site title: " + read -r nz_site_title + printf "Please enter the site access port: (default 8008)" + read -r nz_site_port + printf "Please enter the RPC port to be used for Agent access: (default 5555)" + read -r 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 + err "All options cannot be empty" + 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 + if [ "$IS_DOCKER_NEZHA" = 1 ]; then + 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 + elif [ "$IS_DOCKER_NEZHA" = 0 ]; then + sed -i "s/80/${nz_site_port}/" /tmp/nezha-config.yaml + fi + + sudo mkdir -p $NZ_DASHBOARD_PATH/data + sudo mv -f /tmp/nezha-config.yaml ${NZ_DASHBOARD_PATH}/data/config.yaml + if [ "$IS_DOCKER_NEZHA" = 1 ]; then + sudo mv -f /tmp/nezha-docker-compose.yaml ${NZ_DASHBOARD_PATH}/docker-compose.yaml + fi + + if [ "$IS_DOCKER_NEZHA" = 0 ]; then + echo "Downloading service file" + if [ "$os_alpine" != 1 ]; then + _download="sudo wget -t 2 -T 60 -O $NZ_DASHBOARD_SERVICE https://${GITHUB_RAW_URL}/script/nezha-dashboard.service >/dev/null 2>&1" + else + _download="sudo wget -t 2 -T 60 -O $NZ_DASHBOARD_SERVICERC https://${GITHUB_RAW_URL}/script/nezha-dashboard >/dev/null 2>&1" + if ! eval "$_download"; then + err "File failed to get, please check if the network can link ${GITHUB_RAW_URL}" + return 0 + fi + sudo chmod +x $NZ_DASHBOARD_SERVICERC + fi + fi + + success "Dashboard configuration modified successfully, please wait for Dashboard self-restart to take effect" + + restart_and_update + + if [ $# = 0 ]; then + before_show_menu + fi +} + +restart_and_update() { + echo "> Restart and Update Dashboard" + + if [ "$IS_DOCKER_NEZHA" = 1 ]; then + _cmd="restart_and_update_docker" + elif [ "$IS_DOCKER_NEZHA" = 0 ]; then + _cmd="restart_and_update_standalone" + fi + + if eval "$_cmd"; then + success "Nezha Monitoring Restart Successful" + info "Default Dashboard address: domain:site_access_port" + else + err "The restart failed, probably because the boot time exceeded two seconds, please check the log information later" + fi + + if [ $# = 0 ]; then + before_show_menu + fi +} + +restart_and_update_docker() { + sudo $DOCKER_COMPOSE_COMMAND -f ${NZ_DASHBOARD_PATH}/docker-compose.yaml pull + sudo $DOCKER_COMPOSE_COMMAND -f ${NZ_DASHBOARD_PATH}/docker-compose.yaml down + sudo $DOCKER_COMPOSE_COMMAND -f ${NZ_DASHBOARD_PATH}/docker-compose.yaml up -d +} + +restart_and_update_standalone() { + _version=$(curl -m 10 -sL "https://api.github.com/repos/naiba/nezha/releases/latest" | grep "tag_name" | head -n 1 | awk -F ":" '{print $2}' | sed 's/\"//g;s/,//g;s/ //g') + if [ -z "$_version" ]; then + _version=$(curl -m 10 -sL "https://gitee.com/api/v5/repos/naibahq/nezha/releases/latest" | awk -F '"' '{for(i=1;i<=NF;i++){if($i=="tag_name"){print $(i+2)}}}') + fi + if [ -z "$_version" ]; then + _version=$(curl -m 10 -sL "https://fastly.jsdelivr.net/gh/naiba/nezha/" | grep "option\.value" | awk -F "'" '{print $2}' | sed 's/naiba\/nezha@/v/g') + fi + if [ -z "$_version" ]; then + _version=$(curl -m 10 -sL "https://gcore.jsdelivr.net/gh/naiba/nezha/" | grep "option\.value" | awk -F "'" '{print $2}' | sed 's/naiba\/nezha@/v/g') + fi + + if [ -z "$_version" ]; then + err "Fail to obtain agent version, please check if the network can link https://api.github.com/repos/nezhahq/agent/releases/latest" + return 1 + else + echo "The current latest version is: ${_version}" + fi + + if [ "$os_alpine" != 1 ]; then + sudo systemctl daemon-reload + sudo systemctl stop nezha-dashboard + else + sudo rc-service nezha-dashboard stop + fi + + if [ -z "$CN" ]; then + NZ_DASHBOARD_URL="https://${GITHUB_URL}/naiba/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 + + sudo wget -qO $NZ_DASHBOARD_PATH/app.zip "$NZ_DASHBOARD_URL" >/dev/null 2>&1 && sudo unzip -qq -o $NZ_DASHBOARD_PATH/app.zip -d $NZ_DASHBOARD_PATH && sudo mv $NZ_DASHBOARD_PATH/dashboard-linux-$os_arch $NZ_DASHBOARD_PATH/app && sudo rm $NZ_DASHBOARD_PATH/app.zip + + if [ "$os_alpine" != 1 ]; then + sudo systemctl enable nezha-dashboard + sudo systemctl restart nezha-dashboard + else + sudo rc-update add nezha-dashboard + sudo rc-service nezha-dashboard restart + fi +} + +start_dashboard() { + echo "> Start Dashboard" + + if [ "$IS_DOCKER_NEZHA" = 1 ]; then + _cmd="start_dashboard_docker" + elif [ "$IS_DOCKER_NEZHA" = 0 ]; then + _cmd="start_dashboard_standalone" + fi + + if eval "$_cmd"; then + success "Nezha Monitoring Start Successful" + else + err "Failed to start, please check the log message later" + fi + + if [ $# = 0 ]; then + before_show_menu + fi +} + +start_dashboard_docker() { + sudo $DOCKER_COMPOSE_COMMAND -f ${NZ_DASHBOARD_PATH}/docker-compose.yaml up -d +} + +start_dashboard_standalone() { + if [ "$os_alpine" != 1 ]; then + sudo systemctl start nezha-dashboard + else + sudo rc-service nezha-dashboard start + fi +} + +stop_dashboard() { + echo "> Stop Dashboard" + + if [ "$IS_DOCKER_NEZHA" = 1 ]; then + _cmd="stop_dashboard_docker" + elif [ "$IS_DOCKER_NEZHA" = 0 ]; then + _cmd="stop_dashboard_standalone" + fi + + if eval "$_cmd"; then + success "Nezha Monitoring Stop Successful" + else + err "Failed to stop, please check the log message later" + fi + + if [ $# = 0 ]; then + before_show_menu + fi +} + +stop_dashboard_docker() { + sudo $DOCKER_COMPOSE_COMMAND -f ${NZ_DASHBOARD_PATH}/docker-compose.yaml down +} + +stop_dashboard_standalone() { + if [ "$os_alpine" != 1 ]; then + sudo systemctl stop nezha-dashboard + else + sudo rc-service nezha-dashboard stop + fi +} + +show_dashboard_log() { + echo "> View Dashboard Log" + + if [ "$IS_DOCKER_NEZHA" = 1 ]; then + show_dashboard_log_docker + elif [ "$IS_DOCKER_NEZHA" = 0 ]; then + show_dashboard_log_standalone + fi + + if [ $# = 0 ]; then + before_show_menu + fi +} + +show_dashboard_log_docker() { + sudo $DOCKER_COMPOSE_COMMAND -f ${NZ_DASHBOARD_PATH}/docker-compose.yaml logs -f +} + +show_dashboard_log_standalone() { + if [ "$os_alpine" != 1 ]; then + sudo journalctl -xf -u nezha-dashboard.service + else + sudo tail -n 10 /var/log/nezha-dashboard.err + fi +} + +uninstall_dashboard() { + echo "> Uninstall Dashboard" + + if [ "$IS_DOCKER_NEZHA" = 1 ]; then + uninstall_dashboard_docker + elif [ "$IS_DOCKER_NEZHA" = 0 ]; then + uninstall_dashboard_standalone + fi + + clean_all + + if [ $# = 0 ]; then + before_show_menu + fi +} + +uninstall_dashboard_docker() { + sudo $DOCKER_COMPOSE_COMMAND -f ${NZ_DASHBOARD_PATH}/docker-compose.yaml down + sudo rm -rf $NZ_DASHBOARD_PATH + sudo docker rmi -f ghcr.io/naiba/nezha-dashboard >/dev/null 2>&1 + sudo docker rmi -f registry.cn-shanghai.aliyuncs.com/naibahq/nezha-dashboard >/dev/null 2>&1 +} + +uninstall_dashboard_standalone() { + sudo rm -rf $NZ_DASHBOARD_PATH + + if [ "$os_alpine" != 1 ]; then + sudo systemctl disable nezha-dashboard + sudo systemctl stop nezha-dashboard + else + sudo rc-update del nezha-dashboard + sudo rc-service nezha-dashboard stop + fi + + if [ "$os_alpine" != 1 ]; then + sudo rm $NZ_DASHBOARD_SERVICE + else + sudo rm $NZ_DASHBOARD_SERVICERC + fi +} + +show_agent_log() { + echo "> View Agent Log" + + if [ "$os_alpine" != 1 ]; then + sudo journalctl -xf -u nezha-agent.service + else + sudo tail -n 10 /var/log/nezha-agent.err + fi + + if [ $# = 0 ]; then + before_show_menu + fi +} + +uninstall_agent() { + echo "> Uninstall Agent" + + sudo ${NZ_AGENT_PATH}/nezha-agent service uninstall + + sudo rm -rf $NZ_AGENT_PATH + clean_all + + if [ $# = 0 ]; then + before_show_menu + fi +} + +restart_agent() { + echo "> Restart Agent" + + sudo ${NZ_AGENT_PATH}/nezha-agent service restart + + if [ $# = 0 ]; then + before_show_menu + fi +} + +clean_all() { + if [ -z "$(ls -A ${NZ_BASE_PATH})" ]; then + sudo rm -rf ${NZ_BASE_PATH} + fi +} + +show_usage() { + echo "Nezha Monitor Management Script Usage: " + echo "--------------------------------------------------------" + echo "./nezha.sh - Show Menu" + echo "./nezha.sh install_dashboard - Install Dashboard" + echo "./nezha.sh modify_dashboard_config - Modify Dashboard Configuration" + echo "./nezha.sh start_dashboard - Start Dashboard" + echo "./nezha.sh stop_dashboard - Stop Dashboard" + echo "./nezha.sh restart_and_update - Restart and Update the Dashboard" + echo "./nezha.sh show_dashboard_log - View Dashboard Log" + echo "./nezha.sh uninstall_dashboard - Uninstall Dashboard" + echo "--------------------------------------------------------" + echo "./nezha.sh install_agent - Install Agent" + echo "./nezha.sh modify_agent_config - Modify Agent Configuration" + echo "./nezha.sh show_agent_log - View Agent Log" + echo "./nezha.sh uninstall_agent - Uninstall Agent" + echo "./nezha.sh restart_agent - Restart Agent" + echo "./nezha.sh update_script - Update Script" + echo "--------------------------------------------------------" +} + +show_menu() { + printf " + ${green}Nezha Monitor Management Script${plain} ${red}${NZ_VERSION}${plain} + --- https://github.com/naiba/nezha --- + ${green}1.${plain} Install Dashboard + ${green}2.${plain} Modify Dashbaord Configuration + ${green}3.${plain} Start Dashboard + ${green}4.${plain} Stop Dashboard + ${green}5.${plain} Restart and Update Dashboard + ${green}6.${plain} View Dashboard Log + ${green}7.${plain} Uninstall Dashboard + ————————————————- + ${green}8.${plain} Install Agent + ${green}9.${plain} Modify Agent Configuration + ${green}10.${plain} View Agent Log + ${green}11.${plain} Uninstall Agent + ${green}12.${plain} Restart Agent + ————————————————- + ${green}13.${plain} Update Script + ————————————————- + ${green}0.${plain} Exit Script + " + echo && printf "Please enter [0-13]: " && read -r num + 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) + update_script + ;; + *) + err "Please enter the correct number [0-13]" + ;; + esac +} + +pre_check +installation_check + +if [ $# -gt 0 ]; 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 + ;; + "update_script") + update_script 0 + ;; + *) show_usage ;; + esac +else + select_version + show_menu +fi