feat(props): add onConnectEnd function #451

src/components/Handle/BaseHandle.tsx

@@ -9,6 +9,7 @@ import {
   OnConnectFunc,
   OnConnectStartFunc,
   OnConnectStopFunc,
+  OnConnectEndFunc,
   Connection,
   SetConnectionId,
 } from '../../types';
@@ -22,6 +23,7 @@ interface BaseHandleProps {
   onConnect: OnConnectFunc;
   onConnectStart?: OnConnectStartFunc;
   onConnectStop?: OnConnectStopFunc;
+  onConnectEnd?: OnConnectEndFunc;
   position: Position;
   setConnectionNodeId: SetSourceIdFunc;
   setPosition: (pos: XYPosition) => void;
@@ -47,7 +49,8 @@ function onMouseDown(
   isTarget: boolean,
   isValidConnection: ValidConnectionFunc,
   onConnectStart?: OnConnectStartFunc,
-  onConnectStop?: OnConnectStopFunc
+  onConnectStop?: OnConnectStopFunc,
+  onConnectEnd?: OnConnectEndFunc
 ): void {
   const reactFlowNode = document.querySelector('.react-flow');
 
@@ -142,6 +145,10 @@ function onMouseDown(
       onConnect(connection);
     }
 
+    if (onConnectEnd) {
+      onConnectEnd(event);
+    }
+
     resetRecentHandle();
     setConnectionNodeId({ connectionNodeId: null, connectionHandleType: null });
 
@@ -159,6 +166,7 @@ const BaseHandle = ({
   onConnect,
   onConnectStart,
   onConnectStop,
+  onConnectEnd,
   position,
   setConnectionNodeId,
   setPosition,
@@ -196,7 +204,8 @@ const BaseHandle = ({
           isTarget,
           isValidConnection,
           onConnectStart,
-          onConnectStop
+          onConnectStop,
+          onConnectEnd
         )
       }
       {...rest}

src/components/Handle/index.tsx

@@ -23,6 +23,8 @@ const Handle = ({
   const onConnectAction = useStoreState((state) => state.onConnect);
   const onConnectStart = useStoreState((state) => state.onConnectStart);
   const onConnectStop = useStoreState((state) => state.onConnectStop);
+  const onConnectEnd = useStoreState((state) => state.onConnectEnd);
+
   const onConnectExtended = (params: Connection) => {
     if (onConnectAction) {
       onConnectAction(params);
@@ -44,6 +46,7 @@ const Handle = ({
       onConnect={onConnectExtended}
       onConnectStart={onConnectStart}
       onConnectStop={onConnectStop}
+      onConnectEnd={onConnectEnd}
       type={type}
       position={position}
       isValidConnection={isValidConnection}