在WKWebView中注入javascript以获取所单击图像的src属性

import WebKit

class ViewController: UIViewController, WKNavigationDelegate, WKUIDelegate {
    var webView: WKWebView!
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        let configuration = WKWebViewConfiguration()
        webView = WKWebView(frame: view.bounds, configuration: configuration)
        webView.navigationDelegate = self
        webView.uiDelegate = self
        
        view.addSubview(webView)
        
        if let url = URL(string: "https://example.com") {
            let request = URLRequest(url: url)
            webView.load(request)
        }
    }
    
    // 拦截网页的点击事件
    func webView(_ webView: WKWebView, decidePolicyFor navigationAction: WKNavigationAction, decisionHandler: @escaping (WKNavigationActionPolicy) -> Void) {
        if navigationAction.navigationType == .linkActivated {
            // 获取点击的元素
            let element = navigationAction.targetFrame?.request.url?.absoluteString
            
            // 判断是否为图片
            if let element = element, element.hasSuffix(".jpg") || element.hasSuffix(".png") {
                // 注入JavaScript获取图片的src属性
                let script = "var imgSrc = document.elementFromPoint(\(navigationAction.location.x), \(navigationAction.location.y)).src; imgSrc;"
                webView.evaluateJavaScript(script) { (result, error) in
                    if let src = result as? String {
                        // 处理或展示获取到的src属性
                        print("点击的图片src属性为：\(src)")
                    }
                }
            }
        }
        
        decisionHandler(.allow)
    }
}