client_events.go

package obsws

import (
	"errors"
)

// ErrUnknownEventType is returned when a handler is added for an unknown event.
var ErrUnknownEventType = errors.New("unknown event type")

// AddEventHandler adds a handler function for a given event type.
func (c *Client) AddEventHandler(eventType string, handler func(Event)) error {
	if eventMap[eventType] == nil {
		return ErrUnknownEventType
	}
	c.handlers[eventType] = handler
	return nil
}

// MustAddEventHandler adds a handler function for a given event type. Panics if eventType is of an unknown type.
func (c *Client) MustAddEventHandler(eventType string, handler func(Event)) {
	err := c.AddEventHandler(eventType, handler)
	if err != nil {
		panic(err)
	}
}

// RemoveEventHandler removes the handler for a given event type.
func (c *Client) RemoveEventHandler(eventType string) {
	delete(c.handlers, eventType)
}

// handleEvent runs an event's handler if it exists.
func (c *Client) handleEvent(m map[string]interface{}) {
	t := m["update-type"].(string)

	eventFn, ok := eventMap[t]
	if !ok {
		Logger.Println("unknown event type:", m["update-type"])
		return
	}
	event := eventFn()

	handler, ok := c.handlers[t]
	if !ok {
		return
	}

	if err := mapToStruct(m, event); err != nil {
		Logger.Println("event handler failed:", err)
		return
	}

	go handler(derefEvent(event))
}