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 } 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", err) continue } logCTX = logCTX.WithField("from", msg.From.Full()).WithField("msg", 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") } } }