package extension import ( "encoding/xml" "dev.sum7.eu/genofire/yaja/database" "dev.sum7.eu/genofire/yaja/xmpp" "dev.sum7.eu/genofire/yaja/server/utils" ) type IQRoster struct { IQExtension Database *database.State } func (ex *IQRoster) Spaces() []string { return []string{"jabber:iq:roster"} } func (ex *IQRoster) Get(msg *xmpp.IQClient, client *utils.Client) bool { log := client.Log.WithField("extension", "roster").WithField("id", msg.ID) // query encode type query struct { XMLName xml.Name `xml:"jabber:iq:roster query"` Version string `xml:"ver,attr"` Body []byte `xml:",innerxml"` } q := &query{} if err := xml.Unmarshal(msg.Body, q); err != nil { return false } // answer query q.Body = []byte{} q.Version = "1" // build answer body type item struct { XMLName xml.Name `xml:"item"` JID string `xml:"jid,attr"` } if acc := ex.Database.GetAccount(client.JID); acc != nil { for jid, _ := range acc.Roster { itemByte, err := xml.Marshal(&item{ JID: jid, }) if err != nil { log.Warn(err) continue } q.Body = append(q.Body, itemByte...) } } // decode query queryByte, err := xml.Marshal(q) if err != nil { log.Warn(err) return false } // reply client.Messages <- &xmpp.IQClient{ Type: xmpp.IQTypeResult, To: client.JID.String(), From: client.JID.Domain, ID: msg.ID, Body: queryByte, } log.Debug("send") return true }