Reset Map behavior when canceling undo.

GitOrigin-RevId: 18415676379e4ec33f10fc17e7fb9213d4411c1a
nkzw-tech · Nov 29, 2024 · 95979df · 95979df
1 parent 84f1fab
commit 95979df
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/hera/ui/GameActions.tsx b/hera/ui/GameActions.tsx
@@ -478,9 +478,10 @@ export default function GameActions({
   inset?: number;
   setZoom?: SetZoomFn;
   subscribe?: (map: MapData) => Promise<void>;
-  undoTurn?: () => void | Promise<void>;
+  undoTurn?: (onCancel: () => void) => void | Promise<void>;
   zoom: number;
 }) {
+  const { update } = actions;
   const {
     currentViewer,
     inlineUI,
@@ -499,6 +500,7 @@ export default function GameActions({
     currentViewer != null ? map.maybeGetPlayer(currentViewer) : null;
   const canUndo =
     undoTurn &&
+    !hasEnded &&
     viewerPlayer?.isHumanPlayer() &&
     viewerPlayer.crystal == null &&
     playerCanEndTurn &&
@@ -508,10 +510,10 @@ export default function GameActions({
   const undo = useCallback(() => {
     if (canUndo && undoTurn) {
       AudioPlayer.playSound('UI/Accept');
-      actions.update(resetBehavior(NullBehavior));
-      undoTurn();
+      update(resetBehavior(NullBehavior));
+      undoTurn(() => update(resetBehavior()));
     }
-  }, [actions, canUndo, undoTurn]);
+  }, [canUndo, undoTurn, update]);
   useInput('undo', undo);
 
   const replayBar = !hasEnded && (