如何将iOS的WKWebView截图代码转为Mac OS下NSView截图？

没问题，我帮你把iOS的WKWebView截图逻辑转换成适配Mac AppKit的版本，替换掉你当前截取整个窗口的实现：

转换后的AppKit版本代码 #

extension WKWebView {
    var screenshot: NSImage {
        // 创建与WebView尺寸匹配的位图图像表示
        let bitmapRep = NSBitmapImageRep(bitmapDataPlanes: nil,
                                         pixelsWide: Int(bounds.width),
                                         pixelsHigh: Int(bounds.height),
                                         bitsPerSample: 8,
                                         samplesPerPixel: 4,
                                         hasAlpha: true,
                                         isPlanar: false,
                                         colorSpaceName: .calibratedRGB,
                                         bytesPerRow: 0,
                                         bitsPerPixel: 0)!
        
        // 保存当前图形状态，创建位图上下文并绘制WebView层级
        NSGraphicsContext.saveGraphicsState()
        NSGraphicsContext.current = NSGraphicsContext(bitmapImageRep: bitmapRep)
        drawHierarchy(in: bounds, afterScreenUpdates: true)
        NSGraphicsContext.restoreGraphicsState()
        
        // 基于位图表示生成NSImage并返回
        let image = NSImage(size: bounds.size)
        image.addRepresentation(bitmapRep)
        return image
    }
}

关键说明 #

• 和iOS版本的核心逻辑一致，都是通过drawHierarchy(in:afterScreenUpdates:)来绘制视图的完整渲染层级，确保WebView的内容被正确截取
• AppKit中使用NSBitmapImageRep来创建绘图上下文，这是Mac平台处理像素级绘图的标准方式，参数配置保证了截图的色彩和透明度正确
• 这个实现只会截取当前WKWebView的bounds区域，而不是整个屏幕，完全满足你针对特定视图截图的需求

内容的提问来源于stack exchange，提问作者Grant Emerson