需求

iOS 项目中 wkwebview 实现的 web 容器,需要监听 web 容器内的所有网络请求

实现

在 iOS 项目中使用 WKWebView 实现的 Web 容器,监听 Web 容器内的网络请求是一个常见需求。可以通过实现 WKURLSchemeHandler 协议来处理自定义的 URL scheme,从而拦截和监听网络请求。以下是一个优秀的实现方案:

实现步骤

  1. 创建自定义 URL Scheme Handler

首先,创建一个新的类实现 WKURLSchemeHandler 协议,例如 CustomURLSchemeHandler

import WebKit

class CustomURLSchemeHandler: NSObject, WKURLSchemeHandler {
    func webView(_ webView: WKWebView, start urlSchemeTask: WKURLSchemeTask) {
        let url = urlSchemeTask.request.url!
        
        // 打印或处理请求
        print("Request URL: \(url.absoluteString)")
        
        // 这里可以进行自定义处理,如加载本地资源或转发请求
        // 示例:直接返回一个空白页面
        let response = HTTPURLResponse(url: url, statusCode: 200, httpVersion: nil, headerFields: nil)!
        let data = Data()
        urlSchemeTask.didReceive(response)
        urlSchemeTask.didReceive(data)
        urlSchemeTask.didFinish()
    }
    
    func webView(_ webView: WKWebView, stop urlSchemeTask: WKURLSchemeTask) {
        // 处理请求取消逻辑
    }
}
  1. 配置 WKWebViewConfiguration 并注册自定义 Scheme Handler

创建并配置 WKWebViewConfiguration,然后注册自定义的 URL Scheme Handler:

let config = WKWebViewConfiguration()
let schemeHandler = CustomURLSchemeHandler()
config.setURLSchemeHandler(schemeHandler, forURLScheme: "https")

let webView = WKWebView(frame: .zero, configuration: config)
  1. 加载 Web 内容

最后,使用 WKWebView 加载需要的 Web 内容:

let request = URLRequest(url: URL(string: "https://example.com")!)
webView.load(request)

完整示例代码

以下是完整的实现代码:

import UIKit
import WebKit

class ViewController: UIViewController {
    var webView: WKWebView!
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        let config = WKWebViewConfiguration()
        let schemeHandler = CustomURLSchemeHandler()
        config.setURLSchemeHandler(schemeHandler, forURLScheme: "https")
        
        webView = WKWebView(frame: self.view.bounds, configuration: config)
        self.view.addSubview(webView)
        
        let request = URLRequest(url: URL(string: "https://example.com")!)
        webView.load(request)
    }
}

class CustomURLSchemeHandler: NSObject, WKURLSchemeHandler {
    func webView(_ webView: WKWebView, start urlSchemeTask: WKURLSchemeTask) {
        let url = urlSchemeTask.request.url!
        
        print("Request URL: \(url.absoluteString)")
        
        // 这里可以进行自定义处理,如加载本地资源或转发请求
        // 示例:直接返回一个空白页面
        let response = HTTPURLResponse(url: url, statusCode: 200, httpVersion: nil, headerFields: nil)!
        let data = Data()
        urlSchemeTask.didReceive(response)
        urlSchemeTask.didReceive(data)
        urlSchemeTask.didFinish()
    }
    
    func webView(_ webView: WKWebView, stop urlSchemeTask: WKURLSchemeTask) {
        // 处理请求取消逻辑
    }
}

总结

通过实现 WKURLSchemeHandler 并注册自定义的 URL scheme,可以轻松监听并处理 WKWebView 内的所有网络请求。这种方法不仅可以用于调试和日志记录,还可以实现高级的请求拦截和处理逻辑。


更新:这是一个错误的方案

上面的代码有一个问题:WKWebViewsetURLSchemeHandler(_:forURLScheme:) 方法不允许处理 httpshttp URL scheme。这是因为这些 URL schemes 是由系统处理的,不能被自定义的 scheme handler 覆盖。

这里是修正的代码示例,使用一个自定义的 URL scheme,例如 customscheme

import WebKit

class CustomURLSchemeHandler: NSObject, WKURLSchemeHandler {
    func webView(_ webView: WKWebView, start urlSchemeTask: WKURLSchemeTask) {
        // 自定义处理 URL 请求的逻辑
    }
    
    func webView(_ webView: WKWebView, stop urlSchemeTask: WKURLSchemeTask) {
        // 自定义停止处理 URL 请求的逻辑
    }
}

let config = WKWebViewConfiguration()
let schemeHandler = CustomURLSchemeHandler()
config.setURLSchemeHandler(schemeHandler, forURLScheme: "customscheme")

let webView = WKWebView(frame: .zero, configuration: config)

// 测试加载自定义 scheme 的 URL
if let url = URL(string: "customscheme://example") {
    let request = URLRequest(url: url)
    webView.load(request)
}

在这个示例中,我们使用了一个自定义的 URL scheme (customscheme),并创建了一个自定义的 WKURLSchemeHandler 来处理这个 scheme 的 URL 请求。这样就避免了覆盖系统处理的 httpshttp URL schemes 的问题。

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐