refactor: add idx property to handle types in GenericNode and reactflowUtils

This commit is contained in:
anovazzi1 2024-05-28 17:57:52 -03:00
commit 45011e8fda
3 changed files with 6 additions and 0 deletions

View file

@ -589,6 +589,7 @@ export default function GenericNode({
baseClasses: data.node!.base_classes,
id: data.id,
dataType: data.type,
idx: 0,
}}
type={data.node?.base_classes.join("|")}
left={false}
@ -848,6 +849,7 @@ export default function GenericNode({
baseClasses: [output.selected ?? output.types[0]],
id: data.id,
dataType: data.type,
idx: idx,
}}
type={output.types.join("|")}
left={false}

View file

@ -58,6 +58,7 @@ export type sourceHandleType = {
dataType: string;
id: string;
baseClasses: string[];
idx: number;
};
//left side
export type targetHandleType = {

View file

@ -79,6 +79,8 @@ export function cleanEdges(nodes: Node[], edges: Edge[]) {
id: sourceNode.data.id,
baseClasses: sourceNode.data.node!.base_classes,
dataType: sourceNode.data.type,
idx:
sourceNode.data.node!.outputs[scapeJSONParse(sourceHandle).idx] ?? 0,
};
if (scapedJSONStringfy(id) !== sourceHandle) {
newEdges = newEdges.filter((e) => e.id !== edge.id);
@ -397,6 +399,7 @@ export function updateEdgesHandleIds({
id: sourceNode.data.id,
baseClasses: sourceNode.data.node!.base_classes,
dataType: sourceNode.data.type,
idx: 0,
};
}
edge.sourceHandle = scapedJSONStringfy(newSource!);