使用takeSnapshot的WKWebView的间歇性空白/部分屏幕截图

时间:2019-03-31 10:32:34

标签: swift macos cocoa screenshot wkwebview

我正在使用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函数的完成处理程序中,内容也常常无法完成渲染,并且我得到的是空白屏幕截图或部分屏幕截图(仅存在某些元素)。

如何确保在截屏之前完成渲染(我不想引入固定的延迟)

2 个答案:

答案 0 :(得分:2)

由于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!)")
        })
    }
}

希望它会对您有所帮助。

答案 1 :(得分:1)

您需要等待,直到所有元素都已加载到Web视图中。 您可以在截图之前添加计时器。

RunLoop.main.run(直到:Date.init(timeIntervalSinceNow:0.5))