Skip to content

Commit 8b13399

Browse files
authored
improve: add more checks for agent script (#24)
1 parent e19a559 commit 8b13399

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

agent/install.sh

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -153,12 +153,23 @@ install() {
153153
path=$(printf "%s" "$NZ_AGENT_PATH/config-$random.yml")
154154
fi
155155

156+
if [ -z "$NZ_SERVER" ]; then
157+
err "NZ_SERVER should not be empty"
158+
exit 1
159+
fi
160+
161+
if [ -z "$NZ_CLIENT_SECRET" ]; then
162+
err "NZ_CLIENT_SECRET should not be empty"
163+
exit 1
164+
fi
165+
156166
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"
157167

158168
sudo "${NZ_AGENT_PATH}"/nezha-agent service -c "$path" uninstall >/dev/null 2>&1
159169
_cmd="sudo env $env $NZ_AGENT_PATH/nezha-agent service -c $path install"
160170
if ! eval "$_cmd"; then
161171
err "Install nezha-agent service failed"
172+
sudo "${NZ_AGENT_PATH}"/nezha-agent service -c "$path" uninstall >/dev/null 2>&1
162173
exit 1
163174
fi
164175

0 commit comments

Comments
 (0)