144 lines
3.8 KiB
Go
144 lines
3.8 KiB
Go
package tester
|
|
|
|
import (
|
|
"strings"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"dev.sum7.eu/genofire/yaja/messages"
|
|
)
|
|
|
|
func (t *Tester) StartBot(status *Status) {
|
|
logger := log.New()
|
|
logger.SetLevel(t.LoggingBots)
|
|
logCTX := logger.WithFields(log.Fields{
|
|
"type": "bot",
|
|
"jid": status.client.JID.Full(),
|
|
})
|
|
for {
|
|
|
|
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.Decode(errMSG, element)
|
|
if err == nil {
|
|
logCTX.Errorf("recv stream error: %s: %s", errMSG.Text, messages.XMLChildrenString(errMSG.Any))
|
|
status.client.Close()
|
|
status.Login = false
|
|
return
|
|
}
|
|
|
|
iq := &messages.IQClient{}
|
|
err = status.client.Decode(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.Send(iq)
|
|
} else {
|
|
logCTX.Warnf("recv iq unsupport: %s", messages.XMLChildrenString(iq))
|
|
}
|
|
continue
|
|
}
|
|
|
|
pres := &messages.PresenceClient{}
|
|
err = status.client.Decode(pres, element)
|
|
if err == nil {
|
|
sender := pres.From
|
|
logPres := logCTX.WithField("from", sender.Full())
|
|
if pres.Type == messages.PresenceTypeSubscribe {
|
|
logPres.Debugf("recv presence subscribe")
|
|
pres.Type = messages.PresenceTypeSubscribed
|
|
pres.To = sender
|
|
pres.From = nil
|
|
status.client.Send(pres)
|
|
logPres.Debugf("accept new subscribe")
|
|
|
|
pres.Type = messages.PresenceTypeSubscribe
|
|
pres.ID = ""
|
|
status.client.Send(pres)
|
|
logPres.Info("request also subscribe")
|
|
} else if pres.Type == messages.PresenceTypeSubscribed {
|
|
logPres.Info("recv presence accepted subscribe")
|
|
} else if pres.Type == messages.PresenceTypeUnsubscribe {
|
|
logPres.Info("recv presence remove subscribe")
|
|
} else if pres.Type == messages.PresenceTypeUnsubscribed {
|
|
logPres.Info("recv presence removed subscribe")
|
|
} else if pres.Type == messages.PresenceTypeUnavailable {
|
|
logPres.Debug("recv presence unavailable")
|
|
} else {
|
|
logCTX.Warnf("recv presence unsupported: %s -> %s", pres.Type, messages.XMLChildrenString(pres))
|
|
}
|
|
continue
|
|
}
|
|
|
|
msg := &messages.MessageClient{}
|
|
err = status.client.Decode(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 {
|
|
if msg.Error.Type == "auth" {
|
|
logCTX.Warnf("recv msg with error not auth")
|
|
status.Login = false
|
|
status.client.Close()
|
|
return
|
|
}
|
|
logCTX.Debugf("recv msg with error %s[%s]: %s -> %s -> %s", msg.Error.Code, msg.Error.Type, msg.Error.Text, messages.XMLChildrenString(msg.Error.StanzaErrorGroup), messages.XMLChildrenString(msg.Error.Other))
|
|
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 "disconnect":
|
|
first := true
|
|
allAdmins := ""
|
|
isAdmin := false
|
|
for _, jid := range t.Admins {
|
|
if first {
|
|
first = false
|
|
} else {
|
|
allAdmins += ", "
|
|
}
|
|
allAdmins += jid.Bare()
|
|
if jid.Bare() == msg.From.Bare() {
|
|
isAdmin = true
|
|
status.client.Send(messages.MessageClient{Type: msg.Type, To: jid, Body: "last message, disconnect requested by " + msg.From.Bare()})
|
|
|
|
}
|
|
}
|
|
if isAdmin {
|
|
status.Login = false
|
|
status.client.Close()
|
|
return
|
|
}
|
|
status.client.Send(messages.MessageClient{Type: msg.Type, To: msg.From, Body: "not allowed, ask " + allAdmins})
|
|
|
|
case "checkmsg":
|
|
if len(msgText) == 2 {
|
|
t.UpdateConnectionStatus(msg.From, status.client.JID, msgText[1])
|
|
} else {
|
|
logCTX.Debug("undetect")
|
|
}
|
|
|
|
default:
|
|
logCTX.Debug("undetect")
|
|
}
|
|
}
|
|
}
|