From 2e419944c062f1bf17cbed2e3c55c5b96539b19b Mon Sep 17 00:00:00 2001
From: 0xdcarns <dillon.carns@gmail.com>
Date: Wed, 28 Dec 2022 12:13:07 -0500
Subject: [PATCH] added network role update to host PUT

---
 controllers/hosts.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/controllers/hosts.go b/controllers/hosts.go
index 55695631..47663ccd 100644
--- a/controllers/hosts.go
+++ b/controllers/hosts.go
@@ -85,6 +85,17 @@ func updateHost(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
+	newNetworks := logic.GetHostNetworks(newHost.ID.String())
+	if len(newNetworks) > 0 {
+		if err = mq.ModifyClient(&mq.MqClient{
+			ID:       currHost.ID.String(),
+			Text:     currHost.Name,
+			Networks: newNetworks,
+		}); err != nil {
+			logger.Log(0, r.Header.Get("user"), "failed to update host networks roles in DynSec:", err.Error())
+		}
+	}
+
 	apiHostData := newHost.ConvertNMHostToAPI()
 	logger.Log(2, r.Header.Get("user"), "updated host", newHost.ID.String())
 	w.WriteHeader(http.StatusOK)