improve: add more checks for agent script (#24)

This commit is contained in:
UUBulb 2024-12-02 08:35:52 +08:00 committed by GitHub
parent e19a559b03
commit 8b13399059
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -153,12 +153,23 @@ install() {
path=$(printf "%s" "$NZ_AGENT_PATH/config-$random.yml") path=$(printf "%s" "$NZ_AGENT_PATH/config-$random.yml")
fi 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" 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 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" _cmd="sudo env $env $NZ_AGENT_PATH/nezha-agent service -c $path install"
if ! eval "$_cmd"; then if ! eval "$_cmd"; then
err "Install nezha-agent service failed" err "Install nezha-agent service failed"
sudo "${NZ_AGENT_PATH}"/nezha-agent service -c "$path" uninstall >/dev/null 2>&1
exit 1 exit 1
fi fi