feat: add convenience functions for creating elements with children

Sponsored-by: author

1 files changed, 15 insertions, 5 deletions

diff --git a/src/lib.rs b/src/lib.rs
index d6a7027..a5e61b7 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -227,6 +227,18 @@ impl Element {
         self
     }
 
+    /// Append a child element, when constructing.
+    pub fn with_child(mut self, child: Element) -> Self {
+        self.children.push(Content::Element(child));
+        self
+    }
+
+    /// Append a text child, when constructing.
+    pub fn with_text(mut self, child: &str) -> Self {
+        self.children.push(Content::Text(child.into()));
+        self
+    }
+
     /// Return the [`Tag`] of the element.
     pub fn tag(&self) -> Tag {
         self.tag
@@ -517,11 +529,9 @@ mod test {
 
     #[test]
     fn visits_all_children() {
-        let mut e = Element::new(Tag::P);
-        e.push_text("hello ");
-        let mut world = Element::new(Tag::B);
-        world.push_text("world");
-        e.push_child(&world);
+        let e = Element::new(Tag::P)
+            .with_text("hello ")
+            .with_child(Element::new(Tag::B).with_text("world"));
 
         let mut collector = Collector::default();
         collector.visit(&e);