1 files changed, 86 insertions, 0 deletions

diff --git a/stanza/src/client/message.rs b/stanza/src/client/message.rs
index 41761d2..3a7be9a 100644
--- a/stanza/src/client/message.rs
+++ b/stanza/src/client/message.rs
@@ -11,6 +11,12 @@ use crate::xep_0131::Headers;
 use crate::xep_0172::Nick;
 #[cfg(feature = "xep_0203")]
 use crate::xep_0203::Delay;
+#[cfg(feature = "xep_0280")]
+use crate::xep_0280::{Private, Received, Sent};
+#[cfg(feature = "xep_0297")]
+use crate::xep_0297::Forwarded;
+#[cfg(feature = "xep_0334")]
+use crate::xep_0334::{NoCopy, NoPermanentStore, NoStore, Store};
 
 use super::XMLNS;
 
@@ -34,6 +40,22 @@ pub struct Message {
     pub nick: Option<Nick>,
     #[cfg(feature = "xep_0060")]
     pub event: Option<Event>,
+    #[cfg(feature = "xep_0297")]
+    pub forwarded: Option<Forwarded>,
+    #[cfg(feature = "xep_0280")]
+    pub sent: Option<Sent>,
+    #[cfg(feature = "xep_0280")]
+    pub received: Option<Received>,
+    #[cfg(feature = "xep_0280")]
+    pub private: Option<Private>,
+    #[cfg(feature = "xep_0334")]
+    pub no_permanent_store: Option<NoPermanentStore>,
+    #[cfg(feature = "xep_0334")]
+    pub no_store: Option<NoStore>,
+    #[cfg(feature = "xep_0334")]
+    pub no_copy: Option<NoCopy>,
+    #[cfg(feature = "xep_0334")]
+    pub store: Option<Store>,
 }
 
 impl FromElement for Message {
@@ -63,6 +85,30 @@ impl FromElement for Message {
         #[cfg(feature = "xep_0060")]
         let event = element.child_opt()?;
 
+        #[cfg(feature = "xep_0297")]
+        let forwarded = element.child_opt()?;
+
+        #[cfg(feature = "xep_0280")]
+        let sent = element.child_opt()?;
+
+        #[cfg(feature = "xep_0280")]
+        let received = element.child_opt()?;
+
+        #[cfg(feature = "xep_0280")]
+        let private = element.child_opt()?;
+
+        #[cfg(feature = "xep_0334")]
+        let no_permanent_store = element.child_opt()?;
+
+        #[cfg(feature = "xep_0334")]
+        let no_store = element.child_opt()?;
+
+        #[cfg(feature = "xep_0334")]
+        let no_copy = element.child_opt()?;
+
+        #[cfg(feature = "xep_0334")]
+        let store = element.child_opt()?;
+
         Ok(Message {
             from,
             id,
@@ -80,6 +126,22 @@ impl FromElement for Message {
             nick,
             #[cfg(feature = "xep_0060")]
             event,
+            #[cfg(feature = "xep_0297")]
+            forwarded,
+            #[cfg(feature = "xep_0280")]
+            sent,
+            #[cfg(feature = "xep_0280")]
+            received,
+            #[cfg(feature = "xep_0280")]
+            private,
+            #[cfg(feature = "xep_0334")]
+            no_permanent_store,
+            #[cfg(feature = "xep_0334")]
+            no_store,
+            #[cfg(feature = "xep_0334")]
+            no_copy,
+            #[cfg(feature = "xep_0334")]
+            store,
         })
     }
 }
@@ -114,6 +176,30 @@ impl IntoElement for Message {
         #[cfg(feature = "xep_0060")]
         let builder = builder.push_child_opt(self.event.clone());
 
+        #[cfg(feature = "xep_0297")]
+        let builder = builder.push_child_opt(self.forwarded.clone());
+
+        #[cfg(feature = "xep_0280")]
+        let builder = builder.push_child_opt(self.sent.clone());
+
+        #[cfg(feature = "xep_0280")]
+        let builder = builder.push_child_opt(self.received.clone());
+
+        #[cfg(feature = "xep_0280")]
+        let builder = builder.push_child_opt(self.private.clone());
+
+        #[cfg(feature = "xep_0334")]
+        let builder = builder.push_child_opt(self.no_permanent_store);
+
+        #[cfg(feature = "xep_0334")]
+        let builder = builder.push_child_opt(self.no_store);
+
+        #[cfg(feature = "xep_0334")]
+        let builder = builder.push_child_opt(self.no_copy);
+
+        #[cfg(feature = "xep_0334")]
+        let builder = builder.push_child_opt(self.store);
+
         builder
     }
 }