forked from GitHub-Mirror/riotX-android
96 lines
3.7 KiB
Kotlin
96 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.mapper
|
|
|
|
import com.squareup.moshi.JsonDataException
|
|
import im.vector.matrix.android.api.session.crypto.MXCryptoError
|
|
import im.vector.matrix.android.api.session.events.model.Event
|
|
import im.vector.matrix.android.api.session.events.model.UnsignedData
|
|
import im.vector.matrix.android.internal.crypto.algorithms.olm.OlmDecryptionResult
|
|
import im.vector.matrix.android.internal.database.model.EventEntity
|
|
import im.vector.matrix.android.internal.di.MoshiProvider
|
|
import timber.log.Timber
|
|
|
|
internal object EventMapper {
|
|
|
|
|
|
fun map(event: Event, roomId: String): EventEntity {
|
|
val uds = if (event.unsignedData == null) null
|
|
else MoshiProvider.providesMoshi().adapter(UnsignedData::class.java).toJson(event.unsignedData)
|
|
val eventEntity = EventEntity()
|
|
eventEntity.eventId = event.eventId ?: ""
|
|
eventEntity.roomId = event.roomId ?: roomId
|
|
eventEntity.content = ContentMapper.map(event.content)
|
|
val resolvedPrevContent = event.prevContent ?: event.unsignedData?.prevContent
|
|
eventEntity.prevContent = ContentMapper.map(resolvedPrevContent)
|
|
eventEntity.stateKey = event.stateKey
|
|
eventEntity.type = event.getClearType()
|
|
eventEntity.sender = event.senderId
|
|
eventEntity.originServerTs = event.originServerTs
|
|
eventEntity.redacts = event.redacts
|
|
eventEntity.age = event.unsignedData?.age ?: event.originServerTs
|
|
eventEntity.unsignedData = uds
|
|
return eventEntity
|
|
}
|
|
|
|
fun map(eventEntity: EventEntity): Event {
|
|
val ud = if (eventEntity.unsignedData.isNullOrBlank()) {
|
|
null
|
|
} else {
|
|
try {
|
|
MoshiProvider.providesMoshi().adapter(UnsignedData::class.java).fromJson(eventEntity.unsignedData)
|
|
} catch (t: JsonDataException) {
|
|
null
|
|
}
|
|
|
|
}
|
|
return Event(
|
|
type = eventEntity.type,
|
|
eventId = eventEntity.eventId,
|
|
content = ContentMapper.map(eventEntity.content),
|
|
prevContent = ContentMapper.map(eventEntity.prevContent),
|
|
originServerTs = eventEntity.originServerTs,
|
|
senderId = eventEntity.sender,
|
|
stateKey = eventEntity.stateKey,
|
|
roomId = eventEntity.roomId,
|
|
unsignedData = ud,
|
|
redacts = eventEntity.redacts
|
|
).also {
|
|
it.sendState = eventEntity.sendState
|
|
eventEntity.decryptionResultJson?.let { json ->
|
|
try {
|
|
it.mxDecryptionResult = MoshiProvider.providesMoshi().adapter(OlmDecryptionResult::class.java).fromJson(json)
|
|
} catch (t: JsonDataException) {
|
|
Timber.e(t, "Failed to parse decryption result")
|
|
}
|
|
}
|
|
//TODO get the full crypto error object
|
|
it.mCryptoError = eventEntity.decryptionErrorCode?.let { MXCryptoError.ErrorType.valueOf(it) }
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
internal fun EventEntity.asDomain(): Event {
|
|
return EventMapper.map(this)
|
|
}
|
|
|
|
internal fun Event.toEntity(roomId: String): EventEntity {
|
|
return EventMapper.map(this, roomId)
|
|
}
|
|
|