APP(iOS)端 从网页拿到第一张图片

class GetWebviewImageHelper:NSObject, WKScriptMessageHandler  {

    

    

    var didGetImg: ((UIImage?) -> (Void))?

    

    init(view:WKWebView) {

        super.init()

        self.initWebket(view)

    }

    

//初始化 webview 并添加 对 "getimage"(此处是死的) 监听

    privatefunc initWebket(webView:WKWebView) {

        webView.configuration.userContentController.addScriptMessageHandler(self,name: "getimage")

    }

    ///获取网页图片 //(js 代码) 

    func getWebViewImages(view:WKWebView) {

        let url ="javascript:(function(){var imgs = document.getElementsByTagName(\"img\");if (imgs.length > 0) {" +

            "window.webkit.messageHandlers.getimage.postMessage(imgs[0].src);" +

            "}" +"})()"

        view.evaluateJavaScript(url) { (obj, err)in

            iflet error = err {

                debugPrint(error.domain)

            }

        }

    }

    

    

    ///受到回调

    func userContentController(userContentController:WKUserContentController, didReceiveScriptMessage message:WKScriptMessage) {

        

        iflet urlStr = message.bodyas? String {

            

            let manager =SDWebImageDownloader()

            manager.downloadImageWithURL(NSURL(string: urlStr), options:SDWebImageDownloaderOptions(), progress:nil, completed: { (img, data, error,bol) in

               self.didGetImg?(img)

            })

        }

        

    }

    

    //移除监听

    func releaseMessageHandler(webView:WKWebView?) {

        iflet wb = webView {

            wb.configuration.userContentController.removeScriptMessageHandlerForName("getimage")

        }

    }

    

}

上面的类是获取 image类 需要在 wekwebview 实例化

然后在webview 的 代理方法里面调用  getWebViewImages 方法

   func webView(webView:WKWebView, didFinishNavigation navigation:WKNavigation!) {

        iflet title = webView.title {

            webTitle = title

            netLoadFinsh?(title)

            

            iflet h = helper {

                h.getWebViewImages(webView)

            }

        }

    }

同时在deinit 时 移除对webview的监听

  deinit {

        iflet h = helper {

            h.releaseMessageHandler(webView)

        }

    }