From 533e37e624596153637ac4b9d0eecaef758c80e3 Mon Sep 17 00:00:00 2001
From: braks <78412429+bcakmakoglu@users.noreply.github.com>
Date: Sat, 3 Feb 2024 18:05:42 +0100
Subject: [PATCH] chore(core,composables): cleanup `useNodesData` types

---
 packages/core/src/composables/useNodesData.ts | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/packages/core/src/composables/useNodesData.ts b/packages/core/src/composables/useNodesData.ts
index 0df67dea..c9eeadac 100644
--- a/packages/core/src/composables/useNodesData.ts
+++ b/packages/core/src/composables/useNodesData.ts
@@ -1,7 +1,10 @@
+import type { ComputedRef } from 'vue'
 import { computed } from 'vue'
 import type { GraphNode, Node } from '../types'
 import { useVueFlow } from './useVueFlow'
 
+type NodeData<NodeType extends Node = GraphNode> = NonNullable<NodeType['data']>
+
 /**
  * Composable for receiving data of one or multiple nodes
  *
@@ -10,12 +13,12 @@ import { useVueFlow } from './useVueFlow'
  * @param guard - Optional guard function to narrow down the node type
  * @returns An array of data objects
  */
-export function useNodesData<NodeType extends Node = GraphNode>(nodeId: string): NodeType['data'] | null
-export function useNodesData<NodeType extends Node = GraphNode>(nodeIds: string[]): NodeType['data'][]
+export function useNodesData<NodeType extends Node = GraphNode>(nodeId: string): ComputedRef<NodeData<NodeType> | null>
+export function useNodesData<NodeType extends Node = GraphNode>(nodeIds: string[]): ComputedRef<NodeData<NodeType>[]>
 export function useNodesData<NodeType extends Node = GraphNode>(
   nodeIds: string[],
   guard: (node: Node) => node is NodeType,
-): NodeType['data'][]
+): ComputedRef<NodeData<NodeType>[]>
 export function useNodesData(nodeIds: any): any {
   const { findNode } = useVueFlow()