from .key_types.layer import Hold, Toggle


class LayerManager:
    holds: list[Hold]
    toggles: list[Toggle]

    def __init__(self, number_of_layers: int):
        self.holds = [Hold(i) for i in range(0, number_of_layers)]
        self.toggles = [Toggle(i) for i in range(0, number_of_layers)]

    def toggle(self, layer) -> Toggle:
        return self.toggles[layer]

    def hold(self, layer) -> Hold:
        return self.holds[layer]