Add support for downstream batch & chathistory

This commit is contained in:
Simon Ser 2020-04-15 11:29:15 +02:00
parent 2194259124
commit 8e852cc7e4
No known key found for this signature in database
GPG Key ID: 0FDE7BE0E88F5E48
1 changed files with 24 additions and 2 deletions

View File

@ -427,7 +427,7 @@ func (dc *downstreamConn) handleCapCommand(cmd string, args []string) error {
}
}
caps := []string{"message-tags", "server-time", "echo-message"}
caps := []string{"message-tags", "server-time", "echo-message", "batch"}
if dc.capVersion >= 302 {
caps = append(caps, "sasl=PLAIN")
@ -480,7 +480,7 @@ func (dc *downstreamConn) handleCapCommand(cmd string, args []string) error {
}
switch name {
case "sasl", "message-tags", "server-time", "echo-message":
case "sasl", "message-tags", "server-time", "echo-message", "batch":
dc.caps[name] = enable
default:
ack = false
@ -716,6 +716,15 @@ func (dc *downstreamConn) sendNetworkHistory(net *network) {
return
}
batchRef := "history"
if dc.caps["batch"] {
dc.SendMessage(&irc.Message{
Prefix: dc.srv.prefix(),
Command: "BATCH",
Params: []string{"+" + batchRef, "chathistory", dc.marshalEntity(uc, target)},
})
}
for {
msg := consumer.Consume()
if msg == nil {
@ -734,8 +743,21 @@ func (dc *downstreamConn) sendNetworkHistory(net *network) {
continue
}
if dc.caps["batch"] {
msg = msg.Copy()
msg.Tags["batch"] = irc.TagValue(batchRef)
}
dc.SendMessage(dc.marshalMessage(msg, uc))
}
if dc.caps["batch"] {
dc.SendMessage(&irc.Message{
Prefix: dc.srv.prefix(),
Command: "BATCH",
Params: []string{"-" + batchRef},
})
}
}
}