Fixed truncate issue while creating nodes

Author: roopeshsn

README.md

@@ -1,9 +1,13 @@
 # Netmap
 
-Netmap stands for Network Mapper, a visualizer for your inventory of network devices. Netmap starts collecting LLDP information with a single device credential and recursively collects neighbor of neighbors information. Built with love by Roopesh and friends in Go.
+Netmap stands for Network Mapper, a visualizer for your inventory of network devices. Netmap starts collecting LLDP information with a single device credential and recursively lookup to create a graph. At the moment the tool supports only Arista devices. 
+
+Built with love by Roopesh and friends in Go.
 
 ## Usage
 
+The graph can be created using the command `netmap create`,
+
 ```
 roopesh:~/ $ netmap create --help                                                                                                                                                                         
 Create topology diagram
@@ -21,5 +25,5 @@ roopesh:~/ $ netmap create -n ok270 -u admin -p password
 
 <figure>
   <img src="./graph.png" alt="Graph created by netmap">
-  <figcaption>Graph created by Netmap</figcaption>
+  <figcaption>The above graph was created by Netmap. The labels of nodes and edges are truncated for privacy concerns.</figcaption>
 </figure>

go.mod

@@ -3,13 +3,13 @@ module opennetworktools/netmap
 go 1.23.1
 
 require (
+	aqwari.net/xml v0.0.0-20210331023308-d9421b293817
 	github.com/aristanetworks/goeapi v1.0.0
 	github.com/goccy/go-graphviz v0.2.9
 	github.com/spf13/cobra v1.9.1
 )
 
 require (
-	aqwari.net/xml v0.0.0-20210331023308-d9421b293817 // indirect
 	github.com/disintegration/imaging v1.6.2 // indirect
 	github.com/flopp/go-findfont v0.1.0 // indirect
 	github.com/fogleman/gg v1.3.0 // indirect

graph.png

@@ -8,6 +8,7 @@ import (
 	"opennetworktools/netmap/internal/utils"
 	"os"
 	"path/filepath"
+	"strings"
 
 	"aqwari.net/xml/xmltree"
 	"github.com/goccy/go-graphviz"
@@ -42,11 +43,12 @@ func SaveTopologyWithGraphviz(ctx context.Context, networkMap *NetworkMap, times
 
 	// Create nodes
 	for device := range networkMap.Devices {
-		node, err := graph.CreateNodeByName(utils.TruncateString(device, 8))
+		node, err := graph.CreateNodeByName(device)
 		if err != nil {
 			fmt.Println("Error creating node:", err)
 			continue
 		}
+		node.SetLabel(utils.TruncateString(device, 3))
 		node.SetShape(cgraph.EllipseShape)
 		node.SetStyle(cgraph.FilledNodeStyle)
 		nodes[device] = node
@@ -57,11 +59,12 @@ func SaveTopologyWithGraphviz(ctx context.Context, networkMap *NetworkMap, times
 		for _, edge := range edges {
 			neighbor := edge.Neighbor
 			if nodes[neighbor] == nil {
-				node, err := graph.CreateNodeByName(utils.TruncateString(neighbor, 8))
+				node, err := graph.CreateNodeByName(neighbor)
 				if err != nil {
 					fmt.Println("Error creating node:", err)
 					continue
 				}
+				node.SetLabel(utils.TruncateString(neighbor, 3))
 				node.SetShape(cgraph.EllipseShape)
 				node.SetStyle(cgraph.FilledNodeStyle)
 				nodes[neighbor] = node
@@ -72,7 +75,7 @@ func SaveTopologyWithGraphviz(ctx context.Context, networkMap *NetworkMap, times
 				fmt.Println("Error creating edge:", err)
 				continue
 			}
-			e.SetLabel(fmt.Sprintf("%s ↔ %s", edge.LocalPort, edge.NeighborPort))
+			e.SetLabel(fmt.Sprintf("%s - %s", formatEdgeName(edge.LocalPort), formatEdgeName(edge.NeighborPort)))
 		}
 	}
 
@@ -167,3 +170,11 @@ func renderOutputToFilename(filePath string, outData []byte) error {
 
 	return nil
 }
+
+func formatEdgeName(name string) string {
+	if strings.Contains(name, ".") {
+		parts := strings.Split(name, ".")
+		return parts[0]
+	}
+	return name
+}