Generic flow to answer

libraries/coroutines/src/main/java/com/raxdenstudios/commons/coroutines/ext/AnswerExtension.kt

@@ -74,9 +74,15 @@ suspend fun <T, E> Flow<Answer<T, E>>.collectResult(
     onFailure: (E) -> Unit,
 ) = collect { result -> result.fold(onSuccess, onFailure) }
 
-fun <T> Flow<T>.toAnswer() =
-    map { data -> Answer.Success(data) }
-        .catch { error -> Answer.Failure(error) }
+fun <T> Flow<T>.toAnswer(): Flow<Answer<T, Throwable>> =
+    toAnswer { error -> error }
+
+fun <T, E> Flow<T>.toAnswer(
+    mapFailure: (Throwable) -> E,
+): Flow<Answer<T, E>> {
+    val answers: Flow<Answer<T, E>> = map { data -> Answer.Success(data) }
+    return answers.catch { error -> emit(Answer.Failure(mapFailure(error))) }
+}
 
 @Suppress("UNCHECKED_CAST")
 suspend fun <T1, T2, R, E> Answer<T1, E>.coCombine(

libraries/coroutines/src/test/java/com/raxdenstudios/commons/coroutines/ext/AnswerExtensionTest.kt

@@ -221,6 +221,20 @@ internal class AnswerExtensionTest {
             val result1 = awaitItem()
             assertThat(result1.isSuccess).isTrue()
             assertThat(result1).isEqualTo(Answer.Success("value"))
+
+            val result2 = awaitItem()
+            assertThat(result2.isFailure).isTrue()
+            assertThat((result2 as Answer.Failure).value.message).isEqualTo("Test error")
+            awaitComplete()
+        }
+    }
+
+    @Test
+    fun `use toAnswer with mapped failure`() = runTest {
+        val flowData = flow<String> { error("Test error") }
+
+        flowData.toAnswer { error -> "Mapped: ${error.message}" }.test {
+            assertThat(awaitItem()).isEqualTo(Answer.Failure("Mapped: Test error"))
             awaitComplete()
         }
     }