snapshot View to image in swift

view_snapshot.swift

/// Renders `View` as `UIImage`.
extension View {
    func snapshot(renderOnce: UnsafeMutablePointer<Bool>? = nil) -> UIImage? {
        if let flag = renderOnce, flag.pointee {
            return nil
        }
        
        renderOnce?.pointee = true
        let renderer        = ImageRenderer(content: self)
        renderer.scale      = UIScreen.main.scale
        
        return renderer.uiImage
    }
}

@MainActor
final class ViewSnapshotter<V: View> {
    private let view: V
    private var didRender = false

    init(view: V) {
        self.view = view
    }
    
    /// Continiously renders the view as a UIImage.
    @inlinable
    func render() -> UIImage? {
        view.snapshot()
    }
    
    /// Renders the view as a UIImage once and then never again.
    @inlinable
    func renderOnce() -> UIImage? {
        view.snapshot(renderOnce: &didRender)
    }
}