From 1beb8136fb309cd60f79d2e7ab5c4a94b2fee5dc Mon Sep 17 00:00:00 2001
From: Philippe Merle <philippe.merle@inria.fr>
Date: Sat, 19 Apr 2025 09:55:24 +0200
Subject: [PATCH] Add graphical attributes to configurable clusters

---
 bin/kube-diagrams | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/bin/kube-diagrams b/bin/kube-diagrams
index 41f8693..67194d3 100755
--- a/bin/kube-diagrams
+++ b/bin/kube-diagrams
@@ -193,6 +193,7 @@ class ResourceCluster:
         """
         self.resources = {} # dict<str, resource>
         self.clusters = {} # dict<str, Cluster>
+        self.graph_attr = {}
 
     def get_or_create_cluster(self, name):
         """
@@ -872,6 +873,7 @@ def process_resource(resource):
                     cluster = cluster.get_or_create_cluster(
                                 f"{cluster_config['title']}: {labels[label]}"
                     )
+                    cluster.graph_attr = cluster_config.get("graph_attr", {})
                     clusters.remove(cluster_config)
                     return process_clusters(cluster, resource, clusters)
         return cluster
@@ -1006,7 +1008,7 @@ def create_nodes(cluster):
         if get_node_config(resource).get("show", True):
             diagram_nodes[rid] = create_diagram_node(resource)
     for cid, sub_cluster in cluster.clusters.items():
-        with Cluster(cid):
+        with Cluster(cid, graph_attr=sub_cluster.graph_attr):
             create_nodes(sub_cluster)
 
 def create_custom_node(node_id, node_def):