forked from GitHub-Mirror/riotX-android
86 lines
3.7 KiB
Kotlin
86 lines
3.7 KiB
Kotlin
/*
|
|
* Copyright 2019 New Vector Ltd
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
package im.vector.matrix.android.internal.database.helper
|
|
|
|
import im.vector.matrix.android.api.session.events.model.EventType
|
|
import im.vector.matrix.android.api.session.events.model.toModel
|
|
import im.vector.matrix.android.api.session.room.model.RoomMember
|
|
import im.vector.matrix.android.internal.database.mapper.ContentMapper
|
|
import im.vector.matrix.android.internal.database.model.*
|
|
import im.vector.matrix.android.internal.database.query.next
|
|
import im.vector.matrix.android.internal.database.query.prev
|
|
import im.vector.matrix.android.internal.database.query.where
|
|
import im.vector.matrix.android.internal.extensions.assertIsManaged
|
|
import im.vector.matrix.android.internal.session.room.membership.RoomMembers
|
|
import io.realm.Realm
|
|
import io.realm.RealmList
|
|
import io.realm.RealmQuery
|
|
|
|
|
|
internal fun TimelineEventEntity.updateSenderData() {
|
|
assertIsManaged()
|
|
val roomEntity = RoomEntity.where(realm, roomId = roomId).findFirst() ?: return
|
|
val stateIndex = root?.stateIndex ?: return
|
|
val senderId = root?.sender ?: return
|
|
val chunkEntity = chunk?.firstOrNull() ?: return
|
|
val isUnlinked = chunkEntity.isUnlinked()
|
|
var senderMembershipEvent: EventEntity?
|
|
var senderRoomMemberContent: String?
|
|
when {
|
|
stateIndex <= 0 -> {
|
|
senderMembershipEvent = chunkEntity.timelineEvents.buildQuery(senderId, isUnlinked).next(from = stateIndex)?.root
|
|
senderRoomMemberContent = senderMembershipEvent?.prevContent
|
|
}
|
|
else -> {
|
|
senderMembershipEvent = chunkEntity.timelineEvents.buildQuery(senderId, isUnlinked).prev(since = stateIndex)?.root
|
|
senderRoomMemberContent = senderMembershipEvent?.content
|
|
}
|
|
}
|
|
|
|
// We fallback to untimelinedStateEvents if we can't find membership events in timeline
|
|
if (senderMembershipEvent == null) {
|
|
senderMembershipEvent = roomEntity.untimelinedStateEvents
|
|
.where()
|
|
.equalTo(EventEntityFields.STATE_KEY, senderId)
|
|
.equalTo(EventEntityFields.TYPE, EventType.STATE_ROOM_MEMBER)
|
|
.prev(since = stateIndex)
|
|
senderRoomMemberContent = senderMembershipEvent?.content
|
|
}
|
|
val senderRoomMember: RoomMember? = ContentMapper.map(senderRoomMemberContent).toModel()
|
|
this.senderAvatar = senderRoomMember?.avatarUrl
|
|
this.senderName = senderRoomMember?.displayName
|
|
this.isUniqueDisplayName = RoomMembers(realm, roomId).isUniqueDisplayName(senderRoomMember?.displayName)
|
|
this.senderMembershipEvent = senderMembershipEvent
|
|
}
|
|
|
|
internal fun TimelineEventEntity.Companion.nextId(realm: Realm): Long{
|
|
val currentIdNum = realm.where(TimelineEventEntity::class.java).max(TimelineEventEntityFields.LOCAL_ID)
|
|
return if (currentIdNum == null) {
|
|
1
|
|
} else {
|
|
currentIdNum.toLong() + 1
|
|
}
|
|
}
|
|
|
|
private fun RealmList<TimelineEventEntity>.buildQuery(sender: String, isUnlinked: Boolean): RealmQuery<TimelineEventEntity> {
|
|
return where()
|
|
.equalTo(TimelineEventEntityFields.ROOT.STATE_KEY, sender)
|
|
.equalTo(TimelineEventEntityFields.ROOT.TYPE, EventType.STATE_ROOM_MEMBER)
|
|
.equalTo(TimelineEventEntityFields.ROOT.IS_UNLINKED, isUnlinked)
|
|
}
|
|
|