From 62216634480906a3fd949f36b38be0c9a04e7167 Mon Sep 17 00:00:00 2001
From: braks <78412429+bcakmakoglu@users.noreply.github.com>
Date: Tue, 9 Aug 2022 16:38:26 +0200
Subject: [PATCH] fix(nodes): pass node as prop to wrapper

# What's changed?

* instead of computed getter use prop to pass node
* avoids error when node gets deleting from custom node with `removeNodes`
---
 packages/vue-flow/src/components/Nodes/NodeWrapper.vue     | 7 +++++--
 .../vue-flow/src/container/NodeRenderer/NodeRenderer.vue   | 1 +
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/packages/vue-flow/src/components/Nodes/NodeWrapper.vue b/packages/vue-flow/src/components/Nodes/NodeWrapper.vue
index af22087f..3b2b8c48 100644
--- a/packages/vue-flow/src/components/Nodes/NodeWrapper.vue
+++ b/packages/vue-flow/src/components/Nodes/NodeWrapper.vue
@@ -1,6 +1,7 @@
 <script lang="ts" setup>
+import { useVModel } from '@vueuse/core'
 import { useDrag, useNodeHooks, useVueFlow } from '../../composables'
-import type { NodeComponent, SnapGrid, XYZPosition } from '../../types'
+import type { GraphNode, NodeComponent, SnapGrid, XYZPosition } from '../../types'
 import { NodeId } from '../../context'
 import { getConnectedEdges, getHandleBounds, getXYZPos, handleNodeClick } from '../../utils'
 
@@ -12,6 +13,7 @@ const { id, type, name, draggable, selectable, connectable, snapGrid, ...props }
   snapGrid?: SnapGrid
   type: NodeComponent | Function | Object | false
   name: string
+  node: GraphNode
 }>()
 
 provide(NodeId, id)
@@ -31,7 +33,8 @@ const {
   onUpdateNodeInternals,
 } = $(useVueFlow())
 
-const node = $computed(() => getNode(id)!)
+const node = $(useVModel(props, 'node'))
+
 const parentNode = $computed(() => (node.parentNode ? getNode(node.parentNode) : undefined))
 
 const nodeElement = ref()
diff --git a/packages/vue-flow/src/container/NodeRenderer/NodeRenderer.vue b/packages/vue-flow/src/container/NodeRenderer/NodeRenderer.vue
index 29957380..755e8904 100644
--- a/packages/vue-flow/src/container/NodeRenderer/NodeRenderer.vue
+++ b/packages/vue-flow/src/container/NodeRenderer/NodeRenderer.vue
@@ -64,6 +64,7 @@ export default {
       :draggable="draggable(node.draggable)"
       :selectable="selectable(node.selectable)"
       :connectable="connectable(node.connectable)"
+      :node="node"
     />
   </div>
 </template>