Ensure toast closures are called on the main thread (#3347)

duckduckgo · Sep 12, 2024 · 1b4e29a · 1b4e29a
1 parent 90b9d3d
commit 1b4e29a
Showing 1 changed file with 11 additions and 7 deletions.
diff --git a/DuckDuckGo/MainViewController.swift b/DuckDuckGo/MainViewController.swift
@@ -1381,14 +1381,18 @@ class MainViewController: UIViewController {
 
     private func makeBrokenSitePromptViewHostingController(event: UserBehaviorEvent) -> UIHostingController<BrokenSitePromptView> {
         let viewModel = BrokenSitePromptViewModel(onDidDismiss: { [weak self] in
-            self?.hideNotification()
-            self?.brokenSitePromptLimiter.didDismissToast()
-            self?.brokenSitePromptViewHostingController = nil
+            Task { @MainActor in
+                self?.hideNotification()
+                self?.brokenSitePromptLimiter.didDismissToast()
+                self?.brokenSitePromptViewHostingController = nil
+            }
         }, onDidSubmit: { [weak self] in
-            self?.segueToReportBrokenSite(entryPoint: .prompt(event.rawValue))
-            self?.hideNotification()
-            self?.brokenSitePromptLimiter.didOpenReport()
-            self?.brokenSitePromptViewHostingController = nil
+            Task { @MainActor in
+                self?.segueToReportBrokenSite(entryPoint: .prompt(event.rawValue))
+                self?.hideNotification()
+                self?.brokenSitePromptLimiter.didOpenReport()
+                self?.brokenSitePromptViewHostingController = nil
+            }
         })
         return UIHostingController(rootView: BrokenSitePromptView(viewModel: viewModel), ignoreSafeArea: true)
     }