Fix "Engagement Ended" after the survey if the attachment view is opened

MUIC-775
salemove · May 13, 2022 · ed2692b · ed2692b
1 parent 738378e
commit ed2692b
Showing 1 changed file with 17 additions and 9 deletions.
diff --git a/GliaWidgets/ViewModel/EngagementViewModel.swift b/GliaWidgets/ViewModel/EngagementViewModel.swift
@@ -63,15 +63,23 @@ class EngagementViewModel {
 
     func viewWillAppear() {
         if interactor.state == .ended(.byOperator) {
-            EngagementViewModel.alertPresenters.insert(self)
-            engagementAction?(
-                .showSingleActionAlert(
-                    alertConfiguration.operatorEndedEngagement,
-                    actionTapped: { [weak self] in
-                        self?.endSession()
-                    }
-                )
-            )
+            interactor.currentEngagement?.getSurvey { [weak self] result in
+                guard let self = self else { return }
+                switch result {
+                case .success(let survey) where survey != nil:
+                    return
+                default:
+                    EngagementViewModel.alertPresenters.insert(self)
+                    self.engagementAction?(
+                        .showSingleActionAlert(
+                            self.alertConfiguration.operatorEndedEngagement,
+                            actionTapped: { [weak self] in
+                                self?.endSession()
+                            }
+                        )
+                    )
+                }
+            }
         }
     }