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 allAdmins += jid.Bare() } else { 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") } } }