From 449c3055e0e7e8d56ef845029070ff522049dec5 Mon Sep 17 00:00:00 2001
From: braks <78412429+bcakmakoglu@users.noreply.github.com>
Date: Fri, 24 Mar 2023 22:08:05 +0100
Subject: [PATCH] chore(core): cleanup

---
 packages/core/src/components/Edges/wrapEdge.tsx    | 12 ++++++++----
 packages/core/src/container/EdgeRenderer/index.tsx |  2 +-
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/packages/core/src/components/Edges/wrapEdge.tsx b/packages/core/src/components/Edges/wrapEdge.tsx
index a5014af9..cb7d987f 100644
--- a/packages/core/src/components/Edges/wrapEdge.tsx
+++ b/packages/core/src/components/Edges/wrapEdge.tsx
@@ -207,7 +207,8 @@ export default (EdgeComponent: ComponentType<EdgeProps>) => {
         )}
         {isUpdatable && (
           <>
-            {isUpdatable === 'source' && <EdgeAnchor
+            {(isUpdatable === 'source' || isUpdatable === true) && (
+              <EdgeAnchor
                 position={sourcePosition}
                 centerX={sourceX}
                 centerY={sourceY}
@@ -216,8 +217,10 @@ export default (EdgeComponent: ComponentType<EdgeProps>) => {
                 onMouseEnter={onEdgeUpdaterMouseEnter}
                 onMouseOut={onEdgeUpdaterMouseOut}
                 type="source"
-            />}
-            {isUpdatable === 'target' && <EdgeAnchor
+              />
+            )}
+            {(isUpdatable === 'target' || isUpdatable === true) && (
+              <EdgeAnchor
                 position={targetPosition}
                 centerX={targetX}
                 centerY={targetY}
@@ -226,7 +229,8 @@ export default (EdgeComponent: ComponentType<EdgeProps>) => {
                 onMouseEnter={onEdgeUpdaterMouseEnter}
                 onMouseOut={onEdgeUpdaterMouseOut}
                 type="target"
-            />}
+              />
+            )}
           </>
         )}
       </g>
diff --git a/packages/core/src/container/EdgeRenderer/index.tsx b/packages/core/src/container/EdgeRenderer/index.tsx
index 26a12c1c..ed32e62b 100644
--- a/packages/core/src/container/EdgeRenderer/index.tsx
+++ b/packages/core/src/container/EdgeRenderer/index.tsx
@@ -115,7 +115,7 @@ const EdgeRenderer = ({
               const sourcePosition = sourceHandle?.position || Position.Bottom;
               const targetPosition = targetHandle?.position || Position.Top;
               const isFocusable = !!(edge.focusable || (edgesFocusable && typeof edge.focusable === 'undefined'));
-              const isUpdatable = edge.updatable || (edgesUpdatable && typeof edge.updatable === 'undefined');
+              const isUpdatable = edge.updatable || typeof onEdgeUpdate !== 'undefined' || (edgesUpdatable && typeof edge.updatable === 'undefined');
 
               if (!sourceHandle || !targetHandle) {
                 onError?.('008', errorMessages['error008'](sourceHandle, edge));