Merge pull request #241 from klaviyo/ab/CHNL-14729/handle-close-events

handle WKWebView `close` event
klaviyo · Dec 18, 2024 · 2fbebac · 2fbebac
2 parents 18fcbd5 + 8ba7de1
commit 2fbebac
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 2 deletions.
diff --git a/Sources/KlaviyoUI/KlaviyoWebView/KlaviyoWebViewModel.swift b/Sources/KlaviyoUI/KlaviyoWebView/KlaviyoWebViewModel.swift
@@ -27,7 +27,10 @@ class KlaviyoWebViewModel: KlaviyoWebViewModeling {
     private static func initializeLoadScripts() -> [String: WKUserScript] {
         var scripts: [String: WKUserScript] = [:]
 
-        // TODO: initialize scripts
+        if let closeHandlerScript = try? FileIO.getFileContents(path: "closeHandler", type: "js") {
+            let script = WKUserScript(source: closeHandlerScript, injectionTime: .atDocumentEnd, forMainFrameOnly: true)
+            scripts["closeHandler"] = script
+        }
 
         return scripts
     }
@@ -39,6 +42,9 @@ class KlaviyoWebViewModel: KlaviyoWebViewModeling {
     }
 
     func handleScriptMessage(_ message: WKScriptMessage) {
-        // TODO: handle script message
+        if message.name == "closeHandler" {
+            // TODO: handle close button tap
+            print("user tapped close button")
+        }
     }
 }
diff --git a/Sources/KlaviyoUI/KlaviyoWebView/Resources/Scripts/closeHandler.js b/Sources/KlaviyoUI/KlaviyoWebView/Resources/Scripts/closeHandler.js
@@ -0,0 +1,13 @@
+//
+//  closeHandler.js
+//  klaviyo-swift-sdk
+//
+//  Created by Andrew Balmer on 12/18/24.
+//
+
+
+window.addEventListener("klaviyoForms", function(e) {
+  if (e.detail.type == 'close') {
+    window.webkit?.messageHandlers?.closeHandler?.postMessage('close');
+  }
+});