allow for loops and yielding collections

CSBiology · Feb 18, 2022 · 50fbcf3 · 50fbcf3
1 parent ba9f96a
commit 50fbcf3
Showing 1 changed file with 20 additions and 1 deletion.
diff --git a/src/LitXml/ElementBuilder.fs b/src/LitXml/ElementBuilder.fs
@@ -25,7 +25,21 @@ type ElementBuilder(name: string) =
 
     member inline _.Yield(n: #Element) = n
 
-    member inline this.Yield(b: ElementBuilder) =
+    member inline _.YieldFrom(ns: seq<#Element>) = 
+        Element(fun tb ->
+            Seq.iter (fun (n: #Element) -> n.Invoke(tb)) ns
+        )
+
+    member inline _.YieldFrom(bs: seq<ElementBuilder>) = 
+        Element(fun tb ->
+            bs
+            |> Seq.iter (fun b ->
+                tb.WriteStartElement(b.Name)
+                tb.WriteEndElement()
+            )           
+        )
+
+    member inline _.Yield(b: ElementBuilder) =
         Element(fun tb ->
             tb.WriteStartElement(b.Name)
             tb.WriteEndElement()
@@ -46,6 +60,11 @@ type ElementBuilder(name: string) =
 
     member inline _.Delay(n: unit -> Element) = n()
 
+    member inline _.For(ns: 'T seq, ex: 'T -> Element) = 
+        Element(fun tb ->
+            Seq.iter (fun n -> (ex n).Invoke(tb)) ns
+        )
+
     static member WriteTo(stream : System.IO.MemoryStream, element : Element) = 
         let writer = XmlWriter.Create(stream)
         element.Invoke(writer) |> ignore