From 7bc2f98c359f1e847a5e9af47add01d631882ce0 Mon Sep 17 00:00:00 2001
From: djazouli <gael.macherel@gmail.com>
Date: Mon, 3 Aug 2020 10:56:16 +0200
Subject: [PATCH] Add edge handle to edge overlay

---
 src/components/edge.js | 28 +++++++++++++++++-----------
 1 file changed, 17 insertions(+), 11 deletions(-)

diff --git a/src/components/edge.js b/src/components/edge.js
index 2c9ed8d4..d5069cb6 100644
--- a/src/components/edge.js
+++ b/src/components/edge.js
@@ -158,17 +158,23 @@ function Edge({
       </g>
       <g className="edge-mouse-handler">
         <title>{data.handleTooltipText}</title>
-        <path
-          className="edge-overlay-path"
-          ref={edgeOverlayRef}
-          id={id}
-          data-source={data.source}
-          data-target={data.target}
-          d={pathDescription || undefined}
-          style={{
-            ...isBeingDraggedStyle,
-          }}
-        />
+        <g id={id} className="edge-overlay-path" ref={edgeOverlayRef}>
+          <path
+            data-source={data.source}
+            data-target={data.target}
+            d={pathDescription || undefined}
+          />
+          <use
+            href={getShapeId(edgeTypes, data)}
+            width={edgeHandleSize}
+            height={edgeHandleSize}
+            transform={edgeHandleTransformation}
+            style={{
+              transform: edgeHandleTransformation,
+              ...isBeingDraggedStyle,
+            }}
+          />
+        </g>
       </g>
     </g>
   );