package coffee.liz.ecs.input

import coffee.liz.ecs.TickedSystem
import coffee.liz.ecs.World
import kotlin.time.Duration

data class InputState(
    val activeInputs: Set<String>,
)

interface InputProvider {
    val inputState: InputState
}

class InputSystem<Outside : InputProvider>(
    pollRate: Duration,
) : TickedSystem<Outside>(pollRate) {
    private val activeInputs = mutableMapOf<String, Int>()

    override fun update(
        world: World<Outside>,
        state: Outside,
        ticks: Int,
    ) {
        state.inputState.activeInputs.forEach {
            activeInputs[it] = (activeInputs[it] ?: 0) + ticks
        }
        activeInputs.keys.retainAll(state.inputState.activeInputs)

        world.query(Controlled::class).forEach {
            it.get(Controlled::class).handleInput(activeInputs.toMap())
        }
    }
}