feat(rg): removeElements helper

2019-07-25 18:27:41 +02:00
parent d587b1340d
commit bb17c2fcb6
4 changed files with 36 additions and 17 deletions
@@ -1,6 +1,19 @@
 export const isEdge = element => element.source && element.target;
+
 export const isNode = element => !element.source && !element.target;

+export const removeElements = (elements, elementsToRemove) => {
+  const nodeIdsToRemove = elementsToRemove.filter(isNode).map(n => n.id);
+
+  return elements.filter(e => {
+    return (
+      !nodeIdsToRemove.includes(e.id) &&
+      !nodeIdsToRemove.includes(e.target) &&
+      !nodeIdsToRemove.includes(e.source)
+    );
+  });
+}
+
 export const parseElements = e => {
  e.type = e.type || 'default';