feat: add TagStart attribute functions

1 files changed, 43 insertions, 1 deletions

diff --git a/src/tagged.rs b/src/tagged.rs
index 0150697..1fca458 100644
--- a/src/tagged.rs
+++ b/src/tagged.rs
@@ -5,7 +5,7 @@ use std::str::Utf8Error;
 
 use quick_xml::events::BytesStart;
 
-use crate::attribute::Iter as AttributeIter;
+use crate::attribute::{Attribute, Error as AttributeError, Iter as AttributeIter};
 
 /// The start tag of a tagged element.
 ///
@@ -26,6 +26,19 @@ impl<'a> TagStart<'a>
         }
     }
 
+    /// Returns `Self` with the specified attributes.
+    #[must_use]
+    pub fn with_attributes<Attrs>(mut self, attrs: Attrs) -> Self
+    where
+        Attrs: IntoIterator<Item = Attribute<'a>>,
+    {
+        self.inner = self
+            .inner
+            .with_attributes(attrs.into_iter().map(Attribute::into_inner));
+
+        self
+    }
+
     /// Returns the name.
     ///
     /// # Errors
@@ -50,6 +63,26 @@ impl<'a> TagStart<'a>
     {
         AttributeIter::new(self.inner.attributes())
     }
+
+    /// Returns a attribute.
+    ///
+    /// # Errors
+    /// Returns `Err` if a invalid attribute is found.
+    pub fn get_attribute(
+        &self,
+        attr_name: &str,
+    ) -> Result<Option<Attribute>, TagStartError>
+    {
+        for attr_result in self.inner.attributes().with_checks(false) {
+            let attr = attr_result.map_err(AttributeError::from)?;
+
+            if attr.key.as_ref() == attr_name.as_bytes() {
+                return Ok(Some(Attribute::from_inner(attr)));
+            }
+        }
+
+        Ok(None)
+    }
 }
 
 // Crate-local functions
@@ -60,3 +93,12 @@ impl<'a> TagStart<'a>
         Self { inner }
     }
 }
+
+/// `TagStart` error.
+#[derive(Debug, thiserror::Error)]
+pub enum TagStartError
+{
+    /// Invalid attribute.
+    #[error("Invalid attribute")]
+    InvalidAttribute(#[from] AttributeError),
+}