Add very initial snomasks

irc/snomanager.go

@@ -0,0 +1,117 @@
+package irc
+
+import (
+	"fmt"
+	"sync"
+
+	"github.com/DanielOaks/girc-go/ircfmt"
+	"github.com/DanielOaks/oragono/irc/sno"
+)
+
+// SnoManager keeps track of which clients to send snomasks to.
+type SnoManager struct {
+	sendListMutex sync.RWMutex
+	sendLists     map[sno.Mask]map[*Client]bool
+}
+
+// NewSnoManager returns a new SnoManager
+func NewSnoManager() *SnoManager {
+	var m SnoManager
+	m.sendLists = make(map[sno.Mask]map[*Client]bool)
+	return &m
+}
+
+// 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 currentClientList == nil || 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 currentClientList == nil || 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 currentClientList == nil || 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)
+	}
+}
+
+// String returns the snomasks currently enabled.
+func (m *SnoManager) String(client *Client) string {
+	m.sendListMutex.RLock()
+	defer m.sendListMutex.RUnlock()
+
+	var masks string
+	for mask, clients := range m.sendLists {
+		for c := range clients {
+			if c == client {
+				masks += string(mask)
+				break
+			}
+		}
+	}
+	return masks
+}