🎨 Store shells instead of full posts in in-memory database
[Sone.git] / src / main / kotlin / net / pterodactylus / sone / database / memory / MemoryPost.kt
1 /*
2  * Sone - MemoryPost.kt - Copyright Â© 2010–2020 David Roden
3  *
4  * This program is free software: you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation, either version 3 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
16  */
17 package net.pterodactylus.sone.database.memory
18
19 import net.pterodactylus.sone.data.Post
20 import net.pterodactylus.sone.data.Sone
21 import net.pterodactylus.sone.database.PostBuilder
22 import net.pterodactylus.sone.database.SoneProvider
23 import net.pterodactylus.sone.utils.asOptional
24
25 /**
26  * A post is a short message that a user writes in his Sone to let other users
27  * know what is going on.
28  */
29 class MemoryPost(
30                 private val postDatabase: MemoryDatabase,
31                 private val soneProvider: SoneProvider,
32                 override val id: String,
33                 private val soneId: String,
34                 private val recipientId: String?,
35                 private val time: Long,
36                 private val text: String
37 ) : Post {
38
39         override fun isLoaded() = true
40
41         override fun getSone(): Sone = soneProvider.getSone(soneId)!!
42
43         override fun getRecipientId() = recipientId.asOptional()
44
45         override fun getRecipient() = recipientId?.let(soneProvider::getSone).asOptional()
46
47         override fun getTime() = time
48
49         override fun getText() = text
50
51         override fun isKnown() = postDatabase.isPostKnown(this)
52
53         override fun setKnown(known: Boolean) = apply {
54                 postDatabase.setPostKnown(this, known)
55         }
56
57         override fun hashCode() = id.hashCode()
58
59         override fun equals(other: Any?) = (other is MemoryPost) && (other.id == id)
60
61         override fun toString() = "${javaClass.name}[id=$id,sone=$soneId,recipient=$recipientId,time=$time,text=$text]"
62
63         data class Shell(val id: String, val soneId: String, val recipientId: String?, val time: Long, val text: String) {
64
65                 fun build(postBuilder: PostBuilder) =
66                                 postBuilder.withId(id).from(soneId).let { if (recipientId != null) it.to(recipientId) else it }.withTime(time).withText(text).build()
67
68         }
69
70 }
71
72 fun Post.toShell() = MemoryPost.Shell(id, sone!!.id, recipient.orNull()?.id, time, text)