Grinch-AP/worlds/stardew_valley/bundles/bundle_item.py

from __future__ import annotations

from abc import ABC, abstractmethod
from dataclasses import dataclass

from ..content import StardewContent, content_packs
from ..options import StardewValleyOptions, FestivalLocations
from ..strings.crop_names import Fruit
from ..strings.currency_names import Currency
from ..strings.quality_names import CropQuality, FishQuality, ForageQuality


class BundleItemSource(ABC):
    @abstractmethod
    def can_appear(self, content: StardewContent, options: StardewValleyOptions) -> bool:
        ...


class VanillaItemSource(BundleItemSource):
    def can_appear(self, content: StardewContent, options: StardewValleyOptions) -> bool:
        return True


class IslandItemSource(BundleItemSource):
    def can_appear(self, content: StardewContent, options: StardewValleyOptions) -> bool:
        return content_packs.ginger_island_content_pack.name in content.registered_packs


class FestivalItemSource(BundleItemSource):
    def can_appear(self, content: StardewContent, options: StardewValleyOptions) -> bool:
        return options.festival_locations != FestivalLocations.option_disabled


# FIXME remove this once recipes are in content packs
class MasteryItemSource(BundleItemSource):
    def can_appear(self, content: StardewContent, options: StardewValleyOptions) -> bool:
        return content.features.skill_progression.are_masteries_shuffled


class ContentItemSource(BundleItemSource):
    """This is meant to be used for items that are managed by the content packs."""

    def can_appear(self, content: StardewContent, options: StardewValleyOptions) -> bool:
        raise ValueError("This should not be called, check if the item is in the content instead.")


@dataclass(frozen=True, order=True)
class BundleItem:
    class Sources:
        vanilla = VanillaItemSource()
        island = IslandItemSource()
        festival = FestivalItemSource()
        masteries = MasteryItemSource()
        content = ContentItemSource()

    item_name: str
    amount: int = 1
    quality: str = CropQuality.basic
    source: BundleItemSource = Sources.vanilla
    flavor: str = None
    can_have_quality: bool = True

    @staticmethod
    def money_bundle(amount: int) -> BundleItem:
        return BundleItem(Currency.money, amount)

    def get_item(self) -> str:
        if self.flavor is None:
            return self.item_name
        return f"{self.item_name} [{self.flavor}]"

    def as_amount(self, amount: int) -> BundleItem:
        return BundleItem(self.item_name, amount, self.quality, self.source, self.flavor)

    def as_quality(self, quality: str) -> BundleItem:
        if self.can_have_quality:
            return BundleItem(self.item_name, self.amount, quality, self.source, self.flavor)
        return BundleItem(self.item_name, self.amount, self.quality, self.source, self.flavor)

    def as_quality_crop(self) -> BundleItem:
        amount = 5
        difficult_crops = [Fruit.sweet_gem_berry, Fruit.ancient_fruit]
        if self.item_name in difficult_crops:
            amount = 1
        return self.as_quality(CropQuality.gold).as_amount(amount)

    def as_quality_fish(self) -> BundleItem:
        return self.as_quality(FishQuality.gold)

    def as_quality_forage(self) -> BundleItem:
        return self.as_quality(ForageQuality.gold)

    def __repr__(self):
        quality = "" if self.quality == CropQuality.basic else self.quality
        return f"{self.amount} {quality} {self.get_item()}"

    def can_appear(self, content: StardewContent, options: StardewValleyOptions) -> bool:
        if isinstance(self.source, ContentItemSource):
            return self.get_item() in content.game_items

        return self.source.can_appear(content, options)