X-Git-Url: https://git.nmode.ca/signal-cli/blobdiff_plain/a54fc92c05c5c8b532e029e82eedd73f9440e138..43face8ead95344e3d0dcbc94c6cf85959f26203:/src/main/java/org/asamk/signal/json/JsonSyncMessage.java

diff --git a/src/main/java/org/asamk/signal/json/JsonSyncMessage.java b/src/main/java/org/asamk/signal/json/JsonSyncMessage.java
index 31c39a3f..802c07d5 100644
--- a/src/main/java/org/asamk/signal/json/JsonSyncMessage.java
+++ b/src/main/java/org/asamk/signal/json/JsonSyncMessage.java
@@ -1,11 +1,11 @@
 package org.asamk.signal.json;
 
-import org.asamk.Signal;
-import org.whispersystems.signalservice.api.messages.multidevice.ReadMessage;
-import org.whispersystems.signalservice.api.messages.multidevice.SignalServiceSyncMessage;
-import org.whispersystems.signalservice.api.push.SignalServiceAddress;
+import com.fasterxml.jackson.annotation.JsonInclude;
+
+import org.asamk.signal.manager.api.MessageEnvelope;
+import org.asamk.signal.manager.api.RecipientAddress;
+import org.asamk.signal.manager.groups.GroupId;
 
-import java.util.ArrayList;
 import java.util.List;
 
 enum JsonSyncMessageType {
@@ -14,37 +14,50 @@ enum JsonSyncMessageType {
     REQUEST_SYNC
 }
 
-class JsonSyncMessage {
-
-    JsonSyncDataMessage sentMessage;
-    List<String> blockedNumbers;
-    List<ReadMessage> readMessages;
-    JsonSyncMessageType type;
-
-    JsonSyncMessage(SignalServiceSyncMessage syncMessage) {
-        if (syncMessage.getSent().isPresent()) {
-            this.sentMessage = new JsonSyncDataMessage(syncMessage.getSent().get());
-        }
-        if (syncMessage.getBlockedList().isPresent()) {
-            this.blockedNumbers = new ArrayList<>(syncMessage.getBlockedList().get().getAddresses().size());
-            for (SignalServiceAddress address : syncMessage.getBlockedList().get().getAddresses()) {
-                this.blockedNumbers.add(address.getLegacyIdentifier());
-            }
-        }
-        if (syncMessage.getRead().isPresent()) {
-            this.readMessages = syncMessage.getRead().get();
+record JsonSyncMessage(
+        @JsonInclude(JsonInclude.Include.NON_NULL) JsonSyncDataMessage sentMessage,
+        @JsonInclude(JsonInclude.Include.NON_NULL) JsonSyncStoryMessage sentStoryMessage,
+        @JsonInclude(JsonInclude.Include.NON_NULL) List<String> blockedNumbers,
+        @JsonInclude(JsonInclude.Include.NON_NULL) List<String> blockedGroupIds,
+        @JsonInclude(JsonInclude.Include.NON_NULL) List<JsonSyncReadMessage> readMessages,
+        @JsonInclude(JsonInclude.Include.NON_NULL) JsonSyncMessageType type
+) {
+
+    static JsonSyncMessage from(MessageEnvelope.Sync syncMessage) {
+        final var sentMessage = syncMessage.sent().isPresent() && syncMessage.sent().get().story().isEmpty()
+                ? JsonSyncDataMessage.from(syncMessage.sent().get())
+                : null;
+        final var sentStoryMessage = syncMessage.sent().isPresent() && syncMessage.sent().get().story().isPresent()
+                ? JsonSyncStoryMessage.from(syncMessage.sent().get())
+                : null;
+        final List<String> blockedNumbers;
+        final List<String> blockedGroupIds;
+        if (syncMessage.blocked().isPresent()) {
+            blockedNumbers = syncMessage.blocked()
+                    .get()
+                    .recipients()
+                    .stream()
+                    .map(RecipientAddress::getLegacyIdentifier)
+                    .toList();
+            blockedGroupIds = syncMessage.blocked().get().groupIds().stream().map(GroupId::toBase64).toList();
+        } else {
+            blockedNumbers = null;
+            blockedGroupIds = null;
         }
 
-        if (syncMessage.getContacts().isPresent()) {
-            this.type = JsonSyncMessageType.CONTACTS_SYNC;
-        } else if (syncMessage.getGroups().isPresent()) {
-            this.type = JsonSyncMessageType.GROUPS_SYNC;
-        } else if (syncMessage.getRequest().isPresent()) {
-            this.type = JsonSyncMessageType.REQUEST_SYNC;
+        final var readMessages = syncMessage.read().size() > 0 ? syncMessage.read()
+                .stream()
+                .map(JsonSyncReadMessage::from)
+                .toList() : null;
+
+        final JsonSyncMessageType type;
+        if (syncMessage.contacts().isPresent()) {
+            type = JsonSyncMessageType.CONTACTS_SYNC;
+        } else if (syncMessage.groups().isPresent()) {
+            type = JsonSyncMessageType.GROUPS_SYNC;
+        } else {
+            type = null;
         }
-    }
-
-    JsonSyncMessage(Signal.SyncMessageReceived messageReceived) {
-        sentMessage = new JsonSyncDataMessage(messageReceived);
+        return new JsonSyncMessage(sentMessage, sentStoryMessage, blockedNumbers, blockedGroupIds, readMessages, type);
     }
 }