From 07e597fbb17c8b78797d5f119c1ada182b4d22b8 Mon Sep 17 00:00:00 2001
From: moklick <info@moritzklack.com>
Date: Thu, 13 Aug 2020 09:28:02 +0200
Subject: [PATCH] refactor(edges): pass data in onClick handler closes #430

---
 src/components/Edges/wrapEdge.tsx | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/components/Edges/wrapEdge.tsx b/src/components/Edges/wrapEdge.tsx
index e1c0c357..21b1d471 100644
--- a/src/components/Edges/wrapEdge.tsx
+++ b/src/components/Edges/wrapEdge.tsx
@@ -63,7 +63,13 @@ export default (EdgeComponent: ComponentType<EdgeCompProps>) => {
       setSelectedElements({ id, source, target });
 
       if (onClick) {
-        onClick(event, { id, source, target, type });
+        const edgeElement: Edge = { id, source, target, type };
+
+        if (typeof data !== 'undefined') {
+          edgeElement.data = data;
+        }
+
+        onClick(event, edgeElement);
       }
     };