2018-10-17 16:21:09 +00:00
|
|
|
package im.vector.matrix.android.internal.session.sync
|
2018-10-11 18:01:08 +00:00
|
|
|
|
2018-10-18 09:16:02 +00:00
|
|
|
import com.zhuinden.monarchy.Monarchy
|
2018-10-17 16:21:09 +00:00
|
|
|
import im.vector.matrix.android.api.session.events.model.Event
|
2018-10-24 16:11:01 +00:00
|
|
|
import im.vector.matrix.android.api.session.events.model.EventType
|
2018-10-23 16:25:28 +00:00
|
|
|
import im.vector.matrix.android.api.session.room.model.MyMembership
|
2018-11-13 18:17:59 +00:00
|
|
|
import im.vector.matrix.android.internal.database.helper.add
|
2018-10-12 17:26:22 +00:00
|
|
|
import im.vector.matrix.android.internal.database.model.ChunkEntity
|
|
|
|
import im.vector.matrix.android.internal.database.model.RoomEntity
|
2018-10-23 16:25:28 +00:00
|
|
|
import im.vector.matrix.android.internal.database.model.RoomSummaryEntity
|
2018-10-18 16:38:11 +00:00
|
|
|
import im.vector.matrix.android.internal.database.query.findAllIncludingEvents
|
2018-10-22 13:17:50 +00:00
|
|
|
import im.vector.matrix.android.internal.database.query.findLastLiveChunkFromRoom
|
2018-10-18 16:38:11 +00:00
|
|
|
import im.vector.matrix.android.internal.database.query.where
|
2018-11-13 18:17:59 +00:00
|
|
|
import im.vector.matrix.android.internal.session.room.timeline.PaginationDirection
|
2018-11-14 18:32:42 +00:00
|
|
|
import im.vector.matrix.android.internal.session.sync.model.*
|
2018-10-15 17:42:13 +00:00
|
|
|
import io.realm.Realm
|
2018-10-26 10:31:47 +00:00
|
|
|
import io.realm.kotlin.createObject
|
2018-10-11 18:01:08 +00:00
|
|
|
|
|
|
|
|
2018-10-24 16:11:01 +00:00
|
|
|
internal class RoomSyncHandler(private val monarchy: Monarchy,
|
|
|
|
private val stateEventsChunkHandler: StateEventsChunkHandler,
|
|
|
|
private val readReceiptHandler: ReadReceiptHandler) {
|
2018-10-11 18:01:08 +00:00
|
|
|
|
2018-10-15 17:42:13 +00:00
|
|
|
sealed class HandlingStrategy {
|
|
|
|
data class JOINED(val data: Map<String, RoomSync>) : HandlingStrategy()
|
|
|
|
data class INVITED(val data: Map<String, InvitedRoomSync>) : HandlingStrategy()
|
|
|
|
data class LEFT(val data: Map<String, RoomSync>) : HandlingStrategy()
|
|
|
|
}
|
2018-10-11 18:01:08 +00:00
|
|
|
|
2018-10-30 17:22:29 +00:00
|
|
|
fun handle(roomsSyncResponse: RoomsSyncResponse) {
|
|
|
|
monarchy.runTransactionSync { realm ->
|
2018-11-05 13:31:45 +00:00
|
|
|
handleRoomSync(realm, RoomSyncHandler.HandlingStrategy.JOINED(roomsSyncResponse.join))
|
|
|
|
handleRoomSync(realm, RoomSyncHandler.HandlingStrategy.INVITED(roomsSyncResponse.invite))
|
|
|
|
handleRoomSync(realm, RoomSyncHandler.HandlingStrategy.LEFT(roomsSyncResponse.leave))
|
2018-10-30 17:22:29 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// PRIVATE METHODS *****************************************************************************
|
|
|
|
|
2018-11-05 13:31:45 +00:00
|
|
|
private fun handleRoomSync(realm: Realm, handlingStrategy: HandlingStrategy) {
|
|
|
|
val rooms = when (handlingStrategy) {
|
2018-11-14 18:32:42 +00:00
|
|
|
is HandlingStrategy.JOINED -> handlingStrategy.data.map { handleJoinedRoom(realm, it.key, it.value) }
|
2018-11-05 13:31:45 +00:00
|
|
|
is HandlingStrategy.INVITED -> handlingStrategy.data.map { handleInvitedRoom(realm, it.key, it.value) }
|
2018-11-14 18:32:42 +00:00
|
|
|
is HandlingStrategy.LEFT -> handlingStrategy.data.map { handleLeftRoom(it.key, it.value) }
|
2018-10-12 17:26:22 +00:00
|
|
|
}
|
2018-11-05 13:31:45 +00:00
|
|
|
realm.insertOrUpdate(rooms)
|
2018-10-24 16:11:01 +00:00
|
|
|
}
|
2018-10-15 17:42:13 +00:00
|
|
|
|
|
|
|
private fun handleJoinedRoom(realm: Realm,
|
|
|
|
roomId: String,
|
|
|
|
roomSync: RoomSync): RoomEntity {
|
|
|
|
|
2018-10-26 10:31:47 +00:00
|
|
|
val roomEntity = RoomEntity.where(realm, roomId).findFirst()
|
2018-11-14 18:32:42 +00:00
|
|
|
?: RoomEntity(roomId)
|
2018-10-15 17:42:13 +00:00
|
|
|
|
2018-10-23 16:25:28 +00:00
|
|
|
if (roomEntity.membership == MyMembership.INVITED) {
|
2018-10-15 17:42:13 +00:00
|
|
|
roomEntity.chunks.deleteAllFromRealm()
|
2018-10-12 17:26:22 +00:00
|
|
|
}
|
2018-10-15 17:42:13 +00:00
|
|
|
|
2018-10-23 16:25:28 +00:00
|
|
|
roomEntity.membership = MyMembership.JOINED
|
|
|
|
|
2018-10-15 17:42:13 +00:00
|
|
|
if (roomSync.state != null && roomSync.state.events.isNotEmpty()) {
|
2018-11-14 18:32:42 +00:00
|
|
|
val chunkEntity = stateEventsChunkHandler.handle(realm, roomId, roomSync.state.events, PaginationDirection.FORWARDS)
|
2018-10-15 17:42:13 +00:00
|
|
|
if (!roomEntity.chunks.contains(chunkEntity)) {
|
|
|
|
roomEntity.chunks.add(chunkEntity)
|
|
|
|
}
|
|
|
|
}
|
2018-11-13 18:17:59 +00:00
|
|
|
|
2018-10-15 17:42:13 +00:00
|
|
|
if (roomSync.timeline != null && roomSync.timeline.events.isNotEmpty()) {
|
2018-10-26 10:31:47 +00:00
|
|
|
val chunkEntity = handleTimelineEvents(realm, roomId, roomSync.timeline.events, roomSync.timeline.prevToken, isLimited = roomSync.timeline.limited)
|
2018-10-15 17:42:13 +00:00
|
|
|
if (!roomEntity.chunks.contains(chunkEntity)) {
|
|
|
|
roomEntity.chunks.add(chunkEntity)
|
|
|
|
}
|
2018-10-12 17:26:22 +00:00
|
|
|
}
|
2018-11-05 13:31:45 +00:00
|
|
|
|
2018-10-29 13:57:36 +00:00
|
|
|
if (roomSync.summary != null) {
|
|
|
|
handleRoomSummary(realm, roomId, roomSync.summary)
|
|
|
|
}
|
2018-11-05 13:31:45 +00:00
|
|
|
|
|
|
|
if (roomSync.ephemeral != null && roomSync.ephemeral.events.isNotEmpty()) {
|
|
|
|
handleEphemeral(realm, roomId, roomSync.ephemeral)
|
|
|
|
}
|
2018-10-15 17:42:13 +00:00
|
|
|
return roomEntity
|
|
|
|
}
|
2018-10-12 17:26:22 +00:00
|
|
|
|
2018-10-15 17:42:13 +00:00
|
|
|
private fun handleInvitedRoom(realm: Realm,
|
|
|
|
roomId: String,
|
|
|
|
roomSync:
|
|
|
|
InvitedRoomSync): RoomEntity {
|
|
|
|
val roomEntity = RoomEntity()
|
|
|
|
roomEntity.roomId = roomId
|
2018-10-23 16:25:28 +00:00
|
|
|
roomEntity.membership = MyMembership.INVITED
|
2018-10-15 17:42:13 +00:00
|
|
|
if (roomSync.inviteState != null && roomSync.inviteState.events.isNotEmpty()) {
|
2018-10-26 10:31:47 +00:00
|
|
|
val chunkEntity = handleTimelineEvents(realm, roomId, roomSync.inviteState.events)
|
2018-10-15 17:42:13 +00:00
|
|
|
if (!roomEntity.chunks.contains(chunkEntity)) {
|
|
|
|
roomEntity.chunks.add(chunkEntity)
|
|
|
|
}
|
2018-10-12 17:26:22 +00:00
|
|
|
}
|
|
|
|
return roomEntity
|
|
|
|
}
|
2018-10-11 18:01:08 +00:00
|
|
|
|
2018-10-15 17:42:13 +00:00
|
|
|
// TODO : handle it
|
|
|
|
private fun handleLeftRoom(roomId: String,
|
|
|
|
roomSync: RoomSync): RoomEntity {
|
|
|
|
return RoomEntity().apply {
|
|
|
|
this.roomId = roomId
|
2018-10-23 16:25:28 +00:00
|
|
|
this.membership = MyMembership.LEFT
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
private fun handleRoomSummary(realm: Realm,
|
|
|
|
roomId: String,
|
|
|
|
roomSummary: RoomSyncSummary) {
|
|
|
|
|
|
|
|
val roomSummaryEntity = RoomSummaryEntity.where(realm, roomId).findFirst()
|
2018-11-14 18:32:42 +00:00
|
|
|
?: RoomSummaryEntity(roomId)
|
2018-10-23 16:25:28 +00:00
|
|
|
|
|
|
|
if (roomSummary.heroes.isNotEmpty()) {
|
|
|
|
roomSummaryEntity.heroes.clear()
|
|
|
|
roomSummaryEntity.heroes.addAll(roomSummary.heroes)
|
|
|
|
}
|
|
|
|
if (roomSummary.invitedMembersCount != null) {
|
|
|
|
roomSummaryEntity.invitedMembersCount = roomSummary.invitedMembersCount
|
|
|
|
}
|
|
|
|
if (roomSummary.joinedMembersCount != null) {
|
|
|
|
roomSummaryEntity.joinedMembersCount = roomSummary.joinedMembersCount
|
2018-10-15 17:42:13 +00:00
|
|
|
}
|
2018-10-23 16:25:28 +00:00
|
|
|
realm.insertOrUpdate(roomSummaryEntity)
|
2018-10-15 17:42:13 +00:00
|
|
|
}
|
|
|
|
|
2018-10-26 10:31:47 +00:00
|
|
|
private fun handleTimelineEvents(realm: Realm,
|
|
|
|
roomId: String,
|
|
|
|
eventList: List<Event>,
|
|
|
|
prevToken: String? = null,
|
|
|
|
nextToken: String? = null,
|
|
|
|
isLimited: Boolean = true): ChunkEntity {
|
2018-11-13 09:50:53 +00:00
|
|
|
|
|
|
|
val lastChunk = ChunkEntity.findLastLiveChunkFromRoom(realm, roomId)
|
2018-10-15 17:42:13 +00:00
|
|
|
val chunkEntity = if (!isLimited) {
|
2018-11-13 09:50:53 +00:00
|
|
|
lastChunk
|
2018-10-15 17:42:13 +00:00
|
|
|
} else {
|
2018-10-16 13:52:30 +00:00
|
|
|
val eventIds = eventList.filter { it.eventId != null }.map { it.eventId!! }
|
2018-10-18 16:38:11 +00:00
|
|
|
ChunkEntity.findAllIncludingEvents(realm, eventIds).firstOrNull()
|
2018-10-26 10:31:47 +00:00
|
|
|
} ?: realm.createObject<ChunkEntity>().apply { this.prevToken = prevToken }
|
2018-10-15 17:42:13 +00:00
|
|
|
|
2018-11-13 09:50:53 +00:00
|
|
|
lastChunk?.isLast = false
|
|
|
|
chunkEntity.isLast = true
|
2018-10-13 09:18:49 +00:00
|
|
|
chunkEntity.nextToken = nextToken
|
2018-11-13 18:17:59 +00:00
|
|
|
|
|
|
|
var currentStateIndex = chunkEntity.nextStateIndex
|
|
|
|
eventList.forEach { event ->
|
2018-11-14 18:32:42 +00:00
|
|
|
if (event.isStateEvent()) {
|
2018-11-13 18:17:59 +00:00
|
|
|
currentStateIndex += 1
|
|
|
|
}
|
|
|
|
chunkEntity.add(event, currentStateIndex, PaginationDirection.FORWARDS)
|
|
|
|
}
|
|
|
|
chunkEntity.nextStateIndex = currentStateIndex
|
2018-10-13 09:18:49 +00:00
|
|
|
return chunkEntity
|
|
|
|
}
|
|
|
|
|
2018-10-24 16:11:01 +00:00
|
|
|
private fun handleEphemeral(realm: Realm,
|
|
|
|
roomId: String,
|
2018-11-05 13:31:45 +00:00
|
|
|
ephemeral: RoomSyncEphemeral) {
|
2018-10-24 16:11:01 +00:00
|
|
|
ephemeral.events
|
|
|
|
.filter { it.type == EventType.RECEIPT }
|
|
|
|
.map { it.content<ReadReceiptContent>() }
|
|
|
|
.flatMap { readReceiptHandler.handle(realm, roomId, it) }
|
|
|
|
}
|
2018-10-11 18:01:08 +00:00
|
|
|
}
|