ganfra
4 months ago
4 changed files with 144 additions and 65 deletions
@ -0,0 +1,124 @@
@@ -0,0 +1,124 @@
|
||||
/* |
||||
* Copyright (c) 2024 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 io.element.android.libraries.matrix.impl.room |
||||
|
||||
import io.element.android.appconfig.TimelineConfig |
||||
import io.element.android.libraries.core.coroutine.CoroutineDispatchers |
||||
import io.element.android.libraries.matrix.api.core.RoomId |
||||
import io.element.android.libraries.matrix.api.core.SessionId |
||||
import io.element.android.libraries.matrix.api.notificationsettings.NotificationSettingsService |
||||
import io.element.android.libraries.matrix.api.room.MatrixRoom |
||||
import io.element.android.libraries.matrix.api.roomlist.RoomListService |
||||
import io.element.android.libraries.matrix.api.roomlist.awaitLoaded |
||||
import io.element.android.libraries.matrix.impl.roomlist.fullRoomWithTimeline |
||||
import io.element.android.libraries.matrix.impl.roomlist.roomOrNull |
||||
import io.element.android.libraries.sessionstorage.api.SessionData |
||||
import io.element.android.services.toolbox.api.systemclock.SystemClock |
||||
import kotlinx.coroutines.CoroutineScope |
||||
import kotlinx.coroutines.ExperimentalCoroutinesApi |
||||
import kotlinx.coroutines.sync.Mutex |
||||
import kotlinx.coroutines.sync.withLock |
||||
import kotlinx.coroutines.withContext |
||||
import org.matrix.rustcomponents.sdk.FilterTimelineEventType |
||||
import org.matrix.rustcomponents.sdk.Room |
||||
import org.matrix.rustcomponents.sdk.RoomListException |
||||
import org.matrix.rustcomponents.sdk.RoomListItem |
||||
import org.matrix.rustcomponents.sdk.TimelineEventTypeFilter |
||||
import timber.log.Timber |
||||
import org.matrix.rustcomponents.sdk.RoomListService as InnerRoomListService |
||||
|
||||
class RustRoomFactory( |
||||
private val sessionId: SessionId, |
||||
private val notificationSettingsService: NotificationSettingsService, |
||||
private val sessionCoroutineScope: CoroutineScope, |
||||
private val dispatchers: CoroutineDispatchers, |
||||
private val systemClock: SystemClock, |
||||
private val roomContentForwarder: RoomContentForwarder, |
||||
private val roomListService: RoomListService, |
||||
private val innerRoomListService: InnerRoomListService, |
||||
private val isKeyBackupEnabled: suspend () -> Boolean, |
||||
private val getSessionData: suspend () -> SessionData, |
||||
) { |
||||
@OptIn(ExperimentalCoroutinesApi::class) |
||||
private val createRoomDispatcher = dispatchers.io.limitedParallelism(1) |
||||
private val mutex = Mutex() |
||||
|
||||
private val matrixRoomInfoMapper = MatrixRoomInfoMapper() |
||||
|
||||
private val roomSyncSubscriber: RoomSyncSubscriber = RoomSyncSubscriber(innerRoomListService, dispatchers) |
||||
|
||||
private val eventFilters = TimelineConfig.excludedEvents |
||||
.takeIf { it.isNotEmpty() } |
||||
?.let { listStateEventType -> |
||||
TimelineEventTypeFilter.exclude( |
||||
listStateEventType.map { stateEventType -> |
||||
FilterTimelineEventType.State(stateEventType.map()) |
||||
} |
||||
) |
||||
} |
||||
|
||||
suspend fun create(roomId: RoomId): MatrixRoom? = withContext(createRoomDispatcher) { |
||||
var cachedPairOfRoom: Pair<RoomListItem, Room>? |
||||
mutex.withLock { |
||||
// Check if already in memory... |
||||
cachedPairOfRoom = pairOfRoom(roomId) |
||||
if (cachedPairOfRoom == null) { |
||||
// ... otherwise, lets wait for the SS to load all rooms and check again. |
||||
roomListService.allRooms.awaitLoaded() |
||||
cachedPairOfRoom = pairOfRoom(roomId) |
||||
} |
||||
} |
||||
if (cachedPairOfRoom == null) { |
||||
Timber.d("No room found for $roomId") |
||||
return@withContext null |
||||
} |
||||
cachedPairOfRoom?.let { (roomListItem, fullRoom) -> |
||||
RustMatrixRoom( |
||||
sessionId = sessionId, |
||||
isKeyBackupEnabled = isKeyBackupEnabled(), |
||||
roomListItem = roomListItem, |
||||
innerRoom = fullRoom, |
||||
innerTimeline = fullRoom.timeline(), |
||||
notificationSettingsService = notificationSettingsService, |
||||
sessionCoroutineScope = sessionCoroutineScope, |
||||
coroutineDispatchers = dispatchers, |
||||
systemClock = systemClock, |
||||
roomContentForwarder = roomContentForwarder, |
||||
sessionData = getSessionData(), |
||||
roomSyncSubscriber = roomSyncSubscriber, |
||||
matrixRoomInfoMapper = matrixRoomInfoMapper, |
||||
) |
||||
} |
||||
} |
||||
|
||||
private suspend fun pairOfRoom(roomId: RoomId): Pair<RoomListItem, Room>? { |
||||
val cachedRoomListItem = innerRoomListService.roomOrNull(roomId.value) |
||||
val fullRoom = try { |
||||
cachedRoomListItem?.fullRoomWithTimeline(filter = eventFilters) |
||||
} catch (e: RoomListException) { |
||||
Timber.e(e, "Failed to get full room with timeline for $roomId") |
||||
null |
||||
} |
||||
return if (cachedRoomListItem == null || fullRoom == null) { |
||||
Timber.d("No room cached for $roomId") |
||||
null |
||||
} else { |
||||
Timber.d("Found room cached for $roomId") |
||||
Pair(cachedRoomListItem, fullRoom) |
||||
} |
||||
} |
||||
} |
||||
Loading…
Reference in new issue