xmpp/filter.go
author Chris Jones <christian.jones@sri.com>
Sun, 08 Sep 2013 14:40:35 -0700
changeset 134 80b764fa2f08
parent 133 c4e0b897db83
child 136 f35f853a52b6
permissions -rw-r--r--
Fixed some logic errors and initialized some things that needed to be initialized.

package xmpp

// Manages the stack of filters that can read and modify stanzas on
// their way from the remote to the application.

// Receive new filters on filterAdd; those new filters get added to
// the top of the stack. Receive stanzas at the bottom of the stack on
// input. Send stanzas out the top of the stack on output.
func filterMgr(filterAdd <-chan Filter, input <-chan Stanza, output chan<- Stanza) {
	botFiltIn := output
	topFiltOut := input

loop:
	for {
		select {
		case stan, ok := <-input:
			if !ok {
				break loop
			}
			botFiltIn <- stan

		case stan, ok := <-topFiltOut:
			if !ok {
				break loop
			}
			output <- stan

		case filt := <-filterAdd:
			newTop := make(chan Stanza)
			go filt(topFiltOut, newTop)
			topFiltOut = newTop
		}
	}
	close(botFiltIn)
}

// AddRecvFilter adds a new filter to the top of the stack through which
// incoming stanzas travel on their way up to the client.
func (cl *Client) AddRecvFilter(filt Filter) {
	if filt == nil {
		return
	}
	cl.recvFilterAdd <- filt
}

// AddSendFilter adds a new filter to the top of the stack through
// which outgoing stanzas travel on their way down from the client to
// the network.
func (cl *Client) AddSendFilter(filt Filter) {
	if filt == nil {
		return
	}
	cl.sendFilterAdd <- filt
}