From a3f307708b8e01c4b1ebff1109a6f05550deac64 Mon Sep 17 00:00:00 2001
From: bcakmakoglu <78412429+bcakmakoglu@users.noreply.github.com>
Date: Fri, 27 May 2022 22:16:09 +0200
Subject: [PATCH] fix(edges): missing source / target node

---
 packages/vue-flow/src/store/actions.ts | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/packages/vue-flow/src/store/actions.ts b/packages/vue-flow/src/store/actions.ts
index 22f65ba0..4fb4c7aa 100644
--- a/packages/vue-flow/src/store/actions.ts
+++ b/packages/vue-flow/src/store/actions.ts
@@ -175,6 +175,8 @@ export default (state: State, getters: ComputedGetters): Actions => {
       res.push(
         shallowReactive({
           ...parseEdge(edge, Object.assign({}, storedEdge, state.defaultEdgeOptions)),
+          sourceNode,
+          targetNode,
         }),
       )
 
@@ -220,7 +222,15 @@ export default (state: State, getters: ComputedGetters): Actions => {
         if (missingTarget) console.warn(`[vueflow]: Couldn't create edge for target id: ${edge.target}; edge id: ${edge.id}`)
         if (missingTarget || missingSource) return acc
 
-        acc.push(createAdditionChange<GraphEdge>(shallowReactive(edge)))
+        acc.push(
+          createAdditionChange<GraphEdge>(
+            shallowReactive({
+              ...edge,
+              sourceNode,
+              targetNode,
+            }),
+          ),
+        )
       }
 
       return acc