Grinch-AP/worlds/stardew_valley/logic/crafting_logic.py

from Utils import cache_self1
from .base_logic import BaseLogicMixin, BaseLogic
from .. import options
from ..data.craftable_data import CraftingRecipe
from ..data.recipe_source import CutsceneSource, ShopTradeSource, ArchipelagoSource, LogicSource, SpecialOrderSource, \
    FestivalShopSource, QuestSource, StarterSource, ShopSource, SkillSource, MasterySource, FriendshipSource, SkillCraftsanitySource
from ..options import Craftsanity, SpecialOrderLocations
from ..stardew_rule import StardewRule, True_, False_
from ..strings.region_names import Region


class CraftingLogicMixin(BaseLogicMixin):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.crafting = CraftingLogic(*args, **kwargs)


class CraftingLogic(BaseLogic):
    @cache_self1
    def can_craft(self, recipe: CraftingRecipe = None) -> StardewRule:
        if recipe is None:
            return True_()

        learn_rule = self.logic.crafting.knows_recipe(recipe)
        ingredients_rule = self.logic.has_all(*recipe.ingredients)
        return learn_rule & ingredients_rule

    @cache_self1
    def knows_recipe(self, recipe: CraftingRecipe) -> StardewRule:
        if isinstance(recipe.source, ArchipelagoSource):
            return self.logic.received_all(*recipe.source.ap_item)
        if isinstance(recipe.source, FestivalShopSource):
            if self.options.festival_locations == options.FestivalLocations.option_disabled:
                return self.logic.crafting.can_learn_recipe(recipe)
            else:
                return self.logic.crafting.received_recipe(recipe.item)
        if isinstance(recipe.source, QuestSource):
            if self.options.quest_locations.has_no_story_quests():
                return self.logic.crafting.can_learn_recipe(recipe)
            else:
                return self.logic.crafting.received_recipe(recipe.item)
        if self.options.craftsanity == Craftsanity.option_none:
            return self.logic.crafting.can_learn_recipe(recipe)
        if isinstance(recipe.source, (StarterSource, ShopTradeSource, ShopSource, SkillCraftsanitySource)):
            return self.logic.crafting.received_recipe(recipe.item)
        if isinstance(recipe.source, SpecialOrderSource) and self.options.special_order_locations & SpecialOrderLocations.option_board:
            return self.logic.crafting.received_recipe(recipe.item)
        return self.logic.crafting.can_learn_recipe(recipe)

    @cache_self1
    def can_learn_recipe(self, recipe: CraftingRecipe) -> StardewRule:
        if isinstance(recipe.source, StarterSource):
            return True_()
        if isinstance(recipe.source, ArchipelagoSource):
            return self.logic.received_all(*recipe.source.ap_item)
        if isinstance(recipe.source, ShopTradeSource):
            return self.logic.money.can_trade_at(recipe.source.region, recipe.source.currency, recipe.source.price)
        if isinstance(recipe.source, ShopSource):
            return self.logic.money.can_spend_at(recipe.source.region, recipe.source.price)
        if isinstance(recipe.source, SkillCraftsanitySource):
            return self.logic.skill.has_level(recipe.source.skill, recipe.source.level) & self.logic.skill.can_earn_level(recipe.source.skill,
                                                                                                                          recipe.source.level)
        if isinstance(recipe.source, SkillSource):
            return self.logic.skill.has_level(recipe.source.skill, recipe.source.level)
        if isinstance(recipe.source, MasterySource):
            return self.logic.skill.has_mastery(recipe.source.skill)
        if isinstance(recipe.source, CutsceneSource):
            return self.logic.region.can_reach(recipe.source.region) & self.logic.relationship.has_hearts(recipe.source.friend, recipe.source.hearts)
        if isinstance(recipe.source, FriendshipSource):
            return self.logic.relationship.has_hearts(recipe.source.friend, recipe.source.hearts)
        if isinstance(recipe.source, QuestSource):
            return self.logic.quest.can_complete_quest(recipe.source.quest)
        if isinstance(recipe.source, SpecialOrderSource):
            if self.options.special_order_locations & SpecialOrderLocations.option_board:
                return self.logic.crafting.received_recipe(recipe.item)
            return self.logic.special_order.can_complete_special_order(recipe.source.special_order)
        if isinstance(recipe.source, LogicSource):
            if recipe.source.logic_rule == "Cellar":
                return self.logic.region.can_reach(Region.cellar)

        return False_()

    @cache_self1
    def received_recipe(self, item_name: str):
        return self.logic.received(f"{item_name} Recipe")