from __future__ import annotations

from pathlib import Path

from PySide2.QtCore import Qt
from PySide2.QtGui import QIcon, QPixmap
from PySide2.QtWidgets import (
    QDialog,
    QGridLayout,
    QLabel,
    QTextBrowser,
    QFrame,
)

from game.dcs.aircrafttype import AircraftType
from game.dcs.groundunittype import GroundUnitType
from game.dcs.unittype import UnitType
from game.game import Game

AIRCRAFT_BANNERS_BASE = Path("resources/ui/units/aircrafts/banners")
VEHICLE_BANNERS_BASE = Path("resources/ui/units/vehicles/banners")
MISSING_BANNER_PATH = AIRCRAFT_BANNERS_BASE / "Missing.jpg"


def aircraft_banner_for(unit_type: AircraftType) -> Path:
    if unit_type.dcs_id in {
        "Mirage-F1CT",
        "Mirage-F1EE",
        "Mirage-F1M-EE",
        "Mirage-F1EQ",
    }:
        name = "Mirage-F1C-200"
    elif unit_type.dcs_id in {"Mirage-F1CE", "Mirage-F1M-CE"}:
        name = "Mirage-F1C"
    elif unit_type.dcs_id in {"Su-30MKA", "Su-30MKI", "Su-30MKM"}:
        name = "Su-30SM"
    elif unit_type.dcs_id == "F-15ESE":
        name = "F-15E"
    else:
        name = unit_type.dcs_id
    return AIRCRAFT_BANNERS_BASE / f"{name}.jpg"


def vehicle_banner_for(unit_type: GroundUnitType) -> Path:
    if unit_type.dcs_id == "(IDF Mods Project) BM-21 Grad 122mm":
        return VEHICLE_BANNERS_BASE / "Grad-URAL.jpg"
    elif unit_type.dcs_id == "(IDF Mods Project) Urgan BM-27 220mm":
        return VEHICLE_BANNERS_BASE / "Uragan_BM-27.jpg"
    elif unit_type.dcs_id == "(IDF Mods Project) 9A52 Smerch CM 300mm":
        return VEHICLE_BANNERS_BASE / "Smerch_HE.jpg"
    else:
        return VEHICLE_BANNERS_BASE / f"{unit_type.dcs_id}.jpg"


def banner_path_for(unit_type: UnitType) -> Path:
    if isinstance(unit_type, AircraftType):
        return aircraft_banner_for(unit_type)
    if isinstance(unit_type, GroundUnitType):
        return vehicle_banner_for(unit_type)
    raise NotImplementedError(f"Unhandled UnitType subclass: {unit_type.__class__}")


class QUnitInfoWindow(QDialog):
    def __init__(self, game: Game, unit_type: UnitType) -> None:
        super().__init__()
        self.setModal(True)
        self.game = game
        self.unit_type = unit_type
        self.name = unit_type.name
        self.setWindowTitle(f"Unit Info: {self.name}")
        self.setWindowIcon(QIcon("./resources/icon.png"))
        self.setMinimumHeight(570)
        self.setMaximumWidth(640)
        self.setWindowFlags(Qt.WindowStaysOnTopHint)

        self.layout = QGridLayout()

        header = QLabel(self)
        header.setGeometry(0, 0, 720, 360)

        banner_path = banner_path_for(unit_type)
        if not banner_path.exists():
            banner_path = MISSING_BANNER_PATH
        pixmap = QPixmap(banner_path.as_posix())
        header.setPixmap(pixmap.scaled(header.width(), header.height()))
        self.layout.addWidget(header, 0, 0)

        self.gridLayout = QGridLayout()

        # Build the topmost details grid.
        self.details_grid = QFrame()
        self.details_grid_layout = QGridLayout()
        self.details_grid_layout.setMargin(0)

        self.name_box = QLabel(
            f"<b>Name:</b> {unit_type.manufacturer} {unit_type.name}"
        )
        self.name_box.setProperty("style", "info-element")

        self.country_box = QLabel(
            f"<b>Country of Origin:</b> {unit_type.country_of_origin}"
        )
        self.country_box.setProperty("style", "info-element")

        self.role_box = QLabel(f"<b>Role:</b> {unit_type.role}")
        self.role_box.setProperty("style", "info-element")

        self.year_box = QLabel(
            f"<b>Variant Introduction:</b> {unit_type.year_introduced}"
        )
        self.year_box.setProperty("style", "info-element")

        self.details_grid_layout.addWidget(self.name_box, 0, 0)
        self.details_grid_layout.addWidget(self.country_box, 0, 1)
        self.details_grid_layout.addWidget(self.role_box, 1, 0)
        self.details_grid_layout.addWidget(self.year_box, 1, 1)

        self.details_grid.setLayout(self.details_grid_layout)

        self.gridLayout.addWidget(self.details_grid, 1, 0)

        # If it's an aircraft, include the task list.
        if isinstance(unit_type, AircraftType):
            tasks = ", ".join(str(t) for t in unit_type.iter_task_capabilities())
            self.tasks_box = QLabel(f"<b>In-Game Tasks:</b> {tasks}")
            self.tasks_box.setProperty("style", "info-element")
            self.gridLayout.addWidget(self.tasks_box, 2, 0)

        # Finally, add the description box.
        self.details_text = QTextBrowser()
        self.details_text.setProperty("style", "info-desc")
        self.details_text.setText(unit_type.description)
        self.details_text.setOpenExternalLinks(
            True
        )  # in aircrafttype.py and groundunittype.py, for the descriptions, if No Data. including a google search link
        self.gridLayout.addWidget(self.details_text, 3, 0)

        self.layout.addLayout(self.gridLayout, 1, 0)
        self.setLayout(self.layout)