98 lines
2.4 KiB
Go
98 lines
2.4 KiB
Go
package xmpp
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/bdlm/log"
|
|
"gosrc.io/xmpp"
|
|
)
|
|
|
|
func (out *Output) recvMessage(s xmpp.Sender, p xmpp.Packet) {
|
|
before := time.Now()
|
|
|
|
msg, ok := p.(xmpp.Message)
|
|
if !ok {
|
|
log.Errorf("blame gosrc.io/xmpp for routing: %s", p)
|
|
return
|
|
}
|
|
log.WithFields(map[string]interface{}{
|
|
"sender": msg.From,
|
|
"request": msg.Body,
|
|
}).Debug("handling bot message")
|
|
from, err := xmpp.NewJid(msg.From)
|
|
if err != nil {
|
|
log.Errorf("blame gosrc.io/xmpp for jid encoding: %s", msg.From)
|
|
return
|
|
}
|
|
|
|
fromBare := from.Bare()
|
|
fromLogmania := ""
|
|
if msg.Type == xmpp.MessageTypeGroupchat {
|
|
fromLogmania = protoGroup + ":" + fromBare
|
|
} else {
|
|
fromLogmania = proto + ":" + fromBare
|
|
}
|
|
|
|
answer := out.bot.Handle(fromLogmania, msg.Body)
|
|
if answer == "" {
|
|
return
|
|
}
|
|
reply := xmpp.Message{Attrs: xmpp.Attrs{To: fromBare, Type: msg.Type}, Body: answer}
|
|
s.Send(reply)
|
|
|
|
after := time.Now()
|
|
delta := after.Sub(before)
|
|
|
|
log.WithFields(map[string]interface{}{
|
|
"sender": fromLogmania,
|
|
"request": msg.Body,
|
|
"answer": answer,
|
|
"ms": float64(delta) / float64(time.Millisecond),
|
|
}).Debug("handled xmpp bot message")
|
|
}
|
|
|
|
func (out *Output) recvPresence(s xmpp.Sender, p xmpp.Packet) {
|
|
pres, ok := p.(xmpp.Presence)
|
|
if !ok {
|
|
log.Errorf("blame gosrc.io/xmpp for routing: %s", p)
|
|
return
|
|
}
|
|
from, err := xmpp.NewJid(pres.From)
|
|
if err != nil {
|
|
log.Errorf("blame gosrc.io/xmpp for jid encoding: %s", pres.From)
|
|
return
|
|
}
|
|
fromBare := from.Bare()
|
|
logPres := logger.WithField("from", from)
|
|
|
|
switch pres.Type {
|
|
case xmpp.PresenceTypeSubscribe:
|
|
logPres.Debugf("recv presence subscribe")
|
|
s.Send(xmpp.Presence{Attrs: xmpp.Attrs{
|
|
Type: xmpp.PresenceTypeSubscribed,
|
|
To: fromBare,
|
|
Id: pres.Id,
|
|
}})
|
|
logPres.Debugf("accept new subscribe")
|
|
|
|
s.Send(xmpp.Presence{Attrs: xmpp.Attrs{
|
|
Type: xmpp.PresenceTypeSubscribe,
|
|
To: fromBare,
|
|
}})
|
|
logPres.Info("request also subscribe")
|
|
case xmpp.PresenceTypeSubscribed:
|
|
logPres.Info("recv presence accepted subscribe")
|
|
case xmpp.PresenceTypeUnsubscribe:
|
|
logPres.Info("recv presence remove subscribe")
|
|
case xmpp.PresenceTypeUnsubscribed:
|
|
logPres.Info("recv presence removed subscribe")
|
|
case xmpp.PresenceTypeUnavailable:
|
|
logPres.Debug("recv presence unavailable")
|
|
case "":
|
|
logPres.Debug("recv empty presence, maybe from joining muc")
|
|
return
|
|
default:
|
|
logPres.Warnf("recv presence unsupported: %s -> %s", pres.Type, pres)
|
|
}
|
|
}
|