🚚 Move shells closer to the interfaces

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

diff --git a/src/main/kotlin/net/pterodactylus/sone/data/Post.kt b/src/main/kotlin/net/pterodactylus/sone/data/Post.kt
index 955196a..5781508 100644 (file)
--- a/src/main/kotlin/net/pterodactylus/sone/data/Post.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/data/Post.kt
@@ -1,5 +1,6 @@
 package net.pterodactylus.sone.data
 
+import net.pterodactylus.sone.database.PostBuilder
 import java.util.Comparator.comparing
 
 /**
@@ -12,5 +13,14 @@ val noFuturePost: (Post) -> Boolean = { it.time <= System.currentTimeMillis() }
 /**
  * Comparator that orders posts by their time, newest posts first.
  */
-@get:JvmName("newestFirst")
-val newestFirst: Comparator<Post> = comparing(Post::getTime).reversed()
+@get:JvmName("newestPostFirst")
+val newestPostFirst: Comparator<Post> = comparing(Post::getTime).reversed()
+
+data class PostShell(val id: String, val soneId: String, val recipientId: String?, val time: Long, val text: String) {
+
+       fun build(postBuilder: PostBuilder) =
+                       postBuilder.withId(id).from(soneId).let { if (recipientId != null) it.to(recipientId) else it }.withTime(time).withText(text).build()
+
+}
+
+fun Post.toShell() = PostShell(id, sone!!.id, recipient.orNull()?.id, time, text)