From f37200170345c25b29befdd7d753c686eb2867fa Mon Sep 17 00:00:00 2001
From: Jason Michalski <armooo@armooo.new>
Date: Sat, 15 Jun 2024 19:46:52 -0700
Subject: [PATCH] Color ROUTER nodes in the network graph.

---
 meshview/web.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/meshview/web.py b/meshview/web.py
index c9b2dbb..1efb5f0 100644
--- a/meshview/web.py
+++ b/meshview/web.py
@@ -628,7 +628,7 @@ async def graph_traceroute(request):
     )
 
 @routes.get("/graph/network")
-async def graph_traceroute(request):
+async def graph_network(request):
     nodes = {}
     node_ids = set()
 
@@ -688,14 +688,18 @@ async def graph_traceroute(request):
     graph = pydot.Dot('network', graph_type="digraph", layout="sfdp", overlap="prism", quadtree="normal", repulsiveforce="1.5", k="1")
     for node_id in used_nodes:
         node = await nodes[node_id]
+        color = '#000000'
         if not node:
             node_name = node_id_to_hex(node_id)
         else:
             node_name = f'[{node.short_name}] {node.long_name}\n{node_id_to_hex(node_id)}'
+            if node.role in ('ROUTER', 'ROUTER_CLIENT', 'REPEATER'):
+                color = '#0000FF'
         graph.add_node(pydot.Node(
             str(node_id),
             label=node_name,
             shape='box',
+            color=color,
         ))
 
     if edges: