iosswiftuikitscreenshotuialertcontroller

How to programmatically take a screenshot of an app in iOS with UIAlertController as the top controller?


extension UIViewController {
    static func takeScreenshot() -> UIImage? {
        guard let topController = UIApplication.getTopController() else { return nil }
        
        if let view = topController.view {
            UIGraphicsBeginImageContextWithOptions(view.bounds.size, false, UIScreen.main.scale)
            
            defer { UIGraphicsEndImageContext() }
            
            if let context = UIGraphicsGetCurrentContext() {
                view.layer.render(in: context)
                let screenshot = UIGraphicsGetImageFromCurrentImageContext()
                return screenshot
            }
        }

        return nil
    }
}

This code works perfectly but the issue is when we have UIAlertController on top. I want to take a screenshot of the entire screen (not the status bar of course) but the controller as well as any alert controller that is presented on top. Right now the code is working fine if there is no alert controller on top.


Solution

  • This code did the job, for iOS 13 with iPad not supporting multiple scenes.

    func takeScreenshot() -> UIImage? {
        
        guard let window = UIApplication.shared.windows.first else {
            return nil
        }
        
        let renderer = UIGraphicsImageRenderer(bounds: window.bounds)
        let image = renderer.image { _ in
            window.drawHierarchy(in: window.bounds, afterScreenUpdates: true)
        }
        
        return image
    }