Handle situations where the dropped unit is not available.

GitOrigin-RevId: db782946b9fbe98f1049af2c22c457eee797eb58
nkzw-tech · May 25, 2024 · 3ddb233 · 3ddb233
1 parent 98b195d
commit 3ddb233
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/hera/behavior/drop/dropUnitAction.tsx b/hera/behavior/drop/dropUnitAction.tsx
@@ -15,7 +15,15 @@ export default function dropUnitAction(
   const { from, to } = actionResponse;
   const { vision } = state;
   const originalUnitA = state.map.units.get(from)!;
+  const newUnitB = map.units.get(to);
   const unitB = originalUnitA.getTransportedUnit(actionResponse.index)!;
+  if (!newUnitB) {
+    return {
+      map,
+      ...resetBehavior(),
+    };
+  }
+
   return {
     animations: addMoveAnimation(state.animations, {
       from,
@@ -47,7 +55,7 @@ export default function dropUnitAction(
     map: map.copy({
       units: map.units
         .set(from, originalUnitA.drop(unitB).recover())
-        .set(to, map.units.get(to)!.recover()),
+        .set(to, newUnitB.recover()),
     }),
     ...resetBehavior(),
     behavior: new NullBehavior(),