웹뷰 전체 내역 캡쳐 및 앨범 저장

웹뷰 extension

extension WKWebview {
	func createImage(webView: WKWebView, completion: @escaping (UIImage?) -> ()) {
    	let originalFrame = webView.frame
        let originalConstraints = webView.constraints
        let originalScrollViewOffset = webView.scrollView.contentOffset
        
        let newSize = webView.scrollView.contentSize
        
        webView.removeConstraints(originalConstraints)
        webView.translatesAutoresizingMaskIntroConstraints = true
        webView.frame = CGRect(origin: .zero, size: newSize)
        webView.scrollView.contentOffset = .zero
        
        DispatchQueue.main.asyncAfter(deadline: .now() + 4.0) {
        	defer {
            	UIGraphicsEndImageContext()
                }
            UIGraphicsBeginImageContextWithOptions(newSize, false, 0)
            if let context = UIGraphicsGetCurrentContext() {
            	webView.scrollView.layer.render(in: context)
                
                webView.frame = originalFrame
                webView.translatesAutoresizingMaskIntroConstraints = false
                webView.addConstraints(originalConstraints)
                webView.scrollView.contentOffset = originalScrollViewOffset
                
                if let image = UIGraphicsGetImageFromCurrentImageContext() {
                	completion(image)
                } else {
                	completion(nil)
                }
            }
        }
    }
}

 

함수 호출 및 앨범 저장

self.webView.createImage(webView: webView) { image in
	if let img = image {
    	UIImageWriteToSavedPhotoAlbum(img, self, nil, nil)
    }
}