From 22691450afc79e1d5e0268f265d03ea68c036f84 Mon Sep 17 00:00:00 2001
From: Sergey Mosin <sergey@srgdev.com>
Date: Thu, 10 Aug 2023 17:20:58 -0400
Subject: [PATCH] improve: if applicable set '\\answered' or '$forwarded' flag
 after a message is sent so the proper icon is shown in the message list view

---
 dev/View/Popup/Compose.js | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/dev/View/Popup/Compose.js b/dev/View/Popup/Compose.js
index 183e5b831..e1557d268 100644
--- a/dev/View/Popup/Compose.js
+++ b/dev/View/Popup/Compose.js
@@ -493,6 +493,19 @@ export class ComposePopupView extends AbstractViewPopup {
 										|| getNotification(Notifications.CantSendMessage));
 								}
 							} else {
+								if (arrayLength(this.aDraftInfo) > 0) {
+									const flag = {
+										'reply': '\\answered',
+										'forward': '$forwarded'
+									}[this.aDraftInfo[0]];
+									if (flag) {
+										const aFlags = MessageUserStore.message().flags();
+										if (aFlags.indexOf(flag) === -1) {
+											aFlags.push(flag);
+											MessageUserStore.message().flags(aFlags);
+										}
+									}
+								}
 								this.close();
 							}
 							setFolderETag(this.draftsFolder(), '');