From 62f78e0795e805c7acca8424da966e23c9eb8de5 Mon Sep 17 00:00:00 2001
From: Lucas Oliveira <lucas.edu.oli@hotmail.com>
Date: Fri, 7 Jun 2024 13:58:03 -0300
Subject: [PATCH] Made component not be updatable if it's been edited before

---
 src/frontend/src/CustomNodes/hooks/use-check-code-validity.tsx | 3 ++-
 src/frontend/src/CustomNodes/hooks/use-update-node-code.tsx    | 1 +
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/frontend/src/CustomNodes/hooks/use-check-code-validity.tsx b/src/frontend/src/CustomNodes/hooks/use-check-code-validity.tsx
index 0b83429f8..4cbfbd8df 100644
--- a/src/frontend/src/CustomNodes/hooks/use-check-code-validity.tsx
+++ b/src/frontend/src/CustomNodes/hooks/use-check-code-validity.tsx
@@ -26,7 +26,8 @@ const useCheckCodeValidity = (
     const componentsToIgnore = ["Custom Component", "Prompt"];
     if (
       currentCode !== thisNodesCode &&
-      !componentsToIgnore.includes(data.node!.display_name)
+      !componentsToIgnore.includes(data.node!.display_name) &&
+      !(data.node?.edited ?? false)
     ) {
       setIsOutdated(true);
     } else {
diff --git a/src/frontend/src/CustomNodes/hooks/use-update-node-code.tsx b/src/frontend/src/CustomNodes/hooks/use-update-node-code.tsx
index 28220d141..f1593597f 100644
--- a/src/frontend/src/CustomNodes/hooks/use-update-node-code.tsx
+++ b/src/frontend/src/CustomNodes/hooks/use-update-node-code.tsx
@@ -19,6 +19,7 @@ const useUpdateNodeCode = (
           node: newNodeClass,
           description: newNodeClass.description ?? dataNode.description,
           display_name: newNodeClass.display_name ?? dataNode.display_name,
+          edited: false,
         };
 
         newNode.data.node.template[name].value = code;