From 352c2ddc56378ff4dddc8eddda05675da95d6f69 Mon Sep 17 00:00:00 2001
From: zhexu14 <64713351+zhexu14@users.noreply.github.com>
Date: Wed, 3 May 2023 23:48:16 +1000
Subject: [PATCH] make BAI plannable against vehicles only

---
 game/theater/theatergroundobject.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/game/theater/theatergroundobject.py b/game/theater/theatergroundobject.py
index 8d1dc2b7..e302c94b 100644
--- a/game/theater/theatergroundobject.py
+++ b/game/theater/theatergroundobject.py
@@ -164,7 +164,6 @@ class TheaterGroundObject(MissionTarget, SidcDescribable, ABC):
         else:
             yield from [
                 FlightType.STRIKE,
-                FlightType.BAI,
                 FlightType.REFUELING,
             ]
         yield from super().mission_types(for_player)
@@ -584,6 +583,13 @@ class VehicleGroupGroundObject(TheaterGroundObject):
     def should_head_to_conflict(self) -> bool:
         return True
 
+    def mission_types(self, for_player: bool) -> Iterator[FlightType]:
+        from game.ato import FlightType
+
+        if not self.is_friendly(for_player):
+            yield FlightType.BAI
+        yield from super().mission_types(for_player)
+
 
 class EwrGroundObject(IadsGroundObject):
     def __init__(