使用 takeSnapshot 的 WKWebView 间歇性 blank/partial 截图

Question

我正在截取 javascript 评估的完成处理程序中的 WKVebView：

func loadPage(){
    let fileURL = URL(fileURLWithPath: "...")
    let baseUrl = URL(fileURLWithPath: "...")
    webView?.navigationDelegate = self
    webView.loadFileURL(fileURL, allowingReadAccessTo: baseUrl)
}

func webView(_ webView: WKWebView, didFinish navigation: WKNavigation!) {
    webView.evaluateJavaScript("....") { (result, error) in
        let configuration = WKSnapshotConfiguration()
        configuration.rect = CGRect(origin: .zero, size: (self.webView?.frame.size)!)
        self.webView!.takeSnapshot(with: configuration, completionHandler: { (image, error) in
            // do something with the image
        })
}

即使它在 webView.evaluateJavaScript 函数的完成处理程序中，内容如果经常没有完成渲染，我得到的要么是空白屏幕截图，要么是部分屏幕截图（只有一些元素存在）。

如何保证截图前渲染完成（我不想引入固定延迟）

Answer 1

由于WebView的内容比屏幕大，所以我们先获取内容的高度和宽度，然后再截图

  func webView(_ webView: WKWebView, didFinish navigation: WKNavigation!) {

    // Will get Height and Width of the Content

    self.webView.evaluateJavaScript("document.readyState", completionHandler: { (complete, error) in
        if complete != nil {
            self.webView.evaluateJavaScript("document.body.scrollHeight", completionHandler: { (height, error) in
                guard let contentHeight = height as? CGFloat else { print("Content height could not be obtained"); return }
                self.webView.evaluateJavaScript("document.body.scrollWidth", completionHandler: { [weak self](width, error) in
                    let contentWidth = width as! CGFloat
                    let rect = CGRect(x: 0, y: 0, width: contentWidth, height: contentHeight)
                    self?.takeScreenshot(rect)
                })
            })
        }
    })

}

一旦我们获得了高度和宽度，我们将调用下面的方法来捕获屏幕截图

func takeScreenshot(_ rect: CGRect) {
    webView.evaluateJavaScript(".....") { (result, error) in
        let configuration = WKSnapshotConfiguration()
        configuration.rect = rect//CGRect(origin: .zero, size: (self.webView?.frame.size)!)
        self.webView!.takeSnapshot(with: configuration, completionHandler: { (image, error) in
            // do something with the image
            print("image:\(image!)")
        })
    }
}

希望对您有所帮助。

Answer 2

您需要等到所有元素都加载到 webview 中。您可以在截图前添加定时器。

RunLoop.main.run（直到：Date.init（timeIntervalSinceNow：0.5））

使用 takeSnapshot 的 WKWebView 间歇性 blank/partial 截图

intermittent blank/partial screenshots of WKWebView using takeSnapshot

macos

cocoa

screenshot

swift

wkwebview