From 9feef8da9931419653010876654f8c6fa0a9a4c2 Mon Sep 17 00:00:00 2001
From: dab246 <tdvu@linagora.com>
Date: Thu, 26 Mar 2026 15:46:23 +0700
Subject: [PATCH] TF-4392 [Team Mailbox] Edit email in Drafts folder as new in
 the composer

---
 .../extensions/presentation_mailbox_extension.dart | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/model/lib/extensions/presentation_mailbox_extension.dart b/model/lib/extensions/presentation_mailbox_extension.dart
index 87a334ea8..a0f977725 100644
--- a/model/lib/extensions/presentation_mailbox_extension.dart
+++ b/model/lib/extensions/presentation_mailbox_extension.dart
@@ -53,7 +53,19 @@ extension PresentationMailboxExtension on PresentationMailbox {
         name?.name.toLowerCase() == PresentationMailbox.trashRole.toLowerCase();
   }
 
-  bool get isDrafts => role == PresentationMailbox.roleDrafts;
+  bool get isDrafts {
+    if (isPersonal) {
+      return role == PresentationMailbox.roleDrafts;
+    } else {
+      return isDraftsTeamMailbox;
+    }
+  }
+
+  bool get isDraftsTeamMailbox {
+    return isChildOfTeamMailboxes &&
+        name?.name.toLowerCase() ==
+            PresentationMailbox.draftsRole.toLowerCase();
+  }
 
   bool get isTemplates {
     if (isPersonal) {