From 9f9e709aa6ba295c3f4a09b9e33f1f9d7f8ae59d Mon Sep 17 00:00:00 2001
From: Shailendra Ahir <shailendraahir@hotmail.com>
Date: Tue, 22 Mar 2022 23:39:50 +0530
Subject: [PATCH] #8850 Added renderNode and renderLabel props to support
 custom node and custom label

Signed-off-by: Shailendra Ahir <shailendraahir@hotmail.com>
---
 .../EntityRelationsGraph/EntityRelationsGraph.tsx         | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/plugins/catalog-graph/src/components/EntityRelationsGraph/EntityRelationsGraph.tsx b/plugins/catalog-graph/src/components/EntityRelationsGraph/EntityRelationsGraph.tsx
index c5c83a54cd..19ca5398a6 100644
--- a/plugins/catalog-graph/src/components/EntityRelationsGraph/EntityRelationsGraph.tsx
+++ b/plugins/catalog-graph/src/components/EntityRelationsGraph/EntityRelationsGraph.tsx
@@ -79,6 +79,8 @@ export const EntityRelationsGraph = ({
   relationPairs = ALL_RELATION_PAIRS,
   className,
   zoom = 'enabled',
+  renderNode,
+  renderLabel,
 }: {
   rootEntityNames: CompoundEntityRef | CompoundEntityRef[];
   maxDepth?: number;
@@ -91,6 +93,8 @@ export const EntityRelationsGraph = ({
   relationPairs?: RelationPairs;
   className?: string;
   zoom?: 'enabled' | 'disabled' | 'enable-on-click';
+  renderNode?: any;
+  renderLabel?: any;
 }) => {
   const theme = useTheme();
   const classes = useStyles();
@@ -127,8 +131,8 @@ export const EntityRelationsGraph = ({
         <DependencyGraph
           nodes={nodes}
           edges={edges}
-          renderNode={CustomNode}
-          renderLabel={CustomLabel}
+          renderNode={renderNode || CustomNode}
+          renderLabel={renderLabel || CustomLabel}
           direction={direction}
           className={classes.graph}
           paddingX={theme.spacing(4)}