X-Git-Url: https://git.nmode.ca/signal-cli/blobdiff_plain/e04c45766d50d6a4e2d3c084145e2115b38bef5c..ae41d0c5026fe868c6198e1005344fc78b6e0a2c:/src/main/java/org/asamk/signal/json/JsonAttachment.java

diff --git a/src/main/java/org/asamk/signal/json/JsonAttachment.java b/src/main/java/org/asamk/signal/json/JsonAttachment.java
new file mode 100644
index 00000000..8a405fc4
--- /dev/null
+++ b/src/main/java/org/asamk/signal/json/JsonAttachment.java
@@ -0,0 +1,31 @@
+package org.asamk.signal.json;
+
+import org.whispersystems.signalservice.api.messages.SignalServiceAttachment;
+import org.whispersystems.signalservice.api.messages.SignalServiceAttachmentPointer;
+
+class JsonAttachment {
+
+    String contentType;
+    String filename;
+    String id;
+    int size;
+
+    JsonAttachment(SignalServiceAttachment attachment) {
+        this.contentType = attachment.getContentType();
+
+        final SignalServiceAttachmentPointer pointer = attachment.asPointer();
+        if (attachment.isPointer()) {
+            this.id = String.valueOf(pointer.getId());
+            if (pointer.getFileName().isPresent()) {
+                this.filename = pointer.getFileName().get();
+            }
+            if (pointer.getSize().isPresent()) {
+                this.size = pointer.getSize().get();
+            }
+        }
+    }
+
+    JsonAttachment(String filename) {
+        this.filename = filename;
+    }
+}