customer-engineering/wallarm
3
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: check node.yaml for success, ignore env.list exit code

Browse source
This commit is contained in:
admin 2026-08-01 18:57:24 +00:00
parent 8484ca46a7
commit f7689b4d6a
1 changed files with 19 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
internal/native/native.go
View file

@ -78,12 +78,12 @@ func InstallNode(node state.Node, apiToken, apiHost, labels string) error {
}
}()
// 6. Run setup.sh
// 6. Run setup.sh (ignore env.list error, check for real success)
fmt.Printf("[%s] Running setup...\n", node.Name)
cmd = exec.Command("bash", filepath.Join(DeployTo, "setup.sh"),
"--batch", "--token", apiToken, "--cloud", cloudFromHost(apiHost),
"--custom-ngx-build",
)
// Use +e to prevent premature exit on env.list sed failure
cmd = exec.Command("bash", "-c",
fmt.Sprintf("set +e; bash %s/setup.sh --batch --token %s --cloud %s --custom-ngx-build",
DeployTo, apiToken, cloudFromHost(apiHost)))
cmd.Dir = DeployTo
logFile, _ := os.Create(filepath.Join(DeployTo, "install.log"))
@ -96,6 +96,19 @@ func InstallNode(node state.Node, apiToken, apiHost, labels string) error {
if logFile != nil {
logFile.Close()
}
// Success check: node.yaml exists = node registered
nodeYaml := filepath.Join(DeployTo, "etc", "wallarm", "node.yaml")
if _, err := os.Stat(nodeYaml); err == nil {
// Move /opt/wallarm → /opt/fw/{name}/wallarm
os.MkdirAll(filepath.Join(BaseDir, node.Name), 0755)
os.RemoveAll(instanceDir)
os.Rename(DeployTo, instanceDir)
fmt.Printf("[%s] Done. Installed to %s\n", node.Name, instanceDir)
return nil
}
// Failed: show log
if runErr != nil {
if data, _ := os.ReadFile(filepath.Join(DeployTo, "install.log")); len(data) > 0 {
lines := strings.Split(string(data), "\n")
@ -107,16 +120,7 @@ func InstallNode(node state.Node, apiToken, apiHost, labels string) error {
}
return fmt.Errorf("setup failed: %v", runErr)
}
// 6. Move /opt/wallarm → /opt/fw/{name}/wallarm
os.MkdirAll(BaseDir+"/"+node.Name, 0755)
os.RemoveAll(instanceDir)
if err := os.Rename(DeployTo, instanceDir); err != nil {
return fmt.Errorf("move to instance dir: %w", err)
}
fmt.Printf("[%s] Done. Installed to %s\n", node.Name, instanceDir)
return nil
return fmt.Errorf("node registration failed — no node.yaml found")
}
func copyNginx(dir string) {