如何像处理菜单栏App视图一样将Mac桌面（含菜单栏）渲染为PDF？

问题描述 #

我正在为Mac App Store制作一款菜单栏应用的截图，常规截图分辨率极低。目前我已实现将自身应用视图导出为PDF，再转换为SVG、PNG以获取高分辨率截图。现希望采用类似方式，将包含菜单栏的桌面视图渲染为PDF。以下是我当前处理应用视图的Swift代码：

let url = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask).first!.appendingPathComponent("\(Int(Date().timeIntervalSince1970 * 1000)).pdf") as CFURL
let size = Defaults[.windowSize]
var mediaBox = NSRect(origin: .zero, size: size)
let gc = CGContext(url, mediaBox: &mediaBox, nil)!
let nsgc = NSGraphicsContext(cgContext: gc, flipped: false)
NSGraphicsContext.current = nsgc
gc.beginPDFPage(nil); do {
  let view = contentViewController!.view as! NSHostingView<ContentView>
  gc.saveGState(); do {
    gc.drawFlipped(rect: mediaBox) {
      view.layer?.render(in: gc)
    }
  }; gc.restoreGState()

}; gc.endPDFPage()
NSGraphicsContext.current = nil
gc.closePDF()

解决方案 #

不能直接替换contentViewController!.view来获取包含菜单栏的桌面视图——Mac App Store应用受沙箱限制，无法直接访问系统级的桌面渲染内容。不过可以通过ScreenCaptureKit（MacOS 12+推荐）或CGDisplayStream捕获屏幕内容，再将捕获的内容渲染到PDF中。

使用ScreenCaptureKit实现（推荐） #

ScreenCaptureKit是Apple提供的现代屏幕捕获API，适配MacOS 12及以上，权限处理更规范，适合App Store上架应用。

1. 首先导入所需框架：

import ScreenCaptureKit
import AppKit

2. 实现捕获并渲染到PDF的逻辑：

func captureDesktopWithMenuBarToPDF() {
    // 获取主显示器
    guard let mainDisplay = SCShareableContent.displayList.first else { return }
    
    // 配置捕获参数，匹配屏幕分辨率
    let configuration = SCStreamConfiguration()
    configuration.width = Int(NSScreen.main?.frame.width ?? 0)
    configuration.height = Int(NSScreen.main?.frame.height ?? 0)
    configuration.pixelFormat = .RGBA8888
    
    do {
        // 创建捕获流
        let stream = try SCStream(shareableContent: SCShareableContent(display: mainDisplay), 
                                 configuration: configuration, 
                                 delegate: self)
        try stream.startCapture()
    } catch {
        print("屏幕捕获失败: \(error.localizedDescription)")
    }
}

// 实现SCStreamDelegate协议，处理捕获到的帧数据
extension YourViewController: SCStreamDelegate {
    func stream(_ stream: SCStream, didOutputSampleBuffer sampleBuffer: CMSampleBuffer, of type: SCStreamOutputType) {
        // 只处理屏幕捕获帧
        guard type == .screen, let imageBuffer = CMSampleBufferGetImageBuffer(sampleBuffer) else { return }
        
        // 将捕获的帧转换为NSImage
        let ciImage = CIImage(cvImageBuffer: imageBuffer)
        let imageRep = NSCIImageRep(ciImage: ciImage)
        let desktopImage = NSImage(size: imageRep.size)
        desktopImage.addRepresentation(imageRep)
        
        // 停止捕获，避免重复触发
        stream.stopCapture()
        
        // 把NSImage渲染到PDF，复用你原有的PDF生成逻辑
        let pdfURL = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask)
            .first!
            .appendingPathComponent("\(Int(Date().timeIntervalSince1970 * 1000)).pdf") as CFURL
        
        let mediaBox = NSRect(origin: .zero, size: desktopImage.size)
        guard let pdfContext = CGContext(pdfURL, mediaBox: &mediaBox, nil) else { return }
        
        let graphicsContext = NSGraphicsContext(cgContext: pdfContext, flipped: false)
        NSGraphicsContext.current = graphicsContext
        
        pdfContext.beginPDFPage(nil)
        pdfContext.saveGState()
        
        // 翻转上下文以匹配Mac的坐标系统
        pdfContext.drawFlipped(rect: mediaBox) {
            desktopImage.draw(in: mediaBox)
        }
        
        pdfContext.restoreGState()
        pdfContext.endPDFPage()
        
        NSGraphicsContext.current = nil
        pdfContext.closePDF()
    }
    
    func stream(_ stream: SCStream, didStopWithError error: Error) {
        print("捕获流停止: \(error.localizedDescription)")
    }
}

关键注意事项 #

• 权限配置：Mac App Store应用必须在Info.plist中添加NSDesktopFolderUsageDescription（说明屏幕捕获的用途），并在Xcode的「Signing & Capabilities」面板中添加「Screen Capture」权限，否则会被拒绝上架。
• 兼容性：如果需要支持MacOS 11及以下版本，可以替换为CGDisplayStream实现，但ScreenCaptureKit的API更简洁稳定。
• 沙箱限制：所有屏幕捕获操作必须获得用户授权，首次调用时系统会弹出权限请求对话框，用户同意后才能正常工作。

内容的提问来源于stack exchange，提问作者Peter R