1 files changed, 137 insertions, 0 deletions
diff --git a/stanza/src/stream_error.rs b/stanza/src/stream_error.rs
new file mode 100644
index 0000000..5ae04a6
--- /dev/null
+++ b/stanza/src/stream_error.rs
@@ -0,0 +1,137 @@
+use peanuts::{
+    element::{FromElement, IntoElement},
+    DeserializeError, Element, XML_NS,
+};
+
+pub const XMLNS: &str = "urn:ietf:params:xml:ns:xmpp-streams";
+
+#[derive(Clone, Debug)]
+pub enum Error {
+    BadFormat,
+    BadNamespacePrefix,
+    Conflict,
+    ConnectionTimeout,
+    HostGone,
+    HostUnknown,
+    ImproperAddressing,
+    InternalServerError,
+    InvalidFrom,
+    InvalidId,
+    InvalidNamespace,
+    InvalidXml,
+    NotAuthorized,
+    NotWellFormed,
+    PolicyViolation,
+    RemoteConnectionFailed,
+    Reset,
+    ResourceConstraint,
+    RestrictedXml,
+    SeeOtherHost(Option<String>),
+    SystemShutdown,
+    UndefinedCondition,
+    UnsupportedEncoding,
+    UnsupportedStanzaType,
+    UnsupportedVersion,
+}
+
+impl FromElement for Error {
+    fn from_element(mut element: peanuts::Element) -> peanuts::element::DeserializeResult<Self> {
+        let error;
+        match element.identify() {
+            (Some(XMLNS), "bad-format") => error = Error::BadFormat,
+            (Some(XMLNS), "bad-namespace-prefix") => error = Error::BadNamespacePrefix,
+            (Some(XMLNS), "conflict") => error = Error::Conflict,
+            (Some(XMLNS), "connection-timeout") => error = Error::ConnectionTimeout,
+            (Some(XMLNS), "host-gone") => error = Error::HostGone,
+            (Some(XMLNS), "host-unknown") => error = Error::HostUnknown,
+            (Some(XMLNS), "improper-addressing") => error = Error::ImproperAddressing,
+            (Some(XMLNS), "internal-server-error") => error = Error::InternalServerError,
+            (Some(XMLNS), "invalid-from") => error = Error::InvalidFrom,
+            (Some(XMLNS), "invalid-id") => error = Error::InvalidId,
+            (Some(XMLNS), "invalid-namespace") => error = Error::InvalidNamespace,
+            (Some(XMLNS), "invalid-xml") => error = Error::InvalidXml,
+            (Some(XMLNS), "not-authorized") => error = Error::NotAuthorized,
+            (Some(XMLNS), "not-well-formed") => error = Error::NotWellFormed,
+            (Some(XMLNS), "policy-violation") => error = Error::PolicyViolation,
+            (Some(XMLNS), "remote-connection-failed") => error = Error::RemoteConnectionFailed,
+            (Some(XMLNS), "reset") => error = Error::Reset,
+            (Some(XMLNS), "resource-constraint") => error = Error::ResourceConstraint,
+            (Some(XMLNS), "restricted-xml") => error = Error::RestrictedXml,
+            (Some(XMLNS), "see-other-host") => {
+                return Ok(Error::SeeOtherHost(element.pop_value_opt()?))
+            }
+            (Some(XMLNS), "system-shutdown") => error = Error::SystemShutdown,
+            (Some(XMLNS), "undefined-condition") => error = Error::UndefinedCondition,
+            (Some(XMLNS), "unsupported-encoding") => error = Error::UnsupportedEncoding,
+            (Some(XMLNS), "unsupported-stanza-type") => error = Error::UnsupportedStanzaType,
+            (Some(XMLNS), "unsupported-version") => error = Error::UnsupportedVersion,
+            _ => return Err(DeserializeError::UnexpectedElement(element)),
+        }
+        element.no_more_content()?;
+        return Ok(error);
+    }
+}
+
+impl IntoElement for Error {
+    fn builder(&self) -> peanuts::element::ElementBuilder {
+        match self {
+            Error::BadFormat => Element::builder("bad-format", Some(XMLNS)),
+            Error::BadNamespacePrefix => Element::builder("bad-namespace-prefix", Some(XMLNS)),
+            Error::Conflict => Element::builder("conflict", Some(XMLNS)),
+            Error::ConnectionTimeout => Element::builder("connection-timeout", Some(XMLNS)),
+            Error::HostGone => Element::builder("host-gone", Some(XMLNS)),
+            Error::HostUnknown => Element::builder("host-unknown", Some(XMLNS)),
+            Error::ImproperAddressing => Element::builder("improper-addressing", Some(XMLNS)),
+            Error::InternalServerError => Element::builder("internal-server-error", Some(XMLNS)),
+            Error::InvalidFrom => Element::builder("invalid-from", Some(XMLNS)),
+            Error::InvalidId => Element::builder("invalid-id", Some(XMLNS)),
+            Error::InvalidNamespace => Element::builder("invalid-namespace", Some(XMLNS)),
+            Error::InvalidXml => Element::builder("invalid-xml", Some(XMLNS)),
+            Error::NotAuthorized => Element::builder("not-authorized", Some(XMLNS)),
+            Error::NotWellFormed => Element::builder("not-well-formed", Some(XMLNS)),
+            Error::PolicyViolation => Element::builder("policy-violation", Some(XMLNS)),
+            Error::RemoteConnectionFailed => {
+                Element::builder("remote-connection-failed", Some(XMLNS))
+            }
+            Error::Reset => Element::builder("reset", Some(XMLNS)),
+            Error::ResourceConstraint => Element::builder("resource-constraint", Some(XMLNS)),
+            Error::RestrictedXml => Element::builder("restricted-xml", Some(XMLNS)),
+            Error::SeeOtherHost(h) => {
+                Element::builder("see-other-host", Some(XMLNS)).push_text_opt(h.clone())
+            }
+            Error::SystemShutdown => Element::builder("system-shutdown", Some(XMLNS)),
+            Error::UndefinedCondition => Element::builder("undefined-condition", Some(XMLNS)),
+            Error::UnsupportedEncoding => Element::builder("unsupported-encoding", Some(XMLNS)),
+            Error::UnsupportedStanzaType => {
+                Element::builder("unsupported-stanza-type", Some(XMLNS))
+            }
+            Error::UnsupportedVersion => Element::builder("unsupported-version", Some(XMLNS)),
+        }
+    }
+}
+
+#[derive(Clone, Debug)]
+pub struct Text {
+    text: Option<String>,
+    lang: Option<String>,
+}
+
+impl FromElement for Text {
+    fn from_element(mut element: peanuts::Element) -> peanuts::element::DeserializeResult<Self> {
+        element.check_name("text")?;
+        element.check_name(XMLNS)?;
+
+        let lang = element.attribute_opt_namespaced("lang", XML_NS)?;
+        let text = element.pop_value_opt()?;
+
+        Ok(Text { lang, text })
+    }
+}
+
+impl IntoElement for Text {
+    fn builder(&self) -> peanuts::element::ElementBuilder {
+        Element::builder("text", Some(XMLNS))
+            .push_attribute_opt_namespaced(XML_NS, "lang", self.lang.clone())
+            .push_text_opt(self.text.clone())
+    }
+}