[ScreenShottable] Screenshot a view using UIGraphicsImageRenderer #swift #uikit

ScreenShottable.swift

protocol Screenshottable {
    func screenshot() -> UIImage?
}

extension UIView: Screenshottable {
    func screenshot() -> UIImage? {
        let renderer = UIGraphicsImageRenderer(bounds: bounds)

        return renderer.image { _ in
            drawHierarchy(in: self.bounds, afterScreenUpdates: false)
        }
    }
}