From 1a5f33ddd459eeeb510f9c6e2fbe6d0c13528be4 Mon Sep 17 00:00:00 2001
From: anovazzi1 <otavio2204@gmail.com>
Date: Wed, 19 Jul 2023 18:14:22 -0300
Subject: [PATCH] chore(codeAreaModal): import cleanEdges function from
 reactflowUtils to improve code readability and maintainability
 feat(codeAreaModal): add typesContext to access reactFlowInstance for
 cleaning edges before processing code

---
 src/frontend/src/modals/codeAreaModal/index.tsx | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/src/frontend/src/modals/codeAreaModal/index.tsx b/src/frontend/src/modals/codeAreaModal/index.tsx
index 4da1e5d24..1548943d2 100644
--- a/src/frontend/src/modals/codeAreaModal/index.tsx
+++ b/src/frontend/src/modals/codeAreaModal/index.tsx
@@ -11,8 +11,10 @@ import { Button } from "../../components/ui/button";
 import { CODE_PROMPT_DIALOG_SUBTITLE } from "../../constants";
 import { alertContext } from "../../contexts/alertContext";
 import { darkContext } from "../../contexts/darkContext";
+import { typesContext } from "../../contexts/typesContext";
 import { postCustomComponent, postValidateCode } from "../../controllers/API";
 import { APIClassType } from "../../types/api";
+import { cleanEdges } from "../../util/reactflowUtils";
 import BaseModal from "../baseModal";
 
 export default function CodeAreaModal({
@@ -32,6 +34,7 @@ export default function CodeAreaModal({
 }) {
   const [code, setCode] = useState(value);
   const { dark } = useContext(darkContext);
+  const { reactFlowInstance } = useContext(typesContext);
   const [height, setHeight] = useState(null);
   const { setErrorData, setSuccessData } = useContext(alertContext);
   const [error, setError] = useState<{
@@ -111,6 +114,13 @@ export default function CodeAreaModal({
   }
 
   function handleClick() {
+    cleanEdges({
+      flow: {
+        nodes: reactFlowInstance.getNodes(),
+        edges: reactFlowInstance.getEdges(),
+      },
+      updateEdge: reactFlowInstance.setEdges,
+    });
     processCode();
   }