1 files changed, 9 insertions, 41 deletions

diff --git a/juick-core/src/main/java/com/juick/Message.java b/juick-core/src/main/java/com/juick/Message.java
index 25463c46..8f903e1f 100644
--- a/juick-core/src/main/java/com/juick/Message.java
+++ b/juick-core/src/main/java/com/juick/Message.java
@@ -27,16 +27,13 @@ import org.apache.commons.lang3.builder.ToStringBuilder;
 import javax.xml.bind.annotation.*;
 import javax.xml.bind.annotation.adapters.XmlJavaTypeAdapter;
 import java.time.Instant;
-import java.util.ArrayList;
-import java.util.List;
-import java.util.Set;
-import java.util.TreeSet;
+import java.util.*;
 
 /**
  * @author Ugnich Anton
  */
 @XmlRootElement(name = "juick", namespace = "http://juick.com/message")
-@XmlAccessorType(XmlAccessType.PUBLIC_MEMBER)
+@XmlAccessorType()
 public class Message implements Comparable {
     private int mid = 0;
     private int rid = 0;
@@ -67,6 +64,8 @@ public class Message implements Comparable {
     private String attachmentType;
     @XmlTransient
     private Photo photo;
+    @XmlTransient
+    private Attachment attachment;
     public String Video = null;
     public Place Place = null;
     private int likes;
@@ -159,25 +158,6 @@ public class Message implements Comparable {
     }
 
     @JsonIgnore
-    public String getAttachmentURL() {
-        if (attachmentType != null) {
-            StringBuilder builder = new StringBuilder();
-
-            builder.append("http://i.juick.com/");
-            builder.append(attachmentType.equals("mp4") ? "video" : "photos-1024");
-            builder.append("/").append(getMid());
-
-            if (getRid() > 0)
-                builder.append("-").append(getRid());
-
-            builder.append(".").append(attachmentType);
-
-            return builder.toString();
-        }
-        return null;
-    }
-
-    @JsonIgnore
     public String getTagsString() {
         StringBuilder builder = new StringBuilder();
         if (!tags.isEmpty()) {
@@ -352,22 +332,10 @@ public class Message implements Comparable {
         this.repliesBy = repliesBy;
     }
 
-    @JsonProperty("photo")
-    @XmlTransient
-    public Photo getPhotoURLs() {
-        if (StringUtils.isNotBlank(attachmentType)) {
-            Photo photo = new Photo();
-            if (rid > 0) {
-                photo.setSmall(String.format("https://i.juick.com/photos-512/%d-%d.%s", mid, rid, attachmentType));
-                photo.setMedium(String.format("https://i.juick.com/photos-1024/%d-%d.%s", mid, rid, attachmentType));
-                photo.setThumbnail(String.format("https://i.juick.com/ps/%d-%d.%s", mid, rid, attachmentType));
-            } else {
-                photo.setSmall(String.format("https://i.juick.com/photos-512/%d.%s", mid, attachmentType));
-                photo.setMedium(String.format("https://i.juick.com/photos-1024/%d.%s", mid, attachmentType));
-                photo.setThumbnail(String.format("https://i.juick.com/ps/%d.%s", mid, attachmentType));
-            }
-            return photo;
-        }
-        return null;
+    public Attachment getAttachment() {
+        return attachment;
+    }
+    public void setAttachment(Attachment attachment) {
+        this.attachment = attachment;
     }
 }