feat(nodes): add useNode composable

This commit is contained in:
braks
2022-10-06 18:22:22 +02:00
committed by Braks
parent a29d7c86f0
commit 70871df469

View File

@@ -0,0 +1,32 @@
import useVueFlow from './useVueFlow'
import { NodeId, NodeRef } from '~/context'
import type { CustomEvent, ElementData } from '~/types'
import { getConnectedEdges } from '~/utils'
/**
* Access a node, it's parent (if one exists) and connected edges
*
* If no node id is provided, the node id is injected from context
*
* Meaning if you do not provide an id, this composable has to be called in a child of your custom node component, or it will throw
*/
export function useNode<Data = ElementData, CustomEvents extends Record<string, CustomEvent> = any>(id?: string) {
const nodeId = id ?? inject(NodeId, '')
const nodeEl = inject(NodeRef, null)
const { findNode, getEdges } = useVueFlow()
const node = findNode<Data, CustomEvents>(nodeId)
if (!node) {
throw new Error(`[vue-flow]: useNode - Node with id ${nodeId} not found!`)
}
return {
id: nodeId,
node,
nodeEl,
parentNode: node.parentNode ? findNode(node.parentNode) : undefined,
connectedEdges: computed(() => getConnectedEdges([node], getEdges.value)),
}
}