Grinch-AP/worlds/stardew_valley/mods/logic/magic_logic.py

from typing import Union

from ...logic.base_logic import BaseLogicMixin, BaseLogic
from ...logic.has_logic import HasLogicMixin
from ...logic.received_logic import ReceivedLogicMixin
from ...logic.region_logic import RegionLogicMixin
from ...mods.mod_data import ModNames
from ...stardew_rule import StardewRule, False_
from ...strings.ap_names.skill_level_names import ModSkillLevel
from ...strings.region_names import MagicRegion
from ...strings.spells import MagicSpell


class MagicLogicMixin(BaseLogicMixin):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.magic = MagicLogic(*args, **kwargs)


# TODO add logic.mods.magic for altar
class MagicLogic(BaseLogic[Union[RegionLogicMixin, ReceivedLogicMixin, HasLogicMixin]]):
    def can_use_clear_debris_instead_of_tool_level(self, level: int) -> StardewRule:
        if ModNames.magic not in self.options.mods:
            return False_()
        return self.logic.received(MagicSpell.clear_debris) & self.can_use_altar() & self.logic.received(ModSkillLevel.magic_level, level)

    def can_use_altar(self) -> StardewRule:
        if ModNames.magic not in self.options.mods:
            return False_()
        return self.logic.region.can_reach(MagicRegion.altar)

    def has_any_spell(self) -> StardewRule:
        if ModNames.magic not in self.options.mods:
            return False_()
        return self.can_use_altar()

    def has_attack_spell_count(self, count: int) -> StardewRule:
        attack_spell_rule = [self.logic.received(MagicSpell.fireball), self.logic.received(MagicSpell.frostbite), self.logic.received(MagicSpell.shockwave),
                             self.logic.received(MagicSpell.spirit), self.logic.received(MagicSpell.meteor)]
        return self.logic.count(count, *attack_spell_rule)

    def has_support_spell_count(self, count: int) -> StardewRule:
        support_spell_rule = [self.can_use_altar(), self.logic.received(ModSkillLevel.magic_level, 2),
                              self.logic.received(MagicSpell.descend), self.logic.received(MagicSpell.heal),
                              self.logic.received(MagicSpell.tendrils)]
        return self.logic.count(count, *support_spell_rule)

    def has_decent_spells(self) -> StardewRule:
        if ModNames.magic not in self.options.mods:
            return False_()
        magic_resource_rule = self.can_use_altar() & self.logic.received(ModSkillLevel.magic_level, 2)
        magic_attack_options_rule = self.has_attack_spell_count(1)
        return magic_resource_rule & magic_attack_options_rule

    def has_good_spells(self) -> StardewRule:
        if ModNames.magic not in self.options.mods:
            return False_()
        magic_resource_rule = self.can_use_altar() & self.logic.received(ModSkillLevel.magic_level, 4)
        magic_attack_options_rule = self.has_attack_spell_count(2)
        magic_support_options_rule = self.has_support_spell_count(1)
        return magic_resource_rule & magic_attack_options_rule & magic_support_options_rule

    def has_great_spells(self) -> StardewRule:
        if ModNames.magic not in self.options.mods:
            return False_()
        magic_resource_rule = self.can_use_altar() & self.logic.received(ModSkillLevel.magic_level, 6)
        magic_attack_options_rule = self.has_attack_spell_count(3)
        magic_support_options_rule = self.has_support_spell_count(1)
        return magic_resource_rule & magic_attack_options_rule & magic_support_options_rule

    def has_amazing_spells(self) -> StardewRule:
        if ModNames.magic not in self.options.mods:
            return False_()
        magic_resource_rule = self.can_use_altar() & self.logic.received(ModSkillLevel.magic_level, 8)
        magic_attack_options_rule = self.has_attack_spell_count(4)
        magic_support_options_rule = self.has_support_spell_count(2)
        return magic_resource_rule & magic_attack_options_rule & magic_support_options_rule

    def can_blink(self) -> StardewRule:
        if ModNames.magic not in self.options.mods:
            return False_()
        return self.logic.received(MagicSpell.blink) & self.can_use_altar()