From f6302e4ed3dbf3504de26dbb0b8826ce336782a9 Mon Sep 17 00:00:00 2001
From: Benoit Marty <benoit@matrix.org>
Date: Tue, 25 Apr 2023 16:02:38 +0200
Subject: [PATCH] better api

---
 .../login/impl/oidc/customtab/DefaultOidcActionFlow.kt       | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/features/login/impl/src/main/kotlin/io/element/android/features/login/impl/oidc/customtab/DefaultOidcActionFlow.kt b/features/login/impl/src/main/kotlin/io/element/android/features/login/impl/oidc/customtab/DefaultOidcActionFlow.kt
index 41ec484298..17dfa8418f 100644
--- a/features/login/impl/src/main/kotlin/io/element/android/features/login/impl/oidc/customtab/DefaultOidcActionFlow.kt
+++ b/features/login/impl/src/main/kotlin/io/element/android/features/login/impl/oidc/customtab/DefaultOidcActionFlow.kt
@@ -21,6 +21,7 @@ import io.element.android.features.login.api.oidc.OidcAction
 import io.element.android.features.login.api.oidc.OidcActionFlow
 import io.element.android.libraries.di.AppScope
 import io.element.android.libraries.di.SingleIn
+import kotlinx.coroutines.flow.FlowCollector
 import kotlinx.coroutines.flow.MutableStateFlow
 import javax.inject.Inject
 
@@ -33,8 +34,8 @@ class DefaultOidcActionFlow @Inject constructor() : OidcActionFlow {
         mutableStateFlow.value = oidcAction
     }
 
-    suspend fun collect(lambda: suspend (OidcAction?) -> Unit) {
-        mutableStateFlow.collect(lambda)
+    suspend fun collect(collector: FlowCollector<OidcAction?>) {
+        mutableStateFlow.collect(collector)
     }
 
     fun reset() {