Reply to IQs.
parent
8840a53ab7
commit
f0a4bcf10c
@ -1,35 +0,0 @@
|
||||
// Copyright 2020 - 2021 Martin Dosch.
|
||||
// Use of this source code is governed by the BSD-2-clause
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"github.com/mattn/go-xmpp" // BSD-3-Clause
|
||||
)
|
||||
|
||||
func sendIQ(client *xmpp.Client, iqc chan xmpp.IQ, target string,
|
||||
iQtype string, content string) (xmpp.IQ, error) {
|
||||
var iq xmpp.IQ
|
||||
id := getID()
|
||||
c := make(chan xmpp.IQ)
|
||||
go getIQ(client, id, c, iqc)
|
||||
_, err := client.RawInformation(client.JID(), target, id,
|
||||
iQtype, content + "\n")
|
||||
if err != nil {
|
||||
return iq, err
|
||||
}
|
||||
iq = <-c
|
||||
return iq, nil
|
||||
}
|
||||
|
||||
func getIQ(client *xmpp.Client, id string, c chan xmpp.IQ,
|
||||
iqc chan xmpp.IQ) {
|
||||
for {
|
||||
iq := <-iqc
|
||||
if iq.ID == id {
|
||||
c <- iq
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
@ -0,0 +1,127 @@
|
||||
// Copyright 2020 - 2021 Martin Dosch.
|
||||
// Use of this source code is governed by the BSD-2-clause
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"log"
|
||||
|
||||
"github.com/beevik/etree" // BSD-2-clause
|
||||
"github.com/mattn/go-xmpp" // BSD-3-Clause
|
||||
)
|
||||
|
||||
func sendIQ(client *xmpp.Client, iqc chan xmpp.IQ, target string,
|
||||
iQtype string, content string) (xmpp.IQ, error) {
|
||||
var iq xmpp.IQ
|
||||
id := getID()
|
||||
c := make(chan xmpp.IQ)
|
||||
go getIQ(client, id, c, iqc)
|
||||
_, err := client.RawInformation(client.JID(), target, id,
|
||||
iQtype, content+"\n")
|
||||
if err != nil {
|
||||
return iq, err
|
||||
}
|
||||
iq = <-c
|
||||
return iq, nil
|
||||
}
|
||||
|
||||
func getIQ(client *xmpp.Client, id string, c chan xmpp.IQ,
|
||||
iqc chan xmpp.IQ) {
|
||||
for {
|
||||
iq := <-iqc
|
||||
if iq.ID == id {
|
||||
c <- iq
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func rcvStanzas(client *xmpp.Client, iqc chan xmpp.IQ, msgc chan xmpp.Chat) {
|
||||
for {
|
||||
received, err := client.Recv()
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
switch v := received.(type) {
|
||||
case xmpp.Chat:
|
||||
msgc <- v
|
||||
case xmpp.IQ:
|
||||
switch v.Type {
|
||||
case "get":
|
||||
iq := etree.NewDocument()
|
||||
err = iq.ReadFromBytes(v.Query)
|
||||
if err != nil {
|
||||
log.Println("Couldn't parse IQ:",
|
||||
string(v.Query), err)
|
||||
}
|
||||
query := iq.SelectElement("query")
|
||||
xmlns := query.SelectAttr("xmlns")
|
||||
switch xmlns.Value {
|
||||
case nsDiscoInfo:
|
||||
root := etree.NewDocument()
|
||||
reply := root.CreateElement("iq")
|
||||
reply.CreateAttr("type", "result")
|
||||
reply.CreateAttr("from", client.JID())
|
||||
reply.CreateAttr("to", v.From)
|
||||
reply.CreateAttr("id", v.ID)
|
||||
replyQuery := reply.CreateElement("query")
|
||||
replyQuery.CreateAttr("xmlns", nsDiscoInfo)
|
||||
identity := replyQuery.CreateElement("identity")
|
||||
identity.CreateAttr("category", "client")
|
||||
identity.CreateAttr("type", "bot")
|
||||
identity.CreateAttr("name", "go-sendxmpp")
|
||||
feat := replyQuery.CreateElement("feature")
|
||||
feat.CreateAttr("var", nsDiscoInfo)
|
||||
xmlString, err := root.WriteToString()
|
||||
if err == nil {
|
||||
_, err = client.SendOrg(xmlString)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
}
|
||||
default:
|
||||
root := etree.NewDocument()
|
||||
reply := root.CreateElement("iq")
|
||||
reply.CreateAttr("type", "error")
|
||||
reply.CreateAttr("from", client.JID())
|
||||
reply.CreateAttr("to", v.From)
|
||||
reply.CreateAttr("id", v.ID)
|
||||
errorReply := reply.CreateElement("error")
|
||||
errorReply.CreateAttr("type", "cancel")
|
||||
su := errorReply.CreateElement("service-unavailable")
|
||||
su.CreateAttr("xmlns", nsXMPPStanzas)
|
||||
xmlString, err := root.WriteToString()
|
||||
if err == nil {
|
||||
_, err = client.SendOrg(xmlString)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
case "set":
|
||||
root := etree.NewDocument()
|
||||
reply := root.CreateElement("iq")
|
||||
reply.CreateAttr("type", "error")
|
||||
reply.CreateAttr("from", client.JID())
|
||||
reply.CreateAttr("to", v.From)
|
||||
reply.CreateAttr("id", v.ID)
|
||||
errorReply := reply.CreateElement("error")
|
||||
errorReply.CreateAttr("type", "cancel")
|
||||
su := errorReply.CreateElement("service-unavailable")
|
||||
su.CreateAttr("xmlns", nsXMPPStanzas)
|
||||
xmlString, err := root.WriteToString()
|
||||
if err == nil {
|
||||
_, err = client.SendOrg(xmlString)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
}
|
||||
case "reply", "error":
|
||||
iqc <- v
|
||||
default:
|
||||
iqc <- v
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue