from collections.abc import Callable
from typing import TYPE_CHECKING

from BaseClasses import CollectionState
from .player_logic import AccessRequirements

if TYPE_CHECKING:
    from . import Lingo2World


def lingo2_can_satisfy_requirements(state: CollectionState, reqs: AccessRequirements, world: "Lingo2World") -> bool:
    if not all(state.has(item, world.player) for item in reqs.items):
        return False

    if not all(state.has(item, world.player, amount) for item, amount in reqs.progressives.items()):
        return False

    if not all(state.can_reach_region(region_name, world.player) for region_name in reqs.rooms):
        return False

    for letter_key, letter_level in reqs.letters.items():
        if not state.has(letter_key, world.player, letter_level):
            return False

    if reqs.cyans:
        if not any(state.has(letter, world.player, amount)
                   for letter, amount in world.player_logic.double_letter_amount.items()):
            return False

    if len(reqs.or_logic) > 0:
        if not all(any(lingo2_can_satisfy_requirements(state, sub_reqs, world) for sub_reqs in subjunction)
                   for subjunction in reqs.or_logic):
            return False

    if reqs.complete_at is not None:
        completed = 0
        checked = 0
        for possibility in reqs.possibilities:
            checked += 1
            if lingo2_can_satisfy_requirements(state, possibility, world):
                completed += 1
                if completed >= reqs.complete_at:
                    break
            elif len(reqs.possibilities) - checked + completed < reqs.complete_at:
                # There aren't enough remaining possibilities for the check to pass.
                return False
        if completed < reqs.complete_at:
            return False

    return True

def make_location_lambda(reqs: AccessRequirements, world: "Lingo2World") -> Callable[[CollectionState], bool]:
    return lambda state: lingo2_can_satisfy_requirements(state, reqs, world)