Fix leak (#179)

EmergeTools · Aug 23, 2024 · 22cf568 · 22cf568
1 parent 08a5806
commit 22cf568
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/Sources/SnapshotPreviewsCore/AppKitRenderingStrategy.swift b/Sources/SnapshotPreviewsCore/AppKitRenderingStrategy.swift
@@ -43,9 +43,13 @@ public class AppKitRenderingStrategy: RenderingStrategy {
  }
  let vc = AppKitContainer(rootView: wrappedView)
  vc.setupView(layout: preview.layout)
- vc.rendered = { mode, precision, accessibilityEnabled in
+ // Reset the window size to default before adding the new view controller
+ window.contentViewController = NSViewController()
+ window.setContentSize(AppKitContainer.defaultSize)
+ window.contentViewController = vc
+ vc.rendered = { [weak vc] mode, precision, accessibilityEnabled in
  DispatchQueue.main.async {
- let image = vc.view.snapshot()
+ let image = vc?.view.snapshot()
  completion(
  SnapshotResult(
  image: image != nil ? .success(image!) : .failure(SwiftUIRenderingError.renderingError),
@@ -55,10 +59,6 @@ public class AppKitRenderingStrategy: RenderingStrategy {
  colorScheme: _colorScheme))
  }
  }
- // Reset the window size to default before adding the new view controller
- window.contentViewController = NSViewController()
- window.setContentSize(AppKitContainer.defaultSize)
- window.contentViewController = vc
  }
 }