Entrance.py

from __future__ import annotations
from typing import TYPE_CHECKING, Optional, Any

if TYPE_CHECKING:
    from Region import Region
    from RulesCommon import AccessRule
    from World import World


class Entrance:
    def __init__(self, name: str = '', parent: Optional[Region] = None) -> None:
        self.name: str = name
        self.parent_region: Optional[Region] = parent
        self.world: Optional[World] = parent.world if parent is not None else None
        self.connected_region: Optional[Region] = None
        self.access_rule: AccessRule = lambda state, **kwargs: True
        self.access_rules: list[AccessRule] = []
        self.reverse: Optional[Entrance] = None
        self.replaces: Optional[Entrance] = None
        self.assumed: Optional[Entrance] = None
        self.type: Optional[str] = None
        self.shuffled: bool = False
        self.data: Optional[dict[str, Any]] = None
        self.primary: bool = False
        self.always: bool = False
        self.never: bool = False
        self.rule_string: Optional[str] = None

    def copy(self, *, copy_dict: Optional[dict[int, Any]] = None) -> Entrance:
        copy_dict = {} if copy_dict is None else copy_dict
        if (new_entrance := copy_dict.get(id(self), None)) and isinstance(new_entrance, Entrance):
            return new_entrance

        new_entrance = Entrance(self.name, self.parent_region.copy(copy_dict=copy_dict) if self.parent_region else None)
        copy_dict[id(self)] = new_entrance

        if self.connected_region is not None:
            new_entrance.connected_region = self.connected_region.copy(copy_dict=copy_dict)
        new_entrance.access_rule = self.access_rule
        new_entrance.access_rules = list(self.access_rules)
        if self.reverse:
            new_entrance.reverse = self.reverse.copy(copy_dict=copy_dict)
        if self.replaces:
            new_entrance.replaces = self.replaces.copy(copy_dict=copy_dict)
        if self.assumed:
            new_entrance.assumed = self.assumed.copy(copy_dict=copy_dict)
        new_entrance.type = self.type
        new_entrance.shuffled = self.shuffled
        new_entrance.data = self.data
        new_entrance.primary = self.primary
        new_entrance.always = self.always
        new_entrance.never = self.never

        return new_entrance

    def add_rule(self, lambda_rule: AccessRule) -> None:
        if self.always:
            self.set_rule(lambda_rule)
            self.always = False
            return
        if self.never:
            return
        self.access_rules.append(lambda_rule)
        self.access_rule = lambda state, **kwargs: all(rule(state, **kwargs) for rule in self.access_rules)

    def set_rule(self, lambda_rule: AccessRule) -> None:
        self.access_rule = lambda_rule
        self.access_rules = [lambda_rule]

    def connect(self, region: Region) -> None:
        self.connected_region = region
        region.entrances.append(self)

    def disconnect(self) -> Optional[Region]:
        if self.connected_region is None:
            raise Exception(f"`disconnect()` called without a valid `connected_region` for entrance {self.name}.")
        self.connected_region.entrances.remove(self)
        previously_connected = self.connected_region
        self.connected_region = None
        return previously_connected

    def bind_two_way(self, other_entrance: Entrance) -> None:
        self.reverse = other_entrance
        other_entrance.reverse = self

    def get_new_target(self) -> Entrance:
        if self.world is None:
            raise Exception(f"`get_new_target()` called without a valid `world` for entrance {self.name}.")
        if self.connected_region is None:
            raise Exception(f"`get_new_target()` called without a valid `connected_region` for entrance {self.name}.")
        root = self.world.get_region('Root Exits')
        target_entrance = Entrance('Root -> ' + self.connected_region.name, root)
        target_entrance.connect(self.connected_region)
        target_entrance.replaces = self
        root.exits.append(target_entrance)
        return target_entrance

    def assume_reachable(self) -> Entrance:
        if self.assumed is None:
            self.assumed = self.get_new_target()
            self.disconnect()
        return self.assumed

    def __str__(self) -> str:
        return str(self.__unicode__())

    def __unicode__(self) -> str:
        return '%s' % self.name