From b1135b070b751998cb98eded6d29bdd7b800ecb9 Mon Sep 17 00:00:00 2001
From: Bohdan Horbeshko <bodqhrohro@gmail.com>
Date: Wed, 8 Mar 2023 07:51:19 -0500
Subject: Cut fallback quotes out

---
 xmpp/extensions/extensions.go | 47 +++++++++++++++++++++++++++++++++++++++++++
 xmpp/gateway/gateway.go       |  5 +++++
 2 files changed, 52 insertions(+)

(limited to 'xmpp')

diff --git a/xmpp/extensions/extensions.go b/xmpp/extensions/extensions.go
index ce27656..7e44bae 100644
--- a/xmpp/extensions/extensions.go
+++ b/xmpp/extensions/extensions.go
@@ -2,6 +2,7 @@ package extensions
 
 import (
 	"encoding/xml"
+	"strconv"
 
 	"gosrc.io/xmpp/stanza"
 )
@@ -118,6 +119,28 @@ type Reply struct {
 	Id      string   `xml:"id,attr"`
 }
 
+// Fallback is from XEP-0428
+type Fallback struct {
+	XMLName xml.Name `xml:"urn:xmpp:fallback:0 fallback"`
+	For     string   `xml:"for,attr"`
+	Body    []FallbackBody
+	Subject []FallbackSubject
+}
+
+// FallbackBody is from XEP-0428
+type FallbackBody struct {
+	XMLName xml.Name `xml:"body"`
+	Start   string   `xml:"start,attr"`
+	End     string   `xml:"end,attr"`
+}
+
+// FallbackSubject is from XEP-0428
+type FallbackSubject struct {
+	XMLName xml.Name `xml:"subject"`
+	Start   string   `xml:"start,attr"`
+	End     string   `xml:"end,attr"`
+}
+
 // Namespace is a namespace!
 func (c PresenceNickExtension) Namespace() string {
 	return c.XMLName.Space
@@ -143,6 +166,24 @@ func (c Reply) Namespace() string {
 	return c.XMLName.Space
 }
 
+// Namespace is a namespace!
+func (c Fallback) Namespace() string {
+	return c.XMLName.Space
+}
+
+// NewReplyFallback initializes a fallback range
+func NewReplyFallback(start uint64, end uint64) Fallback {
+	return Fallback{
+		For: "urn:xmpp:reply:0",
+		Body: []FallbackBody{
+			FallbackBody{
+				Start: strconv.FormatUint(start, 10),
+				End: strconv.FormatUint(end, 10),
+			},
+		},
+	}
+}
+
 func init() {
 	// presence nick
 	stanza.TypeRegistry.MapExtension(stanza.PKTPresence, xml.Name{
@@ -167,4 +208,10 @@ func init() {
 		"urn:xmpp:reply:0",
 		"reply",
 	}, Reply{})
+
+	// fallback
+	stanza.TypeRegistry.MapExtension(stanza.PKTMessage, xml.Name{
+		"urn:xmpp:fallback:0",
+		"fallback",
+	}, Fallback{})
 }
diff --git a/xmpp/gateway/gateway.go b/xmpp/gateway/gateway.go
index d309ade..de8b495 100644
--- a/xmpp/gateway/gateway.go
+++ b/xmpp/gateway/gateway.go
@@ -16,6 +16,8 @@ import (
 type Reply struct {
 	Author string
 	Id     string
+	Start  uint64
+	End    uint64
 }
 
 const NSNick string = "http://jabber.org/protocol/nick"
@@ -89,6 +91,9 @@ func sendMessageWrapper(to string, from string, body string, id string, componen
 			To: reply.Author,
 			Id: reply.Id,
 		})
+		if reply.End > 0 {
+			message.Extensions = append(message.Extensions, extensions.NewReplyFallback(reply.Start, reply.End))
+		}
 	}
 
 	sendMessage(&message, component)
-- 
cgit v1.2.3