Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[CHNL-16564] update script and message handling #261

Merged
merged 5 commits into from
Jan 27, 2025
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
updated KlaviyoWebViewModel
ab1470 committed Jan 27, 2025
commit 4d1388c596e8f6cee897626271f86e44565f6c04
27 changes: 18 additions & 9 deletions Sources/KlaviyoUI/KlaviyoWebView/KlaviyoWebViewModel.swift
Original file line number Diff line number Diff line change
@@ -23,23 +23,29 @@ public protocol KlaviyoWebViewDelegate: AnyObject {

@_spi(KlaviyoPrivate)
public class KlaviyoWebViewModel: KlaviyoWebViewModeling {
public let url: URL
public let loadScripts: [String: WKUserScript]?
private enum MessageHandler: String, CaseIterable {
case closeHandler
}

public weak var delegate: KlaviyoWebViewDelegate?

public let url: URL
public let loadScripts: Set<WKUserScript>? = KlaviyoWebViewModel.initializeLoadScripts()
public var messageHandlers: Set<String>? = Set(MessageHandler.allCases.map(\.rawValue))

public let (navEventStream, navEventContinuation) = AsyncStream.makeStream(of: WKNavigationEvent.self)

public init(url: URL) {
self.url = url
loadScripts = KlaviyoWebViewModel.initializeLoadScripts()
// loadScripts = KlaviyoWebViewModel.initializeLoadScripts()
}

private static func initializeLoadScripts() -> [String: WKUserScript] {
var scripts: [String: WKUserScript] = [:]
private static func initializeLoadScripts() -> Set<WKUserScript> {
var scripts = Set<WKUserScript>()

if let closeHandlerScript = try? ResourceLoader.getResourceContents(path: "closeHandler", type: "js") {
let script = WKUserScript(source: closeHandlerScript, injectionTime: .atDocumentEnd, forMainFrameOnly: true)
scripts["closeHandler"] = script
scripts.insert(script)
}

return scripts
@@ -48,10 +54,13 @@ public class KlaviyoWebViewModel: KlaviyoWebViewModeling {
// MARK: handle WKWebView events

public func handleScriptMessage(_ message: WKScriptMessage) {
if message.name == "closeHandler" {
// TODO: handle close button tap
print("user tapped close button")
guard let handler = MessageHandler(rawValue: message.name) else {
// script message has no handler
return
}

switch handler {
case .closeHandler:
Task {
await delegate?.dismiss()
}