from __future__ import annotations

import json
import argparse
from pathlib import Path
from typing import List, Tuple, Union, Dict

from dcs.terrain import Caucasus, PersianGulf, Syria, Nevada, Normandy, TheChannel
from dcs import Mission

Terrain = Union[Caucasus, PersianGulf, Syria, Nevada, Normandy, TheChannel]

SAVE_PATH = Path("resources/frontlines")


def validate_miz(file_path: Path) -> bool:
    return bool(file_path.suffix == ".miz" and file_path.exists())


def validate_airports(airports: Tuple[int], terrain: Terrain):
    for airport in airports:
        if terrain.airport_by_id(airport) is None:
            print(f"Cannot load airport for invalid id {airport}")


def load_files(files) -> List[Mission]:
    missions = []
    for file in files:
        if validate_miz(file):
            mission = Mission()
            mission.load_file(file)
            missions.append(mission)
        else:
            print(f"Error: {file} doesn't look like a valid mission file.")
    return missions


def create_frontline_dict(mission: Mission) -> Dict[str, Dict]:
    frontline_dict = {}
    for group in mission.country("USA").vehicle_group:
        groupname = str(group.name).replace(group.name.id, "").replace(":", "")
        control_points = groupname.split("|")
        frontline_dict[groupname] = {
            "points": [(i.position.x, i.position.y) for i in group.points],
            "start_cp": int(control_points[0]),
        }
    return frontline_dict


def process_missions(missions: List[Mission]) -> None:
    for mission in missions:
        frontline_dict = create_frontline_dict(mission)
        write_json(frontline_dict, mission.terrain.name.lower())


def write_json(frontline_dict: Dict[str, Dict], terrain_name: str) -> None:
    with open(SAVE_PATH.joinpath(terrain_name + ".json"), "w") as file:
        json.dump(frontline_dict, file)


if __name__ == "__main__":
    parser = argparse.ArgumentParser(
        description="Process a miz file to create json descriptions of multi-segment frontlines"
    )
    parser.add_argument(
        "files",
        metavar="N",
        type=Path,
        nargs="+",
        help="A list of space separated .miz files to extract frontlines from",
    )

    args = parser.parse_args()
    missions = load_files(args.files)
    process_missions(missions)
    # frontline_dict = create_frontline_dict(missions[0])

    print("Done")