From b0b2b02a49b62c43e4103c5e69ff6903f2c6234e Mon Sep 17 00:00:00 2001
From: Clark <clark@signal.org>
Date: Mon, 24 Apr 2023 14:50:28 -0400
Subject: [PATCH] Fix draft edit messages not being cleared.

---
 .../securesms/conversation/ConversationParentFragment.java  | 1 +
 .../securesms/conversation/drafts/DraftViewModel.kt         | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/app/src/main/java/org/thoughtcrime/securesms/conversation/ConversationParentFragment.java b/app/src/main/java/org/thoughtcrime/securesms/conversation/ConversationParentFragment.java
index 7eb83bcf3d..b4ee6fae7d 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/conversation/ConversationParentFragment.java
+++ b/app/src/main/java/org/thoughtcrime/securesms/conversation/ConversationParentFragment.java
@@ -4170,6 +4170,7 @@ public class ConversationParentFragment extends Fragment
   @Override
   public void onExitEditMode() {
     updateToggleButtonState();
+    draftViewModel.deleteMessageEditDraft();
   }
 
   @Override
diff --git a/app/src/main/java/org/thoughtcrime/securesms/conversation/drafts/DraftViewModel.kt b/app/src/main/java/org/thoughtcrime/securesms/conversation/drafts/DraftViewModel.kt
index a33a56cd41..491e94dae3 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/conversation/drafts/DraftViewModel.kt
+++ b/app/src/main/java/org/thoughtcrime/securesms/conversation/drafts/DraftViewModel.kt
@@ -84,6 +84,12 @@ class DraftViewModel @JvmOverloads constructor(
     }
   }
 
+  fun deleteMessageEditDraft() {
+    store.update {
+      saveDrafts(it.copy(textDraft = null, bodyRangesDraft = null, messageEditDraft = null))
+    }
+  }
+
   fun setTextDraft(text: String, mentions: List<Mention>, styleBodyRanges: BodyRangeList?) {
     store.update {
       val mentionRanges: BodyRangeList? = MentionUtil.mentionsToBodyRangeList(mentions)