From b6ea9bb1b3c1d7a006da23e51ab9a1448635135c Mon Sep 17 00:00:00 2001
From: Ivan Vorobei <hello@ivanvorobei.io>
Date: Wed, 13 Sep 2023 11:52:23 +0300
Subject: [PATCH] Restructed.

---
 .../Compability/WidgetViewCompability.swift   | 24 +++++++++++++++++++
 .../Extensions/ViewExtension.swift            | 24 -------------------
 2 files changed, 24 insertions(+), 24 deletions(-)

diff --git a/Sources/SwiftUIExtension/Compability/WidgetViewCompability.swift b/Sources/SwiftUIExtension/Compability/WidgetViewCompability.swift
index 98d32b6..4d48d8e 100644
--- a/Sources/SwiftUIExtension/Compability/WidgetViewCompability.swift
+++ b/Sources/SwiftUIExtension/Compability/WidgetViewCompability.swift
@@ -6,6 +6,10 @@ extension View {
     public func containerBackgroundForWidgetCompability<Background>(background: @escaping () -> Background) -> some View where Background: View {
         modifier(ContainerBackgroundForWidgetCompabilityModifier(background: background))
     }
+
+    public func containerBackgroundForWidget<Background>(@ViewBuilder background: @escaping () -> Background) -> some View where Background: View {
+        modifier(ContainerBackgroundForWidgetModifier(background: background))
+    }
 }
 
 struct ContainerBackgroundForWidgetCompabilityModifier<Background>: ViewModifier where Background: View {
@@ -23,3 +27,23 @@ struct ContainerBackgroundForWidgetCompabilityModifier<Background>: 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()
+            }
+        }
+    }
+}
\ No newline at end of file
diff --git a/Sources/SwiftUIExtension/Extensions/ViewExtension.swift b/Sources/SwiftUIExtension/Extensions/ViewExtension.swift
index 554bcf1..6945505 100644
--- a/Sources/SwiftUIExtension/Extensions/ViewExtension.swift
+++ b/Sources/SwiftUIExtension/Extensions/ViewExtension.swift
@@ -5,30 +5,6 @@ extension View {
     public func onReceive(notification name: Notification.Name, perform action: @escaping (NotificationCenter.Publisher.Output) -> Void) -> some View {
         self.onReceive(NotificationCenter.default.publisher(for: name), perform: action)
     }
-    
-    public func containerBackgroundForWidget<Background>(@ViewBuilder background: @escaping () -> Background) -> some View where Background: View {
-        modifier(ContainerBackgroundForWidgetModifier(background: background))
-    }
-}
-
-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()
-            }
-        }
-    }
 }
 
 // MARK: - Only iOS