From dc06b5266f8d45cd4982f908a84cc801035790d3 Mon Sep 17 00:00:00 2001
From: ganfra <francoisg@element.io>
Date: Tue, 27 Jun 2023 12:52:06 +0200
Subject: [PATCH] RoomList: branch invites

---
 .../impl/room/RustRoomSummaryDataSource.kt    | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

diff --git a/libraries/matrix/impl/src/main/kotlin/io/element/android/libraries/matrix/impl/room/RustRoomSummaryDataSource.kt b/libraries/matrix/impl/src/main/kotlin/io/element/android/libraries/matrix/impl/room/RustRoomSummaryDataSource.kt
index ed1a060086..45e399e2d5 100644
--- a/libraries/matrix/impl/src/main/kotlin/io/element/android/libraries/matrix/impl/room/RustRoomSummaryDataSource.kt
+++ b/libraries/matrix/impl/src/main/kotlin/io/element/android/libraries/matrix/impl/room/RustRoomSummaryDataSource.kt
@@ -23,6 +23,7 @@ import kotlinx.coroutines.CoroutineScope
 import kotlinx.coroutines.flow.Flow
 import kotlinx.coroutines.flow.MutableStateFlow
 import kotlinx.coroutines.flow.StateFlow
+import kotlinx.coroutines.flow.first
 import kotlinx.coroutines.flow.launchIn
 import kotlinx.coroutines.flow.map
 import kotlinx.coroutines.flow.onEach
@@ -34,6 +35,7 @@ import org.matrix.rustcomponents.sdk.RoomListInput
 import org.matrix.rustcomponents.sdk.RoomListLoadingState
 import org.matrix.rustcomponents.sdk.RoomListRange
 import org.matrix.rustcomponents.sdk.RoomListService
+import org.matrix.rustcomponents.sdk.RoomListServiceState
 import timber.log.Timber
 
 internal class RustRoomSummaryDataSource(
@@ -48,18 +50,31 @@ internal class RustRoomSummaryDataSource(
 
     private val allRoomsLoadingState: MutableStateFlow<RoomSummaryDataSource.LoadingState> = MutableStateFlow(RoomSummaryDataSource.LoadingState.NotLoaded)
     private val allRoomsListProcessor = RoomSummaryListProcessor(allRooms, roomListService, roomSummaryDetailsFactory)
+    private val inviteRoomsListProcessor = RoomSummaryListProcessor(inviteRooms, roomListService, roomSummaryDetailsFactory)
 
     init {
         sessionCoroutineScope.launch(coroutineDispatchers.computation) {
             val allRooms = roomListService.allRooms()
-            allRooms.observeEntriesWithProcessor(allRoomsListProcessor)
+            allRooms
+                .observeEntriesWithProcessor(allRoomsListProcessor)
                 .launchIn(this)
 
-            allRooms.loadingStateFlow()
+            allRooms
+                .loadingStateFlow()
                 .map { it.toRoomSummaryDataSourceLoadingState() }
                 .onEach {
                     allRoomsLoadingState.value = it
                 }.launchIn(this)
+
+            launch {
+                // Wait until running, as invites is only available after that
+                roomListService.stateFlow().first {
+                    it == RoomListServiceState.RUNNING
+                }
+                roomListService.invites()
+                    .observeEntriesWithProcessor(inviteRoomsListProcessor)
+                    .launchIn(this)
+            }
         }
     }