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, i=i: _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("NPC Trinket", 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)