From 75fa7c147bd8efbef82711d8fadd74ed8a211c30 Mon Sep 17 00:00:00 2001
From: igorrCarvalho <igorsilvabhz6@gmail.com>
Date: Mon, 10 Jun 2024 00:29:48 -0300
Subject: [PATCH] Fix: state not reseting properly

---
 src/frontend/src/CustomNodes/GenericNode/index.tsx         | 6 ++++--
 .../src/pages/SettingsPage/pages/ShortcutsPage/index.tsx   | 7 +++++++
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/frontend/src/CustomNodes/GenericNode/index.tsx b/src/frontend/src/CustomNodes/GenericNode/index.tsx
index 431327d8e..dea103b37 100644
--- a/src/frontend/src/CustomNodes/GenericNode/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/index.tsx
@@ -34,10 +34,9 @@ import useUpdateValidationStatus from "../hooks/use-update-validation-status";
 import useValidationStatusString from "../hooks/use-validation-status-string";
 import getFieldTitle from "../utils/get-field-title";
 import sortFields from "../utils/sort-fields";
-import isWrappedWithClass from "../../pages/FlowPage/components/PageComponent/utils/is-wrapped-with-class";
 import ParameterComponent from "./components/parameterComponent";
 import { postCustomComponent } from "../../controllers/API";
-import { cloneDeep } from "lodash";
+import { useShortcutsStore } from "../../stores/shortcuts";
 
 export default function GenericNode({
   data,
@@ -234,6 +233,8 @@ export default function GenericNode({
     }
   };
 
+  const shortcuts = useShortcutsStore((state) => state.shortcuts);
+
   const memoizedNodeToolbarComponent = useMemo(() => {
     return (
       <NodeToolbar>
@@ -271,6 +272,7 @@ export default function GenericNode({
     updateNodeCode,
     isOutdated,
     selected,
+    shortcuts,
     //    openWDoubleCLick,
     //    setOpenWDoubleCLick,
   ]);
diff --git a/src/frontend/src/pages/SettingsPage/pages/ShortcutsPage/index.tsx b/src/frontend/src/pages/SettingsPage/pages/ShortcutsPage/index.tsx
index 83d51ee53..9a6b4ea8f 100644
--- a/src/frontend/src/pages/SettingsPage/pages/ShortcutsPage/index.tsx
+++ b/src/frontend/src/pages/SettingsPage/pages/ShortcutsPage/index.tsx
@@ -40,9 +40,16 @@ export default function ShortcutsPage() {
 
   const combinationToEdit = shortcuts.filter((s) => s.name === selectedRows[0]);
   const [open, setOpen] = useState(false);
+  const updateUniqueShortcut = useShortcutsStore(
+    (state) => state.updateUniqueShortcut,
+  );
 
   function handleRestore() {
     setShortcuts(defaultShortcuts);
+    defaultShortcuts.forEach(({ name, shortcut }) => {
+      const fixedName = name.split(" ")[0].toLowerCase();
+      updateUniqueShortcut(fixedName, shortcut);
+    });
     localStorage.removeItem("langflow-shortcuts");
   }