Grinch-AP/worlds/stardew_valley/content/feature/building_progression.py

from abc import ABC
from dataclasses import dataclass
from typing import ClassVar, Set, Tuple

from ...strings.building_names import Building

progressive_house = "Progressive House"

# This assumes that the farm house is always available, which might not be true forever...
progressive_house_by_upgrade_name = {
    Building.farm_house: 0,
    Building.kitchen: 1,
    Building.kids_room: 2,
    Building.cellar: 3
}


def to_progressive_item(building: str) -> Tuple[str, int]:
    """Return the name of the progressive item and its quantity required to unlock the building.
    """
    if building in [Building.coop, Building.barn, Building.shed]:
        return f"Progressive {building}", 1
    elif building.startswith("Big"):
        return f"Progressive {building[building.index(' ') + 1:]}", 2
    elif building.startswith("Deluxe"):
        return f"Progressive {building[building.index(' ') + 1:]}", 3
    elif building in progressive_house_by_upgrade_name:
        return progressive_house, progressive_house_by_upgrade_name[building]

    return building, 1


def to_location_name(building: str) -> str:
    return f"{building} Blueprint"


@dataclass(frozen=True)
class BuildingProgressionFeature(ABC):
    is_progressive: ClassVar[bool]
    starting_buildings: Set[str]

    to_progressive_item = staticmethod(to_progressive_item)
    progressive_house = progressive_house

    to_location_name = staticmethod(to_location_name)


class BuildingProgressionVanilla(BuildingProgressionFeature):
    is_progressive = False


class BuildingProgressionProgressive(BuildingProgressionFeature):
    is_progressive = True