Max size (#110)

EmergeTools · Mar 26, 2024 · 8a82556 · 8a82556
1 parent af5c23f
commit 8a82556
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/Sources/SnapshotPreviewsCore/View+Snapshot.swift b/Sources/SnapshotPreviewsCore/View+Snapshot.swift
@@ -13,6 +13,7 @@ import AccessibilitySnapshotCore
 
 public enum RenderingError: Error {
   case failedRendering(CGSize)
+  case maxSize(CGSize)
 }
 
 public struct SnapshotResult {
@@ -117,7 +118,8 @@ extension View {
     layout: PreviewLayout,
     renderingMode: EmergeRenderingMode?,
     rootVC: UIViewController,
-    targetView: UIView) -> Result<UIImage, RenderingError>
+    targetView: UIView,
+    maxSize: Double = 1_000_000) -> Result<UIImage, RenderingError>
   {
     let view = targetView
     let drawCode: (CGContext) -> Void
@@ -147,6 +149,9 @@ extension View {
         success = rootVC.view.render(size: targetSize, mode: renderingMode, context: ctx)
       }
     }
+    if targetSize.height > maxSize || targetSize.width > maxSize {
+      return .failure(RenderingError.maxSize(targetSize))
+    }
     let renderer = UIGraphicsImageRenderer(size: targetSize)
     let image = renderer.image { context in
       drawCode(context.cgContext)