1 files changed, 21 insertions, 12 deletions

diff --git a/stanza/src/client/message.rs b/stanza/src/client/message.rs
index 78258ca..13ee6f1 100644
--- a/stanza/src/client/message.rs
+++ b/stanza/src/client/message.rs
@@ -1,10 +1,7 @@
 use std::str::FromStr;
 
 use jid::JID;
-use peanuts::{
-    element::{FromElement, IntoElement},
-    DeserializeError, Element, XML_NS,
-};
+use peanuts::{DeserializeError, Element, FromElement, IntoElement, XML_NS};
 
 #[cfg(feature = "xep_0060")]
 use crate::xep_0060::event::Event;
@@ -14,6 +11,8 @@ use crate::xep_0131::Headers;
 use crate::xep_0172::Nick;
 #[cfg(feature = "xep_0203")]
 use crate::xep_0203::Delay;
+#[cfg(feature = "xep_0297")]
+use crate::xep_0297::Forwarded;
 
 use super::XMLNS;
 
@@ -37,10 +36,12 @@ pub struct Message {
     pub nick: Option<Nick>,
     #[cfg(feature = "xep_0060")]
     pub event: Option<Event>,
+    #[cfg(feature = "xep_0297")]
+    pub forwarded: Option<Forwarded>,
 }
 
 impl FromElement for Message {
-    fn from_element(mut element: Element) -> peanuts::element::DeserializeResult<Self> {
+    fn from_element(mut element: Element) -> peanuts::DeserializeResult<Self> {
         element.check_name("message")?;
         element.check_namespace(XMLNS)?;
 
@@ -66,6 +67,9 @@ impl FromElement for Message {
         #[cfg(feature = "xep_0060")]
         let event = element.child_opt()?;
 
+        #[cfg(feature = "xep_0297")]
+        let forwarded = element.child_opt()?;
+
         Ok(Message {
             from,
             id,
@@ -83,12 +87,14 @@ impl FromElement for Message {
             nick,
             #[cfg(feature = "xep_0060")]
             event,
+            #[cfg(feature = "xep_0297")]
+            forwarded,
         })
     }
 }
 
 impl IntoElement for Message {
-    fn builder(&self) -> peanuts::element::ElementBuilder {
+    fn builder(&self) -> peanuts::ElementBuilder {
         let builder = Element::builder("message", Some(XMLNS))
             .push_attribute_opt("from", self.from.clone())
             .push_attribute_opt("id", self.id.clone())
@@ -117,6 +123,9 @@ 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());
+
         builder
     }
 }
@@ -166,7 +175,7 @@ pub struct Body {
 }
 
 impl FromElement for Body {
-    fn from_element(mut element: peanuts::Element) -> peanuts::element::DeserializeResult<Self> {
+    fn from_element(mut element: peanuts::Element) -> peanuts::DeserializeResult<Self> {
         element.check_name("body")?;
         element.check_namespace(XMLNS)?;
 
@@ -178,7 +187,7 @@ impl FromElement for Body {
 }
 
 impl IntoElement for Body {
-    fn builder(&self) -> peanuts::element::ElementBuilder {
+    fn builder(&self) -> peanuts::ElementBuilder {
         Element::builder("body", Some(XMLNS))
             .push_attribute_opt_namespaced(XML_NS, "lang", self.lang.clone())
             .push_text_opt(self.body.clone())
@@ -192,7 +201,7 @@ pub struct Subject {
 }
 
 impl FromElement for Subject {
-    fn from_element(mut element: peanuts::Element) -> peanuts::element::DeserializeResult<Self> {
+    fn from_element(mut element: peanuts::Element) -> peanuts::DeserializeResult<Self> {
         element.check_name("subject")?;
         element.check_namespace(XMLNS)?;
 
@@ -204,7 +213,7 @@ impl FromElement for Subject {
 }
 
 impl IntoElement for Subject {
-    fn builder(&self) -> peanuts::element::ElementBuilder {
+    fn builder(&self) -> peanuts::ElementBuilder {
         Element::builder("subject", Some(XMLNS))
             .push_attribute_opt_namespaced(XML_NS, "lang", self.lang.clone())
             .push_text_opt(self.subject.clone())
@@ -218,7 +227,7 @@ pub struct Thread {
 }
 
 impl FromElement for Thread {
-    fn from_element(mut element: peanuts::Element) -> peanuts::element::DeserializeResult<Self> {
+    fn from_element(mut element: peanuts::Element) -> peanuts::DeserializeResult<Self> {
         element.check_name("thread")?;
         element.check_namespace(XMLNS)?;
 
@@ -230,7 +239,7 @@ impl FromElement for Thread {
 }
 
 impl IntoElement for Thread {
-    fn builder(&self) -> peanuts::element::ElementBuilder {
+    fn builder(&self) -> peanuts::ElementBuilder {
         Element::builder("thread", Some(XMLNS))
             .push_attribute_opt("parent", self.parent.clone())
             .push_text_opt(self.thread.clone())