mirror of https://github.com/ergochat/ergo.git
120 lines
2.5 KiB
Go
120 lines
2.5 KiB
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()
|
|
}
|