Page MenuHomePhabricator
Differential D8475

[Keyserver] Refactor identity utilities
ClosedPublic
Actions

Authored by • jon on Jul 11 2023, 10:03 AM.
Tags
None
Referenced Files
F5083401: D8475.id.diff
Sat, Mar 29, 11:40 AM
Unknown Object (File)
Tue, Mar 11, 12:54 PM
Unknown Object (File)
Wed, Mar 5, 4:57 PM
Unknown Object (File)
Wed, Mar 5, 2:53 AM
Unknown Object (File)
Sat, Mar 1, 7:55 PM
Unknown Object (File)
Sat, Mar 1, 7:55 PM
Unknown Object (File)
Sat, Mar 1, 7:55 PM
Unknown Object (File)
Sat, Mar 1, 7:55 PM
View All Files
Subscribers
ashoat
tomek

Details

Reviewers
kamil
michal
ashoat
Commits
rCOMM4b40142ee014: [Keyserver] Refactor identity utilities
Summary

These are useful when interacting with identity service,
pull them out of login.js to make them more widely available.

Part of https://linear.app/comm/issue/ENG-3944

Depends on D8474

Test Plan

Launch keyserver with login workflow (missing identity info)

Diff Detail

Repository
rCOMM Comm
Lint
No Lint Coverage
Unit
No Test Coverage

Event Timeline

• jon created this revision.Jul 11 2023, 10:03 AM
Herald added subscribers: tomek, ashoat. · View Herald TranscriptJul 11 2023, 10:03 AM
• jon added a child revision: D8476: [Keyserver] Publish prekeys to identity service for cron job.Jul 11 2023, 10:09 AM
Harbormaster completed remote builds in B20870: Diff 28590.Jul 11 2023, 10:29 AM
• jon requested review of this revision.Jul 11 2023, 10:29 AM
• jon added inline comments.Jul 14 2023, 3:33 PM
keyserver/src/user/identity.js
27 ↗(On Diff #28590)

Apply https://phab.comm.dev/D8405#inline-54532 to this refactor

• jon updated this revision to Diff 28697.Jul 14 2023, 3:37 PM

Remove extra async

Harbormaster completed remote builds in B20941: Diff 28697.Jul 14 2023, 3:54 PM
kamil accepted this revision.Jul 17 2023, 2:59 AM
kamil added inline comments.
keyserver/src/user/login.js
11–12 ↗(On Diff #28697)

nit: merge imports

This revision is now accepted and ready to land.Jul 17 2023, 2:59 AM
• jon updated this revision to Diff 28967.Jul 24 2023, 6:35 AM

Rebase on master

• jon added a parent revision: D8473: [Keyserver/rust] Add publish_prekeys implementation.Jul 24 2023, 6:41 AM
• jon removed a parent revision: D8474: [Keyserver] Refactor prekey lifecycle logic.
Harbormaster completed remote builds in B21145: Diff 28967.Jul 24 2023, 7:09 AM
• jon updated this revision to Diff 30014.Aug 17 2023, 8:19 AM
• jon added a parent revision: D8474: [Keyserver] Refactor prekey lifecycle logic.

Use userID when able to

• jon updated this revision to Diff 30015.Aug 17 2023, 8:21 AM
• jon marked an inline comment as done.

Combine imports

• jon added a comment.Aug 17 2023, 8:22 AM

Address Kamil's feedback

• jon added a reviewer: ashoat.Aug 17 2023, 8:22 AM
This revision now requires review to proceed.Aug 17 2023, 8:22 AM
Harbormaster completed remote builds in B21842: Diff 30014.Aug 17 2023, 8:38 AM
Harbormaster completed remote builds in B21843: Diff 30015.
ashoat accepted this revision.Aug 18 2023, 3:23 PM
This revision is now accepted and ready to land.Aug 18 2023, 3:23 PM
Closed by commit rCOMM4b40142ee014: [Keyserver] Refactor identity utilities (authored by • jon). · Explain WhyAug 20 2023, 11:47 PM
This revision was automatically updated to reflect the committed changes.
• jon added a commit: rCOMM4b40142ee014: [Keyserver] Refactor identity utilities.

Revision Contents
Changeset List

PathSize
keyserver/
src/
user/
37 lines
32 lines

Diff 28967

View Options

keyserver/src/user/identity.js

Show All 17 Lines
import app.comm.android.ExpoUtils; import app.comm.android.ExpoUtils;
import app.comm.android.MainActivity; import app.comm.android.MainActivity;
import app.comm.android.R; import app.comm.android.R;
import app.comm.android.fbjni.CommSecureStore; import app.comm.android.fbjni.CommSecureStore;
import app.comm.android.fbjni.GlobalDBSingleton; import app.comm.android.fbjni.GlobalDBSingleton;
import app.comm.android.fbjni.MessageOperationsUtilities; import app.comm.android.fbjni.MessageOperationsUtilities;
import app.comm.android.fbjni.NetworkModule; import app.comm.android.fbjni.NetworkModule;
import app.comm.android.fbjni.NotificationsCryptoModule; import app.comm.android.fbjni.NotificationsCryptoModule;
import app.comm.android.fbjni.StaffUtils;
import app.comm.android.fbjni.ThreadOperations; import app.comm.android.fbjni.ThreadOperations;
import com.google.firebase.messaging.FirebaseMessagingService; import com.google.firebase.messaging.FirebaseMessagingService;
import com.google.firebase.messaging.RemoteMessage; import com.google.firebase.messaging.RemoteMessage;
import java.io.File; import java.io.File;
import java.lang.StringBuilder;
import java.util.ArrayList;
import me.leolin.shortcutbadger.ShortcutBadger; import me.leolin.shortcutbadger.ShortcutBadger;
import org.json.JSONException; import org.json.JSONException;
import org.json.JSONObject; import org.json.JSONObject;
public class CommNotificationsHandler extends FirebaseMessagingService { public class CommNotificationsHandler extends FirebaseMessagingService {
private static final String BADGE_KEY = "badge"; private static final String BADGE_KEY = "badge";
private static final String BADGE_ONLY_KEY = "badgeOnly"; private static final String BADGE_ONLY_KEY = "badgeOnly";
private static final String BACKGROUND_NOTIF_TYPE_KEY = "backgroundNotifType"; private static final String BACKGROUND_NOTIF_TYPE_KEY = "backgroundNotifType";
private static final String SET_UNREAD_STATUS_KEY = "setUnreadStatus"; private static final String SET_UNREAD_STATUS_KEY = "setUnreadStatus";
private static final String NOTIF_ID_KEY = "id"; private static final String NOTIF_ID_KEY = "id";
private static final String ENCRYPTED_PAYLOAD_KEY = "encryptedPayload"; private static final String ENCRYPTED_PAYLOAD_KEY = "encryptedPayload";
private static final String ENCRYPTION_FAILED_KEY = "encryptionFailed"; private static final String ENCRYPTION_FAILED_KEY = "encryptionFailed";
private static final String GROUP_NOTIF_IDS_KEY = "groupNotifIDs";
private static final String CHANNEL_ID = "default"; private static final String CHANNEL_ID = "default";
private static final long[] VIBRATION_SPEC = {500, 500}; private static final long[] VIBRATION_SPEC = {500, 500};
private Bitmap displayableNotificationLargeIcon; private Bitmap displayableNotificationLargeIcon;
private NotificationManager notificationManager; private NotificationManager notificationManager;
private LocalBroadcastManager localBroadcastManager; private LocalBroadcastManager localBroadcastManager;
public static final String RESCIND_KEY = "rescind"; public static final String RESCIND_KEY = "rescind";
public static final String RESCIND_ID_KEY = "rescindID"; public static final String RESCIND_ID_KEY = "rescindID";
▲ Show 20 Lines • Show All 132 Lines • ▼ Show 20 Lines private void handleNotificationRescind(RemoteMessage message) {
for (StatusBarNotification notification : for (StatusBarNotification notification :
notificationManager.getActiveNotifications()) { notificationManager.getActiveNotifications()) {
String tag = notification.getTag(); String tag = notification.getTag();
boolean isGroupMember = boolean isGroupMember =
threadID.equals(notification.getNotification().getGroup()); threadID.equals(notification.getNotification().getGroup());
boolean isGroupSummary = boolean isGroupSummary =
(notification.getNotification().flags & (notification.getNotification().flags &
Notification.FLAG_GROUP_SUMMARY) == Notification.FLAG_GROUP_SUMMARY; Notification.FLAG_GROUP_SUMMARY) == Notification.FLAG_GROUP_SUMMARY;
if (tag != null && tag.equals(rescindID)) { if (tag != null && tag.equals(rescindID)) {
notificationManager.cancel(notification.getTag(), notification.getId()); notificationManager.cancel(notification.getTag(), notification.getId());
} else if (isGroupMember && isGroupSummary) { } else if (isGroupMember && isGroupSummary) {
groupSummaryPresent = true; groupSummaryPresent = true;
removeNotificationFromGroupSummary(threadID, rescindID, notification);
} else if (isGroupMember) { } else if (isGroupMember) {
threadGroupPresent = true; threadGroupPresent = true;
} else if (isGroupSummary) {
checkForUnmatchedRescind(threadID, rescindID, notification);
} }
} }
if (groupSummaryPresent && !threadGroupPresent) { if (groupSummaryPresent && !threadGroupPresent) {
notificationManager.cancel(threadID, threadID.hashCode()); notificationManager.cancel(threadID, threadID.hashCode());
} }
} }
Show All 9 Lines private void addToThreadGroupAndDisplay(
NotificationCompat.Builder groupSummaryNotificationBuilder = NotificationCompat.Builder groupSummaryNotificationBuilder =
new NotificationCompat.Builder(this.getApplicationContext()) new NotificationCompat.Builder(this.getApplicationContext())
.setChannelId(CHANNEL_ID) .setChannelId(CHANNEL_ID)
.setSmallIcon(R.drawable.notif_icon) .setSmallIcon(R.drawable.notif_icon)
.setContentIntent( .setContentIntent(
this.createStartMainActivityAction(threadID, threadID)) this.createStartMainActivityAction(threadID, threadID))
.setGroup(threadID) .setGroup(threadID)
.setGroupSummary(true) .setGroupSummary(true)
.setAutoCancel(true) .setAutoCancel(false)
ashoatUnsubmitted
Not Done
Inline Actions

Do we really want to make this change in production? Should we consider gating it on isStaffRelease?

ashoat: Do we really want to make this change in production? Should we consider gating it on…
marcinAuthorUnsubmitted
Done
Inline Actions

This is my oversight. We want it false for staff to avoid accidental evidence removal. However we don't want it for public users. I will update the diff shortly.

marcin: This is my oversight. We want it false for staff to avoid accidental evidence removal. However…
.setGroupAlertBehavior(NotificationCompat.GROUP_ALERT_CHILDREN); .setGroupAlertBehavior(NotificationCompat.GROUP_ALERT_CHILDREN);
if (StaffUtils.isStaffRelease()) {
ArrayList<String> groupNotifIDs =
recordNotificationInGroupSummary(threadID, notificationID);
String notificationSummaryBody = groupNotifIDs.stream().reduce(
"Notif IDs: ", (acc, el) -> acc + System.lineSeparator() + el);
Bundle data = new Bundle();
data.putStringArrayList(GROUP_NOTIF_IDS_KEY, groupNotifIDs);
groupSummaryNotificationBuilder =
groupSummaryNotificationBuilder
.setContentTitle("Summary for thread id " + threadID)
.setExtras(data)
.setStyle(new NotificationCompat.BigTextStyle().bigText(
notificationSummaryBody));
}
notificationManager.notify( notificationManager.notify(
notificationID, notificationID.hashCode(), notificationBuilder.build()); notificationID, notificationID.hashCode(), notificationBuilder.build());
notificationManager.notify( notificationManager.notify(
threadID, threadID.hashCode(), groupSummaryNotificationBuilder.build()); threadID, threadID.hashCode(), groupSummaryNotificationBuilder.build());
} }
private void displayNotification(RemoteMessage message) { private void displayNotification(RemoteMessage message) {
if (message.getData().get(RESCIND_KEY) != null) { if (message.getData().get(RESCIND_KEY) != null) {
▲ Show 20 Lines • Show All 81 Lines • ▼ Show 20 Linespublic class CommNotificationsHandler extends FirebaseMessagingService {
} }
private Bundle private Bundle
serializeMessageDataForIntentAttachment(RemoteMessage message) { serializeMessageDataForIntentAttachment(RemoteMessage message) {
Bundle bundle = new Bundle(); Bundle bundle = new Bundle();
message.getData().forEach(bundle::putString); message.getData().forEach(bundle::putString);
return bundle; return bundle;
} }
private void displayErrorMessageNotification(
String errorMessage,
String errorTitle,
String largeErrorData) {
NotificationCompat.Builder errorNotificationBuilder =
new NotificationCompat.Builder(this.getApplicationContext())
.setDefaults(Notification.DEFAULT_ALL)
.setChannelId(CHANNEL_ID)
.setSmallIcon(R.drawable.notif_icon)
.setLargeIcon(displayableNotificationLargeIcon);
if (errorMessage != null) {
errorNotificationBuilder =
errorNotificationBuilder.setContentText(errorMessage);
}
if (errorTitle != null) {
errorNotificationBuilder =
errorNotificationBuilder.setContentTitle(errorTitle);
}
if (largeErrorData != null) {
errorNotificationBuilder = errorNotificationBuilder.setStyle(
new NotificationCompat.BigTextStyle().bigText(largeErrorData));
}
notificationManager.notify(
errorMessage,
errorMessage.hashCode(),
errorNotificationBuilder.build());
}
private boolean
isGroupSummary(StatusBarNotification notification, String threadID) {
boolean isAnySummary = (notification.getNotification().flags &
Notification.FLAG_GROUP_SUMMARY) != 0;
if (threadID == null) {
return isAnySummary;
}
return isAnySummary &&
threadID.equals(notification.getNotification().getGroup());
}
private ArrayList<String>
recordNotificationInGroupSummary(String threadID, String notificationID) {
ArrayList<String> groupNotifIDs = new ArrayList<>();
for (StatusBarNotification notif :
notificationManager.getActiveNotifications()) {
if (!isGroupSummary(notif, threadID)) {
continue;
}
groupNotifIDs = notif.getNotification().extras.getStringArrayList(
GROUP_NOTIF_IDS_KEY);
break;
}
groupNotifIDs.add(notificationID);
return groupNotifIDs;
}
private void removeNotificationFromGroupSummary(
String threadID,
String notificationID,
StatusBarNotification groupSummaryNotification) {
if (!StaffUtils.isStaffRelease()) {
return;
}
ArrayList<String> groupNotifIDs =
groupSummaryNotification.getNotification().extras.getStringArrayList(
GROUP_NOTIF_IDS_KEY);
if (groupNotifIDs == null) {
displayErrorMessageNotification(
"Empty summary notif for thread ID " + threadID,
"Empty Summary Notif",
"Summary notification for thread ID " + threadID +
" had empty body when rescinding " + notificationID);
}
int notificationIndex = groupNotifIDs.indexOf(notificationID);
if (notificationIndex == -1) {
displayErrorMessageNotification(
"Notif with ID " + notificationID + "not in " + threadID,
"Unrecorded Notif",
"Rescinded notification with id " + notificationID +
" not found in group summary for thread id " + threadID);
return;
}
groupNotifIDs.remove(notificationIndex);
String notificationSummaryBody = groupNotifIDs.stream().reduce(
"Notif IDs: ", (acc, el) -> acc + System.lineSeparator() + el);
Bundle data = new Bundle();
data.putStringArrayList(GROUP_NOTIF_IDS_KEY, groupNotifIDs);
NotificationCompat.Builder groupSummaryNotificationBuilder =
new NotificationCompat.Builder(this.getApplicationContext())
.setChannelId(CHANNEL_ID)
.setSmallIcon(R.drawable.notif_icon)
.setContentIntent(
this.createStartMainActivityAction(threadID, threadID))
.setContentTitle("Summary for thread id " + threadID)
.setExtras(data)
.setStyle(new NotificationCompat.BigTextStyle().bigText(
notificationSummaryBody))
.setGroup(threadID)
.setGroupSummary(true)
.setAutoCancel(false)
.setGroupAlertBehavior(NotificationCompat.GROUP_ALERT_CHILDREN);
notificationManager.notify(
threadID, threadID.hashCode(), groupSummaryNotificationBuilder.build());
}
private void checkForUnmatchedRescind(
String threadID,
String notificationID,
StatusBarNotification anySummaryNotification) {
if (!StaffUtils.isStaffRelease()) {
return;
}
ArrayList<String> anyGroupNotifIDs =
anySummaryNotification.getNotification().extras.getStringArrayList(
GROUP_NOTIF_IDS_KEY);
if (anyGroupNotifIDs == null) {
return;
}
String groupID = anySummaryNotification.getNotification().getGroup();
for (String notifID : anyGroupNotifIDs) {
if (!notificationID.equals(notifID)) {
continue;
}
displayErrorMessageNotification(
"Summary for thread id " + groupID + "has " + notifID,
"Rescind Mismatch",
"Summary notif for thread id " + groupID + " contains notif id " +
notifID + " which was received in rescind with thread id " +
threadID);
}
}
} }
View Options

keyserver/src/user/login.js

Show All 17 Lines
import app.comm.android.ExpoUtils; import app.comm.android.ExpoUtils;
import app.comm.android.MainActivity; import app.comm.android.MainActivity;
import app.comm.android.R; import app.comm.android.R;
import app.comm.android.fbjni.CommSecureStore; import app.comm.android.fbjni.CommSecureStore;
import app.comm.android.fbjni.GlobalDBSingleton; import app.comm.android.fbjni.GlobalDBSingleton;
import app.comm.android.fbjni.MessageOperationsUtilities; import app.comm.android.fbjni.MessageOperationsUtilities;
import app.comm.android.fbjni.NetworkModule; import app.comm.android.fbjni.NetworkModule;
import app.comm.android.fbjni.NotificationsCryptoModule; import app.comm.android.fbjni.NotificationsCryptoModule;
import app.comm.android.fbjni.StaffUtils;
import app.comm.android.fbjni.ThreadOperations; import app.comm.android.fbjni.ThreadOperations;
import com.google.firebase.messaging.FirebaseMessagingService; import com.google.firebase.messaging.FirebaseMessagingService;
import com.google.firebase.messaging.RemoteMessage; import com.google.firebase.messaging.RemoteMessage;
import java.io.File; import java.io.File;
import java.lang.StringBuilder;
import java.util.ArrayList;
import me.leolin.shortcutbadger.ShortcutBadger; import me.leolin.shortcutbadger.ShortcutBadger;
import org.json.JSONException; import org.json.JSONException;
import org.json.JSONObject; import org.json.JSONObject;
public class CommNotificationsHandler extends FirebaseMessagingService { public class CommNotificationsHandler extends FirebaseMessagingService {
private static final String BADGE_KEY = "badge"; private static final String BADGE_KEY = "badge";
private static final String BADGE_ONLY_KEY = "badgeOnly"; private static final String BADGE_ONLY_KEY = "badgeOnly";
private static final String BACKGROUND_NOTIF_TYPE_KEY = "backgroundNotifType"; private static final String BACKGROUND_NOTIF_TYPE_KEY = "backgroundNotifType";
private static final String SET_UNREAD_STATUS_KEY = "setUnreadStatus"; private static final String SET_UNREAD_STATUS_KEY = "setUnreadStatus";
private static final String NOTIF_ID_KEY = "id"; private static final String NOTIF_ID_KEY = "id";
private static final String ENCRYPTED_PAYLOAD_KEY = "encryptedPayload"; private static final String ENCRYPTED_PAYLOAD_KEY = "encryptedPayload";
private static final String ENCRYPTION_FAILED_KEY = "encryptionFailed"; private static final String ENCRYPTION_FAILED_KEY = "encryptionFailed";
private static final String GROUP_NOTIF_IDS_KEY = "groupNotifIDs";
private static final String CHANNEL_ID = "default"; private static final String CHANNEL_ID = "default";
private static final long[] VIBRATION_SPEC = {500, 500}; private static final long[] VIBRATION_SPEC = {500, 500};
private Bitmap displayableNotificationLargeIcon; private Bitmap displayableNotificationLargeIcon;
private NotificationManager notificationManager; private NotificationManager notificationManager;
private LocalBroadcastManager localBroadcastManager; private LocalBroadcastManager localBroadcastManager;
public static final String RESCIND_KEY = "rescind"; public static final String RESCIND_KEY = "rescind";
public static final String RESCIND_ID_KEY = "rescindID"; public static final String RESCIND_ID_KEY = "rescindID";
▲ Show 20 Lines • Show All 132 Lines • ▼ Show 20 Lines private void handleNotificationRescind(RemoteMessage message) {
for (StatusBarNotification notification : for (StatusBarNotification notification :
notificationManager.getActiveNotifications()) { notificationManager.getActiveNotifications()) {
String tag = notification.getTag(); String tag = notification.getTag();
boolean isGroupMember = boolean isGroupMember =
threadID.equals(notification.getNotification().getGroup()); threadID.equals(notification.getNotification().getGroup());
boolean isGroupSummary = boolean isGroupSummary =
(notification.getNotification().flags & (notification.getNotification().flags &
Notification.FLAG_GROUP_SUMMARY) == Notification.FLAG_GROUP_SUMMARY; Notification.FLAG_GROUP_SUMMARY) == Notification.FLAG_GROUP_SUMMARY;
if (tag != null && tag.equals(rescindID)) { if (tag != null && tag.equals(rescindID)) {
notificationManager.cancel(notification.getTag(), notification.getId()); notificationManager.cancel(notification.getTag(), notification.getId());
} else if (isGroupMember && isGroupSummary) { } else if (isGroupMember && isGroupSummary) {
groupSummaryPresent = true; groupSummaryPresent = true;
removeNotificationFromGroupSummary(threadID, rescindID, notification);
} else if (isGroupMember) { } else if (isGroupMember) {
threadGroupPresent = true; threadGroupPresent = true;
} else if (isGroupSummary) {
checkForUnmatchedRescind(threadID, rescindID, notification);
} }
} }
if (groupSummaryPresent && !threadGroupPresent) { if (groupSummaryPresent && !threadGroupPresent) {
notificationManager.cancel(threadID, threadID.hashCode()); notificationManager.cancel(threadID, threadID.hashCode());
} }
} }
Show All 9 Lines private void addToThreadGroupAndDisplay(
NotificationCompat.Builder groupSummaryNotificationBuilder = NotificationCompat.Builder groupSummaryNotificationBuilder =
new NotificationCompat.Builder(this.getApplicationContext()) new NotificationCompat.Builder(this.getApplicationContext())
.setChannelId(CHANNEL_ID) .setChannelId(CHANNEL_ID)
.setSmallIcon(R.drawable.notif_icon) .setSmallIcon(R.drawable.notif_icon)
.setContentIntent( .setContentIntent(
this.createStartMainActivityAction(threadID, threadID)) this.createStartMainActivityAction(threadID, threadID))
.setGroup(threadID) .setGroup(threadID)
.setGroupSummary(true) .setGroupSummary(true)
.setAutoCancel(true) .setAutoCancel(false)
ashoatUnsubmitted
Not Done
Inline Actions

Do we really want to make this change in production? Should we consider gating it on isStaffRelease?

ashoat: Do we really want to make this change in production? Should we consider gating it on…
marcinAuthorUnsubmitted
Done
Inline Actions

This is my oversight. We want it false for staff to avoid accidental evidence removal. However we don't want it for public users. I will update the diff shortly.

marcin: This is my oversight. We want it false for staff to avoid accidental evidence removal. However…
.setGroupAlertBehavior(NotificationCompat.GROUP_ALERT_CHILDREN); .setGroupAlertBehavior(NotificationCompat.GROUP_ALERT_CHILDREN);
if (StaffUtils.isStaffRelease()) {
ArrayList<String> groupNotifIDs =
recordNotificationInGroupSummary(threadID, notificationID);
String notificationSummaryBody = groupNotifIDs.stream().reduce(
"Notif IDs: ", (acc, el) -> acc + System.lineSeparator() + el);
Bundle data = new Bundle();
data.putStringArrayList(GROUP_NOTIF_IDS_KEY, groupNotifIDs);
groupSummaryNotificationBuilder =
groupSummaryNotificationBuilder
.setContentTitle("Summary for thread id " + threadID)
.setExtras(data)
.setStyle(new NotificationCompat.BigTextStyle().bigText(
notificationSummaryBody));
}
notificationManager.notify( notificationManager.notify(
notificationID, notificationID.hashCode(), notificationBuilder.build()); notificationID, notificationID.hashCode(), notificationBuilder.build());
notificationManager.notify( notificationManager.notify(
threadID, threadID.hashCode(), groupSummaryNotificationBuilder.build()); threadID, threadID.hashCode(), groupSummaryNotificationBuilder.build());
} }
private void displayNotification(RemoteMessage message) { private void displayNotification(RemoteMessage message) {
if (message.getData().get(RESCIND_KEY) != null) { if (message.getData().get(RESCIND_KEY) != null) {
▲ Show 20 Lines • Show All 81 Lines • ▼ Show 20 Linespublic class CommNotificationsHandler extends FirebaseMessagingService {
} }
private Bundle private Bundle
serializeMessageDataForIntentAttachment(RemoteMessage message) { serializeMessageDataForIntentAttachment(RemoteMessage message) {
Bundle bundle = new Bundle(); Bundle bundle = new Bundle();
message.getData().forEach(bundle::putString); message.getData().forEach(bundle::putString);
return bundle; return bundle;
} }
private void displayErrorMessageNotification(
String errorMessage,
String errorTitle,
String largeErrorData) {
NotificationCompat.Builder errorNotificationBuilder =
new NotificationCompat.Builder(this.getApplicationContext())
.setDefaults(Notification.DEFAULT_ALL)
.setChannelId(CHANNEL_ID)
.setSmallIcon(R.drawable.notif_icon)
.setLargeIcon(displayableNotificationLargeIcon);
if (errorMessage != null) {
errorNotificationBuilder =
errorNotificationBuilder.setContentText(errorMessage);
}
if (errorTitle != null) {
errorNotificationBuilder =
errorNotificationBuilder.setContentTitle(errorTitle);
}
if (largeErrorData != null) {
errorNotificationBuilder = errorNotificationBuilder.setStyle(
new NotificationCompat.BigTextStyle().bigText(largeErrorData));
}
notificationManager.notify(
errorMessage,
errorMessage.hashCode(),
errorNotificationBuilder.build());
}
private boolean
isGroupSummary(StatusBarNotification notification, String threadID) {
boolean isAnySummary = (notification.getNotification().flags &
Notification.FLAG_GROUP_SUMMARY) != 0;
if (threadID == null) {
return isAnySummary;
}
return isAnySummary &&
threadID.equals(notification.getNotification().getGroup());
}
private ArrayList<String>
recordNotificationInGroupSummary(String threadID, String notificationID) {
ArrayList<String> groupNotifIDs = new ArrayList<>();
for (StatusBarNotification notif :
notificationManager.getActiveNotifications()) {
if (!isGroupSummary(notif, threadID)) {
continue;
}
groupNotifIDs = notif.getNotification().extras.getStringArrayList(
GROUP_NOTIF_IDS_KEY);
break;
}
groupNotifIDs.add(notificationID);
return groupNotifIDs;
}
private void removeNotificationFromGroupSummary(
String threadID,
String notificationID,
StatusBarNotification groupSummaryNotification) {
if (!StaffUtils.isStaffRelease()) {
return;
}
ArrayList<String> groupNotifIDs =
groupSummaryNotification.getNotification().extras.getStringArrayList(
GROUP_NOTIF_IDS_KEY);
if (groupNotifIDs == null) {
displayErrorMessageNotification(
"Empty summary notif for thread ID " + threadID,
"Empty Summary Notif",
"Summary notification for thread ID " + threadID +
" had empty body when rescinding " + notificationID);
}
int notificationIndex = groupNotifIDs.indexOf(notificationID);
if (notificationIndex == -1) {
displayErrorMessageNotification(
"Notif with ID " + notificationID + "not in " + threadID,
"Unrecorded Notif",
"Rescinded notification with id " + notificationID +
" not found in group summary for thread id " + threadID);
return;
}
groupNotifIDs.remove(notificationIndex);
String notificationSummaryBody = groupNotifIDs.stream().reduce(
"Notif IDs: ", (acc, el) -> acc + System.lineSeparator() + el);
Bundle data = new Bundle();
data.putStringArrayList(GROUP_NOTIF_IDS_KEY, groupNotifIDs);
NotificationCompat.Builder groupSummaryNotificationBuilder =
new NotificationCompat.Builder(this.getApplicationContext())
.setChannelId(CHANNEL_ID)
.setSmallIcon(R.drawable.notif_icon)
.setContentIntent(
this.createStartMainActivityAction(threadID, threadID))
.setContentTitle("Summary for thread id " + threadID)
.setExtras(data)
.setStyle(new NotificationCompat.BigTextStyle().bigText(
notificationSummaryBody))
.setGroup(threadID)
.setGroupSummary(true)
.setAutoCancel(false)
.setGroupAlertBehavior(NotificationCompat.GROUP_ALERT_CHILDREN);
notificationManager.notify(
threadID, threadID.hashCode(), groupSummaryNotificationBuilder.build());
}
private void checkForUnmatchedRescind(
String threadID,
String notificationID,
StatusBarNotification anySummaryNotification) {
if (!StaffUtils.isStaffRelease()) {
return;
}
ArrayList<String> anyGroupNotifIDs =
anySummaryNotification.getNotification().extras.getStringArrayList(
GROUP_NOTIF_IDS_KEY);
if (anyGroupNotifIDs == null) {
return;
}
String groupID = anySummaryNotification.getNotification().getGroup();
for (String notifID : anyGroupNotifIDs) {
if (!notificationID.equals(notifID)) {
continue;
}
displayErrorMessageNotification(
"Summary for thread id " + groupID + "has " + notifID,
"Rescind Mismatch",
"Summary notif for thread id " + groupID + " contains notif id " +
notifID + " which was received in rescind with thread id " +
threadID);
}
}
} }