alighasami/vue-flow
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(core): allow passing node or id to getoutgoers/getincomers

Browse Source
This commit is contained in:
braks
2023-07-24 22:11:05 +02:00
committed by Braks
parent c4acb0d0f6
commit 2a597f8605
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
packages/core/src/utils/graph.ts
View File

@@ -152,23 +152,23 @@ export function parseEdge(edge: Edge, defaults: Partial<GraphEdge> = {}): GraphE
}
function getConnectedElements<T extends Elements = FlowElements>(
node: Node,
nodeOrId: Node | { id: string } | string,
elements: T,
dir: 'source' | 'target',
): T extends FlowElements ? GraphNode[] : Node[] {
if (!isNode(node)) {
return []
}
const id = isString(nodeOrId) ? nodeOrId : nodeOrId.id
const origin = dir === 'source' ? 'target' : 'source'
const ids = elements.filter((e) => isEdge(e) && e[origin] === node.id).map((e) => isEdge(e) && e[dir])
const ids = elements.filter((e) => isEdge(e) && e[origin] === id).map((e) => isEdge(e) && e[dir])
return elements.filter((e) => ids.includes(e.id)) as T extends FlowElements ? GraphNode[] : Node[]
}
export function getOutgoers<T extends Elements = FlowElements>(node: Node, elements: T) {
return getConnectedElements(node, elements, 'target')
export function getOutgoers<T extends Elements = FlowElements>(nodeOrId: Node | { id: string } | string, elements: T) {
return getConnectedElements(nodeOrId, elements, 'target')
}
export function getIncomers<T extends Elements = FlowElements>(node: Node, elements: T) {
return getConnectedElements(node, elements, 'source')
export function getIncomers<T extends Elements = FlowElements>(nodeOrId: Node | { id: string } | string, elements: T) {
return getConnectedElements(nodeOrId, elements, 'source')
}
export function getEdgeId({ source, sourceHandle, target, targetHandle }: Connection) {