From fb3d6a7f13fae70512540d55e4a852bd9fc97f80 Mon Sep 17 00:00:00 2001
From: braks <78412429+bcakmakoglu@users.noreply.github.com>
Date: Tue, 12 Dec 2023 19:07:04 +0100
Subject: [PATCH] chore(core): cleanup node wrapper

---
 .../core/src/components/Nodes/NodeWrapper.ts     | 16 ++++------------
 1 file changed, 4 insertions(+), 12 deletions(-)

diff --git a/packages/core/src/components/Nodes/NodeWrapper.ts b/packages/core/src/components/Nodes/NodeWrapper.ts
index 819dbc41..d147ee92 100644
--- a/packages/core/src/components/Nodes/NodeWrapper.ts
+++ b/packages/core/src/components/Nodes/NodeWrapper.ts
@@ -3,15 +3,7 @@ import { until, useVModel } from '@vueuse/core'
 import type { GraphNode, HandleConnectable, NodeComponent } from '~/types'
 import { NodeId, NodeRef } from '~/context'
 import { isInputDOMNode, useDrag, useNodeHooks, useUpdateNodePositions, useVueFlow } from '~/composables'
-import {
-  ARIA_NODE_DESC_KEY,
-  arrowKeyDiffs,
-  calcNextPosition,
-  elementSelectionKeys,
-  getConnectedEdges,
-  getXYZPos,
-  handleNodeClick,
-} from '~/utils'
+import { ARIA_NODE_DESC_KEY, arrowKeyDiffs, calcNextPosition, elementSelectionKeys, getXYZPos, handleNodeClick } from '~/utils'
 
 interface Props {
   id: string
@@ -34,7 +26,6 @@ const NodeWrapper = defineComponent({
 
     const {
       id: vueFlowId,
-      edges,
       noPanClassName,
       selectNodesOnDrag,
       nodesSelectionActive,
@@ -54,15 +45,16 @@ const NodeWrapper = defineComponent({
       snapToGrid,
       snapGrid,
       nodeDragThreshold,
+      getConnectedEdges,
     } = useVueFlow()
 
     const updateNodePositions = useUpdateNodePositions()
 
     const node = useVModel(props, 'node')
 
-    const parentNode = computed(() => findNode(node.value.parentNode))
+    const parentNode = computed(() => findNode(node.value?.parentNode))
 
-    const connectedEdges = computed(() => getConnectedEdges([node.value], edges.value))
+    const connectedEdges = computed(() => getConnectedEdges(node.value ? [node.value] : []))
 
     const nodeElement = ref<HTMLDivElement | null>(null)