Grinch-AP/worlds/noita/rules.py

from typing import List, NamedTuple, Set, TYPE_CHECKING

from BaseClasses import CollectionState
from . import items, locations
from .options import BossesAsChecks, VictoryCondition
from worlds.generic import Rules as GenericRules

if TYPE_CHECKING:
    from . import NoitaWorld


class EntranceLock(NamedTuple):
    source: str
    destination: str
    event: str
    items_needed: int


entrance_locks: List[EntranceLock] = [
    EntranceLock("Mines", "Coal Pits Holy Mountain", "Portal to Holy Mountain 1", 1),
    EntranceLock("Coal Pits", "Snowy Depths Holy Mountain", "Portal to Holy Mountain 2", 2),
    EntranceLock("Snowy Depths", "Hiisi Base Holy Mountain", "Portal to Holy Mountain 3", 3),
    EntranceLock("Hiisi Base", "Underground Jungle Holy Mountain", "Portal to Holy Mountain 4", 4),
    EntranceLock("Underground Jungle", "Vault Holy Mountain", "Portal to Holy Mountain 5", 5),
    EntranceLock("The Vault", "Temple of the Art Holy Mountain", "Portal to Holy Mountain 6", 6),
    EntranceLock("Temple of the Art", "Laboratory Holy Mountain", "Portal to Holy Mountain 7", 7),
]


holy_mountain_regions: List[str] = [
    "Coal Pits Holy Mountain",
    "Snowy Depths Holy Mountain",
    "Hiisi Base Holy Mountain",
    "Underground Jungle Holy Mountain",
    "Vault Holy Mountain",
    "Temple of the Art Holy Mountain",
    "Laboratory Holy Mountain",
]


wand_tiers: List[str] = [
    "Wand (Tier 1)",    # Coal Pits
    "Wand (Tier 2)",    # Snowy Depths
    "Wand (Tier 3)",    # Hiisi Base
    "Wand (Tier 4)",    # Underground Jungle
    "Wand (Tier 5)",    # The Vault
    "Wand (Tier 6)",    # Temple of the Art
]


items_hidden_from_shops: Set[str] = {"Gold (200)", "Gold (1000)", "Potion", "Random Potion", "Secret Potion",
                                     "Chaos Die", "Greed Die", "Kammi", "Refreshing Gourd", "Sädekivi", "Broken Wand",
                                     "Powder Pouch"}

perk_list: List[str] = list(filter(items.item_is_perk, items.item_table.keys()))


# ----------------
# Helper Functions
# ----------------


def has_perk_count(state: CollectionState, player: int, amount: int) -> bool:
    return sum(state.count(perk, player) for perk in perk_list) >= amount


def has_orb_count(state: CollectionState, player: int, amount: int) -> bool:
    return state.count("Orb", player) >= amount


def forbid_items_at_locations(world: "NoitaWorld", shop_locations: Set[str], forbidden_items: Set[str]) -> None:
    for shop_location in shop_locations:
        location = world.multiworld.get_location(shop_location, world.player)
        GenericRules.forbid_items_for_player(location, forbidden_items, world.player)


# ----------------
# Rule Functions
# ----------------


# Prevent gold and potions from appearing as purchasable items in shops (because physics will destroy them)
# def ban_items_from_shops(world: "NoitaWorld") -> None:
#     for location_name in Locations.location_name_to_id.keys():
#         if "Shop Item" in location_name:
#             forbid_items_at_location(world, location_name, items_hidden_from_shops)
def ban_items_from_shops(world: "NoitaWorld") -> None:
    forbid_items_at_locations(world, locations.shop_locations, items_hidden_from_shops)


# Prevent high tier wands from appearing in early Holy Mountain shops
def ban_early_high_tier_wands(world: "NoitaWorld") -> None:
    for i, region_name in enumerate(holy_mountain_regions):
        wands_to_forbid = set(wand_tiers[i+1:])

        locations_in_region = set(locations.location_region_mapping[region_name].keys())
        forbid_items_at_locations(world, locations_in_region, wands_to_forbid)

    # Prevent high tier wands from appearing in the Secret shop
    wands_to_forbid = set(wand_tiers[3:])
    locations_in_region = set(locations.location_region_mapping["Secret Shop"].keys())
    forbid_items_at_locations(world, locations_in_region, wands_to_forbid)


def lock_holy_mountains_into_spheres(world: "NoitaWorld") -> None:
    for lock in entrance_locks:
        location = world.multiworld.get_entrance(f"From {lock.source} To {lock.destination}", world.player)
        GenericRules.set_rule(location, lambda state, evt=lock.event: state.has(evt, world.player))


def holy_mountain_unlock_conditions(world: "NoitaWorld") -> None:
    victory_condition = world.options.victory_condition.value
    for lock in entrance_locks:
        location = world.multiworld.get_location(lock.event, world.player)

        if victory_condition == VictoryCondition.option_greed_ending:
            location.access_rule = lambda state, items_needed=lock.items_needed: (
                has_perk_count(state, world.player, items_needed//2)
            )
        elif victory_condition == VictoryCondition.option_pure_ending:
            location.access_rule = lambda state, items_needed=lock.items_needed: (
                has_perk_count(state, world.player, items_needed//2) and
                has_orb_count(state, world.player, items_needed)
            )
        elif victory_condition == VictoryCondition.option_peaceful_ending:
            location.access_rule = lambda state, items_needed=lock.items_needed: (
                has_perk_count(state, world.player, items_needed//2) and
                has_orb_count(state, world.player, items_needed * 3)
            )


def biome_unlock_conditions(world: "NoitaWorld") -> None:
    lukki_entrances = world.multiworld.get_region("Lukki Lair", world.player).entrances
    magical_entrances = world.multiworld.get_region("Magical Temple", world.player).entrances
    wizard_entrances = world.multiworld.get_region("Wizards' Den", world.player).entrances
    for entrance in lukki_entrances:
        entrance.access_rule = lambda state: state.has("Melee Immunity Perk", world.player) and\
                                             state.has("All-Seeing Eye Perk", world.player)
    for entrance in magical_entrances:
        entrance.access_rule = lambda state: state.has("All-Seeing Eye Perk", world.player)
    for entrance in wizard_entrances:
        entrance.access_rule = lambda state: state.has("All-Seeing Eye Perk", world.player)


def victory_unlock_conditions(world: "NoitaWorld") -> None:
    victory_condition = world.options.victory_condition.value
    victory_location = world.multiworld.get_location("Victory", world.player)

    if victory_condition == VictoryCondition.option_pure_ending:
        victory_location.access_rule = lambda state: has_orb_count(state, world.player, 11)
    elif victory_condition == VictoryCondition.option_peaceful_ending:
        victory_location.access_rule = lambda state: has_orb_count(state, world.player, 33)


# ----------------
# Main Function
# ----------------


def create_all_rules(world: "NoitaWorld") -> None:
    if world.multiworld.players > 1:
        ban_items_from_shops(world)
        ban_early_high_tier_wands(world)
        lock_holy_mountains_into_spheres(world)
        holy_mountain_unlock_conditions(world)
        biome_unlock_conditions(world)
    victory_unlock_conditions(world)

    # Prevent the Map perk (used to find Toveri) from being on Toveri (boss)
    if world.options.bosses_as_checks.value >= BossesAsChecks.option_all_bosses:
        toveri = world.multiworld.get_location("Toveri", world.player)
        GenericRules.forbid_items_for_player(toveri, {"Spatial Awareness Perk"}, world.player)