From babcbd0e1ff68cf57d7c44390cf62e0aa33811c6 Mon Sep 17 00:00:00 2001
From: Braks <78412429+bcakmakoglu@users.noreply.github.com>
Date: Fri, 22 Oct 2021 12:39:55 +0200
Subject: [PATCH] update: Flow instance provides update node internals func

* Rename onLoadParams to FlowInstance
* Pass update node internals to instance
---
 examples/Basic/Basic.vue                      |  6 +-
 examples/CustomNode/CustomNode.vue            |  4 +-
 examples/DragNDrop/DnD.vue                    |  6 +-
 examples/EdgeTypes/EdgeTypesExample.vue       |  4 +-
 examples/EdgeWithButton/EdgeWithButton.vue    |  4 +-
 examples/Edges/EdgesExample.vue               |  4 +-
 examples/Empty/EmptyExample.vue               |  4 +-
 .../NodeTypeChange/NodeTypeChangeExample.vue  |  4 +-
 examples/Overview/Overview.vue                |  4 +-
 examples/Provider/ProviderExample.vue         |  4 +-
 examples/SaveRestore/Controls.vue             |  4 +-
 examples/SaveRestore/SaveRestoreExample.vue   |  4 +-
 examples/Stress/StressExample.vue             |  4 +-
 .../UpdatableEdge/UpdatableEdgeExample.vue    |  4 +-
 examples/UpdateNodeInternals/CustomNode.vue   | 25 ++++++
 .../UpdateNodeInternalsExample.vue            | 77 +++++++++++++++++++
 examples/router.ts                            |  4 +
 src/composables/useHooks.ts                   |  4 +-
 src/composables/useUpdateNodeInternals.ts     | 12 +--
 src/container/ZoomPane/ZoomPane.vue           |  3 +-
 src/index.ts                                  |  1 -
 src/types/hooks.ts                            |  4 +-
 src/types/types.ts                            |  5 +-
 23 files changed, 149 insertions(+), 46 deletions(-)
 create mode 100644 examples/UpdateNodeInternals/CustomNode.vue
 create mode 100644 examples/UpdateNodeInternals/UpdateNodeInternalsExample.vue

diff --git a/examples/Basic/Basic.vue b/examples/Basic/Basic.vue
index 6ce3e7c5..e4ccc33c 100644
--- a/examples/Basic/Basic.vue
+++ b/examples/Basic/Basic.vue
@@ -7,7 +7,7 @@ import Flow, {
   Edge,
   Elements,
   FlowElement,
-  OnLoadParams,
+  FlowInstance,
   addEdge,
   isNode,
   removeElements,
@@ -24,11 +24,11 @@ const elements = ref<Elements>([
   { id: 'e1-2', source: '1', target: '2', animated: true },
   { id: 'e1-3', source: '1', target: '3' },
 ] as Elements)
-const rfInstance = ref<OnLoadParams | null>(null)
+const rfInstance = ref<FlowInstance | null>(null)
 const onElementsRemove = (elementsToRemove: Elements) =>
   (elements.value = removeElements(elementsToRemove, elements.value as Elements))
 const onConnect = (params: Edge | Connection) => (elements.value = addEdge(params, elements.value as Elements))
-const onLoad = (flowInstance: OnLoadParams) => {
+const onLoad = (flowInstance: FlowInstance) => {
   flowInstance.fitView({ padding: 0.1 })
   rfInstance.value = flowInstance
 }
diff --git a/examples/CustomNode/CustomNode.vue b/examples/CustomNode/CustomNode.vue
index ea28e6c4..8ad80906 100644
--- a/examples/CustomNode/CustomNode.vue
+++ b/examples/CustomNode/CustomNode.vue
@@ -8,7 +8,7 @@ import Flow, {
   Controls,
   Node,
   FlowElement,
-  OnLoadParams,
+  FlowInstance,
   Elements,
   Position,
   SnapGrid,
@@ -25,7 +25,7 @@ const nodeTypes = {
   selectorNode: ColorSelectorNode,
 }
 
-const onLoad = (flowInstance: OnLoadParams) => {
+const onLoad = (flowInstance: FlowInstance) => {
   flowInstance.fitView()
   console.log('flow loaded:', flowInstance)
 }
diff --git a/examples/DragNDrop/DnD.vue b/examples/DragNDrop/DnD.vue
index ee8890a4..d63d29c9 100644
--- a/examples/DragNDrop/DnD.vue
+++ b/examples/DragNDrop/DnD.vue
@@ -1,9 +1,9 @@
 <script lang="ts" setup>
 import Sidebar from './Sidebar.vue'
-import Flow, { addEdge, removeElements, Controls, OnLoadParams, Elements, Connection, Edge, ElementId, Node } from '~/index'
+import Flow, { addEdge, removeElements, Controls, FlowInstance, Elements, Connection, Edge, ElementId, Node } from '~/index'
 import './dnd.css'
 
-const flowInstance = ref<OnLoadParams>()
+const flowInstance = ref<FlowInstance>()
 const elements = ref<Elements>([
   {
     id: '1',
@@ -26,7 +26,7 @@ const onDragOver = (event: DragEvent) => {
 const onConnect = (params: Connection | Edge) => (elements.value = addEdge(params, elements.value as Elements))
 const onElementsRemove = (elementsToRemove: Elements) =>
   (elements.value = removeElements(elementsToRemove, elements.value as Elements))
-const onLoad = (instance: OnLoadParams) => (flowInstance.value = instance)
+const onLoad = (instance: FlowInstance) => (flowInstance.value = instance)
 
 const onDrop = (event: DragEvent) => {
   event.preventDefault()
diff --git a/examples/EdgeTypes/EdgeTypesExample.vue b/examples/EdgeTypes/EdgeTypesExample.vue
index 6f0b6235..3c9e1bb9 100644
--- a/examples/EdgeTypes/EdgeTypesExample.vue
+++ b/examples/EdgeTypes/EdgeTypesExample.vue
@@ -3,9 +3,9 @@
  * Example for checking the different edge types and source and target positions
  */
 import { getElements } from './utils'
-import Flow, { removeElements, addEdge, MiniMap, Controls, Background, OnLoadParams, Connection, Edge, Elements } from '~/index'
+import Flow, { removeElements, addEdge, MiniMap, Controls, Background, FlowInstance, Connection, Edge, Elements } from '~/index'
 
-const onLoad = (flowInstance: OnLoadParams) => {
+const onLoad = (flowInstance: FlowInstance) => {
   flowInstance.fitView()
   console.log(flowInstance.getElements())
 }
diff --git a/examples/EdgeWithButton/EdgeWithButton.vue b/examples/EdgeWithButton/EdgeWithButton.vue
index 75f4b7a1..a36328dc 100644
--- a/examples/EdgeWithButton/EdgeWithButton.vue
+++ b/examples/EdgeWithButton/EdgeWithButton.vue
@@ -9,7 +9,7 @@ import Flow, {
   EdgeType,
   Elements,
   MiniMap,
-  OnLoadParams,
+  FlowInstance,
   removeElements,
 } from '~/index'
 const edgeTypes = {
@@ -32,7 +32,7 @@ const elements = ref<Elements>([
   },
 ] as Elements)
 
-const onLoad = (flowInstance: OnLoadParams) => flowInstance.fitView()
+const onLoad = (flowInstance: FlowInstance) => flowInstance.fitView()
 const onElementsRemove = (elementsToRemove: Elements) =>
   (elements.value = removeElements(elementsToRemove, elements.value as Elements))
 const onConnect = (params: Connection | Edge) =>
diff --git a/examples/Edges/EdgesExample.vue b/examples/Edges/EdgesExample.vue
index 3d78e977..b38450da 100644
--- a/examples/Edges/EdgesExample.vue
+++ b/examples/Edges/EdgesExample.vue
@@ -7,7 +7,7 @@ import Flow, {
   Controls,
   Background,
   Elements,
-  OnLoadParams,
+  FlowInstance,
   FlowElement,
   removeElements,
   Connection,
@@ -81,7 +81,7 @@ const edgeTypes: Record<string, any> = {
 
 const elements = ref<Elements>(initialElements as Elements)
 
-const onLoad = (flowInstance: OnLoadParams) => flowInstance.fitView()
+const onLoad = (flowInstance: FlowInstance) => flowInstance.fitView()
 const onNodeDragStop = (node: Node) => console.log('drag stop', node)
 const onElementClick = (element: FlowElement) => console.log('click', element)
 const onElementsRemove = (elementsToRemove: Elements) => (elements.value = removeElements(elementsToRemove, elements.value))
diff --git a/examples/Empty/EmptyExample.vue b/examples/Empty/EmptyExample.vue
index 379b832c..71c7ff4f 100644
--- a/examples/Empty/EmptyExample.vue
+++ b/examples/Empty/EmptyExample.vue
@@ -11,7 +11,7 @@ import Flow, {
   Elements,
   FlowElement,
   Node,
-  OnLoadParams,
+  FlowInstance,
   addEdge,
   removeElements,
 } from '~/index'
@@ -20,7 +20,7 @@ const elements = ref<Elements>([])
 const onElementsRemove = (elementsToRemove: Elements) => (elements.value = removeElements(elementsToRemove, elements.value))
 const onConnect = (params: Connection | Edge) => (elements.value = addEdge(params, elements.value))
 
-const onLoad = (flowInstance: OnLoadParams) => console.log('flow loaded:', flowInstance)
+const onLoad = (flowInstance: FlowInstance) => console.log('flow loaded:', flowInstance)
 const onElementClick = (element: FlowElement) => console.log('click', element)
 const onNodeDragStop = (node: Node) => console.log('drag stop', node)
 
diff --git a/examples/NodeTypeChange/NodeTypeChangeExample.vue b/examples/NodeTypeChange/NodeTypeChangeExample.vue
index 783de9c9..af248a2c 100644
--- a/examples/NodeTypeChange/NodeTypeChangeExample.vue
+++ b/examples/NodeTypeChange/NodeTypeChangeExample.vue
@@ -1,6 +1,6 @@
 <script lang="ts" setup>
 import { CSSProperties } from 'vue'
-import Flow, { addEdge, Connection, Edge, Elements, isEdge, OnLoadParams, Position } from '~/index'
+import Flow, { addEdge, Connection, Edge, Elements, isEdge, FlowInstance, Position } from '~/index'
 
 const initialElements: Elements = [
   {
@@ -27,7 +27,7 @@ const elements = ref<Elements>(initialElements)
 
 const onConnect = (params: Connection | Edge) => (elements.value = addEdge(params, elements.value))
 
-const onLoad = (flowInstance: OnLoadParams) => flowInstance.fitView()
+const onLoad = (flowInstance: FlowInstance) => flowInstance.fitView()
 
 const changeType = () => {
   elements.value = elements.value.map((el) => {
diff --git a/examples/Overview/Overview.vue b/examples/Overview/Overview.vue
index eac4de56..53836962 100644
--- a/examples/Overview/Overview.vue
+++ b/examples/Overview/Overview.vue
@@ -10,7 +10,7 @@ import Flow, {
   Node,
   Elements,
   FlowElement,
-  OnLoadParams,
+  FlowInstance,
   FlowTransform,
   SnapGrid,
   ArrowHeadType,
@@ -35,7 +35,7 @@ const onSelectionContextMenu = (event: MouseEvent, nodes: Node[]) => {
 const onElementClick = (_: MouseEvent, element: FlowElement) =>
   console.log(`${isNode(element) ? 'node' : 'edge'} click:`, element)
 const onSelectionChange = (elements: Elements | null) => console.log('selection change', elements)
-const onLoad = (flowInstance: OnLoadParams) => {
+const onLoad = (flowInstance: FlowInstance) => {
   console.log('flow loaded:', flowInstance)
   flowInstance.fitView()
 }
diff --git a/examples/Provider/ProviderExample.vue b/examples/Provider/ProviderExample.vue
index 3e6ac97a..ef1a5113 100644
--- a/examples/Provider/ProviderExample.vue
+++ b/examples/Provider/ProviderExample.vue
@@ -4,7 +4,7 @@ import Flow, {
   addEdge,
   removeElements,
   Controls,
-  OnLoadParams,
+  FlowInstance,
   FlowElement,
   Connection,
   Edge,
@@ -16,7 +16,7 @@ import Flow, {
 import './provider.css'
 
 const onElementClick = (element: FlowElement) => console.log('click', element)
-const onLoad = (flowInstance: OnLoadParams) => console.log('flow loaded:', flowInstance)
+const onLoad = (flowInstance: FlowInstance) => console.log('flow loaded:', flowInstance)
 
 const initialElements: Elements = [
   { id: '1', type: 'input', data: { label: 'Node 1' }, position: { x: 250, y: 5 } },
diff --git a/examples/SaveRestore/Controls.vue b/examples/SaveRestore/Controls.vue
index 9ca6cd13..05ad7363 100644
--- a/examples/SaveRestore/Controls.vue
+++ b/examples/SaveRestore/Controls.vue
@@ -1,6 +1,6 @@
 <script lang="ts" setup>
 import localforage from 'localforage'
-import { useZoomPanHelper, OnLoadParams, FlowExportObject, Node } from '~/index'
+import { useZoomPanHelper, FlowInstance, FlowExportObject, Node } from '~/index'
 
 localforage.config({
   name: 'vue-flow',
@@ -14,7 +14,7 @@ const getNodeId = () => `randomnode_${+new Date()}`
 const { transform } = useZoomPanHelper()
 
 type ControlsProps = {
-  flowInstance?: OnLoadParams
+  flowInstance?: FlowInstance
 }
 
 const props = defineProps<ControlsProps>()
diff --git a/examples/SaveRestore/SaveRestoreExample.vue b/examples/SaveRestore/SaveRestoreExample.vue
index 38d755d5..71c6ceb5 100644
--- a/examples/SaveRestore/SaveRestoreExample.vue
+++ b/examples/SaveRestore/SaveRestoreExample.vue
@@ -1,6 +1,6 @@
 <script lang="ts" setup>
 import Controls from './Controls.vue'
-import Flow, { addEdge, Connection, Node, Edge, Elements, OnLoadParams, removeElements } from '~/index'
+import Flow, { addEdge, Connection, Node, Edge, Elements, FlowInstance, removeElements } from '~/index'
 
 import './save.css'
 
@@ -15,7 +15,7 @@ const flowInstance = ref()
 
 const onConnect = (params: Connection | Edge) => (elements.value = addEdge(params, elements.value))
 const onElementsRemove = (elementsToRemove: Elements) => (elements.value = removeElements(elementsToRemove, elements.value))
-const onLoad = (instance: OnLoadParams) => (flowInstance.value = instance)
+const onLoad = (instance: FlowInstance) => (flowInstance.value = instance)
 const onRestore = (els: Elements) => (elements.value = els)
 const onAdd = (el: Node) => (elements.value = elements.value.concat(el))
 </script>
diff --git a/examples/Stress/StressExample.vue b/examples/Stress/StressExample.vue
index 4c5b2a5b..eed0dfe3 100644
--- a/examples/Stress/StressExample.vue
+++ b/examples/Stress/StressExample.vue
@@ -8,7 +8,7 @@ import Flow, {
   isNode,
   Controls,
   Background,
-  OnLoadParams,
+  FlowInstance,
   Elements,
   Connection,
   Edge,
@@ -16,7 +16,7 @@ import Flow, {
 
 const buttonWrapperStyles: CSSProperties = { position: 'absolute', right: 10, top: 10, zIndex: 4 }
 
-const onLoad = (flowInstance: OnLoadParams) => {
+const onLoad = (flowInstance: FlowInstance) => {
   flowInstance.fitView()
   console.log(flowInstance.getElements())
 }
diff --git a/examples/UpdatableEdge/UpdatableEdgeExample.vue b/examples/UpdatableEdge/UpdatableEdgeExample.vue
index 03126db1..b7e103bc 100644
--- a/examples/UpdatableEdge/UpdatableEdgeExample.vue
+++ b/examples/UpdatableEdge/UpdatableEdgeExample.vue
@@ -4,7 +4,7 @@ import Flow, {
   updateEdge,
   addEdge,
   Elements,
-  OnLoadParams,
+  FlowInstance,
   Connection,
   Edge,
   removeElements,
@@ -40,7 +40,7 @@ const initialElements: Elements = [
 ] as Elements
 
 const elements = ref(initialElements)
-const onLoad = (flowInstance: OnLoadParams) => flowInstance.fitView()
+const onLoad = (flowInstance: FlowInstance) => flowInstance.fitView()
 const onEdgeUpdateStart = (edge: Edge) => console.log('start update', edge)
 const onEdgeUpdateEnd = (edge: Edge) => console.log('end update', edge)
 const onEdgeUpdate = ({ edge, connection }: FlowEvents['edgeUpdate']) =>
diff --git a/examples/UpdateNodeInternals/CustomNode.vue b/examples/UpdateNodeInternals/CustomNode.vue
new file mode 100644
index 00000000..1f4be0a9
--- /dev/null
+++ b/examples/UpdateNodeInternals/CustomNode.vue
@@ -0,0 +1,25 @@
+<script lang="ts" setup>
+import { CSSProperties } from 'vue'
+import { Handle, Position, NodeProps } from '~/index'
+
+interface CustomNodeProps extends NodeProps {
+  data: any
+}
+const nodeStyles: CSSProperties = { padding: 10, border: '1px solid #ddd' }
+
+const props = defineProps<CustomNodeProps>()
+</script>
+<template>
+  <div :style="nodeStyles">
+    <Handle type="target" :position="Position.Left" />
+    <div>output handle count: {{ props.data.handleCount }}</div>
+    <Handle
+      v-for="i of props.data.handleCount"
+      :id="`handle-${i}`"
+      :key="`handle-${i}`"
+      type="source"
+      :position="Position.Right"
+      :style="{ top: 10 * i + props.data.handlePosition * 10 }"
+    />
+  </div>
+</template>
diff --git a/examples/UpdateNodeInternals/UpdateNodeInternalsExample.vue b/examples/UpdateNodeInternals/UpdateNodeInternalsExample.vue
new file mode 100644
index 00000000..ab2429eb
--- /dev/null
+++ b/examples/UpdateNodeInternals/UpdateNodeInternalsExample.vue
@@ -0,0 +1,77 @@
+<script lang="ts" setup>
+import { CSSProperties } from 'vue'
+import CustomNode from './CustomNode.vue'
+import Flow, {
+  NodeType,
+  addEdge,
+  useZoomPanHelper,
+  Node,
+  Elements,
+  Connection,
+  Edge,
+  ElementId,
+  Position,
+  isEdge,
+  FlowInstance,
+} from '~/index'
+
+const initialHandleCount = 1
+
+const initialElements: Elements = [
+  {
+    id: '1',
+    type: 'custom',
+    data: { label: 'Node 1', handleCount: initialHandleCount, handlePosition: 0 },
+    position: { x: 250, y: 5 },
+  },
+] as Elements
+
+const buttonWrapperStyles: CSSProperties = { position: 'absolute', right: 10, top: 10, zIndex: 10 }
+
+const nodeTypes: Record<string, NodeType> = {
+  custom: CustomNode as NodeType,
+}
+
+let id = 5
+const getId = (): ElementId => `${id++}`
+
+const elements = ref<Elements>(initialElements)
+const flowInstance = ref<FlowInstance>()
+const onConnect = (params: Connection | Edge) => (elements.value = addEdge(params, elements.value))
+const { project } = useZoomPanHelper()
+
+const onPaneClick = (evt: MouseEvent) =>
+  (elements.value = elements.value.concat({
+    id: getId(),
+    position: project({ x: evt.clientX, y: evt.clientY - 40 }),
+    data: { label: 'new node' },
+    targetPosition: Position.Left,
+    sourcePosition: Position.Right,
+  } as Node))
+
+const toggleHandleCount = () =>
+  (elements.value = elements.value.map((el) => {
+    if (isEdge(el)) {
+      return el
+    }
+
+    return { ...el, data: { ...el.data, handleCount: el.data?.handleCount === 1 ? 2 : 1 } }
+  }))
+
+const toggleHandlePosition = () =>
+  (elements.value = elements.value.map((el) => {
+    if (isEdge(el)) return el
+    return { ...el, data: { ...el.data, handlePosition: el.data?.handlePosition === 0 ? 1 : 0 } }
+  }))
+
+const onLoad = (instance: FlowInstance) => (flowInstance.value = instance)
+</script>
+<template>
+  <Flow :elements="elements" :node-types="nodeTypes" @connect="onConnect" @pane-click="onPaneClick" @load="onLoad">
+    <div :style="buttonWrapperStyles">
+      <button @click="toggleHandleCount">toggle handle count</button>
+      <button @click="toggleHandlePosition">toggle handle position</button>
+      <button @click="() => flowInstance.updateNodeInternals('1')">update node internals</button>
+    </div>
+  </Flow>
+</template>
diff --git a/examples/router.ts b/examples/router.ts
index ce1ddb71..4876a964 100644
--- a/examples/router.ts
+++ b/examples/router.ts
@@ -93,6 +93,10 @@ export const routes: RouterOptions['routes'] = [
     path: '/update-node',
     component: () => import('./UpdateNode/UpdateNodeExample.vue'),
   },
+  {
+    path: '/update-node-internals',
+    component: () => import('./UpdateNodeInternals/UpdateNodeInternalsExample.vue'),
+  },
 ]
 
 export const router = createRouter({
diff --git a/src/composables/useHooks.ts b/src/composables/useHooks.ts
index b7300293..ae121c7e 100644
--- a/src/composables/useHooks.ts
+++ b/src/composables/useHooks.ts
@@ -1,4 +1,4 @@
-import { Connection, Edge, Elements, EmitFunc, FlowHooks, FlowTransform, Node, OnConnectStartParams, OnLoadParams } from '~/types'
+import { Connection, Edge, Elements, EmitFunc, FlowHooks, FlowTransform, Node, OnConnectStartParams, FlowInstance } from '~/types'
 import { Hooks } from '~/context'
 
 // flow event hooks
@@ -22,7 +22,7 @@ const hooks = (): FlowHooks => {
     }>(),
     connectStop: createEventHook<MouseEvent>(),
     connectEnd: createEventHook<MouseEvent>(),
-    load: createEventHook<OnLoadParams>(),
+    load: createEventHook<FlowInstance>(),
     move: createEventHook<FlowTransform | undefined>(),
     moveStart: createEventHook<FlowTransform | undefined>(),
     moveEnd: createEventHook<FlowTransform | undefined>(),
diff --git a/src/composables/useUpdateNodeInternals.ts b/src/composables/useUpdateNodeInternals.ts
index 9052709b..97b46095 100644
--- a/src/composables/useUpdateNodeInternals.ts
+++ b/src/composables/useUpdateNodeInternals.ts
@@ -1,14 +1,10 @@
-import { ElementId, UpdateNodeInternals } from '~/types'
-import useStore from '~/composables/useStore'
+import { ElementId, FlowStore, UpdateNodeInternals } from '~/types'
 
-export default (): UpdateNodeInternals => {
-  const store = useStore()
-
-  return (id: ElementId) => {
+export default (store: FlowStore): UpdateNodeInternals =>
+  (id: ElementId) => {
     const nodeElement: HTMLDivElement | null = document.querySelector(`.vue-flow__node[data-id="${id}"]`)
 
     if (nodeElement) {
-      store.updateNodeDimensions([{ id, nodeElement, forceUpdate: true }])
+      store.updateNodeDimensions({ id, nodeElement, forceUpdate: true })
     }
   }
-}
diff --git a/src/container/ZoomPane/ZoomPane.vue b/src/container/ZoomPane/ZoomPane.vue
index fd16500d..e06272ad 100644
--- a/src/container/ZoomPane/ZoomPane.vue
+++ b/src/container/ZoomPane/ZoomPane.vue
@@ -4,7 +4,7 @@ import { D3ZoomEvent, zoom, zoomIdentity, ZoomTransform } from 'd3-zoom'
 import { get } from '@vueuse/core'
 import { pointer, select } from 'd3-selection'
 import { FlowTransform, KeyCode, PanOnScrollMode } from '~/types'
-import { useHooks, useKeyPress, useStore, useZoomPanHelper } from '~/composables'
+import { useHooks, useKeyPress, useStore, useUpdateNodeInternals, useZoomPanHelper } from '~/composables'
 import { clamp, onLoadGetElements, onLoadProject, onLoadToObject } from '~/utils/graph'
 
 interface ZoomPaneProps {
@@ -208,6 +208,7 @@ onMounted(() => {
         zoomOut,
         zoomTo,
         setTransform,
+        updateNodeInternals: useUpdateNodeInternals(store),
         project: onLoadProject(store),
         getElements: onLoadGetElements(store),
         toObject: onLoadToObject(store),
diff --git a/src/index.ts b/src/index.ts
index f62f8034..8fa5a793 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -17,7 +17,6 @@ export {
   isInputDOMNode,
 } from './utils/graph'
 export { default as useZoomPanHelper } from './composables/useZoomPanHelper'
-export { default as useUpdateNodeInternals } from './composables/useUpdateNodeInternals'
 export { default as useStore } from './composables/useStore'
 export { default as useHooks } from './composables/useHooks'
 export { default as useHandle } from './composables/useHandle'
diff --git a/src/types/hooks.ts b/src/types/hooks.ts
index 7384eba6..c2daa00f 100644
--- a/src/types/hooks.ts
+++ b/src/types/hooks.ts
@@ -1,5 +1,5 @@
 import { EventHook } from '@vueuse/core'
-import { Connection, Edge, Elements, FlowTransform, Node, OnConnectStartParams, OnLoadParams } from '../types'
+import { Connection, Edge, Elements, FlowTransform, Node, OnConnectStartParams, FlowInstance } from '../types'
 
 export type FlowHook<T = any> = EventHook<T>
 
@@ -22,7 +22,7 @@ export interface FlowEvents {
   }
   connectStop: MouseEvent
   connectEnd: MouseEvent
-  load: OnLoadParams
+  load: FlowInstance
   move: FlowTransform | undefined
   moveStart: FlowTransform | undefined
   moveEnd: FlowTransform | undefined
diff --git a/src/types/types.ts b/src/types/types.ts
index c41824cf..b4e5bab6 100644
--- a/src/types/types.ts
+++ b/src/types/types.ts
@@ -77,7 +77,7 @@ export type FitViewFunc = (fitViewOptions?: FitViewParams) => void
 export type ProjectFunc = (position: XYPosition) => XYPosition
 export type ToObjectFunc<T = any> = () => FlowExportObject<T>
 
-export type OnLoadParams<T = any> = {
+export type FlowInstance<T = any> = {
   zoomIn: () => void
   zoomOut: () => void
   zoomTo: (zoomLevel: number) => void
@@ -86,9 +86,10 @@ export type OnLoadParams<T = any> = {
   getElements: () => Elements<T>
   setTransform: (transform: FlowTransform) => void
   toObject: ToObjectFunc<T>
+  updateNodeInternals: (id: ElementId) => void
 }
 
-export type OnLoadFunc<T = any> = (params: OnLoadParams<T>) => void
+export type OnLoadFunc<T = any> = (params: FlowInstance<T>) => void
 
 export interface FlowOptions extends Omit<HTMLAttributes, 'onLoad'> {
   elements: Elements