Introduce SessionComponent

app/src/main/java/io/element/android/x/MainViewModel.kt

@@ -8,8 +8,7 @@ import dagger.assisted.AssistedInject
 import io.element.android.x.anvilannotations.ContributesViewModel
 import io.element.android.x.core.di.daggerMavericksViewModelFactory
 import io.element.android.x.di.AppScope
-import io.element.android.x.features.messages.MessagesViewModel
-import io.element.android.x.features.messages.model.MessagesViewState
+import io.element.android.x.di.SessionComponentsOwner
 import io.element.android.x.matrix.Matrix
 import kotlinx.coroutines.flow.first
 import kotlinx.coroutines.launch
@@ -19,10 +18,12 @@ data class MainState(val fake: Boolean = false) : MavericksState
 @ContributesViewModel(AppScope::class)
 class MainViewModel @AssistedInject constructor(
     private val matrix: Matrix,
+    private val sessionComponentsOwner: SessionComponentsOwner,
     @Assisted initialState: MainState
 ) : MavericksViewModel<MainState>(initialState) {
 
-    companion object : MavericksViewModelFactory<MainViewModel, MainState> by daggerMavericksViewModelFactory()
+    companion object :
+        MavericksViewModelFactory<MainViewModel, MainState> by daggerMavericksViewModelFactory()
 
     suspend fun isLoggedIn(): Boolean {
         return matrix.isLoggedIn().first()
@@ -31,19 +32,22 @@ class MainViewModel @AssistedInject constructor(
     fun startSyncIfLogged() {
         viewModelScope.launch {
             if (!isLoggedIn()) return@launch
-            matrix.activeClient().startSync()
         }
     }
 
     fun stopSyncIfLogged() {
         viewModelScope.launch {
             if (!isLoggedIn()) return@launch
-            matrix.activeClient().stopSync()
         }
     }
 
     suspend fun restoreSession() {
-        matrix.restoreSession()
-        matrix.activeClient().startSync()
+        val matrixClient = matrix.restoreSession()
+        if (matrixClient == null) {
+            throw IllegalStateException("Couldn't restore session...")
+        } else {
+            sessionComponentsOwner.create(matrixClient)
+            matrixClient.startSync()
+        }
     }
 }