From 394f9a01b38de3d5dfdf9dfdbf160783c3502a71 Mon Sep 17 00:00:00 2001
From: bcakmakoglu <78412429+bcakmakoglu@users.noreply.github.com>
Date: Wed, 22 Jun 2022 13:37:11 +0200
Subject: [PATCH] feat(nodes): add `useNodeHooks` composable

# What's changed?

* create node hooks for context and global event emitters
---
 .../src/components/Nodes/NodeWrapper.vue      | 27 ++++---
 packages/vue-flow/src/composables/index.ts    |  1 +
 .../vue-flow/src/composables/useNodeHooks.ts  | 74 +++++++++++++++++++
 3 files changed, 90 insertions(+), 12 deletions(-)
 create mode 100644 packages/vue-flow/src/composables/useNodeHooks.ts

diff --git a/packages/vue-flow/src/components/Nodes/NodeWrapper.vue b/packages/vue-flow/src/components/Nodes/NodeWrapper.vue
index 7268a3fa..ca216e1b 100644
--- a/packages/vue-flow/src/components/Nodes/NodeWrapper.vue
+++ b/packages/vue-flow/src/components/Nodes/NodeWrapper.vue
@@ -1,5 +1,5 @@
 <script lang="ts" setup>
-import { useDrag, useVueFlow } from '../../composables'
+import { useDrag, useNodeHooks, useVueFlow } from '../../composables'
 import type { NodeComponent, SnapGrid, XYZPosition } from '../../types'
 import { NodeId } from '../../context'
 import { getConnectedEdges, getHandleBounds, getXYZPos, handleNodeClick } from '../../utils'
@@ -21,7 +21,6 @@ const {
   viewport,
   noDragClassName,
   noPanClassName,
-  emits,
   selectNodesOnDrag,
   setState,
   updateNodeDimensions,
@@ -38,6 +37,8 @@ const parentNode = $computed(() => (node.parentNode ? getNode(node.parentNode) :
 
 const nodeElement = ref()
 
+const { emit, on } = useNodeHooks(node)
+
 const dragging = useDrag({
   id,
   el: nodeElement,
@@ -45,13 +46,13 @@ const dragging = useDrag({
   noDragClassName: $$(noDragClassName) as any,
   handleSelector: node.dragHandle,
   onStart(event, node, nodes) {
-    emits.nodeDragStart({ event, node, nodes })
+    emit.dragStart({ event, node, nodes })
   },
   onDrag(event, node, nodes) {
-    emits.nodeDrag({ event, node, nodes })
+    emit.drag({ event, node, nodes })
   },
   onStop(event, node, nodes) {
-    emits.nodeDragStop({ event, node, nodes })
+    emit.dragStop({ event, node, nodes })
   },
 })
 
@@ -123,38 +124,39 @@ onMounted(() => {
 
 const onMouseEnter = (event: MouseEvent) => {
   if (!dragging) {
-    emits.nodeMouseEnter({ event, node, connectedEdges: getConnectedEdges([node], edges) })
+    emit.mouseEnter({ event, node, connectedEdges: getConnectedEdges([node], edges) })
   }
 }
 
 const onMouseMove = (event: MouseEvent) => {
   if (!dragging) {
-    emits.nodeMouseMove({ event, node, connectedEdges: getConnectedEdges([node], edges) })
+    emit.mouseMove({ event, node, connectedEdges: getConnectedEdges([node], edges) })
   }
 }
 
 const onMouseLeave = (event: MouseEvent) => {
   if (!dragging) {
-    emits.nodeMouseLeave({ event, node, connectedEdges: getConnectedEdges([node], edges) })
+    emit.mouseLeave({ event, node, connectedEdges: getConnectedEdges([node], edges) })
   }
 }
 
 const onContextMenu = (event: MouseEvent) => {
-  emits.nodeContextMenu({
+  emit.contextMenu({
     event,
     node,
     connectedEdges: getConnectedEdges([node], edges),
   })
 }
 
-const onDoubleClick = (event: MouseEvent) =>
-  emits.nodeDoubleClick({ event, node, connectedEdges: getConnectedEdges([node], edges) })
+const onDoubleClick = (event: MouseEvent) => {
+  emit.doubleClick({ event, node, connectedEdges: getConnectedEdges([node], edges) })
+}
 
 const onSelectNode = (event: MouseEvent) => {
   if (selectable && (!selectNodesOnDrag || !draggable)) {
     handleNodeClick(node, multiSelectionActive, addSelectedNodes, removeSelectedElements, setState)
   }
-  emits.nodeClick({ event, node, connectedEdges: getConnectedEdges([node], edges) })
+  emit.click({ event, node, connectedEdges: getConnectedEdges([node], edges) })
 }
 
 const getClass = computed(() => {
@@ -212,6 +214,7 @@ export default {
       :id="node.id"
       :type="node.type"
       :data="node.data"
+      :events="{ ...node.events, ...on }"
       :selected="!!node.selected"
       :connectable="connectable"
       :position="node.position"
diff --git a/packages/vue-flow/src/composables/index.ts b/packages/vue-flow/src/composables/index.ts
index 3748217c..d3626d6c 100644
--- a/packages/vue-flow/src/composables/index.ts
+++ b/packages/vue-flow/src/composables/index.ts
@@ -4,3 +4,4 @@ export { default as useZoomPanHelper } from './useZoomPanHelper'
 export { default as useWindow } from './useWindow'
 export { default as useVueFlow } from './useVueFlow'
 export { default as useDrag } from './useDrag'
+export { default as useNodeHooks } from './useNodeHooks'
diff --git a/packages/vue-flow/src/composables/useNodeHooks.ts b/packages/vue-flow/src/composables/useNodeHooks.ts
new file mode 100644
index 00000000..1303b016
--- /dev/null
+++ b/packages/vue-flow/src/composables/useNodeHooks.ts
@@ -0,0 +1,74 @@
+import useVueFlow from './useVueFlow'
+import type { GraphNode, NodeEventsEmit, NodeEventsOn } from '~/types'
+
+const createNodeHooks = () => ({
+  doubleClick: createEventHook(),
+  click: createEventHook(),
+  mouseEnter: createEventHook(),
+  mouseMove: createEventHook(),
+  mouseLeave: createEventHook(),
+  contextMenu: createEventHook(),
+  dragStart: createEventHook(),
+  drag: createEventHook(),
+  dragStop: createEventHook(),
+})
+
+export default function useNodeEvents(node: GraphNode): { emit: NodeEventsEmit; on: NodeEventsOn } {
+  const { emits } = $(useVueFlow())
+
+  const nodeHooks = createNodeHooks()
+
+  nodeHooks.doubleClick.on((event) => {
+    emits.nodeDoubleClick(event)
+    node.events?.doubleClick?.(event)
+  })
+
+  nodeHooks.click.on((event) => {
+    emits.nodeClick(event)
+    node.events?.click?.(event)
+  })
+
+  nodeHooks.mouseEnter.on((event) => {
+    emits.nodeMouseEnter(event)
+    node.events?.mouseEnter?.(event)
+  })
+
+  nodeHooks.mouseMove.on((event) => {
+    emits.nodeMouseMove(event)
+    node.events?.mouseMove?.(event)
+  })
+
+  nodeHooks.mouseLeave.on((event) => {
+    emits.nodeMouseLeave(event)
+    node.events?.mouseLeave?.(event)
+  })
+
+  nodeHooks.contextMenu.on((event) => {
+    emits.nodeContextMenu(event)
+    node.events?.contextMenu?.(event)
+  })
+
+  nodeHooks.dragStart.on((event) => {
+    emits.nodeDragStart(event)
+    node.events?.dragStart?.(event)
+  })
+
+  nodeHooks.drag.on((event) => {
+    emits.nodeDrag(event)
+    node.events?.drag?.(event)
+  })
+
+  nodeHooks.dragStop.on((event) => {
+    emits.nodeDragStop(event)
+    node.events?.dragStop?.(event)
+  })
+
+  return Object.entries(nodeHooks).reduce(
+    (hooks, [key, value]) => {
+      hooks.emit[key as keyof NodeEventsEmit] = value.trigger
+      hooks.on[key as keyof NodeEventsOn] = value.on
+      return hooks
+    },
+    { emit: {} as NodeEventsEmit, on: {} as NodeEventsOn },
+  )
+}