2017-12-17 13:31:02 +01:00
|
|
|
package extension
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/xml"
|
|
|
|
|
2018-02-07 15:34:18 +01:00
|
|
|
"dev.sum7.eu/genofire/yaja/messages"
|
|
|
|
"dev.sum7.eu/genofire/yaja/server/utils"
|
2017-12-17 13:31:02 +01:00
|
|
|
)
|
|
|
|
|
2017-12-17 15:39:36 +01:00
|
|
|
type IQPrivateRoster struct {
|
|
|
|
iqPrivateExtension
|
2017-12-17 13:31:02 +01:00
|
|
|
}
|
|
|
|
|
2018-02-07 19:32:11 +01:00
|
|
|
func (ex *IQPrivateRoster) Handle(msg *messages.IQClient, q *iqPrivateQuery, client *utils.Client) bool {
|
2017-12-17 13:31:02 +01:00
|
|
|
log := client.Log.WithField("extension", "private").WithField("id", msg.ID)
|
|
|
|
|
|
|
|
// roster encode
|
|
|
|
type roster struct {
|
|
|
|
XMLName xml.Name `xml:"roster:delimiter roster"`
|
|
|
|
Body []byte `xml:",innerxml"`
|
|
|
|
}
|
|
|
|
r := &roster{}
|
2017-12-17 17:50:51 +01:00
|
|
|
if err := xml.Unmarshal(q.Body, r); err != nil {
|
2017-12-17 13:31:02 +01:00
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
|
|
|
rosterByte, err := xml.Marshal(&roster{
|
|
|
|
Body: []byte("::"),
|
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
log.Warn(err)
|
|
|
|
return true
|
|
|
|
}
|
2017-12-17 15:39:36 +01:00
|
|
|
queryByte, err := xml.Marshal(&iqPrivateQuery{
|
2017-12-17 13:31:02 +01:00
|
|
|
Body: rosterByte,
|
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
log.Warn(err)
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
|
|
|
|
// reply
|
2018-02-07 19:32:11 +01:00
|
|
|
client.Messages <- &messages.IQClient{
|
2017-12-17 13:31:02 +01:00
|
|
|
Type: messages.IQTypeResult,
|
|
|
|
To: client.JID.String(),
|
|
|
|
From: client.JID.Domain,
|
|
|
|
ID: msg.ID,
|
|
|
|
Body: queryByte,
|
2017-12-17 15:39:36 +01:00
|
|
|
}
|
2017-12-17 13:31:02 +01:00
|
|
|
|
|
|
|
log.Debug("send")
|
|
|
|
|
|
|
|
return true
|
|
|
|
}
|