Capture a SwiftUI view to a UIImage

UIHostingController+Extension.swift

extension UIHostingController {
    
    func capture() -> UIImage {
        let size = sizeThatFits(in: UIScreen.main.bounds.size)
        view.bounds.size = size
        view.sizeToFit()
        UIGraphicsBeginImageContextWithOptions(view.bounds.size, view.isOpaque, 0)
        view.drawHierarchy(in: view.bounds, afterScreenUpdates: true)
        let snapshotImage: UIImage = UIGraphicsGetImageFromCurrentImageContext()!
            UIGraphicsEndImageContext()
        return snapshotImage
    }
}