from kmk.kmktime import ticks_diff, ticks_ms


def df_pressed(key, state, *args, **kwargs):
    """Switches the default layer"""
    state.active_layers[0] = key.meta.layer
    state.reversed_active_layers = list(reversed(state.active_layers))
    return state


def mo_pressed(key, state, *args, **kwargs):
    """Momentarily activates layer, switches off when you let go"""
    state.active_layers.append(key.meta.layer)
    state.reversed_active_layers = list(reversed(state.active_layers))
    return state


def mo_released(key, state, KC, *args, **kwargs):
    state.active_layers = [
        layer for layer in state.active_layers
        if layer != key.meta.layer
    ]
    state.reversed_active_layers = list(reversed(state.active_layers))
    return state


def lm_pressed(key, state, *args, **kwargs):
    """As MO(layer) but with mod active"""
    state.hid_pending = True
    # Sets the timer start and acts like MO otherwise
    state.start_time['lm'] = ticks_ms()
    state.keys_pressed.add(key.meta.kc)
    return mo_pressed(key, state, *args, **kwargs)


def lm_released(key, state, *args, **kwargs):
    """As MO(layer) but with mod active"""
    state.hid_pending = True
    state.keys_pressed.discard(key.meta.kc)
    state.start_time['lm'] = None
    return mo_released(key, state, *args, **kwargs)


def lt_pressed(key, state, *args, **kwargs):
    # Sets the timer start and acts like MO otherwise
    state.start_time['lt'] = ticks_ms()
    return mo_pressed(key, state, *args, **kwargs)


def lt_released(key, state, *args, **kwargs):
    # On keyup, check timer, and press key if needed.
    if state.start_time['lt'] and (
        ticks_diff(ticks_ms(), state.start_time['lt']) < state.config.tap_time
    ):
        state.hid_pending = True
        state.tap_key(key.meta.kc)

    mo_released(key, state, *args, **kwargs)
    state.start_time['lt'] = None
    return state


def tg_pressed(key, state, *args, **kwargs):
    """Toggles the layer (enables it if not active, and vise versa)"""
    if key.meta.layer in state.active_layers:
        state.active_layers = [
            layer for layer in state.active_layers
            if layer != key.meta.layer
        ]
    else:
        state.active_layers.append(key.meta.layer)

    state.reversed_active_layers = list(reversed(state.active_layers))

    return state


def to_pressed(key, state, *args, **kwargs):
    """Activates layer and deactivates all other layers"""
    state.active_layers = [key.meta.layer]
    state.reversed_active_layers = list(reversed(state.active_layers))

    return state


def tt_pressed(key, state, *args, **kwargs):
    """Momentarily activates layer if held, toggles it if tapped repeatedly"""
    # TODO Make this work with tap dance to function more correctly, but technically works.
    if state.start_time['tt'] is None:
        # Sets the timer start and acts like MO otherwise
        state.start_time['tt'] = ticks_ms()
        return mo_pressed(key, state, *args, **kwargs)
    elif ticks_diff(ticks_ms(), state.start_time['tt']) < state.config.tap_time:
        state.start_time['tt'] = None
        return tg_pressed(key, state, *args, **kwargs)


def tt_released(key, state, *args, **kwargs):
    if (
        state.start_time['tt'] is None or
        ticks_diff(ticks_ms(), state.start_time['tt']) >= state.config.tap_time
    ):
        # On first press, works like MO. On second press, does nothing unless let up within
        # time window, then acts like TG.
        state.start_time['tt'] = None
        return mo_released(key, state, *args, **kwargs)

    return state