Convert XEP-0461 replies to Telegram

calls
Bohdan Horbeshko 1 year ago
parent 7bd165a738
commit 90807b2d9e

@ -15,7 +15,7 @@ import (
goxmpp "gosrc.io/xmpp" goxmpp "gosrc.io/xmpp"
) )
var version string = "1.4.0" var version string = "1.5.0-dev"
var commit string var commit string
var sm *goxmpp.StreamManager var sm *goxmpp.StreamManager

@ -488,7 +488,7 @@ func (c *Client) ProcessChatCommand(chatID int64, cmdline string) (string, bool)
return "Last message is empty", true return "Last message is empty", true
} }
content := c.ProcessOutgoingMessage(0, rawCmdArguments(cmdline, 0), "") content := c.ProcessOutgoingMessage(0, rawCmdArguments(cmdline, 0), "", 0)
if content != nil { if content != nil {
c.client.EditMessageText(&client.EditMessageTextRequest{ c.client.EditMessageText(&client.EditMessageTextRequest{
@ -505,7 +505,7 @@ func (c *Client) ProcessChatCommand(chatID int64, cmdline string) (string, bool)
return "Not enough arguments", true return "Not enough arguments", true
} }
content := c.ProcessOutgoingMessage(0, rawCmdArguments(cmdline, 0), "") content := c.ProcessOutgoingMessage(0, rawCmdArguments(cmdline, 0), "", 0)
if content != nil { if content != nil {
_, err := c.client.SendMessage(&client.SendMessageRequest{ _, err := c.client.SendMessage(&client.SendMessageRequest{
@ -584,7 +584,7 @@ func (c *Client) ProcessChatCommand(chatID int64, cmdline string) (string, bool)
} }
} }
content := c.ProcessOutgoingMessage(0, rawCmdArguments(cmdline, 1), "") content := c.ProcessOutgoingMessage(0, rawCmdArguments(cmdline, 1), "", 0)
if content != nil { if content != nil {
_, err := c.client.SendMessage(&client.SendMessageRequest{ _, err := c.client.SendMessage(&client.SendMessageRequest{

@ -857,7 +857,7 @@ func (c *Client) ProcessIncomingMessage(chatId int64, message *client.Message) {
} }
// ProcessOutgoingMessage executes commands or sends messages to mapped chats // ProcessOutgoingMessage executes commands or sends messages to mapped chats
func (c *Client) ProcessOutgoingMessage(chatID int64, text string, returnJid string) client.InputMessageContent { func (c *Client) ProcessOutgoingMessage(chatID int64, text string, returnJid string, replyId int64) client.InputMessageContent {
if !c.Online() { if !c.Online() {
// we're offline // we're offline
return nil return nil
@ -879,9 +879,13 @@ func (c *Client) ProcessOutgoingMessage(chatID int64, text string, returnJid str
// quotations // quotations
var reply int64 var reply int64
replySlice := replyRegex.FindStringSubmatch(text) if replyId == 0 {
if len(replySlice) > 1 { replySlice := replyRegex.FindStringSubmatch(text)
reply, _ = strconv.ParseInt(replySlice[1], 10, 64) if len(replySlice) > 1 {
reply, _ = strconv.ParseInt(replySlice[1], 10, 64)
}
} else {
reply = replyId
} }
// attach a file // attach a file
@ -949,7 +953,7 @@ func (c *Client) ProcessOutgoingMessage(chatID int64, text string, returnJid str
} }
// remove first line from text // remove first line from text
if file != nil || reply != 0 { if file != nil || (reply != 0 && replyId == 0) {
newlinePos := strings.Index(text, newlineChar) newlinePos := strings.Index(text, newlineChar)
if newlinePos != -1 { if newlinePos != -1 {
text = text[newlinePos+1:] text = text[newlinePos+1:]

@ -121,22 +121,22 @@ type Reply struct {
// Fallback is from XEP-0428 // Fallback is from XEP-0428
type Fallback struct { type Fallback struct {
XMLName xml.Name `xml:"urn:xmpp:fallback:0 fallback"` XMLName xml.Name `xml:"urn:xmpp:fallback:0 fallback"`
For string `xml:"for,attr"` For string `xml:"for,attr"`
Body []FallbackBody Body []FallbackBody `xml:"urn:xmpp:fallback:0 body"`
Subject []FallbackSubject Subject []FallbackSubject `xml:"urn:xmpp:fallback:0 subject"`
} }
// FallbackBody is from XEP-0428 // FallbackBody is from XEP-0428
type FallbackBody struct { type FallbackBody struct {
XMLName xml.Name `xml:"body"` XMLName xml.Name `xml:"urn:xmpp:fallback:0 body"`
Start string `xml:"start,attr"` Start string `xml:"start,attr"`
End string `xml:"end,attr"` End string `xml:"end,attr"`
} }
// FallbackSubject is from XEP-0428 // FallbackSubject is from XEP-0428
type FallbackSubject struct { type FallbackSubject struct {
XMLName xml.Name `xml:"subject"` XMLName xml.Name `xml:"urn:xmpp:fallback:0 subject"`
Start string `xml:"start,attr"` Start string `xml:"start,attr"`
End string `xml:"end,attr"` End string `xml:"end,attr"`
} }

@ -99,7 +99,46 @@ func HandleMessage(s xmpp.Sender, p stanza.Packet) {
toID, ok := toToID(msg.To) toID, ok := toToID(msg.To)
if ok { if ok {
session.ProcessOutgoingMessage(toID, msg.Body, msg.From) var reply extensions.Reply
var fallback extensions.Fallback
msg.Get(&reply)
msg.Get(&fallback)
log.Debugf("reply: %#v", reply)
log.Debugf("fallback: %#v", fallback)
var replyId int64
var err error
text := msg.Body
if len(reply.Id) > 0 {
id := reply.Id
if id[0] == 'e' {
id = id[1:]
}
replyId, err = strconv.ParseInt(id, 10, 64)
if err != nil {
log.Warn(errors.Wrap(err, "Failed to parse message ID!"))
}
if replyId != 0 && fallback.For == "urn:xmpp:reply:0" && len(fallback.Body) > 0 {
body := fallback.Body[0]
var start, end int64
start, err = strconv.ParseInt(body.Start, 10, 64)
if err != nil {
log.WithFields(log.Fields{
"start": body.Start,
}).Warn(errors.Wrap(err, "Failed to parse fallback start!"))
}
end, err = strconv.ParseInt(body.End, 10, 64)
if err != nil {
log.WithFields(log.Fields{
"end": body.End,
}).Warn(errors.Wrap(err, "Failed to parse fallback end!"))
}
text = text[:start] + text[end:]
}
}
session.ProcessOutgoingMessage(toID, text, msg.From, replyId)
return return
} else { } else {
toJid, err := stanza.NewJid(msg.To) toJid, err := stanza.NewJid(msg.To)

Loading…
Cancel
Save