Allow extensions

CHANGELOG

@@ -1,5 +1,8 @@
 # Changelog
 
+### Unreleased
+- Add ability to serialize and deserialize `Extensions` in an `AuthnRequest`
+
 ### 0.0.15
 
 - Updates dependencies

src/schema/authn_request.rs

@@ -1,4 +1,4 @@
-use crate::schema::{Conditions, Issuer, NameIdPolicy, Subject};
+use crate::schema::{Conditions, Extensions, Issuer, NameIdPolicy, Subject};
 use crate::signature::Signature;
 use chrono::prelude::*;
 use quick_xml::events::{BytesDecl, BytesEnd, BytesStart, BytesText, Event};
@@ -30,6 +30,9 @@ pub struct AuthnRequest {
     pub issuer: Option<Issuer>,
     #[serde(rename = "Signature")]
     pub signature: Option<Signature>,
+    #[serde(rename = "Extensions")]
+    #[serde(skip_deserializing)]
+    pub extensions: Option<Extensions>,
     #[serde(rename = "Subject")]
     pub subject: Option<Subject>,
     #[serde(rename = "NameIDPolicy")]
@@ -62,6 +65,7 @@ impl Default for AuthnRequest {
             consent: None,
             issuer: None,
             signature: None,
+            extensions: None,
             subject: None,
             name_id_policy: None,
             conditions: None,
@@ -207,6 +211,10 @@ impl TryFrom<&AuthnRequest> for Event<'_> {
             let event: Event<'_> = signature.try_into()?;
             writer.write_event(event)?;
         }
+        if let Some(extensions) = &value.extensions {
+            let event: Event<'_> = extensions.try_into()?;
+            writer.write_event(event)?;
+        }
         if let Some(subject) = &value.subject {
             let event: Event<'_> = subject.try_into()?;
             writer.write_event(event)?;

src/schema/extensions.rs

@@ -0,0 +1,56 @@
+use std::io::{Cursor, Write};
+
+use quick_xml::{
+    events::{BytesEnd, BytesStart, BytesText, Event},
+    Writer,
+};
+use serde::Deserialize;
+
+const NAME: &str = "saml2p:Extensions";
+
+#[derive(Clone, Debug, Deserialize, Hash, Eq, PartialEq, Ord, PartialOrd)]
+pub struct Extensions(pub Vec<String>);
+
+impl TryFrom<Extensions> for Event<'_> {
+    type Error = Box<dyn std::error::Error>;
+
+    fn try_from(value: Extensions) -> Result<Self, Self::Error> {
+        (&value).try_into()
+    }
+}
+
+impl TryFrom<&Extensions> for Event<'_> {
+    type Error = Box<dyn std::error::Error>;
+
+    fn try_from(value: &Extensions) -> Result<Self, Self::Error> {
+        let mut write_buf = Vec::new();
+        let mut writer = Writer::new(Cursor::new(&mut write_buf));
+        let root = BytesStart::from_content(NAME, NAME.len());
+        writer.write_event(Event::Start(root))?;
+
+        for extension in &value.0 {
+            writer.get_mut().write_all(extension.as_bytes())?;
+        }
+
+        writer.write_event(Event::End(BytesEnd::new(NAME)))?;
+        Ok(Event::Text(BytesText::from_escaped(String::from_utf8(
+            write_buf,
+        )?)))
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::traits::ToXml;
+
+    #[test]
+    fn extensions_xml_serialization() {
+        assert_eq!(
+            r#"<saml2p:Extensions><qqq a="b"/></saml2p:Extensions>"#,
+            Extensions(vec![r#"<qqq a="b"/>"#.to_string()])
+                .to_xml()
+                .unwrap(),
+        )
+    }
+}

src/schema/mod.rs

@@ -1,12 +1,14 @@
 pub mod authn_request;
 mod conditions;
+mod extensions;
 mod issuer;
 mod name_id_policy;
 mod response;
 mod subject;
 
 pub use authn_request::AuthnRequest;
 pub use conditions::*;
+pub use extensions::Extensions;
 pub use issuer::Issuer;
 pub use name_id_policy::NameIdPolicy;
 pub use response::Response;