irc/event.go

package irc

import (
	"context"
	"encoding/json"
	"time"
)

// An Event is any thing that passes through the irc client's event loop. It's not thread safe, because it's processed
// in sequence and should not be used off the goroutine that processed it.
type Event struct {
	kind string
	verb string
	name string

	Time       time.Time
	Nick       string
	User       string
	Host       string
	Args       []string
	Text       string
	Tags       map[string]string
	RenderTags map[string]string

	ctx              context.Context
	cancel           context.CancelFunc
	preventedDefault bool
	hidden           bool

	targets []Target
}

// NewEvent makes a new event with Kind, Verb, Time set and Args and Tags initialized.
func NewEvent(kind, verb string) Event {
	return Event{
		kind: kind,
		verb: verb,
		name: kind + "." + verb,

		Time: time.Now(),
		Args: make([]string, 0, 4),
		Tags: make(map[string]string),

		RenderTags: make(map[string]string),
	}
}

// Kind gets the event's kind
func (event *Event) Kind() string {
	return event.kind
}

// Verb gets the event's verb
func (event *Event) Verb() string {
	return event.verb
}

// Name gets the event name, which is Kind and Verb separated by a dot.
func (event *Event) Name() string {
	return event.name
}

// IsEither returns true if the event has the kind and one of the verbs.
func (event *Event) IsEither(kind string, verbs ...string) bool {
	if event.kind != kind {
		return false
	}

	for i := range verbs {
		if event.verb == verbs[i] {
			return true
		}
	}

	return false
}

// Context gets the event's context if it's part of the loop, or `context.Background` otherwise. client.Emit
// will set this context on its copy and return it.
func (event *Event) Context() context.Context {
	if event.ctx == nil {
		return context.Background()
	}

	return event.ctx
}

// PreventDefault stops propagation of the event. The context will be killed once
// the current event handler returns.
//
// A use case for this is to prevent the default input handler from firing
// on an already processed input event.
func (event *Event) PreventDefault() {
	event.preventedDefault = true
}

// Hide will not stop propagation, but it will allow output handlers to know not to
// render it.
func (event *Event) Hide() {
	event.hidden = true
}

// Hidden returns true if Hide has been called.
func (event *Event) Hidden() bool {
	return event.hidden
}

// Arg gets the argument by index, counting the trailing as the last argument. The rationale
// behind it is that some servers may use it for the last argument in JOINs and such.
func (event *Event) Arg(index int) string {
	if index < 0 || index > len(event.Args) {
		return ""
	}

	if index == len(event.Args) {
		return event.Text
	}

	return event.Args[index]
}

// Target finds the first target with one of the kinds specified. If none
// are specified, the first target will be returned. If more than one
// is provided, the first kinds are prioritized.
func (event *Event) Target(kinds ...string) Target {
	if len(event.targets) == 0 {
		return nil
	}

	if len(kinds) == 0 || kinds[0] == "" {
		return event.targets[0]
	}

	for _, kind := range kinds {
		for _, target := range event.targets {
			if target.Kind() == kind {
				return target
			}
		}
	}

	return nil
}

// ChannelTarget gets the first channel target.
func (event *Event) ChannelTarget() *Channel {
	target := event.Target("channel")
	if target == nil {
		return nil
	}

	return target.(*Channel)
}

// QueryTarget gets the first query target.
func (event *Event) QueryTarget() *Query {
	target := event.Target("query")
	if target == nil {
		return nil
	}

	return target.(*Query)
}

// StatusTarget gets the first status target.
func (event *Event) StatusTarget() *Status {
	target := event.Target("status")
	if target == nil {
		return nil
	}

	return target.(*Status)
}

func (event *Event) TargetIDs() []string {
	ids := make([]string, len(event.targets))
	for _, target := range event.targets {
		ids = append(ids, target.ID())
	}

	return ids
}

// MarshalJSON makes a JSON object from the event.
func (event *Event) MarshalJSON() ([]byte, error) {
	data := eventJSONData{
		Name:       event.Name(),
		Kind:       event.kind,
		Verb:       event.verb,
		Time:       event.Time,
		Nick:       event.Nick,
		User:       event.User,
		Host:       event.Host,
		Args:       event.Args,
		Text:       event.Text,
		Tags:       event.Tags,
		RenderTags: event.RenderTags,
	}

	data.Targets = make([]string, 0, len(event.targets))
	for _, target := range event.targets {
		data.Targets = append(data.Targets, target.ID())
	}

	return json.Marshal(data)
}

func (event *Event) Copy() *Event {
	eventCopy := *event
	if len(event.Args) > 0 {
		eventCopy.Args = append(event.Args[:0:0], event.Args...)
	}
	if len(event.Tags) > 0 {
		eventCopy.Tags = make(map[string]string, len(event.Tags))
		for key, value := range event.Tags {
			eventCopy.Tags[key] = value
		}
	}
	if len(event.RenderTags) > 0 {
		eventCopy.RenderTags = make(map[string]string, len(event.RenderTags))
		for key, value := range event.RenderTags {
			eventCopy.RenderTags[key] = value
		}
	}
	if len(event.targets) > 0 {
		eventCopy.targets = append(event.targets[:0:0], event.targets...)
	}

	return &eventCopy
}

type eventJSONData struct {
	Name       string            `json:"name"`
	Kind       string            `json:"kind"`
	Verb       string            `json:"verb"`
	Time       time.Time         `json:"time"`
	Nick       string            `json:"nick"`
	User       string            `json:"user"`
	Host       string            `json:"host"`
	Args       []string          `json:"args"`
	Text       string            `json:"text"`
	Tags       map[string]string `json:"tags"`
	Targets    []string          `json:"targets"`
	RenderTags map[string]string `json:"renderTags"`
}