feat(core): add useEdgesData composable (#1462)

* feat(core): add `useEdgesData` composable

* chore(changeset): add

* chore(core): add export

packages/core/src/composables/useEdgesData.ts

@@ -0,0 +1,73 @@
+import type { ComputedRef, MaybeRefOrGetter } from 'vue'
+import { computed, toValue } from 'vue'
+import type { Edge, GraphEdge } from '../types'
+import { warn } from '../utils'
+import { useVueFlow } from './useVueFlow'
+
+interface EdgeData<EdgeType extends Edge = GraphEdge> {
+  id: string
+  type: EdgeType['type']
+  data: NonNullable<EdgeType['data']> | null
+}
+
+/**
+ * Composable for receiving data of one or multiple nodes
+ *
+ * @public
+ * @param edgeId - The id (or ids) of the node to get the data from
+ * @param guard - Optional guard function to narrow down the node type
+ * @returns An array of data objects
+ */
+export function useEdgesData<EdgeType extends Edge = GraphEdge>(
+  edgeId: MaybeRefOrGetter<string>,
+): ComputedRef<EdgeData<EdgeType> | null>
+export function useEdgesData<EdgeType extends Edge = GraphEdge>(
+  edgeIds: MaybeRefOrGetter<string[]>,
+): ComputedRef<EdgeData<EdgeType>[]>
+export function useEdgesData<EdgeType extends Edge = GraphEdge>(
+  edgeIds: MaybeRefOrGetter<string[]>,
+  guard: (node: Edge) => node is EdgeType,
+): ComputedRef<EdgeData<EdgeType>[]>
+export function useEdgesData(_edgeIds: any): any {
+  const { findEdge } = useVueFlow()
+
+  return computed({
+    get() {
+      const edgeIds = toValue(_edgeIds)
+
+      if (!Array.isArray(edgeIds)) {
+        const edge = findEdge(edgeIds)
+
+        if (edge) {
+          return {
+            id: edge.id,
+            type: edge.type,
+            data: edge.data ?? null,
+          }
+        }
+
+        return null
+      }
+
+      const data: EdgeData<Edge>[] = []
+
+      for (const edgeId of edgeIds) {
+        const edge = findEdge(edgeId)
+
+        if (edge) {
+          data.push({
+            id: edge.id,
+            type: edge.type,
+            data: edge.data ?? null,
+          })
+        }
+      }
+
+      return data
+    },
+    set() {
+      // noop
+      warn('You are trying to set edge data via useEdgesData. This is not supported.')
+    },
+  })
+}

packages/core/src/index.ts

@@ -71,6 +71,7 @@ export { useNodeId } from './composables/useNodeId'
 export { useConnection } from './composables/useConnection'
 export { useHandleConnections } from './composables/useHandleConnections'
 export { useNodesData } from './composables/useNodesData'
+export { useEdgesData } from './composables/useEdgesData'
 export { useNodesInitialized } from './composables/useNodesInitialized'
 
 export { VueFlowError, ErrorCode, isErrorOfType } from './utils/errors'