From a19474c3416959ba4e56543176310d9b2a58a5bd Mon Sep 17 00:00:00 2001
From: Braks <78412429+bcakmakoglu@users.noreply.github.com>
Date: Wed, 20 Apr 2022 22:56:27 +0200
Subject: [PATCH] feat(edges): Add source/target nodes to graph edges

# What's changed?

* Bind source and target nodes to graph edges for convenience
---
 package/src/store/actions.ts | 4 ++++
 package/src/types/edge.ts    | 2 ++
 2 files changed, 6 insertions(+)

diff --git a/package/src/store/actions.ts b/package/src/store/actions.ts
index 09514f8a..ec806040 100644
--- a/package/src/store/actions.ts
+++ b/package/src/store/actions.ts
@@ -249,6 +249,8 @@ export default (state: State, getters: ComputedGetters): Actions => {
           ...state.defaultEdgeOptions,
           ...storedEdge,
         }),
+        sourceNode,
+        targetNode,
       })
 
       return res
@@ -287,6 +289,8 @@ export default (state: State, getters: ComputedGetters): Actions => {
         state.edges.push({
           ...state.defaultEdgeOptions,
           ...edge,
+          sourceNode,
+          targetNode,
         })
       }
     })
diff --git a/package/src/types/edge.ts b/package/src/types/edge.ts
index 20f5a0f4..36079a7d 100644
--- a/package/src/types/edge.ts
+++ b/package/src/types/edge.ts
@@ -97,6 +97,8 @@ export interface EdgePositions {
 export type GraphEdge<Data = ElementData> = Edge<Data> & {
   selected?: boolean
   z?: number
+  sourceNode: GraphNode
+  targetNode: GraphNode
 } & EdgePositions
 
 /** these props are passed to edge components */