update
This commit is contained in:
commit
eab9b6c02d
198
nezha.sh
Normal file
198
nezha.sh
Normal file
@ -0,0 +1,198 @@
|
|||||||
|
#!/bin/sh
|
||||||
|
|
||||||
|
NZ_BASE_PATH="/opt/nezha"
|
||||||
|
NZ_AGENT_PATH="${NZ_BASE_PATH}/agent"
|
||||||
|
|
||||||
|
red='\033[0;31m'
|
||||||
|
green='\033[0;32m'
|
||||||
|
yellow='\033[0;33m'
|
||||||
|
plain='\033[0m'
|
||||||
|
|
||||||
|
err() {
|
||||||
|
printf "${red}%s${plain}\n" "$*" >&2
|
||||||
|
}
|
||||||
|
|
||||||
|
success() {
|
||||||
|
printf "${green}%s${plain}\n" "$*"
|
||||||
|
}
|
||||||
|
|
||||||
|
info() {
|
||||||
|
printf "${yellow}%s${plain}\n" "$*"
|
||||||
|
}
|
||||||
|
|
||||||
|
sudo() {
|
||||||
|
myEUID=$(id -ru)
|
||||||
|
if [ "$myEUID" -ne 0 ]; then
|
||||||
|
if command -v sudo > /dev/null 2>&1; then
|
||||||
|
command sudo "$@"
|
||||||
|
else
|
||||||
|
err "ERROR: sudo is not installed on the system, the action cannot be proceeded."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
"$@"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
deps_check() {
|
||||||
|
deps="wget unzip grep"
|
||||||
|
set -- "$api_list"
|
||||||
|
for dep in $deps; do
|
||||||
|
if ! command -v "$dep" >/dev/null 2>&1; then
|
||||||
|
err "$dep not found, please install it first."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
}
|
||||||
|
|
||||||
|
geo_check() {
|
||||||
|
api_list="https://blog.cloudflare.com/cdn-cgi/trace https://developers.cloudflare.com/cdn-cgi/trace"
|
||||||
|
ua="Mozilla/5.0 (X11; Linux x86_64; rv:60.0) Gecko/20100101 Firefox/81.0"
|
||||||
|
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
|
||||||
|
}
|
||||||
|
|
||||||
|
env_check() {
|
||||||
|
mach=$(uname -m)
|
||||||
|
case "$mach" in
|
||||||
|
amd64|x86_64)
|
||||||
|
os_arch="amd64"
|
||||||
|
;;
|
||||||
|
i386|i686)
|
||||||
|
os_arch="386"
|
||||||
|
;;
|
||||||
|
aarch64|arm64)
|
||||||
|
os_arch="arm64"
|
||||||
|
;;
|
||||||
|
*arm*)
|
||||||
|
os_arch="arm"
|
||||||
|
;;
|
||||||
|
s390x)
|
||||||
|
os_arch="s390x"
|
||||||
|
;;
|
||||||
|
riscv64)
|
||||||
|
os_arch="riscv64"
|
||||||
|
;;
|
||||||
|
mips)
|
||||||
|
os_arch="mips"
|
||||||
|
;;
|
||||||
|
mipsel|mipsle)
|
||||||
|
os_arch="mipsle"
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
err "Unknown architecture: $uname"
|
||||||
|
exit 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
|
||||||
|
system=$(uname)
|
||||||
|
case "$system" in
|
||||||
|
*Linux*)
|
||||||
|
os="linux"
|
||||||
|
;;
|
||||||
|
*Darwin*)
|
||||||
|
os="darwin"
|
||||||
|
;;
|
||||||
|
*FreeBSD*)
|
||||||
|
os="freebsd"
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
err "Unknown architecture: $system"
|
||||||
|
exit 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
}
|
||||||
|
|
||||||
|
init() {
|
||||||
|
deps_check
|
||||||
|
env_check
|
||||||
|
|
||||||
|
## China_IP
|
||||||
|
if [ -z "$CN" ]; then
|
||||||
|
geo_check
|
||||||
|
if [ -n "$isCN" ]; then
|
||||||
|
CN=true
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -z "$CN" ]; then
|
||||||
|
GITHUB_URL="github.com"
|
||||||
|
else
|
||||||
|
GITHUB_URL="gitee.com"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
install() {
|
||||||
|
echo "Installing..."
|
||||||
|
|
||||||
|
if [ -z "$CN" ]; then
|
||||||
|
NZ_AGENT_URL="https://${GITHUB_URL}/nezhahq/agent/releases/latest/download/nezha-agent_${os}_${os_arch}.zip"
|
||||||
|
else
|
||||||
|
_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)}}}')
|
||||||
|
NZ_AGENT_URL="https://${GITHUB_URL}/naibahq/agent/releases/download/${_version}/nezha-agent_${os}_${os_arch}.zip"
|
||||||
|
fi
|
||||||
|
|
||||||
|
_cmd="wget -t 2 -T 60 -O /tmp/nezha-agent_${os}_${os_arch}.zip $NZ_AGENT_URL >/dev/null 2>&1"
|
||||||
|
if ! eval "$_cmd"; then
|
||||||
|
err "Download nezha-agent release failed, check your network connectivity"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
sudo mkdir -p $NZ_AGENT_PATH
|
||||||
|
|
||||||
|
sudo unzip -qo /tmp/nezha-agent_${os}_${os_arch}.zip -d $NZ_AGENT_PATH &&
|
||||||
|
sudo rm -rf /tmp/nezha-agent_${os}_${os_arch}.zip
|
||||||
|
|
||||||
|
path="$NZ_AGENT_PATH/config.yml"
|
||||||
|
if [ -f "$path" ]; then
|
||||||
|
random=$(LC_ALL=C tr -dc a-z0-9 </dev/urandom | head -c 5)
|
||||||
|
path=$(printf "%s" "$NZ_AGENT_PATH/config-$random.yml")
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -z "$NZ_SERVER" ]; then
|
||||||
|
err "NZ_SERVER should not be empty"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -z "$NZ_CLIENT_SECRET" ]; then
|
||||||
|
err "NZ_CLIENT_SECRET should not be empty"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
env="NZ_SERVER=$NZ_SERVER NZ_CLIENT_SECRET=$NZ_CLIENT_SECRET NZ_TLS=$NZ_TLS NZ_DISABLE_AUTO_UPDATE=$NZ_DISABLE_AUTO_UPDATE NZ_DISABLE_FORCE_UPDATE=$DISABLE_FORCE_UPDATE NZ_DISABLE_COMMAND_EXECUTE=$NZ_DISABLE_COMMAND_EXECUTE NZ_SKIP_CONNECTION_COUNT=$NZ_SKIP_CONNECTION_COUNT"
|
||||||
|
|
||||||
|
sudo "${NZ_AGENT_PATH}"/nezha-agent service -c "$path" uninstall >/dev/null 2>&1
|
||||||
|
_cmd="sudo env $env $NZ_AGENT_PATH/nezha-agent service -c $path install"
|
||||||
|
if ! eval "$_cmd"; then
|
||||||
|
err "Install nezha-agent service failed"
|
||||||
|
sudo "${NZ_AGENT_PATH}"/nezha-agent service -c "$path" uninstall >/dev/null 2>&1
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
success "nezha-agent successfully installed"
|
||||||
|
}
|
||||||
|
|
||||||
|
uninstall() {
|
||||||
|
find "$NZ_AGENT_PATH" -type f -name "*config*.yml" | while read -r file; do
|
||||||
|
sudo "$NZ_AGENT_PATH/nezha-agent" service -c "$file" uninstall
|
||||||
|
sudo rm "$file"
|
||||||
|
done
|
||||||
|
info "Uninstallation completed."
|
||||||
|
}
|
||||||
|
|
||||||
|
if [ "$1" = "uninstall" ]; then
|
||||||
|
uninstall
|
||||||
|
exit
|
||||||
|
fi
|
||||||
|
|
||||||
|
init
|
||||||
|
install
|
||||||
Loading…
Reference in New Issue
Block a user