from decimal import Decimal
from typing import Literal

from ._common import StepBooster


class LegacyMapStepBooster(StepBooster):
    __fragment_boost_multipliers = {
        None: Decimal("1.0"),
        100: Decimal("1.1"),
        250: Decimal("1.25"),
        500: Decimal("1.5"),
    }

    def __init__(
        self,
        stamina: Literal[2, 4, 6],
        fragments: Literal[100, 250, 500, None],
    ):
        self.stamina = stamina
        self.fragments = fragments

    @property
    def stamina(self):
        return self.__stamina

    @stamina.setter
    def stamina(self, value: Literal[2, 4, 6]):
        if value not in [2, 4, 6]:
            raise ValueError("stamina can only be one of [2, 4, 6]")
        self.__stamina = value

    @property
    def fragments(self):
        return self.__fragments

    @fragments.setter
    def fragments(self, value: Literal[100, 250, 500, None]):
        if value not in [100, 250, 500, None]:
            raise ValueError("fragments can only be one of [100, 250, 500, None]")
        self.__fragments = value

    def final_value(self) -> Decimal:
        stamina_multiplier = Decimal(self.stamina)
        fragments_multiplier = self.__fragment_boost_multipliers[self.fragments]
        return stamina_multiplier * fragments_multiplier