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], area_cost_map: 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})
    if world.AreaCostRandomizer[player].value:
        world.random.shuffle(areashuffle)
    area_cost_map.update({(index+1): (value+1) for index, value in enumerate(areashuffle)})
    area_cost_map.update({0:0})

    for i in range(1,5):
        connect_regions(world, player, "Menu", v6areas[area_connections[i]-1], lambda state: _has_trinket_range(state,player,world.DoorCost[player].value*(area_cost_map[i]-1),
                                                                                                                             world.DoorCost[player].value*area_cost_map[i]))

    #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("V",player), lambda state : state.can_reach("Laboratory",'Region',player) or
                                                            ( state.can_reach("The Tower",'Region',player) and
                                                              state.can_reach("Space Station 2",'Region',player) and
                                                              state.can_reach("Warp Zone",'Region',player) ))

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