aboutsummaryrefslogtreecommitdiff
path: root/src/main/kotlin/events/FirmamentEventBus.kt
blob: 71331d159b90f959b08879d067edf3764684ee2c (plain)
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
package moe.nea.firmament.events

import java.util.concurrent.CopyOnWriteArrayList
import org.apache.commons.lang3.reflect.TypeUtils
import moe.nea.firmament.Firmament
import moe.nea.firmament.util.MC

/**
 * A pubsub event bus.
 *
 * [subscribe] to events [publish]ed on this event bus.
 * Subscriptions may not necessarily be delivered in the order of registering.
 */
open class FirmamentEventBus<T : FirmamentEvent> {
    companion object {
        val allEventBuses = mutableListOf<FirmamentEventBus<*>>()
    }

    val eventType = TypeUtils.getTypeArguments(javaClass, FirmamentEventBus::class.java)!!.values.single()

    init {
        allEventBuses.add(this)
    }

    data class Handler<T>(
        val invocation: (T) -> Unit, val receivesCancelled: Boolean,
        var knownErrors: MutableSet<Class<*>> = mutableSetOf(),
        val label: String,
    )

    private val toHandle: MutableList<Handler<T>> = CopyOnWriteArrayList()
    val handlers: List<Handler<T>> get() = toHandle

    fun subscribe(label: String, handle: (T) -> Unit) {
        subscribe(false, label, handle)
    }

    fun subscribe(receivesCancelled: Boolean, label: String, handle: (T) -> Unit) {
        toHandle.add(Handler(handle, receivesCancelled, label = label))
    }

    fun publish(event: T): T {
        for (function in toHandle) {
            if (function.receivesCancelled || event !is FirmamentEvent.Cancellable || !event.cancelled) {
                try {
                    function.invocation(event)
                } catch (e: Exception) {
                    val klass = e.javaClass
                    if (!function.knownErrors.contains(klass) || Firmament.DEBUG) {
                        function.knownErrors.add(klass)
                        Firmament.logger.error("Caught exception during processing event $event by $function", e)
                    }
                }
            }
        }
        return event
    }

    fun publishSync(event: T) {
        MC.onMainThread {
            publish(event)
        }
    }
}