Add more @mainactor

EmergeTools · Jan 29, 2025 · 97d14ad · 97d14ad
1 parent c90f85c
commit 97d14ad
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/Examples/DemoApp/DemoApp/TestViews/PreviewVariants.swift b/Examples/DemoApp/DemoApp/TestViews/PreviewVariants.swift
@@ -14,6 +14,7 @@ extension View {
 
 struct PreviewVariants: View {
 
+  @MainActor
   init(
     modifiers: [NamedViewModifier] = .previewDefault,
     layout: PreviewLayout = .device,

diff --git a/Sources/SnapshottingSwift/Initializer.swift b/Sources/SnapshottingSwift/Initializer.swift
@@ -20,11 +20,15 @@ public final class Initializer: NSObject, Sendable {
     super.init()
 
     #if !canImport(UIKit) || os(watchOS)
-    snapshots = Snapshots()
+    Task { @MainActor [weak self] in
+      guard let self else { return }
+      self.snapshots = Snapshots()
+    }
     #else
     NotificationCenter.default.addObserver(forName: UIApplication.didBecomeActiveNotification, object: nil, queue: .main) { [weak self] notification in
-      Task { @MainActor in
-        self?.snapshots = Snapshots()
+      Task { @MainActor [weak self] in
+        guard let self else { return }
+        self.snapshots = Snapshots()
       }
     }
     #endif