Added widget wrapper.

sparrowcode · Sep 10, 2023 · 87f135f · 87f135f
1 parent d64b4bf
commit 87f135f
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/Sources/SwiftUIExtension/Extensions/ViewExtension.swift b/Sources/SwiftUIExtension/Extensions/ViewExtension.swift
@@ -9,6 +9,10 @@ extension View {
     public func onRotate(perform action: @escaping (UIDeviceOrientation) -> Void) -> some View {
         self.modifier(DeviceRotationViewModifier(action: action))
     }
+
+    public func containerBackgroundForWidget<Background>(background: @escaping () -> Background) -> some View where Background: View {
+        modifier(ContainerBackgroundForWidgetModifier(background: background))
+    }
 }
 
 struct DeviceRotationViewModifier: ViewModifier {
@@ -23,3 +27,23 @@ struct DeviceRotationViewModifier: ViewModifier {
             }
     }
 }
+
+struct ContainerBackgroundForWidgetModifier<Background>: ViewModifier where Background: View {
+
+    let background: () -> Background
+
+    func body(content: Content) -> some View {
+        if #available(iOS 17.0, iOSApplicationExtension 17.0, watchOS 10.0, watchOSApplicationExtension 10.0, macOS 14.0, *) {
+            content
+                .containerBackground(for: .widget) {
+                    background()
+                }
+        } else {
+            ZStack {
+                background()
+                content
+                    .padding()
+            }
+        }
+    }
+}