1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
package proxy
import (
"sync"
"github.com/HimbeerserverDE/mt"
)
var (
modChanSubscribers map[string][]*ClientConn
modChanSubscriberMu sync.RWMutex
)
var (
onCltModChanMsg []func(string, *ClientConn, string) bool
onCltModChanMsgMu sync.RWMutex
onCltModChanMsgOnce sync.Once
)
// SendModChanMsg sends a message to all subscribed clients on a modchannel.
func SendModChanMsg(channel, msg string) {
modChanSubscriberMu.RLock()
defer modChanSubscriberMu.RUnlock()
subs, _ := modChanSubscribers[channel]
for _, cc := range subs {
cc.SendCmd(&mt.ToCltModChanMsg{
Channel: channel,
Msg: msg,
})
}
}
// RegisterOnCltModChanMsg registers a handler that is called
// when a client sends a message on a modchannel.
// If any handler returns true, the message is not forwarded
// to the upstream server.
func RegisterOnCltModChanMsg(handler func(string, *ClientConn, string) bool) {
onCltModChanMsgMu.Lock()
defer onCltModChanMsgMu.Unlock()
onCltModChanMsg = append(onCltModChanMsg, handler)
}
func cltLeaveModChan(cc *ClientConn, channel string) {
modChanSubscriberMu.Lock()
defer modChanSubscriberMu.Unlock()
for i, sub := range modChanSubscribers[channel] {
if sub == cc {
modChanSubscribers[channel] = append(modChanSubscribers[channel][:i], modChanSubscribers[channel][i+1:]...)
break
}
}
}
func cltLeaveModChans(cc *ClientConn) {
modChanSubscriberMu.Lock()
defer modChanSubscriberMu.Unlock()
for ch := range modChanSubscribers {
for i, sub := range modChanSubscribers[ch] {
if sub == cc {
modChanSubscribers[ch] = append(modChanSubscribers[ch][:i], modChanSubscribers[ch][i+1:]...)
break
}
}
}
}
func handleCltModChanMsg(cc *ClientConn, cmd *mt.ToSrvMsgModChan) bool {
onCltModChanMsgMu.RLock()
defer onCltModChanMsgMu.RUnlock()
subs, _ := modChanSubscribers[cmd.Channel]
for _, sub := range subs {
sub.SendCmd(&mt.ToCltModChanMsg{
Channel: cmd.Channel,
Sender: cc.Name(),
Msg: cmd.Msg,
})
}
drop := false
for _, handler := range onCltModChanMsg {
if handler(cmd.Channel, cc, cmd.Msg) {
drop = true
}
}
return drop
}
func init() {
modChanSubscribers = make(map[string][]*ClientConn)
}
|
