from __future__ import annotations

from collections import defaultdict
from enum import Enum
from typing import Optional

from .aoc import Aoc2


class Type(Enum):
    HIGH_CARD = 0
    ONE_PAIR = 1
    TWO_PAIR = 2
    THREE_KIND = 3
    FULL_HOUSE = 4
    FOUR_KIND = 5
    FIVE_KIND = 6

    def __lt__(self, other: Type) -> bool:
        return self.value < other.value


class Hand:
    def __init__(
        self, string: str, translation: str, joker: Optional[str] = None
    ) -> None:
        self.string = [translation.index(x) for x in string]
        self.cards_dict: dict[str, int] = defaultdict(lambda: 0)
        for letter in string:
            self.cards_dict[letter] += 1

        value_amount = list(
            reversed(
                sorted(
                    list([v for k, v in self.cards_dict.items() if k != joker]))
            )
        )
        joker_amount = 0
        if joker is not None:
            joker_amount = self.cards_dict[joker]

            if joker_amount == 5:
                self.type = Type.FIVE_KIND
                return

        match value_amount[0] + joker_amount:
            case 5:
                self.type = Type.FIVE_KIND
            case 4:
                self.type = Type.FOUR_KIND
            case 3:
                if value_amount[1] == 2:
                    self.type = Type.FULL_HOUSE
                else:
                    self.type = Type.THREE_KIND
            case 2:
                if value_amount[1] == 2:
                    self.type = Type.TWO_PAIR
                else:
                    self.type = Type.ONE_PAIR
            case _:
                self.type = Type.HIGH_CARD

    def __lt__(self, other: Hand) -> bool:
        """self < other"""
        return (self.type, self.string) < (other.type, other.string)

    def __repr__(self) -> str:
        return f"Hand({self.type}, {self.string})"


class Day07(Aoc2[list[tuple[Hand, int]], list[tuple[Hand, int]]]):
    def parseinput1(self, inpt: str) -> list[tuple[Hand, int]]:
        lines = inpt.splitlines()
        return [
            (Hand(line.split(" ")[0], "23456789TJQKA"),
             int(line.split(" ")[1]))
            for line in lines
        ]

    def parseinput2(self, inpt: str) -> list[tuple[Hand, int]]:
        lines = inpt.splitlines()
        return [
            (Hand(line.split(" ")[0], "J23456789TQKA", "J"), int(
                line.split(" ")[1]))
            for line in lines
        ]

    def part1(self, inpt: list[tuple[Hand, int]]) -> int:
        return sum(((rnk + 1) * bid) for rnk, (_, bid) in list(enumerate(sorted(inpt))))

    def part2(self, inpt: list[tuple[Hand, int]]) -> int:
        return sum(((rnk + 1) * bid) for rnk, (_, bid) in list(enumerate(sorted(inpt))))


def main() -> None:
    day07 = Day07(2023, 7)
    day07.run()


if __name__ == "__main__":
    main()