feat: Update behavior according to action

Author: valentinperignon

Mail/Components/ActionsPanelButton.swift

@@ -42,7 +42,7 @@ struct ActionsPanelButton<Content: View>: View {
             panelSource: panelSource,
             popoverArrowEdge: popoverArrowEdge
         ) { action in
-            if action == .markAsUnread {
+            if action == .markAsUnread || action == .snooze || action == .modifySnooze {
                 dismiss()
             }
         }

Mail/Views/Bottom sheets/Actions/ActionsPanelViewModifier.swift

@@ -147,6 +147,6 @@ struct ActionsPanelViewModifier: ViewModifier {
             messages: messagesToSnooze,
             initialDate: initialSnoozedDate,
             folder: originFolder?.freezeIfNeeded()
-        ) { completionHandler?(.snooze) }
+        ) { completionHandler?($0) }
     }
 }

Mail/Views/Schedule/SnoozedFloatingPanel.swift

@@ -24,7 +24,7 @@ extension View {
         messages: [Message]?,
         initialDate: Date?,
         folder: Folder?,
-        completionHandler: (() -> Void)? = nil
+        completionHandler: ((Action) -> Void)? = nil
     ) -> some View {
         modifier(
             SnoozedFloatingPanel(
@@ -45,7 +45,7 @@ struct SnoozedFloatingPanel: ViewModifier {
     let messages: [Message]?
     let initialDate: Date?
     let folder: Folder?
-    let completionHandler: (() -> Void)?
+    let completionHandler: ((Action) -> Void)?
 
     func body(content: Content) -> some View {
         content
@@ -64,8 +64,8 @@ struct SnoozedFloatingPanel: ViewModifier {
         guard let messages else { return }
 
         Task {
-            try await actionsManager.performSnooze(messages: messages, date: date, originFolder: folder)
-            completionHandler?()
+            let action = try await actionsManager.performSnooze(messages: messages, date: date, originFolder: folder)
+            completionHandler?(action)
         }
     }
 }

Mail/Views/Thread/SnoozedThreadHeaderView.swift

@@ -48,9 +48,8 @@ struct SnoozedThreadHeaderView: View {
         .snoozedFloatingPanel(
             messages: messagesToSnooze,
             initialDate: date,
-            folder: folder,
-            completionHandler: dismiss.callAsFunction
-        )
+            folder: folder
+        ) { _ in dismiss() }
     }
 
     private func edit() {

MailCore/Cache/Actions/ActionsManager.swift

@@ -291,19 +291,22 @@ public class ActionsManager: ObservableObject {
         }
     }
 
-    public func performSnooze(messages: [Message], date: Date, originFolder: Folder?) async throws {
+    public func performSnooze(messages: [Message], date: Date, originFolder: Folder?) async throws -> Action {
         let messagesToExecuteAction = messages.lastMessagesToExecuteAction(
             currentMailboxEmail: mailboxManager.mailbox.email,
             currentFolder: originFolder
         )
 
-        if messagesToExecuteAction.allSatisfy(\.isSnoozed) {
+        let allMessagesAreSnoozed = messagesToExecuteAction.allSatisfy(\.isSnoozed)
+        if allMessagesAreSnoozed {
             try await mailboxManager.updateSnooze(messages: messagesToExecuteAction, until: date)
         } else {
             try await mailboxManager.snooze(messages: messagesToExecuteAction, until: date)
         }
 
         snackbarPresenter.show(message: MailResourcesStrings.Localizable.snackbarSnoozeSuccess(date.formatted()))
+
+        return allMessagesAreSnoozed ? .modifySnooze : .snooze
     }
 
     @MainActor