from __future__ import annotations

from typing import Union

from dcs import Point
from dcs.mapping import LatLng
from pydantic import BaseModel
from shapely.geometry import LineString, MultiLineString, MultiPolygon, Polygon

from game.theater import ConflictTheater


class LeafletPoint(BaseModel):
    lat: float
    lng: float

    class Config:
        orm_mode = True

        title = "LatLng"


LeafletPoly = list[LeafletPoint]


class ShapelyUtil:
    @staticmethod
    def latlng_to_leaflet(latlng: LatLng) -> LeafletPoint:
        return LeafletPoint(lat=latlng.lat, lng=latlng.lng)

    @classmethod
    def poly_to_leaflet(cls, poly: Polygon, theater: ConflictTheater) -> LeafletPoly:
        if poly.is_empty:
            return []
        return [
            cls.latlng_to_leaflet(Point(x, y, theater.terrain).latlng())
            for x, y in poly.exterior.coords
        ]

    @classmethod
    def polys_to_leaflet(
        cls, poly: Union[Polygon, MultiPolygon], theater: ConflictTheater
    ) -> list[LeafletPoly]:
        if isinstance(poly, MultiPolygon):
            polys = poly.geoms
        else:
            polys = [poly]
        return [cls.poly_to_leaflet(poly, theater) for poly in polys]

    @staticmethod
    def line_to_leaflet(line: LineString, theater: ConflictTheater) -> list[LatLng]:
        return [Point(x, y, theater.terrain).latlng() for x, y in line.coords]

    @classmethod
    def lines_to_leaflet(
        cls, line_string: MultiLineString | LineString, theater: ConflictTheater
    ) -> list[list[LatLng]]:
        if isinstance(line_string, MultiLineString):
            lines = line_string.geoms
        else:
            lines = [line_string]
        return [cls.line_to_leaflet(line, theater) for line in lines]