from enum import IntEnum, StrEnum
from random import choice

from src.utils.LazyPhraseDict import LazyPhraseDict


class Sign(StrEnum):
    """
    Enum for the sign of an ability.
    """

    POSITIVE = "+"
    NEGATIVE = "-"


class DevilFruitAbilityType(IntEnum):
    """
    Enum for the type of Devil Fruit ability type.
    """

    DOC_Q_COOLDOWN_DURATION = 1
    GAME_COOLDOWN_DURATION = 2
    FIGHT_COOLDOWN_DURATION = 3
    FIGHT_IMMUNITY_DURATION = 4
    FIGHT_DEFENSE_BOOST = 5
    PREDICTION_WAGER_REFUND = 6
    GIFT_LOAN_TAX = 7
    INCOME_TAX = 8
    PLUNDER_COOLDOWN_DURATION = 9
    PLUNDER_IMMUNITY_DURATION = 10
    PLUNDER_SENTENCE_DURATION = 11
    FIGHT_SCOUT_FEE = 12
    PLUNDER_SCOUT_FEE = 13
    GAME_GLOBAL_ACCEPT_COOLDOWN_DURATION = 14

    def get_description(self) -> str:
        """
        Get the description of the devil fruit ability type
        :return: The description of the devil fruit ability type
        """

        return DEVIL_FRUIT_ABILITY_TYPE_DESCRIPTION_MAP[self]

    def get_sign(self) -> Sign:
        """
        Get the sign of the devil fruit ability type
        :return: The sign of the devil fruit ability type
        """

        return DEVIL_FRUIT_ABILITY_TYPE_SIGN_MAP[self]

    @staticmethod
    def get_not_allowed_ability_types_from_random() -> list["DevilFruitAbilityType"]:
        """
        Returns the not allowed ability types
        :return: The not allowed ability types
        """

        return [DevilFruitAbilityType.INCOME_TAX]

    @staticmethod
    def get_allowed_ability_types_from_random() -> list["DevilFruitAbilityType"]:
        """
        Returns the allowed ability types
        :return: The allowed ability types
        """
        return [
            ability_type
            for ability_type in DevilFruitAbilityType
            if ability_type
            not in DevilFruitAbilityType.get_not_allowed_ability_types_from_random()
        ]

    @staticmethod
    def get_random_ability() -> "DevilFruitAbilityType":
        """
        Get a random devil fruit ability type
        :return: A random devil fruit ability type
        """

        return choice(list(DevilFruitAbilityType.get_allowed_ability_types_from_random()))


DEVIL_FRUIT_ABILITY_TYPE_DESCRIPTION_MAP = LazyPhraseDict(
    {
        DevilFruitAbilityType.DOC_Q_COOLDOWN_DURATION: "ABILITY_TYPE_DOC_Q_COOLDOWN_DURATION",
        DevilFruitAbilityType.GAME_COOLDOWN_DURATION: "ABILITY_TYPE_GAME_COOLDOWN_DURATION",
        DevilFruitAbilityType.FIGHT_COOLDOWN_DURATION: "ABILITY_TYPE_FIGHT_COOLDOWN_DURATION",
        DevilFruitAbilityType.FIGHT_IMMUNITY_DURATION: "ABILITY_TYPE_FIGHT_IMMUNITY_DURATION",
        DevilFruitAbilityType.FIGHT_DEFENSE_BOOST: "ABILITY_TYPE_FIGHT_DEFENSE_BOOST",
        DevilFruitAbilityType.PREDICTION_WAGER_REFUND: "ABILITY_TYPE_PREDICTION_WAGER_REFUND",
        DevilFruitAbilityType.GIFT_LOAN_TAX: "ABILITY_TYPE_GIFT_LOAN_TAX",
        DevilFruitAbilityType.INCOME_TAX: "ABILITY_TYPE_INCOME_TAX",
        DevilFruitAbilityType.PLUNDER_COOLDOWN_DURATION: (
            "ABILITY_TYPE_PLUNDER_COOLDOWN_DURATION"
        ),
        DevilFruitAbilityType.PLUNDER_IMMUNITY_DURATION: (
            "ABILITY_TYPE_PLUNDER_IMMUNITY_DURATION"
        ),
        DevilFruitAbilityType.PLUNDER_SENTENCE_DURATION: (
            "ABILITY_TYPE_PLUNDER_SENTENCE_DURATION"
        ),
        DevilFruitAbilityType.FIGHT_SCOUT_FEE: "ABILITY_TYPE_FIGHT_SCOUT_FEE",
        DevilFruitAbilityType.PLUNDER_SCOUT_FEE: "ABILITY_TYPE_PLUNDER_SCOUT_FEE",
        DevilFruitAbilityType.GAME_GLOBAL_ACCEPT_COOLDOWN_DURATION: (
            "ABILITY_TYPE_GAME_GLOBAL_ACCEPT_COOLDOWN_DURATION"
        ),
    }
)

DEVIL_FRUIT_ABILITY_TYPE_SIGN_MAP = {
    DevilFruitAbilityType.DOC_Q_COOLDOWN_DURATION: Sign.NEGATIVE,
    DevilFruitAbilityType.GAME_COOLDOWN_DURATION: Sign.NEGATIVE,
    DevilFruitAbilityType.FIGHT_COOLDOWN_DURATION: Sign.NEGATIVE,
    DevilFruitAbilityType.FIGHT_IMMUNITY_DURATION: Sign.POSITIVE,
    DevilFruitAbilityType.FIGHT_DEFENSE_BOOST: Sign.POSITIVE,
    DevilFruitAbilityType.PREDICTION_WAGER_REFUND: Sign.POSITIVE,
    DevilFruitAbilityType.GIFT_LOAN_TAX: Sign.NEGATIVE,
    DevilFruitAbilityType.INCOME_TAX: Sign.NEGATIVE,
    DevilFruitAbilityType.PLUNDER_COOLDOWN_DURATION: Sign.NEGATIVE,
    DevilFruitAbilityType.PLUNDER_IMMUNITY_DURATION: Sign.POSITIVE,
    DevilFruitAbilityType.PLUNDER_SENTENCE_DURATION: Sign.NEGATIVE,
    DevilFruitAbilityType.FIGHT_SCOUT_FEE: Sign.NEGATIVE,
    DevilFruitAbilityType.PLUNDER_SCOUT_FEE: Sign.NEGATIVE,
    DevilFruitAbilityType.GAME_GLOBAL_ACCEPT_COOLDOWN_DURATION: Sign.NEGATIVE,
}