Periodic alarm to check for messages.

2026-04-19 08:09:12 +01:00 · 2021-02-10 14:01:11 -04:00
parent 29d66f2b92
commit 98cb6b457c
14 changed files with 349 additions and 61 deletions
--- a/app/src/main/java/org/thoughtcrime/securesms/messageprocessingalarm/MessageProcessReceiver.java
+++ b/app/src/main/java/org/thoughtcrime/securesms/messageprocessingalarm/MessageProcessReceiver.java
@@ -0,0 +1,81 @@
+package org.thoughtcrime.securesms.messageprocessingalarm;
+
+import android.annotation.SuppressLint;
+import android.app.AlarmManager;
+import android.app.PendingIntent;
+import android.content.BroadcastReceiver;
+import android.content.Context;
+import android.content.Intent;
+import android.os.SystemClock;
+
+import androidx.annotation.NonNull;
+
+import org.signal.core.util.concurrent.SignalExecutors;
+import org.signal.core.util.logging.Log;
+import org.thoughtcrime.securesms.dependencies.ApplicationDependencies;
+import org.thoughtcrime.securesms.jobmanager.JobTracker;
+import org.thoughtcrime.securesms.jobs.PushNotificationReceiveJob;
+import org.thoughtcrime.securesms.util.FeatureFlags;
+import org.whispersystems.libsignal.util.guava.Optional;
+
+import java.util.concurrent.TimeUnit;
+
+/**
+ * On received message, runs a job to poll for messages.
+ */
+public final class MessageProcessReceiver extends BroadcastReceiver {
+
+  private static final String TAG = Log.tag(MessageProcessReceiver.class);
+
+  private static final long FIRST_RUN_DELAY  = TimeUnit.MINUTES.toMillis(3);
+  private static final long FOREGROUND_DELAY = TimeUnit.SECONDS.toMillis(2);
+  private static final long JOB_TIMEOUT      = TimeUnit.SECONDS.toMillis(5);
+
+  public static final String BROADCAST_ACTION = "org.thoughtcrime.securesms.action.PROCESS_MESSAGES";
+
+  @Override
+  @SuppressLint("StaticFieldLeak")
+  public void onReceive(@NonNull Context context, @NonNull Intent intent) {
+    Log.i(TAG, String.format("onReceive(%s)", intent.getAction()));
+
+    if (Intent.ACTION_BOOT_COMPLETED.equals(intent.getAction())) {
+      Log.i(TAG, "Starting Alarm because of boot receiver");
+      startOrUpdateAlarm(context);
+    } else if (BROADCAST_ACTION.equals(intent.getAction())) {
+      PendingResult pendingResult = goAsync();
+
+      SignalExecutors.BOUNDED.submit(() -> {
+        Log.i(TAG, "Running PushNotificationReceiveJob");
+
+        Optional<JobTracker.JobState> jobState = ApplicationDependencies.getJobManager()
+                                                                        .runSynchronously(PushNotificationReceiveJob.withDelayedForegroundService(FOREGROUND_DELAY), JOB_TIMEOUT);
+
+        Log.i(TAG, "PushNotificationReceiveJob ended: " + (jobState.isPresent() ? jobState.get().toString() : "Job did not complete"));
+
+        pendingResult.finish();
+      });
+    }
+  }
+
+  public static void startOrUpdateAlarm(@NonNull Context context) {
+    Intent alarmIntent = new Intent(context, MessageProcessReceiver.class);
+
+    alarmIntent.setAction(MessageProcessReceiver.BROADCAST_ACTION);
+
+    PendingIntent pendingIntent = PendingIntent.getBroadcast(context, 123, alarmIntent, PendingIntent.FLAG_UPDATE_CURRENT);
+    AlarmManager  alarmManager  = (AlarmManager) context.getSystemService(Context.ALARM_SERVICE);
+
+    long interval = FeatureFlags.getBackgroundMessageProcessDelay();
+
+    if (interval < 0) {
+      alarmManager.cancel(pendingIntent);
+      Log.i(TAG, "Alarm cancelled");
+    } else {
+      alarmManager.setRepeating(AlarmManager.ELAPSED_REALTIME_WAKEUP,
+                                SystemClock.elapsedRealtime() + FIRST_RUN_DELAY,
+                                interval,
+                                pendingIntent);
+      Log.i(TAG, "Alarm scheduled to repeat at interval " + interval);
+    }
+  }
+}