From fd3e88707c34178a464249543d1ce3c7a0d44042 Mon Sep 17 00:00:00 2001
From: Cody Henthorne <cody@signal.org>
Date: Thu, 9 Sep 2021 10:06:49 -0400
Subject: [PATCH] Fix preupload in new Media Send flow.

---
 .../securesms/mediasend/v2/MediaSelectionRepository.kt     | 2 +-
 .../securesms/mediasend/v2/MediaSelectionViewModel.kt      | 7 ++++++-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/app/src/main/java/org/thoughtcrime/securesms/mediasend/v2/MediaSelectionRepository.kt b/app/src/main/java/org/thoughtcrime/securesms/mediasend/v2/MediaSelectionRepository.kt
index f5fc91c31..69db057c2 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/mediasend/v2/MediaSelectionRepository.kt
+++ b/app/src/main/java/org/thoughtcrime/securesms/mediasend/v2/MediaSelectionRepository.kt
@@ -142,7 +142,7 @@ class MediaSelectionRepository(context: Context) {
   }
 
   fun isLocalSelfSend(recipient: Recipient?, isSms: Boolean): Boolean {
-    return !MessageSender.isLocalSelfSend(context, recipient, isSms)
+    return MessageSender.isLocalSelfSend(context, recipient, isSms)
   }
 
   @WorkerThread
diff --git a/app/src/main/java/org/thoughtcrime/securesms/mediasend/v2/MediaSelectionViewModel.kt b/app/src/main/java/org/thoughtcrime/securesms/mediasend/v2/MediaSelectionViewModel.kt
index 554737656..07c412ed3 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/mediasend/v2/MediaSelectionViewModel.kt
+++ b/app/src/main/java/org/thoughtcrime/securesms/mediasend/v2/MediaSelectionViewModel.kt
@@ -68,7 +68,12 @@ class MediaSelectionViewModel(
   init {
     val recipientId = destination.getRecipientId()
     if (recipientId != null) {
-      store.update(Recipient.live(recipientId).liveData) { r, s -> s.copy(recipient = r) }
+      store.update(Recipient.live(recipientId).liveData) { r, s ->
+        s.copy(
+          recipient = r,
+          isPreUploadEnabled = shouldPreUpload(s.isMeteredConnection, s.transportOption.isSms, r)
+        )
+      }
     }
 
     if (initialMedia.isNotEmpty()) {