Grinch-AP/worlds/stardew_valley/region_classes.py

from enum import IntFlag
from typing import Optional, List
from dataclasses import dataclass, field

connector_keyword = " to "


class ModificationFlag(IntFlag):
    NOT_MODIFIED = 0
    MODIFIED = 1

class RandomizationFlag(IntFlag):
    NOT_RANDOMIZED = 0b0
    PELICAN_TOWN = 0b11111
    NON_PROGRESSION = 0b11110
    BUILDINGS = 0b11100
    EVERYTHING = 0b11000
    CHAOS = 0b10000
    GINGER_ISLAND = 0b0100000
    LEAD_TO_OPEN_AREA = 0b1000000


@dataclass(frozen=True)
class RegionData:
    name: str
    exits: List[str] = field(default_factory=list)
    flag: ModificationFlag = ModificationFlag.NOT_MODIFIED

    def get_merged_with(self, exits: List[str]):
        merged_exits = []
        merged_exits.extend(self.exits)
        if exits is not None:
            merged_exits.extend(exits)
        merged_exits = list(set(merged_exits))
        return RegionData(self.name, merged_exits)

    def get_without_exit(self, exit_to_remove: str):
        exits = [exit for exit in self.exits if exit != exit_to_remove]
        return RegionData(self.name, exits)

    def get_clone(self):
        return self.get_merged_with(None)


@dataclass(frozen=True)
class ConnectionData:
    name: str
    destination: str
    origin: Optional[str] = None
    reverse: Optional[str] = None
    flag: RandomizationFlag = RandomizationFlag.NOT_RANDOMIZED

    def __post_init__(self):
        if connector_keyword in self.name:
            origin, destination = self.name.split(connector_keyword)
            if self.reverse is None:
                super().__setattr__("reverse", f"{destination}{connector_keyword}{origin}")


@dataclass(frozen=True)
class ModRegionData:
    mod_name: str
    regions: List[RegionData]
    connections: List[ConnectionData]