From dbc3eaa6736955de02579353f651bbd7516ba7a2 Mon Sep 17 00:00:00 2001
From: Lucas Oliveira <lucas.edu.oli@hotmail.com>
Date: Fri, 13 Oct 2023 00:13:59 -0300
Subject: [PATCH] Transformed paste in 3 functions and fixed problem with edges

---
 src/frontend/src/contexts/tabsContext.tsx | 159 +++++++++++++---------
 1 file changed, 92 insertions(+), 67 deletions(-)

diff --git a/src/frontend/src/contexts/tabsContext.tsx b/src/frontend/src/contexts/tabsContext.tsx
index 79b39ad60..4a583c55c 100644
--- a/src/frontend/src/contexts/tabsContext.tsx
+++ b/src/frontend/src/contexts/tabsContext.tsx
@@ -300,7 +300,7 @@ export function TabsProvider({ children }: { children: ReactNode }) {
       let fileData = JSON.parse(text);
       if (fileData.flows) {
         fileData.flows.forEach((flow: FlowType) => {
-          id = addFlow(flow, newProject);
+          id = addFlow(newProject, flow);
         });
       }
       // parse the text into a JSON object
@@ -376,6 +376,7 @@ export function TabsProvider({ children }: { children: ReactNode }) {
     selectionInstance: { nodes: Node[]; edges: Edge[] },
     position: { x: number; y: number; paneX?: number; paneY?: number }
   ) {
+    console.log(selectionInstance);
     let minimumX = Infinity;
     let minimumY = Infinity;
     let idsMap = {};
@@ -395,80 +396,104 @@ export function TabsProvider({ children }: { children: ReactNode }) {
       : reactFlowInstance!.project({ x: position.x, y: position.y });
 
     selectionInstance.nodes.forEach((node: NodeType) => {
-      // Generate a unique node ID
-      let newId = getNodeId(node.data.type);
-      idsMap[node.id] = newId;
-
-      // Create a new node object
-      const newNode: NodeType = {
-        id: newId,
-        type: "genericNode",
-        position: {
-          x: insidePosition.x + node.position!.x - minimumX,
-          y: insidePosition.y + node.position!.y - minimumY,
-        },
-        data: {
-          ..._.cloneDeep(node.data),
-          id: newId,
-        },
-      };
-
-      // Add the new node to the list of nodes in state
-      nodes = nodes
-        .map((node) => ({ ...node, selected: false }))
-        .concat({ ...newNode, selected: false });
+      nodes = updateNodeId(
+        node,
+        idsMap,
+        nodes,
+        insidePosition,
+        minimumX,
+        minimumY
+      );
     });
+
     reactFlowInstance!.setNodes(nodes);
 
     selectionInstance.edges.forEach((edge: Edge) => {
-      let source = idsMap[edge.source];
-      let target = idsMap[edge.target];
-      const sourceHandleObject: sourceHandleType = scapeJSONParse(
-        edge.sourceHandle!
-      );
-      let sourceHandle = scapedJSONStringfy({
-        ...sourceHandleObject,
-        id: source,
-      });
-      sourceHandleObject.id = source;
-      edge.data.sourceHandle = sourceHandleObject;
-      const targetHandleObject: targetHandleType = scapeJSONParse(
-        edge.targetHandle!
-      );
-      let targetHandle = scapedJSONStringfy({
-        ...targetHandleObject,
-        id: target,
-      });
-      targetHandleObject.id = target;
-      edge.data.targetHandle = targetHandleObject;
-      let id =
-        "reactflow__edge-" +
-        source +
-        sourceHandle +
-        "-" +
-        target +
-        targetHandle;
-      edges = addEdge(
-        {
-          source,
-          target,
-          sourceHandle,
-          targetHandle,
-          id,
-          style: { stroke: "#555" },
-          className:
-            targetHandleObject.type === "Text"
-              ? "stroke-gray-800 "
-              : "stroke-gray-900 ",
-          animated: targetHandleObject.type === "Text",
-          selected: false,
-        },
-        edges.map((edge) => ({ ...edge, selected: false }))
-      );
+      edges = updateEdgeId(edge, idsMap, edges);
     });
+
     reactFlowInstance!.setEdges(edges);
   }
 
+  const updateNodeId = (
+    node: NodeType,
+    idsMap,
+    nodes,
+    insidePosition,
+    minimumX,
+    minimumY
+  ) => {
+    // Generate a unique node ID
+    let newId = getNodeId(node.data.type);
+    idsMap[node.id] = newId;
+
+    // Create a new node object
+    const newNode: NodeType = {
+      id: newId,
+      type: "genericNode",
+      position: {
+        x: insidePosition.x + node.position!.x - minimumX,
+        y: insidePosition.y + node.position!.y - minimumY,
+      },
+      data: {
+        ..._.cloneDeep(node.data),
+        id: newId,
+      },
+    };
+
+    // Add the new node to the list of nodes in state
+    nodes = nodes
+      .map((node) => ({ ...node, selected: false }))
+      .concat({ ...newNode, selected: false });
+
+    return nodes;
+  };
+
+  const updateEdgeId = (edge: Edge, idsMap, edges) => {
+    let source = idsMap[edge.source];
+    let target = idsMap[edge.target];
+    const sourceHandleObject: sourceHandleType = scapeJSONParse(
+      edge.sourceHandle!
+    );
+    let sourceHandle = scapedJSONStringfy({
+      ...sourceHandleObject,
+      id: source,
+    });
+    sourceHandleObject.id = source;
+    edge.data.sourceHandle = sourceHandleObject;
+    const targetHandleObject: targetHandleType = scapeJSONParse(
+      edge.targetHandle!
+    );
+    let targetHandle = scapedJSONStringfy({
+      ...targetHandleObject,
+      id: target,
+    });
+    targetHandleObject.id = target;
+    edge.data.targetHandle = targetHandleObject;
+    let id =
+      "reactflow__edge-" + source + sourceHandle + "-" + target + targetHandle;
+    edges = addEdge(
+      {
+        data: edge.data,
+        source,
+        target,
+        sourceHandle,
+        targetHandle,
+        id,
+        style: { stroke: "#555" },
+        className:
+          targetHandleObject.type === "Text"
+            ? "stroke-gray-800 "
+            : "stroke-gray-900 ",
+        animated: targetHandleObject.type === "Text",
+        selected: false,
+      },
+      edges.map((edge) => ({ ...edge, selected: false }))
+    );
+
+    return edges;
+  };
+
   const addFlow = async (
     newProject: Boolean,
     flow?: FlowType