From 366780f6cb5b0967d0b14fed5729ebd0480f94df Mon Sep 17 00:00:00 2001
From: Rashad Sookram <rashad@signal.org>
Date: Wed, 9 Mar 2022 11:55:52 -0500
Subject: [PATCH] Revert "Avoid querying conversation size twice."

This reverts commit fe088c39c708f3085b7a6ebddc8f892821b37b0c.
---
 .../securesms/conversation/ConversationDataSource.java      | 6 ++----
 .../securesms/conversation/ConversationViewModel.java       | 2 +-
 2 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/app/src/main/java/org/thoughtcrime/securesms/conversation/ConversationDataSource.java b/app/src/main/java/org/thoughtcrime/securesms/conversation/ConversationDataSource.java
index 650b86739a..4fb285a605 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/conversation/ConversationDataSource.java
+++ b/app/src/main/java/org/thoughtcrime/securesms/conversation/ConversationDataSource.java
@@ -46,20 +46,18 @@ class ConversationDataSource implements PagedDataSource<MessageId, ConversationM
   private final long               threadId;
   private final MessageRequestData messageRequestData;
   private final boolean            showUniversalExpireTimerUpdate;
-  private final int                baseSize;
 
-  ConversationDataSource(@NonNull Context context, long threadId, @NonNull MessageRequestData messageRequestData, boolean showUniversalExpireTimerUpdate, int baseSize) {
+  ConversationDataSource(@NonNull Context context, long threadId, @NonNull MessageRequestData messageRequestData, boolean showUniversalExpireTimerUpdate) {
     this.context                        = context;
     this.threadId                       = threadId;
     this.messageRequestData             = messageRequestData;
     this.showUniversalExpireTimerUpdate = showUniversalExpireTimerUpdate;
-    this.baseSize                       = baseSize;
   }
 
   @Override
   public int size() {
     long startTime = System.currentTimeMillis();
-    int  size      = baseSize +
+    int  size      = SignalDatabase.mmsSms().getConversationCount(threadId) +
                      (messageRequestData.includeWarningUpdateMessage() ? 1 : 0) +
                      (showUniversalExpireTimerUpdate ? 1 : 0);
 
diff --git a/app/src/main/java/org/thoughtcrime/securesms/conversation/ConversationViewModel.java b/app/src/main/java/org/thoughtcrime/securesms/conversation/ConversationViewModel.java
index 58c054371e..8e3e83eb89 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/conversation/ConversationViewModel.java
+++ b/app/src/main/java/org/thoughtcrime/securesms/conversation/ConversationViewModel.java
@@ -155,7 +155,7 @@ public class ConversationViewModel extends ViewModel {
       ApplicationDependencies.getDatabaseObserver().registerConversationObserver(data.getThreadId(), conversationObserver);
       ApplicationDependencies.getDatabaseObserver().registerMessageInsertObserver(data.getThreadId(), messageInsertObserver);
 
-      ConversationDataSource dataSource = new ConversationDataSource(context, data.getThreadId(), messageRequestData, data.showUniversalExpireTimerMessage(), data.getThreadSize());
+      ConversationDataSource dataSource = new ConversationDataSource(context, data.getThreadId(), messageRequestData, data.showUniversalExpireTimerMessage());
       PagingConfig config = new PagingConfig.Builder().setPageSize(25)
                                                       .setBufferPages(3)
                                                       .setStartIndex(Math.max(startPosition, 0))