fix(tabsContext.tsx): add missing import statement for updateEdgesHandleIds function

fix(tabsContext.tsx): update processFlowEdges function to call updateEdgesHandleIds and update the edges in the flow data
refactor(reactflowUtils.ts): update updateEdgesHandleIds function to return the updated edges instead of using setEdges function
This commit is contained in:
anovazzi1 2023-08-04 14:52:44 -03:00
commit ee196210f2
3 changed files with 6 additions and 5 deletions

View file

@ -27,6 +27,7 @@ import {
import { TabsContextType, TabsState } from "../types/tabs"; import { TabsContextType, TabsState } from "../types/tabs";
import { import {
addVersionToDuplicates, addVersionToDuplicates,
updateEdgesHandleIds,
updateIds, updateIds,
updateTemplate, updateTemplate,
} from "../utils/reactflowUtils"; } from "../utils/reactflowUtils";
@ -149,8 +150,10 @@ export function TabsProvider({ children }: { children: ReactNode }) {
}); });
} }
function processFlowEdges(flow) { function processFlowEdges(flow: FlowType) {
if (!flow.data || !flow.data.edges) return; if (!flow.data || !flow.data.edges) return;
const newEdges = updateEdgesHandleIds(flow.data);
flow.data.edges = newEdges;
flow.data.edges.forEach((edge) => { flow.data.edges.forEach((edge) => {
edge.className = ""; edge.className = "";
edge.style = { stroke: "#555" }; edge.style = { stroke: "#555" };

View file

@ -12,5 +12,4 @@ export type cleanEdgesType = {
export type updateEdgesHandleIdsType = { export type updateEdgesHandleIdsType = {
nodes: NodeType[]; nodes: NodeType[];
edges: Edge[]; edges: Edge[];
setEdges: (edges: Edge[]) => void;
}; };

View file

@ -235,8 +235,7 @@ export function addVersionToDuplicates(flow: FlowType, flows: FlowType[]) {
export function updateEdgesHandleIds({ export function updateEdgesHandleIds({
edges, edges,
nodes, nodes,
setEdges, }: updateEdgesHandleIdsType): Edge[] {
}: updateEdgesHandleIdsType) {
let newEdges = _.cloneDeep(edges); let newEdges = _.cloneDeep(edges);
newEdges.forEach((edge) => { newEdges.forEach((edge) => {
const sourceNodeId = edge.source; const sourceNodeId = edge.source;
@ -268,7 +267,7 @@ export function updateEdgesHandleIds({
edge.sourceHandle = JSON.stringify(newSource); edge.sourceHandle = JSON.stringify(newSource);
edge.targetHandle = JSON.stringify(newTarget); edge.targetHandle = JSON.stringify(newTarget);
}); });
setEdges(newEdges); return newEdges;
} }
export function getConnectedNodes( export function getConnectedNodes(