from collections import defaultdict
from collections.abc import Iterator

from arkhamdb import ArkhamDBCard, ArkhamDBDeck


class Validator:
    cards_by_code: dict[str, ArkhamDBCard]

    def __init__(self, cards: list[ArkhamDBCard]) -> None:
        cards: list[ArkhamDBCard]
        self.cards_by_code = {card["code"]: card for card in cards}

    def validate(self, decks: list[ArkhamDBDeck]) -> Iterator[str]:
        card_total_counts = defaultdict(lambda: 0)
        for deck in decks:
            for code, count in deck["slots"].items():
                card_total_counts[code] += count

        for code, count in card_total_counts.items():
            try:
                card = self.cards_by_code[code]
                quantity = card["quantity"]
                # assume two copies of the core set
                if card["pack_code"] == "core":
                    quantity = quantity * 2

                if count > quantity:
                    in_decks = [deck["id"] for deck in decks if code in deck["slots"]]
                    yield f"Too many copies of {card['name']}! ({count} > {quantity}), in decks {in_decks}"
            except KeyError:
                yield f'Did not find card with code "{code}"'