From e37712bbb99344709c89f440aa0b49749c6cd2ff Mon Sep 17 00:00:00 2001
From: peterkogo <peter@xyflow.com>
Date: Tue, 7 Jan 2025 12:20:54 +0100
Subject: [PATCH] add NodeConnection

---
 packages/react/src/hooks/useNodeConnections.ts      | 6 +++---
 packages/svelte/src/lib/hooks/useNodeConnections.ts | 6 +++---
 packages/system/src/types/general.ts                | 5 +++++
 3 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/packages/react/src/hooks/useNodeConnections.ts b/packages/react/src/hooks/useNodeConnections.ts
index 02b82cd6..038a70a2 100644
--- a/packages/react/src/hooks/useNodeConnections.ts
+++ b/packages/react/src/hooks/useNodeConnections.ts
@@ -1,7 +1,7 @@
 import { useEffect, useMemo, useRef } from 'react';
 import {
   Connection,
-  HandleConnection,
+  NodeConnection,
   HandleType,
   areConnectionMapsEqual,
   handleConnectionChange,
@@ -38,7 +38,7 @@ export function useNodeConnections({
   nodeId,
   onConnect,
   onDisconnect,
-}: UseNodeConnectionsParams): HandleConnection[] {
+}: UseNodeConnectionsParams): NodeConnection[] {
   const _nodeId = useNodeId();
   const currentNodeId = nodeId ?? _nodeId;
 
@@ -46,7 +46,7 @@ export function useNodeConnections({
     throw new Error(error014);
   }
 
-  const prevConnections = useRef<Map<string, HandleConnection> | null>(null);
+  const prevConnections = useRef<Map<string, NodeConnection> | null>(null);
 
   const connections = useStore(
     (state) =>
diff --git a/packages/svelte/src/lib/hooks/useNodeConnections.ts b/packages/svelte/src/lib/hooks/useNodeConnections.ts
index fc5c9933..0393db21 100644
--- a/packages/svelte/src/lib/hooks/useNodeConnections.ts
+++ b/packages/svelte/src/lib/hooks/useNodeConnections.ts
@@ -2,7 +2,7 @@ import { derived } from 'svelte/store';
 import {
   areConnectionMapsEqual,
   errorMessages,
-  type HandleConnection,
+  type NodeConnection,
   type HandleType
 } from '@xyflow/system';
 
@@ -20,7 +20,7 @@ type UseNodeConnectionsParams = {
   //   onDisconnect?: (connections: Connection[]) => void;
 };
 
-const initialConnections: HandleConnection[] = [];
+const initialConnections: NodeConnection[] = [];
 
 /**
  * Hook to retrieve all edges connected to a node. Can be filtered by handle type and id.
@@ -43,7 +43,7 @@ export function useNodeConnections({ type, nodeId, handleId }: UseNodeConnection
     throw new Error(error014);
   }
 
-  let prevConnections: Map<string, HandleConnection> | undefined = undefined;
+  let prevConnections: Map<string, NodeConnection> | undefined = undefined;
 
   return derived(
     [edges, connectionLookup],
diff --git a/packages/system/src/types/general.ts b/packages/system/src/types/general.ts
index 98d8afb6..3c6eed52 100644
--- a/packages/system/src/types/general.ts
+++ b/packages/system/src/types/general.ts
@@ -33,10 +33,15 @@ export type Connection = {
   targetHandle: string | null;
 };
 
+// TODO: remove in next version
 export type HandleConnection = Connection & {
   edgeId: string;
 };
 
+export type NodeConnection = Connection & {
+  edgeId: string;
+};
+
 export enum ConnectionMode {
   Strict = 'strict',
   Loose = 'loose',