from typing import Callable

from BaseClasses import CollectionState, MultiWorld


def get_button_rule(multiworld: MultiWorld, player: int) -> Callable[[CollectionState], bool]:
    if getattr(multiworld, "hard_mode")[player]:
        return lambda state: state.has("Button Activation", player)

    return lambda state: True