TF-3385 Fix move and delete emails not work properly

lib/features/mailbox_dashboard/presentation/extensions/move_emails_to_mailbox_extension.dart

@@ -1,15 +1,15 @@
 import 'package:jmap_dart_client/jmap/mail/email/email.dart';
 import 'package:jmap_dart_client/jmap/mail/mailbox/mailbox.dart';
 import 'package:model/email/presentation_email.dart';
+import 'package:tmail_ui_user/features/email/domain/model/move_action.dart';
 import 'package:tmail_ui_user/features/mailbox_dashboard/presentation/controller/mailbox_dashboard_controller.dart';
 
 extension MoveEmailsToMailboxExtension on MailboxDashBoardController {
   void handleMoveEmailsToMailbox({
     required Map<MailboxId,List<EmailId>> originalMailboxIdsWithEmailIds,
     required MailboxId destinationMailboxId,
+    required MoveAction moveAction,
   }) {
-    if (destinationMailboxId == selectedMailbox.value?.id) return;
-
     final currentEmails = List<PresentationEmail>.from(
       emailsInCurrentMailbox,
     );
@@ -20,7 +20,17 @@ extension MoveEmailsToMailboxExtension on MailboxDashBoardController {
         return emailIds;
       },
     ).toList();
-    currentEmails.removeWhere((email) => movedEmailIds.contains(email.id));
-    updateEmailList(currentEmails);
+    final currentEmailsToBeMoved = currentEmails
+      .where((email) => movedEmailIds.contains(email.id))
+      .toList();
+    if (currentEmailsToBeMoved.isNotEmpty && destinationMailboxId != selectedMailbox.value?.id) {
+      emailsToBeUndo = currentEmailsToBeMoved;
+      currentEmails.removeWhere(currentEmailsToBeMoved.contains);
+      updateEmailList(currentEmails);
+    } else if (moveAction == MoveAction.undo && destinationMailboxId == selectedMailbox.value?.id) {
+      currentEmails.addAll(emailsToBeUndo);
+      currentEmails.sort((a, b) => b.receivedAt?.value.compareTo(a.receivedAt?.value ?? DateTime.now()) ?? -1);
+      updateEmailList(currentEmails);
+    }
   }
 }

lib/features/mailbox_dashboard/presentation/extensions/update_emails_with_new_mailbox_id_extension.dart

@@ -18,13 +18,14 @@ extension UpdateEmailsWithNewMailboxIdExtension on MailboxDashBoardController {
         return emailIds;
       },
     ).toList();
-    for (var email in currentEmails) {
-      if (!movedEmailIds.contains(email.id)) continue;
+    for (int i = 0; i < currentEmails.length; i++) {
+      if (!movedEmailIds.contains(currentEmails[i].id)) continue;
 
-      email = email.copyWith(
+      currentEmails[i] = currentEmails[i].copyWith(
         mailboxIds: {destinationMailboxId: true},
         mailboxContain: mapMailboxById[destinationMailboxId],
       );
     }
+    updateEmailList(currentEmails);
   }
 }