From 9809321f6ff4bf47d740c22ab8dd12056bb58637 Mon Sep 17 00:00:00 2001
From: diamondburned <diamond@arikawa-hi.me>
Date: Wed, 27 Dec 2023 19:19:23 -0800
Subject: [PATCH] handler: Add AllCallersForType for low-level control

This is needed in more silly main thread applications to prevent mutex
deadlocks from uncontrollable recursions.
---
 utils/handler/handler.go | 58 +++++++++++++++++++++++++++-------------
 1 file changed, 40 insertions(+), 18 deletions(-)

diff --git a/utils/handler/handler.go b/utils/handler/handler.go
index bc75676..55f029d 100644
--- a/utils/handler/handler.go
+++ b/utils/handler/handler.go
@@ -38,32 +38,48 @@ func New() *Handler {
 // Call calls all handlers with the given event. This is an internal method; use
 // with care.
 func (h *Handler) Call(ev interface{}) {
+	v := reflect.ValueOf(ev)
 	t := reflect.TypeOf(ev)
 
-	h.mutex.RLock()
-	defer h.mutex.RUnlock()
+	all := h.AllCallersForType(t)
+	all(func(caller Caller) bool {
+		caller.Call(v)
+		return true
+	})
+}
 
-	typedHandlers := h.events[t].Entries
-	anyHandlers := h.events[nil].Entries
+// AllCallersForType returns all callers for the given event type. This is an
+// internal method that is rarely useful for external use and should be used
+// with care.
+func (h *Handler) AllCallersForType(t reflect.Type) func(yield func(Caller) bool) {
+	return func(yield func(Caller) bool) {
+		h.mutex.RLock()
+		defer h.mutex.RUnlock()
 
-	if len(typedHandlers) == 0 && len(anyHandlers) == 0 {
-		return
-	}
+		typedHandlers := h.events[t].Entries
+		anyHandlers := h.events[nil].Entries
 
-	v := reflect.ValueOf(ev)
-
-	for _, entry := range typedHandlers {
-		if entry.isInvalid() {
-			continue
+		if len(typedHandlers) == 0 && len(anyHandlers) == 0 {
+			return
 		}
-		entry.Call(v)
-	}
 
-	for _, entry := range anyHandlers {
-		if entry.isInvalid() || entry.not(t) {
-			continue
+		for _, entry := range typedHandlers {
+			if entry.isInvalid() {
+				continue
+			}
+			if !yield(entry) {
+				return
+			}
+		}
+
+		for _, entry := range anyHandlers {
+			if entry.isInvalid() || entry.not(t) {
+				continue
+			}
+			if !yield(entry) {
+				return
+			}
 		}
-		entry.Call(v)
 	}
 }
 
@@ -239,6 +255,10 @@ func (h *Handler) addHandler(fn interface{}, sync bool) (rm func(), err error) {
 	}, nil
 }
 
+// Caller is an interface that can be used to call a handler.
+// It directly accepts a reflect.Value, which is the event.
+type Caller interface{ Call(ev reflect.Value) }
+
 type handler struct {
 	event     reflect.Type // underlying type; arg0 or chan underlying type
 	callback  reflect.Value
@@ -248,6 +268,8 @@ type handler struct {
 	isOnce    bool
 }
 
+var _ Caller = (*handler)(nil)
+
 // newHandler reflects either a channel or a function into a handler. A function
 // must only have a single argument being the event and no return, and a channel
 // must have the event type as the underlying type.