fix(nested-nodes): fix getSourceTargetNode for childNodes

This commit is contained in:
Christopher Möller
2021-10-21 14:08:39 +02:00
parent b008c7b548
commit b15d3012e0
2 changed files with 58 additions and 18 deletions
+22 -14
View File
@@ -172,18 +172,26 @@ type SourceTargetNode = {
};
export const getSourceTargetNodes = (edge: Edge, nodes: Node[]): SourceTargetNode => {
return nodeHelper(nodes)
.flatten()
.reduce(
(res, node) => {
if (node.id === edge.source) {
res.sourceNode = node;
}
if (node.id === edge.target) {
res.targetNode = node;
}
return res;
},
{ sourceNode: null, targetNode: null } as SourceTargetNode
);
const sourceNode = nodeHelper(nodes).find((n) => n.id === edge.source) || null;
const targetNode = nodeHelper(nodes).find((n) => n.id === edge.target) || null;
return {
sourceNode,
targetNode,
};
// return nodeHelper(nodes)
// .flatten()
// .reduce(
// (res, node) => {
// if (node.id === edge.source) {
// res.sourceNode = node;
// }
// if (node.id === edge.target) {
// res.targetNode = node;
// }
// return res;
// },
// { sourceNode: null, targetNode: null } as SourceTargetNode
// );
};