Archipelago/worlds/v6/Rules.py

import typing
from ..generic.Rules import add_rule
from .Regions import connect_regions, v6areas


def _has_trinket_range(state,player,start,end) -> bool:
    for i in range(start,end):
        if (not state.has("Trinket " + str(i+1).zfill(2), player)):
            return False
    return True


def set_rules(world, player, area_connections: typing.Dict[int, int]):
    areashuffle = list(range(len(v6areas)))
    if world.AreaRandomizer[player].value:
        world.random.shuffle(areashuffle)
    area_connections.update({(index+1): (value+1) for index, value in enumerate(areashuffle)})
    area_connections.update({0:0})

    connect_regions(world, player, "Menu", v6areas[area_connections[1]-1], lambda state: _has_trinket_range(state,player,0,world.DoorCost[player].value))
    connect_regions(world, player, "Menu", v6areas[area_connections[2]-1], lambda state: _has_trinket_range(state,player,world.DoorCost[player].value,world.DoorCost[player].value*2))
    connect_regions(world, player, "Menu", v6areas[area_connections[3]-1], lambda state: _has_trinket_range(state,player,world.DoorCost[player].value*2,world.DoorCost[player].value*3))
    connect_regions(world, player, "Menu", v6areas[area_connections[4]-1], lambda state: _has_trinket_range(state,player,world.DoorCost[player].value*3,world.DoorCost[player].value*4))

    #Special Rule for V
    add_rule(world.get_location("V",player), lambda state : state.can_reach("Laboratory",'Region',player) and
                                                            state.can_reach("The Tower",'Region',player) and
                                                            state.can_reach("Space Station 2",'Region',player) and
                                                            state.can_reach("Warp Zone",'Region',player))

    #Special Rule for NPC Trinket
    add_rule(world.get_location("NPC Trinket",player), lambda state: state.can_reach("Laboratory",'Region',player) or 
                                                                     state.can_reach("Space Station 2",'Region',player))

    world.completion_condition[player] = lambda state: state.can_reach("V",'Location',player)