added dismiss() method to KlaviyoWebViewDelegate

klaviyo · Jan 22, 2025 · 4741810 · 4741810
1 parent 044ae38
commit 4741810
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/Sources/KlaviyoUI/KlaviyoWebView/KlaviyoWebViewController.swift b/Sources/KlaviyoUI/KlaviyoWebView/KlaviyoWebViewController.swift
@@ -81,6 +81,11 @@ class KlaviyoWebViewController: UIViewController, WKUIDelegate, KlaviyoWebViewDe
         loadUrl()
     }
 
+    @MainActor
+    func dismiss() {
+        dismiss(animated: true)
+    }
+
     // MARK: - Scripts
 
     /// Configures the scripts to be injected into the website when the website loads.

diff --git a/Sources/KlaviyoUI/KlaviyoWebView/KlaviyoWebViewModel.swift b/Sources/KlaviyoUI/KlaviyoWebView/KlaviyoWebViewModel.swift
@@ -16,6 +16,9 @@ public protocol KlaviyoWebViewDelegate: AnyObject {
 
     @MainActor
     func evaluateJavaScript(_ script: String) async throws -> Any
+
+    @MainActor
+    func dismiss()
 }
 
 @_spi(KlaviyoPrivate)
@@ -48,6 +51,10 @@ public class KlaviyoWebViewModel: KlaviyoWebViewModeling {
         if message.name == "closeHandler" {
             // TODO: handle close button tap
             print("user tapped close button")
+
+            Task {
+                await delegate?.dismiss()
+            }
         }
     }
 }