feat(reactflow): add isValidConnection prop closes #2876

2023-03-02 17:42:07 +01:00
parent afa38727fa
commit 99ec9f0f03
9 changed files with 45 additions and 13 deletions
@@ -33,8 +33,9 @@ const createRFStore = () =>
      return Array.from(get().nodeInternals.values());
    },
    setEdges: (edges: Edge[]) => {
-      const { defaultEdgeOptions = {} } = get();
-      set({ edges: edges.map((e) => ({ ...defaultEdgeOptions, ...e })) });
+      const { defaultEdgeOptions = null, isValidConnection } = get();
+      const validEdges = isValidConnection ? edges.filter(isValidConnection) : edges;
+      set({ edges: defaultEdgeOptions ? validEdges.map((e) => ({ ...defaultEdgeOptions, ...e })) : edges });
    },
    setDefaultNodesAndEdges: (nodes?: Node[], edges?: Edge[]) => {
      const hasDefaultNodes = typeof nodes !== 'undefined';