92 lines
2.1 KiB
Go
92 lines
2.1 KiB
Go
package tester
|
|
|
|
import (
|
|
"strings"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"dev.sum7.eu/genofire/yaja/messages"
|
|
)
|
|
|
|
func (t *Tester) StartBot(status *Status) {
|
|
for {
|
|
logCTX := log.WithField("jid", status.client.JID.Full())
|
|
|
|
element, err := status.client.Read()
|
|
if err != nil {
|
|
logCTX.Errorf("read client %s", err)
|
|
status.client.Close()
|
|
status.Login = false
|
|
return
|
|
}
|
|
|
|
errMSG := &messages.StreamError{}
|
|
err = status.client.In.DecodeElement(errMSG, element)
|
|
if err == nil {
|
|
logCTX.Errorf("recv stream error: %s: %v", errMSG.Text, errMSG.Any)
|
|
status.client.Close()
|
|
status.Login = false
|
|
return
|
|
}
|
|
|
|
iq := &messages.IQClient{}
|
|
err = status.client.In.DecodeElement(iq, element)
|
|
if err == nil {
|
|
if iq.Ping != nil {
|
|
logCTX.Debug("answer ping")
|
|
iq.Type = messages.IQTypeResult
|
|
iq.To = iq.From
|
|
iq.From = status.client.JID
|
|
status.client.Out.Encode(iq)
|
|
} else {
|
|
logCTX.Warnf("unsupport iq recv: %v", iq)
|
|
}
|
|
continue
|
|
}
|
|
|
|
pres := &messages.PresenceClient{}
|
|
err = status.client.In.DecodeElement(pres, element)
|
|
if err == nil {
|
|
if pres.Type == messages.PresenceTypeSubscribe {
|
|
pres.Type = messages.PresenceTypeSubscribed
|
|
pres.To = pres.From
|
|
status.client.Send(pres)
|
|
logCTX.WithField("from", pres.From.Full()).Info("accept new subscribe")
|
|
} else {
|
|
logCTX.Warnf("unsupported presence recv: %v", pres)
|
|
}
|
|
continue
|
|
}
|
|
|
|
msg := &messages.MessageClient{}
|
|
err = status.client.In.DecodeElement(msg, element)
|
|
if err != nil {
|
|
logCTX.Warnf("unsupport xml recv: %s <-> %v", err, element)
|
|
continue
|
|
}
|
|
logCTX = logCTX.WithField("from", msg.From.Full()).WithField("msg-recv", msg.Body)
|
|
if msg.Error != nil {
|
|
logCTX.Debugf("recv msg with error %s: %s", msg.Error.Code, msg.Error.Text)
|
|
continue
|
|
|
|
}
|
|
|
|
msgText := strings.SplitN(msg.Body, " ", 2)
|
|
switch msgText[0] {
|
|
|
|
case "ping":
|
|
status.client.Send(messages.MessageClient{Type: msg.Type, To: msg.From, Body: "pong"})
|
|
|
|
case "checkmsg":
|
|
if len(msgText) == 2 {
|
|
t.UpdateConnectionStatus(msg.From, status.client.JID, msgText[1])
|
|
} else {
|
|
logCTX.Debug("undetect")
|
|
}
|
|
|
|
default:
|
|
logCTX.Debug("undetect")
|
|
}
|
|
}
|
|
}
|