mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
from uuid import UUID
|
|
|
|
from fastapi import APIRouter, Depends
|
|
from shapely.geometry import LineString, Point as ShapelyPoint
|
|
|
|
from game import Game
|
|
from game.ato.flightplan import CasFlightPlan, PatrollingFlightPlan
|
|
from game.server import GameContext
|
|
from game.server.flights.models import FlightJs
|
|
from game.server.leaflet import LeafletPoly, ShapelyUtil
|
|
|
|
router: APIRouter = APIRouter(prefix="/flights")
|
|
|
|
|
|
@router.get("/")
|
|
def list_flights(
|
|
with_waypoints: bool = False, game: Game = Depends(GameContext.get)
|
|
) -> list[FlightJs]:
|
|
flights = []
|
|
for coalition in game.coalitions:
|
|
for package in coalition.ato.packages:
|
|
for flight in package.flights:
|
|
flights.append(FlightJs.for_flight(flight, with_waypoints))
|
|
return flights
|
|
|
|
|
|
@router.get("/{flight_id}")
|
|
def get_flight(
|
|
flight_id: UUID, with_waypoints: bool = False, game: Game = Depends(GameContext.get)
|
|
) -> FlightJs:
|
|
flight = game.db.flights.get(flight_id)
|
|
return FlightJs.for_flight(flight, with_waypoints)
|
|
|
|
|
|
@router.get("/{flight_id}/commit-boundary")
|
|
def commit_boundary(
|
|
flight_id: UUID, game: Game = Depends(GameContext.get)
|
|
) -> LeafletPoly:
|
|
flight = game.db.flights.get(flight_id)
|
|
if not isinstance(flight.flight_plan, PatrollingFlightPlan):
|
|
return []
|
|
start = flight.flight_plan.patrol_start
|
|
end = flight.flight_plan.patrol_end
|
|
if isinstance(flight.flight_plan, CasFlightPlan):
|
|
center = flight.flight_plan.target.position
|
|
commit_center = ShapelyPoint(center.x, center.y)
|
|
else:
|
|
commit_center = LineString(
|
|
[
|
|
ShapelyPoint(start.x, start.y),
|
|
ShapelyPoint(end.x, end.y),
|
|
]
|
|
)
|
|
bubble = commit_center.buffer(flight.flight_plan.engagement_distance.meters)
|
|
return ShapelyUtil.poly_to_leaflet(bubble, game.theater)
|