fix changes xmpp lib and log error on send

This commit is contained in:
Martin/Geno 2019-06-28 01:48:15 +02:00
parent 3db85b107d
commit e44fb17f40
No known key found for this signature in database
GPG Key ID: 9D7D3C6BFF600C6A
3 changed files with 66 additions and 45 deletions

View File

@ -84,12 +84,12 @@ func Init(configInterface interface{}, db *database.DB, bot *bot.Bot) output.Out
} }
} }
logger.Info("join muc after connect") logger.Info("join muc after connect")
out.client = c
}) })
go func() { go func() {
cm.Run() cm.Run()
log.Panic("closed connection") log.Panic("closed connection")
}() }()
out.client = client
logger.WithField("jid", config.JID).Info("startup") logger.WithField("jid", config.JID).Info("startup")
return out return out
@ -106,11 +106,10 @@ func (out *Output) Close() {
logger.Error("xmpp could generate jid to leave ", jid, " error:", err) logger.Error("xmpp could generate jid to leave ", jid, " error:", err)
} }
toJID.Resource = nickname toJID.Resource = nickname
err = out.client.Send(stanza.Presence{Attrs: stanza.Attrs{ if err = out.client.Send(stanza.Presence{Attrs: stanza.Attrs{
To: toJID.Full(), To: toJID.Full(),
Type: stanza.PresenceTypeUnavailable, Type: stanza.PresenceTypeUnavailable,
}}) }}); err != nil {
if err != nil {
logger.Error("xmpp could not leave ", toJID.Full(), " error:", err) logger.Error("xmpp could not leave ", toJID.Full(), " error:", err)
} }
} }

View File

@ -3,31 +3,32 @@ package xmpp
import ( import (
"time" "time"
"github.com/bdlm/log"
"gosrc.io/xmpp" "gosrc.io/xmpp"
"gosrc.io/xmpp/stanza"
) )
func (out *Output) recvMessage(s xmpp.Sender, p xmpp.Packet) { func (out *Output) recvMessage(s xmpp.Sender, p stanza.Packet) {
before := time.Now() before := time.Now()
msg, ok := p.(xmpp.Message) msg, ok := p.(stanza.Message)
if !ok { if !ok {
log.Errorf("blame gosrc.io/xmpp for routing: %s", p) logger.Errorf("blame gosrc.io/xmpp for routing: %s", p)
return return
} }
log.WithFields(map[string]interface{}{ logger.WithFields(map[string]interface{}{
"sender": msg.From, "sender": msg.From,
"request": msg.Body, "request": msg.Body,
}).Debug("handling bot message") }).Debug("handling bot message")
from, err := xmpp.NewJid(msg.From) from, err := xmpp.NewJid(msg.From)
if err != nil { if err != nil {
log.Errorf("blame gosrc.io/xmpp for jid encoding: %s", msg.From) logger.Errorf("blame gosrc.io/xmpp for jid encoding: %s", msg.From)
return return
} }
fromBare := from.Bare() fromBare := from.Bare()
fromLogmania := "" fromLogmania := ""
if msg.Type == xmpp.MessageTypeGroupchat { if msg.Type == stanza.MessageTypeGroupchat {
fromLogmania = protoGroup + ":" + fromBare fromLogmania = protoGroup + ":" + fromBare
} else { } else {
fromLogmania = proto + ":" + fromBare fromLogmania = proto + ":" + fromBare
@ -37,13 +38,18 @@ func (out *Output) recvMessage(s xmpp.Sender, p xmpp.Packet) {
if answer == "" { if answer == "" {
return return
} }
reply := xmpp.Message{Attrs: xmpp.Attrs{To: fromBare, Type: msg.Type}, Body: answer} if err := s.Send(stanza.Message{Attrs: stanza.Attrs{To: fromBare, Type: msg.Type}, Body: answer}); err != nil {
s.Send(reply) logger.WithFields(map[string]interface{}{
"sender": fromLogmania,
"request": msg.Body,
"answer": answer,
}).Errorf("unable to send bot answer: %s", err)
}
after := time.Now() after := time.Now()
delta := after.Sub(before) delta := after.Sub(before)
log.WithFields(map[string]interface{}{ logger.WithFields(map[string]interface{}{
"sender": fromLogmania, "sender": fromLogmania,
"request": msg.Body, "request": msg.Body,
"answer": answer, "answer": answer,
@ -51,42 +57,48 @@ func (out *Output) recvMessage(s xmpp.Sender, p xmpp.Packet) {
}).Debug("handled xmpp bot message") }).Debug("handled xmpp bot message")
} }
func (out *Output) recvPresence(s xmpp.Sender, p xmpp.Packet) { func (out *Output) recvPresence(s xmpp.Sender, p stanza.Packet) {
pres, ok := p.(xmpp.Presence) pres, ok := p.(stanza.Presence)
if !ok { if !ok {
log.Errorf("blame gosrc.io/xmpp for routing: %s", p) logger.Errorf("blame gosrc.io/xmpp for routing: %s", p)
return return
} }
from, err := xmpp.NewJid(pres.From) from, err := xmpp.NewJid(pres.From)
if err != nil { if err != nil {
log.Errorf("blame gosrc.io/xmpp for jid encoding: %s", pres.From) logger.Errorf("blame gosrc.io/xmpp for jid encoding: %s", pres.From)
return return
} }
fromBare := from.Bare() fromBare := from.Bare()
logPres := logger.WithField("from", from) logPres := logger.WithField("from", from)
switch pres.Type { switch pres.Type {
case xmpp.PresenceTypeSubscribe: case stanza.PresenceTypeSubscribe:
logPres.Debugf("recv presence subscribe") logPres.Debugf("recv presence subscribe")
s.Send(xmpp.Presence{Attrs: xmpp.Attrs{ if err := s.Send(stanza.Presence{Attrs: stanza.Attrs{
Type: xmpp.PresenceTypeSubscribed, Type: stanza.PresenceTypeSubscribed,
To: fromBare, To: fromBare,
Id: pres.Id, Id: pres.Id,
}}) }}); err != nil {
logPres.WithField("user", pres.From).Errorf("answer of subscribe not send: %s", err)
return
}
logPres.Debugf("accept new subscribe") logPres.Debugf("accept new subscribe")
s.Send(xmpp.Presence{Attrs: xmpp.Attrs{ if err := s.Send(stanza.Presence{Attrs: stanza.Attrs{
Type: xmpp.PresenceTypeSubscribe, Type: stanza.PresenceTypeSubscribe,
To: fromBare, To: fromBare,
}}) }}); err != nil {
logPres.WithField("user", pres.From).Errorf("request of subscribe not send: %s", err)
return
}
logPres.Info("request also subscribe") logPres.Info("request also subscribe")
case xmpp.PresenceTypeSubscribed: case stanza.PresenceTypeSubscribed:
logPres.Info("recv presence accepted subscribe") logPres.Info("recv presence accepted subscribe")
case xmpp.PresenceTypeUnsubscribe: case stanza.PresenceTypeUnsubscribe:
logPres.Info("recv presence remove subscribe") logPres.Info("recv presence remove subscribe")
case xmpp.PresenceTypeUnsubscribed: case stanza.PresenceTypeUnsubscribed:
logPres.Info("recv presence removed subscribe") logPres.Info("recv presence removed subscribe")
case xmpp.PresenceTypeUnavailable: case stanza.PresenceTypeUnavailable:
logPres.Debug("recv presence unavailable") logPres.Debug("recv presence unavailable")
case "": case "":
logPres.Debug("recv empty presence, maybe from joining muc") logPres.Debug("recv empty presence, maybe from joining muc")

View File

@ -5,6 +5,7 @@ import (
"github.com/bdlm/log" "github.com/bdlm/log"
"gosrc.io/xmpp" "gosrc.io/xmpp"
"gosrc.io/xmpp/stanza"
"dev.sum7.eu/genofire/logmania/database" "dev.sum7.eu/genofire/logmania/database"
) )
@ -12,19 +13,18 @@ import (
func (out *Output) Join(to string) { func (out *Output) Join(to string) {
toJID, err := xmpp.NewJid(to) toJID, err := xmpp.NewJid(to)
if err != nil { if err != nil {
logger.Error("xmpp could not generate jid to join ", to, " error:", err) logger.Errorf("jid not generate to join muc %s : %s", to, err)
return return
} }
toJID.Resource = nickname toJID.Resource = nickname
err = out.client.Send(xmpp.Presence{Attrs: xmpp.Attrs{To: toJID.Full()}, if err = out.client.Send(stanza.Presence{Attrs: stanza.Attrs{To: toJID.Full()},
Extensions: []xmpp.PresExtension{ Extensions: []stanza.PresExtension{
xmpp.MucPresence{ stanza.MucPresence{
History: xmpp.History{MaxStanzas: xmpp.NewNullableInt(0)}, History: stanza.History{MaxStanzas: stanza.NewNullableInt(0)},
}}, }},
}) }); err != nil {
if err != nil { logger.Errorf("muc not join %s : %s", toJID.Full(), err)
logger.Error("xmpp could not join ", toJID.Full(), " error:", err)
} else { } else {
out.channels[to] = true out.channels[to] = true
} }
@ -39,26 +39,36 @@ func (out *Output) Send(e *log.Entry, to *database.Notify) bool {
html = strings.TrimRight(to.RunReplace(html), "\n") html = strings.TrimRight(to.RunReplace(html), "\n")
text = strings.TrimRight(to.RunReplace(text), "\n") text = strings.TrimRight(to.RunReplace(text), "\n")
msg := xmpp.Message{ msg := stanza.Message{
Attrs: xmpp.Attrs{ Attrs: stanza.Attrs{
To: to.To, To: to.To,
}, },
Body: text, Body: text,
Extensions: []xmpp.MsgExtension{ Extensions: []stanza.MsgExtension{
xmpp.HTML{Body: xmpp.HTMLBody{InnerXML: html}}, stanza.HTML{Body: stanza.HTMLBody{InnerXML: html}},
}, },
} }
if to.Protocol == protoGroup { if to.Protocol == protoGroup {
if _, ok := out.channels[to.To]; ok { if _, ok := out.channels[to.To]; ok {
out.Join(to.To) out.Join(to.To)
} }
msg.Type = xmpp.MessageTypeGroupchat msg.Type = stanza.MessageTypeGroupchat
out.client.Send(msg) if err := out.client.Send(msg); err != nil {
logger.WithFields(map[string]interface{}{
"muc": to.To,
"text": text,
}).Errorf("log message not forwarded: %s", err)
}
return true return true
} }
if to.Protocol == proto { if to.Protocol == proto {
msg.Type = xmpp.MessageTypeChat msg.Type = stanza.MessageTypeChat
out.client.Send(msg) if err := out.client.Send(msg); err != nil {
logger.WithFields(map[string]interface{}{
"user": to.To,
"text": text,
}).Errorf("log message not forwarded: %s", err)
}
return true return true
} }
return false return false