Move retrieval of post likes from Core to Post.

[Sone.git] / src / main / java / net / pterodactylus / sone / data / Post.java

diff --git a/src/main/java/net/pterodactylus/sone/data/Post.java b/src/main/java/net/pterodactylus/sone/data/Post.java
index 759a8ba..474f9d1 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/data/Post.java
+++ b/src/main/java/net/pterodactylus/sone/data/Post.java
@@ -17,8 +17,12 @@
 
 package net.pterodactylus.sone.data;
 
+import java.util.Collections;
 import java.util.Comparator;
+import java.util.List;
+import java.util.Set;
 
+import com.google.common.base.Function;
 import com.google.common.base.Optional;
 import com.google.common.base.Predicate;
 
@@ -50,6 +54,13 @@ public interface Post extends Identified {
 
        };
 
+       public static final Function<Post, List<PostReply>> TO_REPLIES = new Function<Post, List<PostReply>>() {
+               @Override
+               public List<PostReply> apply(Post post) {
+                       return (post == null) ? Collections.<PostReply>emptyList() : post.getReplies();
+               }
+       };
+
        //
        // ACCESSORS
        //
@@ -114,4 +125,11 @@ public interface Post extends Identified {
         */
        public Post setKnown(boolean known);
 
+       public void like(Sone localSone);
+       public void unlike(Sone localSone);
+
+       Set<Sone> getLikes();
+
+       List<PostReply> getReplies();
+
 }