Add downstream support for server-time

This commit is contained in:
Simon Ser 2020-03-31 19:50:31 +02:00
parent 2053e62162
commit 743540665b
No known key found for this signature in database
GPG Key ID: 0FDE7BE0E88F5E48
1 changed files with 14 additions and 5 deletions

View File

@ -444,15 +444,14 @@ func (dc *downstreamConn) handleCapCommand(cmd string, args []string) error {
} }
} }
var caps []string caps := []string{"message-tags", "server-time"}
if dc.capVersion >= 302 { if dc.capVersion >= 302 {
caps = append(caps, "sasl=PLAIN") caps = append(caps, "sasl=PLAIN")
} else { } else {
caps = append(caps, "sasl") caps = append(caps, "sasl")
} }
caps = append(caps, "message-tags")
// TODO: multi-line replies // TODO: multi-line replies
dc.SendMessage(&irc.Message{ dc.SendMessage(&irc.Message{
Prefix: dc.srv.prefix(), Prefix: dc.srv.prefix(),
@ -498,7 +497,7 @@ func (dc *downstreamConn) handleCapCommand(cmd string, args []string) error {
} }
switch name { switch name {
case "sasl", "message-tags": case "sasl", "message-tags", "server-time":
dc.caps[name] = enable dc.caps[name] = enable
default: default:
ack = false ack = false
@ -733,6 +732,7 @@ func (dc *downstreamConn) runNetwork(net *network, loadHistory bool) {
// TODO: can't be enabled/disabled on-the-fly // TODO: can't be enabled/disabled on-the-fly
msgTagsEnabled := dc.caps["message-tags"] msgTagsEnabled := dc.caps["message-tags"]
serverTimeEnabled := dc.caps["server-time"]
consumer, ch := net.ring.NewConsumer(seqPtr) consumer, ch := net.ring.NewConsumer(seqPtr)
go func() { go func() {
@ -773,7 +773,16 @@ func (dc *downstreamConn) runNetwork(net *network, loadHistory bool) {
} }
if !msgTagsEnabled { if !msgTagsEnabled {
msg.Tags = nil for name := range msg.Tags {
supported := false
switch name {
case "time":
supported = serverTimeEnabled
}
if !supported {
delete(msg.Tags, name)
}
}
} }
dc.SendMessage(msg) dc.SendMessage(msg)