From 0623aa38f2ead734dddea4cbad899a868f01cb1e Mon Sep 17 00:00:00 2001
From: Melvin Keskin <melvo@olomono.de>
Date: Thu, 2 Sep 2021 23:41:16 +0200
Subject: Add QXmppOmemoElement
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Co-authored-by: Germán Márquez Mejía <mancho@olomono.de>
---
 src/base/QXmppMessage.cpp | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

(limited to 'src/base/QXmppMessage.cpp')

diff --git a/src/base/QXmppMessage.cpp b/src/base/QXmppMessage.cpp
index f0e74735..f975ae3c 100644
--- a/src/base/QXmppMessage.cpp
+++ b/src/base/QXmppMessage.cpp
@@ -28,6 +28,7 @@
 #include "QXmppBitsOfBinaryDataList.h"
 #include "QXmppConstants_p.h"
 #include "QXmppMixInvitation.h"
+#include "QXmppOmemoElement.h"
 #include "QXmppTrustMessageElement.h"
 #include "QXmppUtils.h"
 
@@ -171,6 +172,9 @@ public:
     bool isSpoiler;
     QString spoilerHint;
 
+    // XEP-0384: OMEMO Encryption
+    std::optional<QXmppOmemoElement> omemoElement;
+
     // XEP-0407: Mediated Information eXchange (MIX): Miscellaneous Capabilities
     std::optional<QXmppMixInvitation> mixInvitation;
 
@@ -1083,6 +1087,26 @@ void QXmppMessage::setSpoilerHint(const QString &spoilerHint)
         d->isSpoiler = true;
 }
 
+///
+/// Returns an included OMEMO element as defined by \xep{0384, OMEMO Encryption}.
+///
+/// \since QXmpp 1.5
+///
+std::optional<QXmppOmemoElement> QXmppMessage::omemoElement() const
+{
+    return d->omemoElement;
+}
+
+///
+/// Sets an OMEMO element as defined by \xep{0384, OMEMO Encryption}.
+///
+/// \since QXmpp 1.5
+///
+void QXmppMessage::setOmemoElement(const std::optional<QXmppOmemoElement> &omemoElement)
+{
+    d->omemoElement = omemoElement;
+}
+
 ///
 /// Returns an included \xep{0369}: Mediated Information eXchange (MIX)
 /// invitation as defined by \xep{0407}: Mediated Information eXchange (MIX):
@@ -1345,6 +1369,11 @@ bool QXmppMessage::parseExtension(const QDomElement &element)
         // XEP-0382: Spoiler messages
         d->isSpoiler = true;
         d->spoilerHint = element.text();
+    } else if (QXmppOmemoElement::isOmemoElement(element)) {
+        // XEP-0384: OMEMO Encryption
+        QXmppOmemoElement omemoElement;
+        omemoElement.parse(element);
+        d->omemoElement = omemoElement;
     } else if (checkElement(element, QStringLiteral("invitation"), ns_mix_misc)) {
         // XEP-0407: Mediated Information eXchange (MIX): Miscellaneous Capabilities
         QXmppMixInvitation mixInvitation;
@@ -1543,6 +1572,11 @@ void QXmppMessage::serializeExtensions(QXmlStreamWriter *xmlWriter) const
         xmlWriter->writeEndElement();
     }
 
+    // XEP-0384: OMEMO Encryption
+    if (d->omemoElement) {
+        d->omemoElement->toXml(xmlWriter);
+    }
+
     // XEP-0407: Mediated Information eXchange (MIX): Miscellaneous Capabilities
     if (d->mixInvitation) {
         d->mixInvitation->toXml(xmlWriter);
-- 
cgit v1.2.3