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

chore(core): cleanup types

Browse Source 
Signed-off-by: braks <78412429+bcakmakoglu@users.noreply.github.com>
This commit is contained in:
braks
2023-03-24 14:35:46 +01:00
committed by Braks
parent 54ea8a0d0c
commit 97914486db
1 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/core/src/utils/graph.ts
+11 -4
View File
@@ -10,6 +10,7 @@ import type {
EdgeMarkerType, EdgeMarkerType,
ElementData, ElementData,
Elements, Elements,
FlowElements,
GraphEdge, GraphEdge,
GraphNode, GraphNode,
MaybeElement, MaybeElement,
@@ -127,15 +128,21 @@ export function parseEdge(edge: Edge, defaults: Partial<GraphEdge> = {}): GraphE
return Object.assign({ id: edge.id.toString() }, edge, defaults) as GraphEdge return Object.assign({ id: edge.id.toString() }, edge, defaults) as GraphEdge
} }
const getConnectedElements = (node: GraphNode, elements: Elements, dir: 'source' | 'target') => { const getConnectedElements = <T extends Elements = FlowElements>(
node: Node,
elements: T,
dir: 'source' | 'target',
): T extends FlowElements ? GraphNode[] : Node[] => {
if (!isNode(node)) return [] if (!isNode(node)) return []
const origin = dir === 'source' ? 'target' : 'source' 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] === node.id).map((e) => isEdge(e) && e[dir])
return elements.filter((e) => ids.includes(e.id)) as GraphEdge[] return elements.filter((e) => ids.includes(e.id)) as T extends FlowElements ? GraphNode[] : Node[]
} }
export const getOutgoers = (node: GraphNode, elements: Elements) => getConnectedElements(node, elements, 'target') export const getOutgoers = <T extends Elements = FlowElements>(node: Node, elements: T) =>
getConnectedElements(node, elements, 'target')
export const getIncomers = (node: GraphNode, elements: Elements) => getConnectedElements(node, elements, 'source') export const getIncomers = <T extends Elements = FlowElements>(node: Node, elements: T) =>
getConnectedElements(node, elements, 'source')
export const getEdgeId = ({ source, sourceHandle, target, targetHandle }: Connection) => export const getEdgeId = ({ source, sourceHandle, target, targetHandle }: Connection) =>
`vueflow__edge-${source}${sourceHandle ?? ''}-${target}${targetHandle ?? ''}` `vueflow__edge-${source}${sourceHandle ?? ''}-${target}${targetHandle ?? ''}`