Added func return container from stack, fix bug axis

rosberry · Apr 26, 2022 · cd85680 · cd85680
1 parent 1a757e6
commit cd85680
Show file tree

Hide file tree

Showing 4 changed files with 56 additions and 0 deletions.
diff --git a/.DS_Store b/.DS_Store
diff --git a/.gitignore b/.gitignore
@@ -63,3 +63,4 @@ fastlane/report.xml
 fastlane/Preview.html
 fastlane/screenshots
 fastlane/test_output
+.DS_Store
diff --git a/Example/.DS_Store b/Example/.DS_Store
diff --git a/Sources/Extensions/Array+Stack.swift b/Sources/Extensions/Array+Stack.swift
@@ -25,6 +25,7 @@ public extension Collection where Iterator.Element: UIView, Self.Index == Int {
 
     func stack(axis: StackAxis, spacing: Number = 0.0, state: AnyHashable = DEFAULT_STATE) {
         for view in self {
+            view.frame = .zero
             guard view.superview != nil else {
                 assertionFailure("Can not configure stack relation without superview.")
                 return
@@ -79,4 +80,58 @@ public extension Collection where Iterator.Element: UIView, Self.Index == Int {
 
         }
     }
+
+    func stackToContainer(in view: UIView, axis: StackAxis, spacing: Number = 0.0, state: AnyHashable = DEFAULT_STATE, installerBlock: (ViewMaker) -> Void) -> UIView {
+        let containerView = UIView()
+        view.addSubview(containerView)
+        Maker.configure(view: containerView, for: state, installerBlock: installerBlock)
+
+        for view in self {
+            view.frame = .zero
+            containerView.addSubview(view)
+        }
+
+        guard containerView.nx_state == state else {
+            return containerView
+        }
+
+        let count = CGFloat(self.count)
+        var prevView = self[0]
+
+        switch axis {
+        case .horizontal:
+            let width = (containerView.bounds.width - (count - 1) * spacing.value) / count
+            let height = containerView.bounds.height
+
+            for (index, view) in self.enumerated() {
+                view.configureFrame { maker in
+                    maker.size(width: width, height: height)
+                    if index == 0 {
+                        maker.left()
+                    }
+                    else {
+                        maker.left(to: prevView.nui_right, inset: spacing)
+                    }
+                }
+                prevView = view
+            }
+        case .vertical:
+            let width = containerView.bounds.width
+            let height = (containerView.bounds.height - (count - 1) * spacing.value) / count
+
+            for (index, view) in self.enumerated() {
+                view.configureFrame { maker in
+                    maker.size(width: width, height: height)
+                    if index == 0 {
+                        maker.top()
+                    }
+                    else {
+                        maker.top(to: prevView.nui_bottom, inset: spacing)
+                    }
+                }
+                prevView = view
+            }
+        }
+        return containerView
+    }
 }