From 3a0dbe6e670f82ddc549378b804d019bedb97924 Mon Sep 17 00:00:00 2001
From: Cody Henthorne <cody@signal.org>
Date: Mon, 6 Feb 2023 19:48:11 -0500
Subject: [PATCH] Use alarm clock for scheduling message sends.

---
 .../securesms/database/MessageTable.java      | 10 ++++---
 .../revealable/ViewOnceMessageManager.java    |  2 +-
 .../service/ExpiringStoriesManager.kt         |  2 +-
 .../service/PendingRetryReceiptManager.java   |  2 +-
 .../service/ScheduledMessageManager.kt        | 27 ++++++++++++++-----
 .../securesms/service/TimedEventManager.java  | 15 +++++------
 .../service/TrimThreadsByDateManager.java     |  2 +-
 7 files changed, 38 insertions(+), 22 deletions(-)

diff --git a/app/src/main/java/org/thoughtcrime/securesms/database/MessageTable.java b/app/src/main/java/org/thoughtcrime/securesms/database/MessageTable.java
index 62b997f54c..c2e424fd53 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/database/MessageTable.java
+++ b/app/src/main/java/org/thoughtcrime/securesms/database/MessageTable.java
@@ -4648,16 +4648,20 @@ public class MessageTable extends DatabaseTable implements MessageTypes, Recipie
     }
   }
 
-  public @Nullable Long getOldestScheduledSendTimestamp() {
+  public @Nullable MessageRecord getOldestScheduledSendTimestamp() {
     String[] columns   = new String[] { SCHEDULED_DATE };
     String   selection = STORY_TYPE + " = ? AND " + PARENT_STORY_ID + " <= ? AND " + SCHEDULED_DATE + " != ?";
     String[] args      = SqlUtil.buildArgs(0, 0, -1);
     String   order     = SCHEDULED_DATE + " ASC, " + ID + " ASC";
     String   limit     = "1";
 
-    try (Cursor cursor = getReadableDatabase().query(TABLE_NAME, columns, selection, args, null, null, order, limit)) {
-      return cursor != null && cursor.moveToNext() ? cursor.getLong(0) : null;
+    try (MmsReader reader = mmsReaderFor(getReadableDatabase().query(TABLE_NAME, MMS_PROJECTION, selection, args, null, null, order, limit))) {
+      if (reader.getNext() != null) {
+        return reader.getCurrent();
+      }
     }
+
+    return null;
   }
 
   public Cursor getMessagesForNotificationState(Collection<DefaultMessageNotifier.StickyThread> stickyThreads) {
diff --git a/app/src/main/java/org/thoughtcrime/securesms/revealable/ViewOnceMessageManager.java b/app/src/main/java/org/thoughtcrime/securesms/revealable/ViewOnceMessageManager.java
index b3147168e4..79ecfde489 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/revealable/ViewOnceMessageManager.java
+++ b/app/src/main/java/org/thoughtcrime/securesms/revealable/ViewOnceMessageManager.java
@@ -68,7 +68,7 @@ public class ViewOnceMessageManager extends TimedEventManager<ViewOnceExpiration
 
   @AnyThread
   @Override
-  protected void scheduleAlarm(@NonNull Application application, long delay) {
+  protected void scheduleAlarm(@NonNull Application application, ViewOnceExpirationInfo event, long delay) {
     setAlarm(application, delay, ViewOnceAlarm.class);
   }
 
diff --git a/app/src/main/java/org/thoughtcrime/securesms/service/ExpiringStoriesManager.kt b/app/src/main/java/org/thoughtcrime/securesms/service/ExpiringStoriesManager.kt
index 343953ecbf..98667b0016 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/service/ExpiringStoriesManager.kt
+++ b/app/src/main/java/org/thoughtcrime/securesms/service/ExpiringStoriesManager.kt
@@ -52,7 +52,7 @@ class ExpiringStoriesManager(
   override fun getDelayForEvent(event: Event): Long = event.delay
 
   @WorkerThread
-  override fun scheduleAlarm(application: Application, delay: Long) {
+  override fun scheduleAlarm(application: Application, event: Event, delay: Long) {
     setAlarm(application, delay, ExpireStoriesAlarm::class.java)
   }
 
diff --git a/app/src/main/java/org/thoughtcrime/securesms/service/PendingRetryReceiptManager.java b/app/src/main/java/org/thoughtcrime/securesms/service/PendingRetryReceiptManager.java
index 7ebd6f26d7..12c3654bdf 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/service/PendingRetryReceiptManager.java
+++ b/app/src/main/java/org/thoughtcrime/securesms/service/PendingRetryReceiptManager.java
@@ -71,7 +71,7 @@ public final class PendingRetryReceiptManager extends TimedEventManager<PendingR
 
   @AnyThread
   @Override
-  protected void scheduleAlarm(@NonNull Application application, long delay) {
+  protected void scheduleAlarm(@NonNull Application application, PendingRetryReceiptModel event, long delay) {
     setAlarm(application, delay, PendingRetryReceiptAlarm.class);
   }
 
diff --git a/app/src/main/java/org/thoughtcrime/securesms/service/ScheduledMessageManager.kt b/app/src/main/java/org/thoughtcrime/securesms/service/ScheduledMessageManager.kt
index cb9bb2eabb..c783ce8d1b 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/service/ScheduledMessageManager.kt
+++ b/app/src/main/java/org/thoughtcrime/securesms/service/ScheduledMessageManager.kt
@@ -1,15 +1,21 @@
 package org.thoughtcrime.securesms.service
 
 import android.app.Application
+import android.app.PendingIntent
 import android.content.BroadcastReceiver
 import android.content.Context
 import android.content.Intent
 import androidx.annotation.WorkerThread
+import org.signal.core.util.PendingIntentFlags
 import org.signal.core.util.logging.Log
+import org.thoughtcrime.securesms.conversation.ConversationIntents
 import org.thoughtcrime.securesms.database.SignalDatabase
+import org.thoughtcrime.securesms.database.model.MediaMmsMessageRecord
 import org.thoughtcrime.securesms.dependencies.ApplicationDependencies
 import org.thoughtcrime.securesms.jobs.IndividualSendJob
 import org.thoughtcrime.securesms.jobs.PushGroupSendJob
+import org.thoughtcrime.securesms.recipients.RecipientId
+import org.thoughtcrime.securesms.util.ServiceUtil
 import kotlin.time.Duration.Companion.seconds
 
 /**
@@ -32,12 +38,12 @@ class ScheduledMessageManager(
   @Suppress("UsePropertyAccessSyntax")
   @WorkerThread
   override fun getNextClosestEvent(): Event? {
-    val oldestTimestamp = messagesTable.getOldestScheduledSendTimestamp() ?: return null
+    val oldestMessage = messagesTable.getOldestScheduledSendTimestamp() as? MediaMmsMessageRecord ?: return null
 
-    val delay = (oldestTimestamp - System.currentTimeMillis()).coerceAtLeast(0)
+    val delay = (oldestMessage.scheduledDate - System.currentTimeMillis()).coerceAtLeast(0)
     Log.i(TAG, "The next scheduled message needs to be sent in $delay ms.")
 
-    return Event(delay)
+    return Event(delay, oldestMessage.recipient.id, oldestMessage.threadId)
   }
 
   @WorkerThread
@@ -60,11 +66,20 @@ class ScheduledMessageManager(
   override fun getDelayForEvent(event: Event): Long = event.delay
 
   @WorkerThread
-  override fun scheduleAlarm(application: Application, delay: Long) {
-    trySetExactAlarm(application, System.currentTimeMillis() + delay, ScheduledMessagesAlarm::class.java)
+  override fun scheduleAlarm(application: Application, event: Event, delay: Long) {
+    val conversationIntent = ConversationIntents.createBuilder(application, event.recipientId, event.threadId).build()
+
+    ServiceUtil.getAlarmManager(application)
+
+    trySetExactAlarm(
+      application,
+      System.currentTimeMillis() + delay,
+      ScheduledMessagesAlarm::class.java,
+      PendingIntent.getActivity(application, 0, conversationIntent, PendingIntentFlags.mutable())
+    )
   }
 
-  data class Event(val delay: Long)
+  data class Event(val delay: Long, val recipientId: RecipientId, val threadId: Long)
 
   class ScheduledMessagesAlarm : BroadcastReceiver() {
 
diff --git a/app/src/main/java/org/thoughtcrime/securesms/service/TimedEventManager.java b/app/src/main/java/org/thoughtcrime/securesms/service/TimedEventManager.java
index 7de4190436..bdeebd7b3e 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/service/TimedEventManager.java
+++ b/app/src/main/java/org/thoughtcrime/securesms/service/TimedEventManager.java
@@ -3,6 +3,7 @@ package org.thoughtcrime.securesms.service;
 import android.app.AlarmManager;
 import android.app.Application;
 import android.app.PendingIntent;
+import android.content.BroadcastReceiver;
 import android.content.Context;
 import android.content.Intent;
 import android.os.Build;
@@ -54,7 +55,7 @@ public abstract class TimedEventManager<E> {
           scheduleIfNecessary();
         }, delay);
 
-        scheduleAlarm(application, delay);
+        scheduleAlarm(application, event, delay);
       }
     });
   }
@@ -82,12 +83,12 @@ public abstract class TimedEventManager<E> {
    * use {@link #setAlarm(Context, long, Class)} as a helper method.
    */
   @AnyThread
-  protected abstract void scheduleAlarm(@NonNull Application application, long delay);
+  protected abstract void scheduleAlarm(@NonNull Application application, E event, long delay);
 
   /**
    * Helper method to set an alarm.
    */
-  protected static void setAlarm(@NonNull Context context, long delay, @NonNull Class alarmClass) {
+  protected static void setAlarm(@NonNull Context context, long delay, @NonNull Class<? extends BroadcastReceiver> alarmClass) {
     Intent        intent        = new Intent(context, alarmClass);
     PendingIntent pendingIntent = PendingIntent.getBroadcast(context, 0, intent, PendingIntentFlags.mutable());
     AlarmManager  alarmManager  = ServiceUtil.getAlarmManager(context);
@@ -96,7 +97,7 @@ public abstract class TimedEventManager<E> {
     alarmManager.set(AlarmManager.RTC_WAKEUP, System.currentTimeMillis() + delay, pendingIntent);
   }
 
-  protected static void trySetExactAlarm(@NonNull Context context, long timestamp, @NonNull Class alarmClass) {
+  protected static void trySetExactAlarm(@NonNull Context context, long timestamp, @NonNull Class<? extends BroadcastReceiver> alarmClass, @NonNull PendingIntent showIntent) {
     Intent        intent        = new Intent(context, alarmClass);
     PendingIntent pendingIntent = PendingIntent.getBroadcast(context, 0, intent, PendingIntentFlags.mutable());
     AlarmManager  alarmManager  = ServiceUtil.getAlarmManager(context);
@@ -106,11 +107,7 @@ public abstract class TimedEventManager<E> {
     boolean hasManagerPermission = Build.VERSION.SDK_INT < 31 || alarmManager.canScheduleExactAlarms();
     if (hasManagerPermission) {
       try {
-        if (Build.VERSION.SDK_INT >= 23) {
-          alarmManager.setExactAndAllowWhileIdle(AlarmManager.RTC_WAKEUP, timestamp, pendingIntent);
-        } else {
-          alarmManager.setExact(AlarmManager.RTC_WAKEUP, timestamp, pendingIntent);
-        }
+        alarmManager.setAlarmClock(new AlarmManager.AlarmClockInfo(timestamp, showIntent), pendingIntent);
         return;
       } catch (Exception e) {
         Log.w(TAG, e);
diff --git a/app/src/main/java/org/thoughtcrime/securesms/service/TrimThreadsByDateManager.java b/app/src/main/java/org/thoughtcrime/securesms/service/TrimThreadsByDateManager.java
index 73c49ef2a8..dfc672c9a0 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/service/TrimThreadsByDateManager.java
+++ b/app/src/main/java/org/thoughtcrime/securesms/service/TrimThreadsByDateManager.java
@@ -75,7 +75,7 @@ public class TrimThreadsByDateManager extends TimedEventManager<TrimThreadsByDat
   }
 
   @Override
-  protected void scheduleAlarm(@NonNull Application application, long delay) {
+  protected void scheduleAlarm(@NonNull Application application, TrimEvent event, long delay) {
     setAlarm(application, delay, TrimThreadsByDateAlarm.class);
   }