X-Git-Url: https://git.nmode.ca/signal-cli/blobdiff_plain/ce7aa580b6f0580cdcf7fd68fcc8efba737d21ed..04de0010b57b4d5ab5cdda5cb79507cf7c7dbabe:/src/main/java/org/asamk/signal/json/JsonQuotedAttachment.java

diff --git a/src/main/java/org/asamk/signal/json/JsonQuotedAttachment.java b/src/main/java/org/asamk/signal/json/JsonQuotedAttachment.java
index a72ba7c6..e5022798 100644
--- a/src/main/java/org/asamk/signal/json/JsonQuotedAttachment.java
+++ b/src/main/java/org/asamk/signal/json/JsonQuotedAttachment.java
@@ -2,21 +2,18 @@ package org.asamk.signal.json;
 
 import com.fasterxml.jackson.annotation.JsonInclude;
 
-import org.whispersystems.signalservice.api.messages.SignalServiceDataMessage;
+import org.asamk.signal.manager.api.MessageEnvelope;
 
 public record JsonQuotedAttachment(
         String contentType, String filename, @JsonInclude(JsonInclude.Include.NON_NULL) JsonAttachment thumbnail
 ) {
 
-    static JsonQuotedAttachment from(SignalServiceDataMessage.Quote.QuotedAttachment quotedAttachment) {
-        final var contentType = quotedAttachment.getContentType();
-        final var filename = quotedAttachment.getFileName();
-        final JsonAttachment thumbnail;
-        if (quotedAttachment.getThumbnail() != null) {
-            thumbnail = JsonAttachment.from(quotedAttachment.getThumbnail());
-        } else {
-            thumbnail = null;
-        }
+    static JsonQuotedAttachment from(MessageEnvelope.Data.Attachment quotedAttachment) {
+        final var contentType = quotedAttachment.contentType();
+        final var filename = quotedAttachment.fileName().orElse(null);
+        final var thumbnail = quotedAttachment.thumbnail().isPresent()
+                ? JsonAttachment.from(quotedAttachment.thumbnail().get())
+                : null;
         return new JsonQuotedAttachment(contentType, filename, thumbnail);
     }
 }