src/main/kotlin/moe/nea/firmament/util/async/input.kt


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

package moe.nea.firmament.util.async

import kotlinx.coroutines.suspendCancellableCoroutine
import kotlin.coroutines.resume
import moe.nea.firmament.events.HandledScreenKeyPressedEvent
import moe.nea.firmament.keybindings.IKeyBinding

private object InputHandler {
    data class KeyInputContinuation(val keybind: IKeyBinding, val onContinue: () -> Unit)

    private val activeContinuations = mutableListOf<KeyInputContinuation>()

    fun registerContinuation(keyInputContinuation: KeyInputContinuation): () -> Unit {
        synchronized(InputHandler) {
            activeContinuations.add(keyInputContinuation)
        }
        return {
            synchronized(this) {
                activeContinuations.remove(keyInputContinuation)
            }
        }
    }

    init {
        HandledScreenKeyPressedEvent.subscribe { event ->
            synchronized(InputHandler) {
                val toRemove = activeContinuations.filter {
                    event.matches(it.keybind)
                }
                toRemove.forEach { it.onContinue() }
                activeContinuations.removeAll(toRemove)
            }
        }
    }
}

suspend fun waitForInput(keybind: IKeyBinding): Unit = suspendCancellableCoroutine { cont ->
    val unregister =
        InputHandler.registerContinuation(InputHandler.KeyInputContinuation(keybind) { cont.resume(Unit) })
    cont.invokeOnCancellation {
        unregister()
    }
}