Merge pull request #4 from trafi/feature/optional-stacking

Accept optional ImageModdable
trafi · Jul 24, 2020 · 0f42ca0 · 0f42ca0
2 parents 5b81ee2 + f5497b5
commit 0f42ca0
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/Sources/ImageMod/ImageModable.swift b/Sources/ImageMod/ImageModable.swift
@@ -79,7 +79,8 @@ public extension ImageModable {
 
     // MARK: Stack
 
-    func hStack(_ image: ImageModable, spacing: CGFloat = 0, alignment: UIStackView.Alignment = .center) -> ImageModable {
+    func hStack(_ image: ImageModable?, spacing: CGFloat = 0, alignment: UIStackView.Alignment = .center) -> ImageModable {
+        guard let image = image else { return self }
         let rect = image.canvasSize.vAlign(in: canvasSize, alignment: alignment)
         return self
             .padded(right: rect.maxX + spacing)
@@ -91,7 +92,8 @@ public extension ImageModable {
         )
     }
 
-    func vStack(_ image: ImageModable, spacing: CGFloat = 0, alignment: UIStackView.Alignment = .center) -> ImageModable {
+    func vStack(_ image: ImageModable?, spacing: CGFloat = 0, alignment: UIStackView.Alignment = .center) -> ImageModable {
+        guard let image = image else { return self }
         let rect = image.canvasSize.hAlign(in: canvasSize, alignment: alignment)
         return self
             .padded(bottom: rect.maxY + spacing)
@@ -103,7 +105,8 @@ public extension ImageModable {
         )
     }
 
-    func zStack(_ overlay: ImageModable, alignment: UIView.ContentMode = .center) -> ImageModable {
+    func zStack(_ overlay: ImageModable?, alignment: UIView.ContentMode = .center) -> ImageModable {
+        guard let overlay = overlay else { return self }
         let rect = overlay.canvasSize.align(in: canvasSize, alignment: alignment)
         return with(overlay
             .scaled(to: rect.size)