|
|
|
@ -26,6 +26,7 @@ const (
|
|
|
|
|
TypeVCard4
|
|
|
|
|
)
|
|
|
|
|
const NodeVCard4 string = "urn:xmpp:vcard4"
|
|
|
|
|
const NSCommand string = "http://jabber.org/protocol/commands"
|
|
|
|
|
|
|
|
|
|
func logPacketType(p stanza.Packet) {
|
|
|
|
|
log.Warnf("Ignoring packet: %T\n", p)
|
|
|
|
@ -53,14 +54,14 @@ func HandleIq(s xmpp.Sender, p stanza.Packet) {
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
_, ok = iq.Payload.(*stanza.DiscoInfo)
|
|
|
|
|
discoInfo, ok := iq.Payload.(*stanza.DiscoInfo)
|
|
|
|
|
if ok {
|
|
|
|
|
go handleGetDiscoInfo(s, iq)
|
|
|
|
|
go handleGetDiscoInfo(s, iq, discoInfo)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
_, ok = iq.Payload.(*stanza.DiscoItems)
|
|
|
|
|
discoItems, ok := iq.Payload.(*stanza.DiscoItems)
|
|
|
|
|
if ok {
|
|
|
|
|
go handleGetDiscoItems(s, iq)
|
|
|
|
|
go handleGetDiscoItems(s, iq, discoItems)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
_, ok = iq.Payload.(*extensions.QueryRegister)
|
|
|
|
@ -74,6 +75,11 @@ func HandleIq(s xmpp.Sender, p stanza.Packet) {
|
|
|
|
|
go handleSetQueryRegister(s, iq, query)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
command, ok := iq.Payload.(*stanza.Command)
|
|
|
|
|
if ok {
|
|
|
|
|
go handleSetQueryCommand(s, iq, command)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
@ -468,7 +474,7 @@ func handleGetVcardIq(s xmpp.Sender, iq *stanza.IQ, typ byte) {
|
|
|
|
|
_ = gateway.ResumableSend(component, &answer)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func handleGetDiscoInfo(s xmpp.Sender, iq *stanza.IQ) {
|
|
|
|
|
func handleGetDiscoInfo(s xmpp.Sender, iq *stanza.IQ, di *stanza.DiscoInfo) {
|
|
|
|
|
answer, err := stanza.NewIQ(stanza.Attrs{
|
|
|
|
|
Type: stanza.IQTypeResult,
|
|
|
|
|
From: iq.To,
|
|
|
|
@ -488,8 +494,20 @@ func handleGetDiscoInfo(s xmpp.Sender, iq *stanza.IQ) {
|
|
|
|
|
disco.AddFeatures(stanza.NSMsgChatMarkers)
|
|
|
|
|
disco.AddFeatures(stanza.NSMsgReceipts)
|
|
|
|
|
} else {
|
|
|
|
|
disco.AddIdentity("Telegram Gateway", "gateway", "telegram")
|
|
|
|
|
disco.AddFeatures("jabber:iq:register")
|
|
|
|
|
if di.Node == "" {
|
|
|
|
|
disco.AddIdentity("Telegram Gateway", "gateway", "telegram")
|
|
|
|
|
disco.AddFeatures("jabber:iq:register")
|
|
|
|
|
disco.AddFeatures(NSCommand)
|
|
|
|
|
} else {
|
|
|
|
|
for name, command := range telegram.GetCommands(telegram.CommandTypeTransport) {
|
|
|
|
|
if di.Node == name {
|
|
|
|
|
answer.Payload = di
|
|
|
|
|
di.AddIdentity(telegram.CommandToHelpString(name, command), "automation", "command-node")
|
|
|
|
|
di.AddFeatures(NSCommand, "jabber:x:data")
|
|
|
|
|
break
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
answer.Payload = disco
|
|
|
|
|
|
|
|
|
@ -504,7 +522,7 @@ func handleGetDiscoInfo(s xmpp.Sender, iq *stanza.IQ) {
|
|
|
|
|
_ = gateway.ResumableSend(component, answer)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func handleGetDiscoItems(s xmpp.Sender, iq *stanza.IQ) {
|
|
|
|
|
func handleGetDiscoItems(s xmpp.Sender, iq *stanza.IQ, di *stanza.DiscoItems) {
|
|
|
|
|
answer, err := stanza.NewIQ(stanza.Attrs{
|
|
|
|
|
Type: stanza.IQTypeResult,
|
|
|
|
|
From: iq.To,
|
|
|
|
@ -517,7 +535,20 @@ func handleGetDiscoItems(s xmpp.Sender, iq *stanza.IQ) {
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
answer.Payload = answer.DiscoItems()
|
|
|
|
|
log.Debugf("discoItems: %#v", di)
|
|
|
|
|
|
|
|
|
|
_, ok := toToID(iq.To)
|
|
|
|
|
if !ok {
|
|
|
|
|
commands := telegram.GetCommands(telegram.CommandTypeTransport)
|
|
|
|
|
if di.Node == NSCommand {
|
|
|
|
|
answer.Payload = di
|
|
|
|
|
for name, command := range commands {
|
|
|
|
|
di.AddItem(iq.To, name, telegram.CommandToHelpString(name, command))
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
answer.Payload = answer.DiscoItems()
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
component, ok := s.(*xmpp.Component)
|
|
|
|
|
if !ok {
|
|
|
|
@ -647,6 +678,57 @@ func handleSetQueryRegister(s xmpp.Sender, iq *stanza.IQ, query *extensions.Quer
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func handleSetQueryCommand(s xmpp.Sender, iq *stanza.IQ, command *stanza.Command) {
|
|
|
|
|
component, ok := s.(*xmpp.Component)
|
|
|
|
|
if !ok {
|
|
|
|
|
log.Error("Not a component")
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
answer, err := stanza.NewIQ(stanza.Attrs{
|
|
|
|
|
Type: stanza.IQTypeResult,
|
|
|
|
|
From: iq.To,
|
|
|
|
|
To: iq.From,
|
|
|
|
|
Id: iq.Id,
|
|
|
|
|
Lang: "en",
|
|
|
|
|
})
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Errorf("Failed to create answer IQ: %v", err)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
defer gateway.ResumableSend(component, answer)
|
|
|
|
|
|
|
|
|
|
log.Debugf("command: %#v", command)
|
|
|
|
|
|
|
|
|
|
if command.Action == "" || command.Action == stanza.CommandActionExecute {
|
|
|
|
|
_, ok := toToID(iq.To)
|
|
|
|
|
if !ok {
|
|
|
|
|
bare, resource, ok := gateway.SplitJID(iq.From)
|
|
|
|
|
if !ok {
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
session, ok := sessions[bare]
|
|
|
|
|
if !ok {
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
response := session.ProcessTransportCommand("/" + command.Node, resource)
|
|
|
|
|
|
|
|
|
|
answer.Payload = &stanza.Command{
|
|
|
|
|
Node: command.Node,
|
|
|
|
|
Status: stanza.CommandStatusCompleted,
|
|
|
|
|
CommandElement: &stanza.Note{
|
|
|
|
|
Text: response,
|
|
|
|
|
Type: stanza.CommandNoteTypeInfo,
|
|
|
|
|
},
|
|
|
|
|
}
|
|
|
|
|
log.Debugf("command response: %#v", answer.Payload)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func iqAnswerSetError(answer *stanza.IQ, payload *extensions.QueryRegister, code int) {
|
|
|
|
|
answer.Type = stanza.IQTypeError
|
|
|
|
|
answer.Payload = *payload
|
|
|
|
|