chore(core,utils): add options obj to updateEdge util

This commit is contained in:
braks
2023-03-02 17:00:19 +01:00
parent 24d4c56343
commit eeed447850
+3 -2
View File
@@ -12,6 +12,7 @@ import {
Rect, Rect,
NodeInternals, NodeInternals,
NodeOrigin, NodeOrigin,
UpdateEdgeOptions,
} from '../types'; } from '../types';
import { errorMessages } from '../contants'; import { errorMessages } from '../contants';
@@ -93,7 +94,7 @@ export const addEdge = (edgeParams: Edge | Connection, edges: Edge[]): Edge[] =>
return edges.concat(edge); return edges.concat(edge);
}; };
export const updateEdge = (oldEdge: Edge, newConnection: Connection, edges: Edge[], shouldReplaceId = true): Edge[] => { export const updateEdge = (oldEdge: Edge, newConnection: Connection, edges: Edge[], options: UpdateEdgeOptions = { shouldReplaceId: true }): Edge[] => {
const { id: oldEdgeId, ...rest } = oldEdge; const { id: oldEdgeId, ...rest } = oldEdge;
if (!newConnection.source || !newConnection.target) { if (!newConnection.source || !newConnection.target) {
@@ -113,7 +114,7 @@ export const updateEdge = (oldEdge: Edge, newConnection: Connection, edges: Edge
// Remove old edge and create the new edge with parameters of old edge. // Remove old edge and create the new edge with parameters of old edge.
const edge = { const edge = {
...rest, ...rest,
id: shouldReplaceId ? getEdgeId(newConnection) : oldEdgeId, id: options.shouldReplaceId ? getEdgeId(newConnection) : oldEdgeId,
source: newConnection.source, source: newConnection.source,
target: newConnection.target, target: newConnection.target,
sourceHandle: newConnection.sourceHandle, sourceHandle: newConnection.sourceHandle,