From d5252b3be078071b7b17274c11b0e862a107a030 Mon Sep 17 00:00:00 2001
From: admin <admin@sechpoint.com>
Date: Sat, 1 Aug 2026 19:32:39 +0000
Subject: [PATCH] feat: --remove and --status flags

---
 cmd/deploy/main.go | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/cmd/deploy/main.go b/cmd/deploy/main.go
index 95d2145..e715dc4 100644
--- a/cmd/deploy/main.go
+++ b/cmd/deploy/main.go
@@ -304,6 +304,8 @@ deployment.  On subsequent runs, it shows your existing deployments.
 	versionFlag := flag.Bool("version", false, "Show version")
 	tunnelFlag := flag.Bool("tunnel", false, "Start reverse SSH tunnel")
 	deployFlag := flag.Bool("deploy", false, "Deploy directly from /opt/fw/.env (skip TUI)")
+	removeFlag := flag.String("remove", "", "Remove a node by name")
+	statusFlag := flag.Bool("status", false, "Show node status")
 	flag.Parse()
 
 	if *help {
@@ -347,6 +349,21 @@ deployment.  On subsequent runs, it shows your existing deployments.
 		return
 	}
 
+	if *removeFlag != "" {
+		if err := native.RemoveNode(*removeFlag); err != nil {
+			fmt.Fprintf(os.Stderr, "Remove failed: %v\n", err)
+			os.Exit(1)
+		}
+		fmt.Printf("Node %s removed.\n", *removeFlag)
+		return
+	}
+
+	if *statusFlag {
+		out, _ := native.Status("")
+		fmt.Println(out)
+		return
+	}
+
 	// ── Default: preflight → TUI wizard/dashboard ────────────────
 	fmt.Println("═══ Wallarm Deployment Manager ═══")
 	fmt.Printf("   version %s\n", version)