Add docs to element mod

src/element/attribute.rs

@@ -8,39 +8,74 @@ use taffy::{
 #[derive(Clone, Copy, PartialEq, Eq)]
 /// Element attribute kind.
 pub enum AttributeKind {
+    /// Size attribute.
     Size,
+
+    /// Flex direction attribute.
     FlexDirection,
+
+    /// Item alignment attribute..
     AlignItems,
+
+    /// Content justification attribute.
     JustifyContent,
+
+    /// Click handler attribute.
     OnClick,
+
+    /// Mouse in handler attribute.
     OnMouseIn,
+
+    /// Mouse out handler attribute.
     OnMouseOut,
+
+    /// Background color attribute.
     BackgroundColor,
 }
 
 /// Element attribute value.
 pub enum AttributeValue {
+    /// Size attribute value.
     Size(Size<Dimension>),
+
+    /// Flex direction attribute value.
     FlexDirection(FlexDirection),
+
+    /// Item alignment attribute value.
     AlignItems(AlignItems),
+
+    /// Content justification attribute value.
     JustifyContent(JustifyContent),
+
+    /// Click handler attribute value.
     OnClick(Box<dyn FnMut(&mut Context, event::Click)>),
+
+    /// Mouse in handler attribute value.
     OnMouseIn(Box<dyn FnMut(&mut Context, event::MouseIn)>),
+
+    // Mouse out handler attribute value.
     OnMouseOut(Box<dyn FnMut(&mut Context, event::MouseOut)>),
+
+    /// Color attribute value.
     Color(Color4f),
 }
 
 /// Element attribute.
 pub struct Attribute {
+    /// Attribute kind.
     pub(super) kind: AttributeKind,
+
+    /// Attribute value.
     pub(super) value: AttributeValue,
 }
 
 impl Attribute {
+    /// Get the kind of this attribute.
     pub fn kind(&self) -> AttributeKind {
         self.kind
     }
 
+    /// Get the value of this attribute.
     pub fn value(&self) -> &AttributeValue {
         &self.value
     }

src/element/data.rs

@@ -78,14 +78,17 @@ pub struct ElementData {
 }
 
 impl ElementData {
+    /// Get a reference to the attribute of this kind if present.
     pub fn attr(&self, kind: AttributeKind) -> Option<&Attribute> {
         self.attributes.iter().find(|attr| attr.kind() == kind)
     }
 
+    /// Get a mutable reference to the attribute of this kind if present.
     pub fn attr_mut(&mut self, kind: AttributeKind) -> Option<&mut Attribute> {
         self.attributes.iter_mut().find(|attr| attr.kind() == kind)
     }
 
+    /// Remove an attribute by kind from this element.
     pub fn remove(&mut self, kind: AttributeKind) -> Option<Attribute> {
         self.attributes
             .iter()