changes (#139)

Sources/SnapshotPreviewsCore/Orientation.swift

@@ -0,0 +1,35 @@
+#if canImport(UIKit) && !os(watchOS) && !os(visionOS) && !os(tvOS)
+import Foundation
+import UIKit
+import SwiftUI
+
+extension UIInterfaceOrientation {
+  func toInterfaceOrientationMask() -> UIInterfaceOrientationMask {
+      switch self {
+      case .portraitUpsideDown:
+          return .portraitUpsideDown
+      case .landscapeLeft:
+          return .landscapeLeft
+      case .landscapeRight:
+          return .landscapeRight
+      default:
+          return .portrait
+      }
+  }
+}
+
+extension InterfaceOrientation {
+  func toInterfaceOrientation() -> UIInterfaceOrientation {
+      switch self {
+      case .portraitUpsideDown:
+          return .portraitUpsideDown
+      case .landscapeLeft:
+          return .landscapeLeft
+      case .landscapeRight:
+          return .landscapeRight
+      default:
+        return .portrait
+      }
+  }
+}
+#endif

Sources/SnapshotPreviewsCore/UIKitRenderingStrategy.swift

@@ -25,16 +25,51 @@ public class UIKitRenderingStrategy: RenderingStrategy {
     self.window = window
   }
 
-  public var windowScene: UIWindowScene? {
+  private var windowScene: UIWindowScene? {
     window.windowScene
   }
 
   private let window: UIWindow
 
-  @MainActor public func render(
+  @MainActor
+  public func render(
     preview: SnapshotPreviewsCore.Preview,
-    completion: @escaping (SnapshotResult) -> Void)
-  {
+    completion: @escaping (SnapshotResult) -> Void
+  ) {
+    let previewOrientation = preview.orientation.toInterfaceOrientation()
+    if windowScene!.interfaceOrientation != previewOrientation {
+      var rotationError: (any Error)? = nil
+      if #available(iOS 16.0, *) {
+        windowScene!.requestGeometryUpdate(.iOS(interfaceOrientations: previewOrientation.toInterfaceOrientationMask())) { error in
+          NSLog("Rotation error handler: \(error) \(self.windowScene!.interfaceOrientation)")
+          rotationError = error
+        }
+      }
+
+      // Wait for rotation to complete or timeout
+      DispatchQueue.main.asyncAfter(deadline: .now() + 2) { [weak self] in
+        if let error = rotationError {
+          let result = SnapshotResult(
+            image: .failure(error),
+            precision: nil,
+            accessibilityEnabled: nil,
+            accessibilityMarkers: nil,
+            colorScheme: nil
+          )
+          completion(result)
+        } else {
+          self?.performRender(preview: preview, completion: completion)
+        }
+      }
+    } else {
+      performRender(preview: preview, completion: completion)
+    }
+  }
+
+  @MainActor private func performRender(
+    preview: SnapshotPreviewsCore.Preview,
+    completion: @escaping (SnapshotResult) -> Void
+  ) {
     UIView.setAnimationsEnabled(false)
     let view = preview.view()
     let controller = view.makeExpandingView(layout: preview.layout, window: window)