WKWebView -在Swift - iOS中重定向时向URL添加参数

import UIKit
import WebKit

class ViewController: UIViewController, WKNavigationDelegate {
    var webView: WKWebView!
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        // 创建WKWebView实例
        webView = WKWebView(frame: view.bounds)
        webView.navigationDelegate = self
        view.addSubview(webView)
        
        // 加载URL
        let url = URL(string: "https://example.com")
        let request = URLRequest(url: url!)
        webView.load(request)
    }
    
    // WKNavigationDelegate方法，用于处理导航动作
    func webView(_ webView: WKWebView, decidePolicyFor navigationAction: WKNavigationAction, decisionHandler: @escaping (WKNavigationActionPolicy) -> Void) {
        if let url = navigationAction.request.url {
            // 解析URL
            var components = URLComponents(url: url, resolvingAgainstBaseURL: false)
            
            // 添加参数
            let newQueryItem = URLQueryItem(name: "param", value: "value")
            if let queryItems = components?.queryItems {
                components?.queryItems = queryItems + [newQueryItem]
            } else {
                components?.queryItems = [newQueryItem]
            }
            
            // 获取重定向后的URL
            if let newURL = components?.url {
                // 加载重定向后的URL
                webView.load(URLRequest(url: newURL))
                decisionHandler(.cancel)
                return
            }
        }
        
        decisionHandler(.allow)
    }
}