监听 web 容器内的网络请求（错误的方案）

news2024/7/6 20:12:20

需求

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

实现

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

实现步骤

创建自定义 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) {
        // 处理请求取消逻辑
    }
}

配置 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)

加载 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 内的所有网络请求。这种方法不仅可以用于调试和日志记录，还可以实现高级的请求拦截和处理逻辑。

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

上面的代码有一个问题：WKWebView 的 setURLSchemeHandler(_:forURLScheme:) 方法不允许处理 https 和 http 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)
}