Browse Source

created comms filter

0xdcarns 3 years ago
parent
commit
0325795bf2
2 changed files with 13 additions and 5 deletions
  1. 12 4
      controllers/network.go
  2. 1 1
      logic/nodes.go

+ 12 - 4
controllers/network.go

@@ -57,9 +57,7 @@ func getNetworks(w http.ResponseWriter, r *http.Request) {
 		for _, network := range networksSlice {
 			netObject, parentErr := logic.GetParentNetwork(network)
 			if parentErr == nil {
-				if netObject.IsComms != "yes" {
-					allnetworks = append(allnetworks, netObject)
-				}
+				allnetworks = append(allnetworks, netObject)
 			}
 		}
 	}
@@ -71,7 +69,7 @@ func getNetworks(w http.ResponseWriter, r *http.Request) {
 	}
 	logger.Log(2, r.Header.Get("user"), "fetched networks.")
 	w.WriteHeader(http.StatusOK)
-	json.NewEncoder(w).Encode(allnetworks)
+	json.NewEncoder(w).Encode(filterCommsNetwork(allnetworks))
 }
 
 // Simple get network function
@@ -361,3 +359,13 @@ func isCommsEdit(w http.ResponseWriter, r *http.Request, netname string) bool {
 	}
 	return false
 }
+
+func filterCommsNetwork(networks []models.Network) []models.Network {
+	var filterdNets []models.Network
+	for i := range networks {
+		if networks[i].IsComms != "yes" && networks[i].NetID != servercfg.GetCommsID() {
+			filterdNets = append(filterdNets, networks[i])
+		}
+	}
+	return filterdNets
+}

+ 1 - 1
logic/nodes.go

@@ -573,7 +573,7 @@ func IsLocalServer(node *models.Node) bool {
 
 // IsNodeInComms returns if node is in comms network or not
 func IsNodeInComms(node *models.Node) bool {
-	return node.Network == servercfg.GetCommsID()
+	return node.Network == servercfg.GetCommsID() && node.IsServer != "yes"
 }
 
 // validateServer - make sure servers dont change port or address