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

feat: edit node — reconfigure port, upstream IP/port

Browse source
This commit is contained in:
admin 2026-08-02 09:07:53 +00:00
parent 221b1968b0
commit c96fb88fbd
1 changed files with 86 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

91
cmd/deploy/main.go
View file

@ -60,9 +60,10 @@ func showMenu() {
fmt.Println("─── Main Menu ───")
fmt.Println(" [1] Deploy a node")
fmt.Println(" [2] Remove a node")
fmt.Println(" [3] Show status")
fmt.Println(" [4] Remote tunnel")
fmt.Println(" [5] Dashboard (TUI)")
fmt.Println(" [3] Edit a node")
fmt.Println(" [4] Show status")
fmt.Println(" [5] Remote tunnel")
fmt.Println(" [6] Dashboard (TUI)")
fmt.Println(" [q] Quit")
fmt.Print("\nChoice: ")
c, _ := reader.ReadString('\n')
@ -72,10 +73,12 @@ func showMenu() {
case "2":
removeMenu()
case "3":
showStatus()
editMenu()
case "4":
startTunnel()
showStatus()
case "5":
startTunnel()
case "6":
if err := ui.Run(); err != nil {
fmt.Println("TUI error:", err)
}
@ -282,6 +285,84 @@ func removeMenu() {
fmt.Printf("✅ %s removed.\n", n.Name)
}
// ─── Edit ─────────────────────────────────────────────────────────────
func editMenu() {
s, _ := state.Load()
if s == nil || len(s.Nodes) == 0 {
fmt.Println("No nodes deployed.")
return
}
fmt.Println("\n─── Edit Node ───")
for i, n := range s.Nodes {
out, _ := exec.Command("systemctl", "is-active", "wallarm-node@"+n.Name).CombinedOutput()
status := "●"
if strings.TrimSpace(string(out)) != "active" { status = "○" }
fmt.Printf(" [%d] %s %s %s → %s:%d\n",
i+1, status, n.Name, n.Address, n.UpstreamIP, n.UpstreamPort)
}
fmt.Print("\nSelect node number: ")
reader := bufio.NewReader(os.Stdin)
choice, _ := reader.ReadString('\n')
idx, err := strconv.Atoi(strings.TrimSpace(choice))
if err != nil || idx < 1 || idx > len(s.Nodes) {
fmt.Println("Invalid choice")
return
}
n := &s.Nodes[idx-1]
fmt.Println("\nCurrent config:")
fmt.Printf(" Name: %s\n", n.Name)
fmt.Printf(" Listen: %s\n", n.Address)
fmt.Printf(" Upstream: %s:%d\n", n.UpstreamIP, n.UpstreamPort)
fmt.Println("\nEnter new values (leave blank to keep current):")
fmt.Print("Listen port [" + strings.TrimPrefix(n.Address, "0.0.0.0:") + "]: ")
port, _ := reader.ReadString('\n')
port = strings.TrimSpace(port)
fmt.Printf("Upstream IP [%s]: ", n.UpstreamIP)
ip, _ := reader.ReadString('\n')
ip = strings.TrimSpace(ip)
fmt.Printf("Upstream port [%d]: ", n.UpstreamPort)
portStr, _ := reader.ReadString('\n')
portStr = strings.TrimSpace(portStr)
if port == "" && ip == "" && portStr == "" {
fmt.Println("No changes.")
return
}
if port != "" { n.Address = "0.0.0.0:" + port }
if ip != "" { n.UpstreamIP = ip }
if portStr != "" {
p, _ := strconv.Atoi(portStr)
if p > 0 { n.UpstreamPort = p }
}
// Update nginx config
base := "/opt/fw/" + n.Name + "/wallarm"
nginxConf := base + "/nginx/conf/nginx.conf"
if _, err := os.Stat(nginxConf); err == nil {
data, _ := os.ReadFile(nginxConf)
newData := string(data)
if port != "" {
newData = strings.ReplaceAll(newData, n.Address, "0.0.0.0:"+port)
}
if ip != "" || portStr != "" {
oldUp := fmt.Sprintf("%s:%d", n.UpstreamIP, n.UpstreamPort)
newUp := fmt.Sprintf("%s:%d", n.UpstreamIP, n.UpstreamPort)
newData = strings.ReplaceAll(newData, oldUp, newUp)
}
os.WriteFile(nginxConf, []byte(newData), 0644)
}
state.Save(s)
fmt.Println("✅ Config updated. Restart the node to apply:")
fmt.Printf(" systemctl restart wallarm-node@%s\n", n.Name)
}
// ─── Status ───────────────────────────────────────────────────────────
func showStatus() {