from abc import ABC, abstractmethod
from dataclasses import dataclass
from typing import ClassVar

from .base import FeatureBase
from ...data.hats_data import HatItem

wear_prefix = "Wear "


def to_location_name(hat: str | HatItem) -> str:
    if isinstance(hat, HatItem):
        hat = hat.name
    return f"{wear_prefix}{hat}"


def extract_hat_from_location_name(location_name: str) -> str | None:
    if not location_name.startswith(wear_prefix):
        return None

    return location_name[len(wear_prefix):]


@dataclass(frozen=True)
class HatsanityFeature(FeatureBase, ABC):
    is_enabled: ClassVar[bool]

    to_location_name = staticmethod(to_location_name)
    extract_hat_from_location_name = staticmethod(extract_hat_from_location_name)

    @abstractmethod
    def is_included(self, hat: HatItem) -> bool:
        ...


class HatsanityNone(HatsanityFeature):
    is_enabled = False

    def is_included(self, hat: HatItem) -> bool:
        return False


@dataclass(frozen=True)
class HatsanityHats(HatsanityFeature):
    is_enabled = True
    enabled_hats: frozenset[str]

    def is_included(self, hat: HatItem) -> bool:
        for difficulty in hat.difficulty:
            if difficulty not in self.enabled_hats:
                return False
        return True