From a0e514dac9aad7bf2018476af6de7ce4b963d6ff Mon Sep 17 00:00:00 2001
From: Clark <clark@signal.org>
Date: Tue, 1 Aug 2023 13:17:25 -0400
Subject: [PATCH] Enqueue download jobs for edit messages.

---
 .../securesms/messages/EditMessageProcessor.kt       | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/app/src/main/java/org/thoughtcrime/securesms/messages/EditMessageProcessor.kt b/app/src/main/java/org/thoughtcrime/securesms/messages/EditMessageProcessor.kt
index fc54ec34bc..ed6881307f 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/messages/EditMessageProcessor.kt
+++ b/app/src/main/java/org/thoughtcrime/securesms/messages/EditMessageProcessor.kt
@@ -11,6 +11,7 @@ import org.thoughtcrime.securesms.database.model.databaseprotos.BodyRangeList
 import org.thoughtcrime.securesms.database.model.toBodyRangeList
 import org.thoughtcrime.securesms.dependencies.ApplicationDependencies
 import org.thoughtcrime.securesms.groups.GroupId
+import org.thoughtcrime.securesms.jobs.AttachmentDownloadJob
 import org.thoughtcrime.securesms.jobs.PushProcessEarlyMessagesJob
 import org.thoughtcrime.securesms.jobs.SendDeliveryReceiptJob
 import org.thoughtcrime.securesms.messages.MessageContentProcessorV2.Companion.log
@@ -160,7 +161,16 @@ object EditMessageProcessor {
       isPushMessage = true
     )
 
-    return SignalDatabase.messages.insertEditMessageInbox(-1, mediaMessage, targetMessage).orNull()
+    val insertResult = SignalDatabase.messages.insertEditMessageInbox(-1, mediaMessage, targetMessage).orNull()
+    if (insertResult?.insertedAttachments != null) {
+      SignalDatabase.runPostSuccessfulTransaction {
+        val downloadJobs: List<AttachmentDownloadJob> = insertResult.insertedAttachments.mapNotNull { (_, attachmentId) ->
+          AttachmentDownloadJob(insertResult.messageId, attachmentId, false)
+        }
+        ApplicationDependencies.getJobManager().addAll(downloadJobs)
+      }
+    }
+    return insertResult
   }
 
   private fun handleEditTextMessage(