ergo/irc/snomanager.go

package irc

import (
	"fmt"
	"sync"

	"github.com/ergochat/ergo/irc/sno"
	"github.com/ergochat/irc-go/ircfmt"
)

// SnoManager keeps track of which clients to send snomasks to.
type SnoManager struct {
	sendListMutex sync.RWMutex // tier 2
	sendLists     map[sno.Mask]map[*Client]bool
}

func (m *SnoManager) Initialize() {
	m.sendLists = make(map[sno.Mask]map[*Client]bool)
}

// AddMasks adds the given snomasks to the client.
func (m *SnoManager) AddMasks(client *Client, masks ...sno.Mask) {
	m.sendListMutex.Lock()
	defer m.sendListMutex.Unlock()

	for _, mask := range masks {
		currentClientList := m.sendLists[mask]

		if currentClientList == nil {
			currentClientList = map[*Client]bool{}
		}

		currentClientList[client] = true

		m.sendLists[mask] = currentClientList
	}
}

// RemoveMasks removes the given snomasks from the client.
func (m *SnoManager) RemoveMasks(client *Client, masks ...sno.Mask) {
	m.sendListMutex.Lock()
	defer m.sendListMutex.Unlock()

	for _, mask := range masks {
		currentClientList := m.sendLists[mask]

		if len(currentClientList) == 0 {
			continue
		}

		delete(currentClientList, client)

		m.sendLists[mask] = currentClientList
	}
}

// RemoveClient removes the given client from all of our lists.
func (m *SnoManager) RemoveClient(client *Client) {
	m.sendListMutex.Lock()
	defer m.sendListMutex.Unlock()

	for mask := range m.sendLists {
		currentClientList := m.sendLists[mask]

		if len(currentClientList) == 0 {
			continue
		}

		delete(currentClientList, client)

		m.sendLists[mask] = currentClientList
	}
}

// Send sends the given snomask to all users signed up for it.
func (m *SnoManager) Send(mask sno.Mask, content string) {
	m.sendListMutex.RLock()
	defer m.sendListMutex.RUnlock()

	currentClientList := m.sendLists[mask]

	if len(currentClientList) == 0 {
		return
	}

	// make the message
	name := sno.NoticeMaskNames[mask]
	if name == "" {
		name = string(mask)
	}
	message := fmt.Sprintf(ircfmt.Unescape("$c[grey]-$r%s$c[grey]-$c %s"), name, content)

	// send it out
	for client := range currentClientList {
		client.Notice(message)
	}
}

// MasksEnabled returns the snomasks currently enabled.
func (m *SnoManager) MasksEnabled(client *Client) (result sno.Masks) {
	m.sendListMutex.RLock()
	defer m.sendListMutex.RUnlock()

	for mask, clients := range m.sendLists {
		for c := range clients {
			if c == client {
				result = append(result, mask)
				break
			}
		}
	}
	result.Sort()
	return
}

func (m *SnoManager) String(client *Client) string {
	masks := m.MasksEnabled(client)
	return masks.String()
}