From 290fc96d02a1119dea2cf247a8a1c6e516b20c13 Mon Sep 17 00:00:00 2001
From: Lucas Oliveira <lucas.edu.oli@hotmail.com>
Date: Sat, 6 Jan 2024 11:57:44 -0300
Subject: [PATCH] Fixed ctrl+z on intComponent

---
 .../src/components/intComponent/index.tsx     | 21 ++-----------------
 src/frontend/src/utils/reactflowUtils.ts      | 13 ++++++++++++
 2 files changed, 15 insertions(+), 19 deletions(-)

diff --git a/src/frontend/src/components/intComponent/index.tsx b/src/frontend/src/components/intComponent/index.tsx
index fd6d91ffd..d9d3504ae 100644
--- a/src/frontend/src/components/intComponent/index.tsx
+++ b/src/frontend/src/components/intComponent/index.tsx
@@ -1,6 +1,6 @@
 import { useEffect } from "react";
 import { IntComponentType } from "../../types/components";
-import { handleKeyDown } from "../../utils/reactflowUtils";
+import { handleKeyDown, handleOnlyIntegerInput } from "../../utils/reactflowUtils";
 import { Input } from "../ui/input";
 
 export default function IntComponent({
@@ -24,24 +24,7 @@ export default function IntComponent({
       <Input
         id={id}
         onKeyDown={(event) => {
-          if (
-            event.key !== "Backspace" &&
-            event.key !== "Enter" &&
-            event.key !== "Delete" &&
-            event.key !== "ArrowLeft" &&
-            event.key !== "ArrowRight" &&
-            event.key !== "Control" &&
-            event.key !== "Meta" &&
-            event.key !== "Shift" &&
-            event.key !== "c" &&
-            event.key !== "v" &&
-            event.key !== "a" &&
-            event.key !== "ArrowUp" &&
-            event.key !== "ArrowDown" &&
-            !/^[-]?\d*$/.test(event.key)
-          ) {
-            event.preventDefault();
-          }
+          handleOnlyIntegerInput(event);
           handleKeyDown(event, value, "0");
         }}
         type="number"
diff --git a/src/frontend/src/utils/reactflowUtils.ts b/src/frontend/src/utils/reactflowUtils.ts
index e4b6b2c85..e382267e9 100644
--- a/src/frontend/src/utils/reactflowUtils.ts
+++ b/src/frontend/src/utils/reactflowUtils.ts
@@ -415,6 +415,19 @@ export function handleKeyDown(
   }
 }
 
+export function handleOnlyIntegerInput(event: React.KeyboardEvent<HTMLInputElement>) {
+  if (
+    event.key === "." ||
+    event.key === "-" ||
+    event.key === "," ||
+    event.key === "e" ||
+    event.key === "E" ||
+    event.key === "+"
+  ) {
+    event.preventDefault();
+  }
+}
+
 export function getConnectedNodes(
   edge: Edge,
   nodes: Array<NodeType>