feat: enhance EditButton functionality to toggle editing state

valory-xyz · Dec 24, 2024 · 9565dcc · 9565dcc
1 parent 6527ee0
commit 9565dcc
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/frontend/components/UpdateAgentPage/index.tsx b/frontend/components/UpdateAgentPage/index.tsx
@@ -97,13 +97,18 @@ const UpdateAgentProvider = ({ children }: PropsWithChildren) => {
 };
 
 const EditButton = () => {
-  const { setIsEditing } = useContext(UpdateAgentContext);
+  const { setIsEditing, isEditing } = useContext(UpdateAgentContext);
+
+  const handleEdit = () => {
+    setIsEditing?.((prev) => !prev);
+  };
+
+  if (isEditing) {
+    return null;
+  }
 
   return (
-    <Button
-      icon={<EditFilled />}
-      onClick={() => setIsEditing?.((prev) => !prev)}
-    >
+    <Button icon={<EditFilled />} onClick={handleEdit}>
       Edit
     </Button>
   );