From ec009a2bba3a5a9fd256aae2fce947241f262e44 Mon Sep 17 00:00:00 2001
From: Ameya Lokare <ameya@signal.org>
Date: Mon, 24 Jun 2024 16:42:34 -0700
Subject: [PATCH] Pass AccountAndAuthenticatedDeviceHolder to spam filter

---
 .../textsecuregcm/controllers/MessageController.java          | 2 +-
 .../textsecuregcm/spam/ReportSpamTokenProvider.java           | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/service/src/main/java/org/whispersystems/textsecuregcm/controllers/MessageController.java b/service/src/main/java/org/whispersystems/textsecuregcm/controllers/MessageController.java
index 854ae7adc..bac576b97 100644
--- a/service/src/main/java/org/whispersystems/textsecuregcm/controllers/MessageController.java
+++ b/service/src/main/java/org/whispersystems/textsecuregcm/controllers/MessageController.java
@@ -327,7 +327,7 @@ public class MessageController {
 
       final Optional<byte[]> spamReportToken = switch (senderType) {
         case SENDER_TYPE_IDENTIFIED ->
-        reportSpamTokenProvider.makeReportSpamToken(context, source.get().getAccount(), destination);
+        reportSpamTokenProvider.makeReportSpamToken(context, source.get(), destination);
         default -> Optional.empty();
       };
 
diff --git a/service/src/main/java/org/whispersystems/textsecuregcm/spam/ReportSpamTokenProvider.java b/service/src/main/java/org/whispersystems/textsecuregcm/spam/ReportSpamTokenProvider.java
index 952f9616a..de785c02b 100644
--- a/service/src/main/java/org/whispersystems/textsecuregcm/spam/ReportSpamTokenProvider.java
+++ b/service/src/main/java/org/whispersystems/textsecuregcm/spam/ReportSpamTokenProvider.java
@@ -1,9 +1,9 @@
 package org.whispersystems.textsecuregcm.spam;
 
+import org.whispersystems.textsecuregcm.auth.AccountAndAuthenticatedDeviceHolder;
 import org.whispersystems.textsecuregcm.storage.Account;
 import javax.ws.rs.container.ContainerRequestContext;
 import java.util.Optional;
-import java.util.function.Function;
 
 /**
  * Generates ReportSpamTokens to be used for spam reports.
@@ -18,7 +18,7 @@ public interface ReportSpamTokenProvider {
    * @param maybeDestination the intended recepient of the message if available
    * @return either a generated token or nothing
    */
-  Optional<byte[]> makeReportSpamToken(ContainerRequestContext context, final Account sender,
+  Optional<byte[]> makeReportSpamToken(ContainerRequestContext context, final AccountAndAuthenticatedDeviceHolder sender,
       final Optional<Account> maybeDestination);
 
   /**