如何为 Mac OS 截图 NSView?

How to Screenshot NSView for Mac OS?

我目前有一个功能可以在 Mac 应用程序中创建整个 window 的屏幕截图。但是,我想针对特定的视图。我知道在 iOS 中您可以执行以下操作:

extension WKWebView {

    var screenshot: UIImage {
        if #available(iOS 10.0, *) {
            return UIGraphicsImageRenderer(size: bounds.size).image { _ in
                drawHierarchy(in: CGRect(origin: .zero, size: bounds.size), afterScreenUpdates: true)
            }
        } else {
            UIGraphicsBeginImageContext(bounds.size)
            drawHierarchy(in: bounds, afterScreenUpdates: true)
            let image = UIGraphicsGetImageFromCurrentImageContext()!
            return image
        }
    }
}

我如何使用 AppKit 而不是 UIKit 将其转换为 Mac OS。下面是我当前计算的属性整个window.

的截图
extension WKWebView {

    var screenshot: NSImage {
        let displayID = CGWindowID() // cgmaindisplayid
        let imageRef = CGDisplayCreateImage(displayID)
        return NSImage(cgImage: imageRef!, size: (NSScreen.main?.frame.size)!)
    }
}

你可以使用NSView方法func dataWithPDF(inside rect: NSRect) -> Data

extension NSView {
    var image: NSImage? {
        return NSImage(data: dataWithPDF(inside: bounds))
    }
}