import logging

import roman

from .cards import Cards
from .ttsdeck import TTSDeck


class Opus(Cards):
    def __init__(self, opus_id: str):

        logger = logging.getLogger(__name__)

        if opus_id.isnumeric():
            name = f"Opus {opus_id}"
            self.__number = opus_id
            params = {"set": [f"Opus {roman.toRoman(int(opus_id)).upper()}"]}

        elif opus_id == "chaos":
            name = "Boss Deck Chaos"
            self.__number = "B"
            params = {"set": [name]}

        elif opus_id == "promo":
            name = "Promo"
            self.__number = "PR"
            params = {"rarity": ["pr"]}

        else:
            name = "?"
            self.__number = "?"
            self.__filename = "?"
            params = {"set": "?"}

        super().__init__(name)
        self._load(params)

        # remove reprints
        for card in self:
            if not card.code.opus == self.__number:
                self.remove(card)

        # sort cards by opus, then serial
        self.sort(key=lambda x: x.code.serial)
        self.sort(key=lambda x: x.code.opus)

        for card in self:
            logger.info(f"imported card {card}")

    @property
    def number(self) -> str:
        return self.__number

    @property
    def elemental_decks(self) -> list[TTSDeck]:
        if self.name in ["Promo", "Boss Deck Chaos"]:
            return [TTSDeck(
                [
                    card.code
                    for card in self
                ],
                f"{self.name}",
                f"All {self.name} Cards in elemental, then alphabetical order"
            )]

        else:
            def element_filter(element: str):
                return lambda card: card.elements == [element]

            # simple cases: create lambdas for base elemental decks
            base_elements = ["Fire", "Ice", "Wind", "Earth", "Lightning", "Water"]
            filters = {elem: element_filter(elem) for elem in base_elements}

            filters |= {
                # light/darkness elemental deck
                "Light/Darkness": lambda card: card.elements == ["Light"] or card.elements == ["Darkness"],
                # multi element deck
                "Multi": lambda card: len(card.elements) > 1,
            }

            return [TTSDeck(
                [
                    card.code
                    for card in self
                    if f(card)
                ],
                f"{self.name} {elem}",
                f"All {self.name} Cards with {elem} element in alphabetical order"
            ) for elem, f in filters.items()]