使用Swift 5 xcode 13将屏幕截图保存为pdf或高分辨率图像

import UIKit
import PDFKit

func captureScreenshot(saveAsPDF: Bool) -> URL? {
    let screenSize = UIScreen.main.bounds.size
    let format = UIGraphicsImageRendererFormat()
    format.scale = UIScreen.main.scale
    let renderer = UIGraphicsImageRenderer(size: screenSize, format: format)
    let image = renderer.image { context in
        UIApplication.shared.keyWindow?.layer.render(in: context.cgContext)
    }
    
    if saveAsPDF {
        let pdfData = NSMutableData()
        let pdfConsumer = CGDataConsumer(data: pdfData as CFMutableData)!
        let mediaBox = CGRect(x: 0, y: 0, width: screenSize.width, height: screenSize.height)
        let pdfContext = CGContext(consumer: pdfConsumer, mediaBox: &mediaBox, nil)!
        
        pdfContext.beginPage(mediaBox: &mediaBox)
        pdfContext.draw(image.cgImage!, in: mediaBox)
        pdfContext.endPage()
        pdfContext.closePDF()
        
        let fileManager = FileManager.default
        let documentsPath = NSSearchPathForDirectoriesInDomains(.documentDirectory, .userDomainMask, true)[0]
        let filePath = (documentsPath as NSString).appendingPathComponent("screenshot.pdf")
        fileManager.createFile(atPath: filePath, contents: pdfData as Data, attributes: nil)
        
        return URL(fileURLWithPath: filePath)
    } else {
        guard let data = image.jpegData(compressionQuality: 1.0) else { return nil }
        
        let fileManager = FileManager.default
        let documentsPath = NSSearchPathForDirectoriesInDomains(.documentDirectory, .userDomainMask, true)[0]
        let filePath = (documentsPath as NSString).appendingPathComponent("screenshot.jpg")
        fileManager.createFile(atPath: filePath, contents: data, attributes: nil)
        
        return URL(fileURLWithPath: filePath)
    }
}

if let screenshotURL = captureScreenshot(saveAsPDF: true) {
    print("屏幕截图已保存为PDF：\(screenshotURL.absoluteString)")
} else {
    print("无法保存屏幕截图")
}