import logging
from typing import Callable, Dict, TypeVar

from PySide2.QtGui import QIcon, QPixmap
from PySide2.QtWidgets import (
    QDialog,
    QGridLayout,
    QGroupBox,
    QLabel,
    QPushButton,
    QVBoxLayout,
)

from game import db
from game.debriefing import Debriefing


T = TypeVar("T")


class LossGrid(QGridLayout):
    def __init__(self, debriefing: Debriefing, player: bool) -> None:
        super().__init__()

        if player:
            country = debriefing.player_country
        else:
            country = debriefing.enemy_country

        self.add_loss_rows(
            debriefing.air_losses.by_type(player),
            lambda u: db.unit_get_expanded_info(country, u, "name"),
        )
        self.add_loss_rows(
            debriefing.front_line_losses_by_type(player),
            lambda u: db.unit_type_name(u),
        )
        self.add_loss_rows(
            debriefing.convoy_losses_by_type(player),
            lambda u: f"{db.unit_type_name(u)} from convoy",
        )
        self.add_loss_rows(
            debriefing.cargo_ship_losses_by_type(player),
            lambda u: f"{db.unit_type_name(u)} from cargo ship",
        )
        self.add_loss_rows(
            debriefing.airlift_losses_by_type(player),
            lambda u: f"{db.unit_type_name(u)} from airlift",
        )
        self.add_loss_rows(
            debriefing.building_losses_by_type(player),
            lambda u: u,
        )

        # TODO: Display dead ground object units and runways.

    def add_loss_rows(self, losses: Dict[T, int], make_name: Callable[[T], str]):
        for unit_type, count in losses.items():
            row = self.rowCount()
            try:
                name = make_name(unit_type)
            except AttributeError:
                logging.exception(f"Could not make unit name for {unit_type}")
                name = unit_type.id
            self.addWidget(QLabel(name), row, 0)
            self.addWidget(QLabel(str(count)), row, 1)


class QDebriefingWindow(QDialog):
    def __init__(self, debriefing: Debriefing):
        super(QDebriefingWindow, self).__init__()
        self.debriefing = debriefing

        self.setModal(True)
        self.setWindowTitle("Debriefing")
        self.setMinimumSize(300, 200)
        self.setWindowIcon(QIcon("./resources/icon.png"))

        layout = QVBoxLayout()
        self.setLayout(layout)

        header = QLabel(self)
        header.setGeometry(0, 0, 655, 106)
        pixmap = QPixmap("./resources/ui/debriefing.png")
        header.setPixmap(pixmap)
        layout.addWidget(header)
        layout.addStretch()

        title = QLabel("<b>Casualty report</b>")
        layout.addWidget(title)

        player_lost_units = QGroupBox(f"{self.debriefing.player_country}'s lost units:")
        player_lost_units.setLayout(LossGrid(debriefing, player=True))
        layout.addWidget(player_lost_units)

        enemy_lost_units = QGroupBox(f"{self.debriefing.enemy_country}'s lost units:")
        enemy_lost_units.setLayout(LossGrid(debriefing, player=False))
        layout.addWidget(enemy_lost_units)

        okay = QPushButton("Okay")
        okay.clicked.connect(self.close)
        layout.addWidget(okay)