Supports for-in loop inside block (#77)

* 🌲 Update * 🌲 Update
FluidGroup · Sep 2, 2022 · 5f00b13 · 5f00b13
1 parent 239bc0b
commit 5f00b13
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 1 deletion.
diff --git a/MondrianLayout/Classic/MondrianArrayBuilder.swift b/MondrianLayout/Classic/MondrianArrayBuilder.swift
@@ -25,6 +25,10 @@ public struct MondrianArrayBuilder<Element> {
   public static func buildEither(second component: [Element]) -> [Element] {
     return component
   }
+
+  public static func buildArray(_ components: [[Element]]) -> [Element] {
+    components.flatMap { $0 }
+  }
 
   public static func buildExpression(_ element: Element?) -> [Element] {
     return element.map { [$0] } ?? []

diff --git a/MondrianLayout/Descriptors/VHStackContentBuilder.swift b/MondrianLayout/Descriptors/VHStackContentBuilder.swift
@@ -153,6 +153,10 @@ public enum VStackContentBuilder {
   public static func buildEither(second component: [Component]) -> [Component] {
     return component
   }
+
+  public static func buildArray(_ components: [[Component]]) -> [Component] {
+    return components.flatMap { $0 }
+  }
 
   public static func buildExpression(_ layoutGuides: [UILayoutGuide]...) -> [Component] {
     return layoutGuides.flatMap { $0 }.map {
@@ -227,6 +231,10 @@ public enum HStackContentBuilder {
   public static func buildEither(second component: [Component]) -> [Component] {
     return component
   }
+
+  public static func buildArray(_ components: [[Component]]) -> [Component] {
+    components.flatMap { $0 }
+  }
 
   public static func buildExpression(_ views: [UIView]...) -> [Component] {
     return views.flatMap { $0 }.map {

diff --git a/MondrianLayout/LayoutBlocks/VGridBlock.swift b/MondrianLayout/LayoutBlocks/VGridBlock.swift
@@ -263,6 +263,10 @@ public enum VGridContentBuilder {
   public static func buildEither(second component: [Component]) -> [Component] {
     return component
   }
+
+  public static func buildArray(_ components: [[Component]]) -> [Component] {
+    components.flatMap { $0 }
+  }
 
   public static func buildExpression(_ views: [UIView]...) -> [Component] {
     return views.flatMap { $0 }.map {

diff --git a/MondrianLayout/LayoutBlocks/ZStackBlock.swift b/MondrianLayout/LayoutBlocks/ZStackBlock.swift
@@ -263,7 +263,11 @@ public enum ZStackContentBuilder {
   public static func buildEither(second component: [Component]) -> [Component] {
     return component
   }
-
+
+  public static func buildArray(_ components: [[Component]]) -> [Component] {
+    components.flatMap { $0 }
+  }
+
   public static func buildExpression(_ layoutGuides: [UILayoutGuide]...) -> [Component] {
     return layoutGuides.flatMap { $0 }.map {
       .init(node: .layoutGuide(.init($0)))