fix: Undo/redo fixes.

zmkfirmware · Jul 12, 2024 · 8b25218 · 8b25218
1 parent 28f2625
commit 8b25218
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/undoRedo.ts b/src/undoRedo.ts
@@ -27,11 +27,14 @@ export function useUndoRedo(): [
     [locked, redoStack]
   );
 
-  const doIt = async (doCb: DoCallback) => {
+  const doIt = async (doCb: DoCallback, preserveRedo?: boolean) => {
     setLocked(true);
     let undo = await doCb();
 
     setUndoStack([[doCb, undo], ...undoStack]);
+    if (!preserveRedo) {
+      setRedoStack([]);
+    }
     setLocked(false);
   };
 
@@ -67,7 +70,7 @@ export function useUndoRedo(): [
 
     setRedoStack(redoStack.slice(1));
 
-    return await doIt(doCb);
+    return await doIt(doCb, true);
   };
 
   const reset = () => {