clean up code

districtr · Nov 9, 2024 · fcfc893 · fcfc893
1 parent 9457b94
commit fcfc893
Show file tree

Hide file tree

Showing 2 changed files with 260 additions and 254 deletions.
diff --git a/app/src/app/components/sidebar/UndoRedoButton.tsx b/app/src/app/components/sidebar/UndoRedoButton.tsx
@@ -3,14 +3,16 @@ import { Button } from "@radix-ui/themes";
 import { ResetIcon } from "@radix-ui/react-icons";
 import type { TemporalState } from "zundo";
 
+import type { MapStore } from "../../store/mapStore";
+
 export function UndoRedoButton({ isRedo = false }) {
-  const mapStore = useMapStore.getState() as TemporalState;
+  const mapStore = useMapStore.temporal.getState() as TemporalState<MapStore>;
 
   const handleClickUndoRedo = () => {
     if (isRedo) {
-      mapStore.redo(1);
+      mapStore.redo();
     } else {
-      mapStore.undo(1);
+      mapStore.undo();
     }
   };