82 lines
2.1 KiB
Python
82 lines
2.1 KiB
Python
import enum
|
|
from abc import ABC
|
|
from dataclasses import dataclass, field
|
|
from types import MappingProxyType
|
|
from typing import List, Iterable, Set, ClassVar, Tuple, Mapping, Callable, Any
|
|
|
|
from ..stardew_rule.protocol import StardewRule
|
|
|
|
DEFAULT_REQUIREMENT_TAGS = MappingProxyType({})
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class Requirement(ABC):
|
|
...
|
|
|
|
|
|
class ItemTag(enum.Enum):
|
|
CROPSANITY_SEED = enum.auto()
|
|
CROPSANITY = enum.auto()
|
|
FISH = enum.auto()
|
|
FRUIT = enum.auto()
|
|
VEGETABLE = enum.auto()
|
|
EDIBLE_MUSHROOM = enum.auto()
|
|
BOOK = enum.auto()
|
|
BOOK_POWER = enum.auto()
|
|
BOOK_SKILL = enum.auto()
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class ItemSource(ABC):
|
|
add_tags: ClassVar[Tuple[ItemTag]] = ()
|
|
|
|
other_requirements: Tuple[Requirement, ...] = field(kw_only=True, default_factory=tuple)
|
|
|
|
@property
|
|
def requirement_tags(self) -> Mapping[str, Tuple[ItemTag, ...]]:
|
|
return DEFAULT_REQUIREMENT_TAGS
|
|
|
|
|
|
@dataclass(frozen=True, kw_only=True)
|
|
class GenericSource(ItemSource):
|
|
regions: Tuple[str, ...] = ()
|
|
"""No region means it's available everywhere."""
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class CustomRuleSource(ItemSource):
|
|
"""Hopefully once everything is migrated to sources, we won't need these custom logic anymore."""
|
|
create_rule: Callable[[Any], StardewRule]
|
|
|
|
|
|
@dataclass(frozen=True, kw_only=True)
|
|
class CompoundSource(ItemSource):
|
|
sources: Tuple[ItemSource, ...] = ()
|
|
|
|
|
|
class Tag(ItemSource):
|
|
"""Not a real source, just a way to add tags to an item. Will be removed from the item sources during unpacking."""
|
|
tag: Tuple[ItemTag, ...]
|
|
|
|
def __init__(self, *tag: ItemTag):
|
|
self.tag = tag # noqa
|
|
|
|
@property
|
|
def add_tags(self):
|
|
return self.tag
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class GameItem:
|
|
name: str
|
|
sources: List[ItemSource] = field(default_factory=list)
|
|
tags: Set[ItemTag] = field(default_factory=set)
|
|
|
|
def add_sources(self, sources: Iterable[ItemSource]):
|
|
self.sources.extend(source for source in sources if type(source) is not Tag)
|
|
for source in sources:
|
|
self.add_tags(source.add_tags)
|
|
|
|
def add_tags(self, tags: Iterable[ItemTag]):
|
|
self.tags.update(tags)
|