From bdc2b00a1d89c20f6fe87bdb6f9eb792afeea2bb Mon Sep 17 00:00:00 2001
From: Linus Jahn <lnj@kaidan.im>
Date: Mon, 17 Oct 2022 16:14:35 +0200
Subject: E2eeExtension: Add isEncrypted() and decryptMessage() functions

---
 src/client/QXmppE2eeExtension.h | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'src/client/QXmppE2eeExtension.h')

diff --git a/src/client/QXmppE2eeExtension.h b/src/client/QXmppE2eeExtension.h
index 6c3556f2..295de1a3 100644
--- a/src/client/QXmppE2eeExtension.h
+++ b/src/client/QXmppE2eeExtension.h
@@ -5,6 +5,7 @@
 #ifndef QXMPPE2EEEXTENSION_H
 #define QXMPPE2EEEXTENSION_H
 
+#include "QXmppError.h"
 #include "QXmppExtension.h"
 #include "QXmppSendResult.h"
 #include "QXmppSendStanzaParams.h"
@@ -25,13 +26,16 @@ public:
     };
 
     using MessageEncryptResult = std::variant<QByteArray, QXmpp::SendError>;
+    using MessageDecryptResult = std::variant<QXmppMessage, NotEncrypted, QXmppError>;
     using IqEncryptResult = std::variant<QByteArray, QXmpp::SendError>;
     using IqDecryptResult = std::variant<QDomElement, NotEncrypted, QXmpp::SendError>;
 
     virtual QFuture<MessageEncryptResult> encryptMessage(QXmppMessage &&, const std::optional<QXmppSendStanzaParams> &) = 0;
-
+    virtual QFuture<MessageDecryptResult> decryptMessage(QXmppMessage &&) = 0;
     virtual QFuture<IqEncryptResult> encryptIq(QXmppIq &&, const std::optional<QXmppSendStanzaParams> &) = 0;
     virtual QFuture<IqDecryptResult> decryptIq(const QDomElement &) = 0;
+    virtual bool isEncrypted(const QDomElement &) = 0;
+    virtual bool isEncrypted(const QXmppMessage &) = 0;
 };
 
 #endif  // QXMPPE2EEEXTENSION_H
-- 
cgit v1.2.3