Swift integration

import WebKit

class AgentChatViewController: UIViewController, WKScriptMessageHandler {
    var webView: WKWebView!

    override func viewDidLoad() {
        super.viewDidLoad()
        let config = WKWebViewConfiguration()
        config.userContentController.add(self, name: "tarsbridge")

        // Forwards every tars: message from the page to the tarsbridge handler
        let relayScript = """
        window.addEventListener('message', function(event) {
            var msg = event.data;
            if (msg && typeof msg.type === 'string' && msg.type.indexOf('tars:') === 0) {
                window.webkit.messageHandlers.tarsbridge.postMessage(msg);
            }
        });
        """
        config.userContentController.addUserScript(
            WKUserScript(source: relayScript, injectionTime: .atDocumentStart, forMainFrameOnly: true)
        )

        let initScript = """
        window.addEventListener('load', function() {
            window.postMessage({
                type: 'tars:init', platform: 'ios', version: '1.0',
                chrome: { showHeader: false, showCloseButton: true }
            }, '*');
        });
        """
        config.userContentController.addUserScript(
            WKUserScript(source: initScript, injectionTime: .atDocumentEnd, forMainFrameOnly: true)
        )

        webView = WKWebView(frame: view.bounds, configuration: config)
        view.addSubview(webView)
        webView.load(URLRequest(url: URL(string: "https://YOUR_WIDGET_HOST/widget/YOUR_AGENT_ID?region=YOUR_REGION")!))
    }

    func userContentController(_ controller: WKUserContentController, didReceive message: WKScriptMessage) {
        guard let body = message.body as? [String: Any],
              let type = body["type"] as? String else { return }
        switch type {
        case "tars:navigate":
            if let route = body["route"] as? String {
                // Navigate to the native route, params in body["params"]
                _ = route
            }
        case "tars:close":
            dismiss(animated: true)
        case "tars:data":
            // body["variables"] holds the captured conversation variables
            break
        default: break
        }
    }
}